mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-10-09 03:07:46 +08:00
dayuan.jiang
69fb66804d
Tabs Models, General, Drawing and About, each with a fixed title row. The models tab becomes three pages: the user's providers with one status each and the server's models; a searchable picker of all 24 providers in four groups; and a provider page in three steps (connection, models, test). A new provider gets its first suggested model, and a passing test offers to use the model in the chat. Bedrock asks for one way of signing in, with an optional session token for temporary access keys. The composer's model picker lists every model, marking untested and failed ones, and can open the provider picker. A chat error that offers model settings opens the provider the request went to.
72 lines
2.7 KiB
TypeScript
72 lines
2.7 KiB
TypeScript
import { create } from "zustand"
|
|
|
|
export type SettingsTab = "models" | "general" | "drawing" | "about"
|
|
/** The models tab: its provider list, the provider picker or a provider */
|
|
export type ModelsPage = "list" | "picker" | { providerId: string }
|
|
export type MobileView = "canvas" | "chat"
|
|
|
|
interface UiState {
|
|
/** Desktop: the floating chat panel is open */
|
|
panelOpen: boolean
|
|
/** The start screen was put away (drawing by hand, an opened file) */
|
|
heroDismissed: boolean
|
|
/** Phones: which half of the app is visible */
|
|
mobileView: MobileView
|
|
settingsOpen: boolean
|
|
settingsTab: SettingsTab
|
|
modelsPage: ModelsPage
|
|
/** Version shown in the compare dialog, or null when closed */
|
|
compareVersionId: string | null
|
|
saveDialogOpen: boolean
|
|
/** Bumped to ask the composer to take focus */
|
|
focusComposerToken: number
|
|
setPanelOpen: (open: boolean) => void
|
|
setHeroDismissed: (dismissed: boolean) => void
|
|
togglePanel: () => void
|
|
setMobileView: (view: MobileView) => void
|
|
/** Opens on the given tab (else the last one); the models tab on its
|
|
* list unless a page is given */
|
|
openSettings: (tab?: SettingsTab, modelsPage?: ModelsPage) => void
|
|
setSettingsOpen: (open: boolean) => void
|
|
setSettingsTab: (tab: SettingsTab) => void
|
|
setModelsPage: (page: ModelsPage) => void
|
|
openCompare: (versionId: string) => void
|
|
closeCompare: () => void
|
|
setSaveDialogOpen: (open: boolean) => void
|
|
focusComposer: () => void
|
|
}
|
|
|
|
export const useUiStore = create<UiState>((set) => ({
|
|
panelOpen: true,
|
|
heroDismissed: false,
|
|
mobileView: "chat",
|
|
settingsOpen: false,
|
|
settingsTab: "models",
|
|
modelsPage: "list",
|
|
compareVersionId: null,
|
|
saveDialogOpen: false,
|
|
focusComposerToken: 0,
|
|
setPanelOpen: (open) => set({ panelOpen: open }),
|
|
setHeroDismissed: (dismissed) => set({ heroDismissed: dismissed }),
|
|
togglePanel: () => set((state) => ({ panelOpen: !state.panelOpen })),
|
|
setMobileView: (view) => set({ mobileView: view }),
|
|
openSettings: (tab, modelsPage = "list") =>
|
|
set((state) => ({
|
|
settingsOpen: true,
|
|
settingsTab: tab ?? state.settingsTab,
|
|
modelsPage,
|
|
})),
|
|
setSettingsOpen: (open) => set({ settingsOpen: open }),
|
|
setSettingsTab: (tab) => set({ settingsTab: tab }),
|
|
setModelsPage: (page) => set({ modelsPage: page }),
|
|
openCompare: (versionId) => set({ compareVersionId: versionId }),
|
|
closeCompare: () => set({ compareVersionId: null }),
|
|
setSaveDialogOpen: (open) => set({ saveDialogOpen: open }),
|
|
focusComposer: () =>
|
|
set((state) => ({
|
|
panelOpen: true,
|
|
mobileView: "chat",
|
|
focusComposerToken: state.focusComposerToken + 1,
|
|
})),
|
|
}))
|