v2 initial commit

src/store/index.tsx

@@ -0,0 +1,51 @@
+import { create } from "zustand"
+
+export type Message = {
+  isBot: boolean
+  message: string
+  sources: any[]
+}
+
+export type ChatHistory = {
+  role: "user" | "assistant" | "system"
+  content: string
+}[]
+
+type State = {
+  messages: Message[]
+  setMessages: (messages: Message[]) => void
+  history: ChatHistory
+  setHistory: (history: ChatHistory) => void
+  streaming: boolean
+  setStreaming: (streaming: boolean) => void
+  isFirstMessage: boolean
+  setIsFirstMessage: (isFirstMessage: boolean) => void
+  historyId: string | null
+  setHistoryId: (history_id: string | null) => void
+  isLoading: boolean
+  setIsLoading: (isLoading: boolean) => void
+  isProcessing: boolean
+  setIsProcessing: (isProcessing: boolean) => void
+  selectedModel: string | null
+  setSelectedModel: (selectedModel: string) => void
+}
+
+export const useStoreMessage = create<State>((set) => ({
+  messages: [],
+  setMessages: (messages) => set({ messages }),
+  history: [],
+  setHistory: (history) => set({ history }),
+  streaming: true,
+  setStreaming: (streaming) => set({ streaming }),
+  isFirstMessage: true,
+  setIsFirstMessage: (isFirstMessage) => set({ isFirstMessage }),
+  historyId: null,
+  setHistoryId: (historyId) => set({ historyId }),
+  isLoading: false,
+  setIsLoading: (isLoading) => set({ isLoading }),
+  isProcessing: false,
+  setIsProcessing: (isProcessing) => set({ isProcessing }),
+  defaultSpeechToTextLanguage: "en-US",
+  selectedModel: null,
+  setSelectedModel: (selectedModel) => set({ selectedModel })
+}))