store.ts 1.3 KB

123456789101112131415161718192021222324252627282930313233343536
  1. import { create } from 'zustand'
  2. import type { App } from '@/types/app'
  3. import type { IChatItem } from '@/app/components/app/chat/type'
  4. type State = {
  5. appDetail?: App
  6. appSidebarExpand: string
  7. currentLogItem?: IChatItem
  8. showPromptLogModal: boolean
  9. showAgentLogModal: boolean
  10. showMessageLogModal: boolean
  11. }
  12. type Action = {
  13. setAppDetail: (appDetail?: App) => void
  14. setAppSiderbarExpand: (state: string) => void
  15. setCurrentLogItem: (item?: IChatItem) => void
  16. setShowPromptLogModal: (showPromptLogModal: boolean) => void
  17. setShowAgentLogModal: (showAgentLogModal: boolean) => void
  18. setShowMessageLogModal: (showMessageLogModal: boolean) => void
  19. }
  20. export const useStore = create<State & Action>(set => ({
  21. appDetail: undefined,
  22. setAppDetail: appDetail => set(() => ({ appDetail })),
  23. appSidebarExpand: '',
  24. setAppSiderbarExpand: appSidebarExpand => set(() => ({ appSidebarExpand })),
  25. currentLogItem: undefined,
  26. setCurrentLogItem: currentLogItem => set(() => ({ currentLogItem })),
  27. showPromptLogModal: false,
  28. setShowPromptLogModal: showPromptLogModal => set(() => ({ showPromptLogModal })),
  29. showAgentLogModal: false,
  30. setShowAgentLogModal: showAgentLogModal => set(() => ({ showAgentLogModal })),
  31. showMessageLogModal: false,
  32. setShowMessageLogModal: showMessageLogModal => set(() => ({ showMessageLogModal })),
  33. }))