vercel__ai-elements/apps/docs/hooks/geistdocs/use-chat.ts

import { atom, useAtom } from "jotai";
import { atomWithStorage } from "jotai/utils";

// Export the prompt atom so it can be used from other parts of the app
export const chatPromptAtom = atom<string>("");

// Export the open state atom for controlling chat visibility
export const chatOpenAtom = atomWithStorage<boolean>(
  "geistdocs:chat-open",
  false
);

export const useChatContext = () => {
  const [prompt, setPrompt] = useAtom(chatPromptAtom);
  const [isOpen, setIsOpen] = useAtom(chatOpenAtom);

  return {
    isOpen,
    prompt,
    setIsOpen,
    setPrompt,
  };
};