110 lines
3.3 KiB
TypeScript
110 lines
3.3 KiB
TypeScript
import { create } from "zustand";
|
|
import type { ModelProvider, PromptEnhancement, PRD, ActionPlan } from "@/types";
|
|
|
|
interface AppState {
|
|
currentPrompt: string;
|
|
enhancedPrompt: string | null;
|
|
prd: PRD | null;
|
|
actionPlan: ActionPlan | null;
|
|
selectedProvider: ModelProvider;
|
|
selectedModels: Record<ModelProvider, string>;
|
|
availableModels: Record<ModelProvider, string[]>;
|
|
apiKeys: Record<ModelProvider, string>;
|
|
qwenTokens?: {
|
|
accessToken: string;
|
|
refreshToken?: string;
|
|
expiresAt?: number;
|
|
};
|
|
isProcessing: boolean;
|
|
error: string | null;
|
|
history: {
|
|
id: string;
|
|
prompt: string;
|
|
timestamp: Date;
|
|
}[];
|
|
|
|
setCurrentPrompt: (prompt: string) => void;
|
|
setEnhancedPrompt: (enhanced: string | null) => void;
|
|
setPRD: (prd: PRD) => void;
|
|
setActionPlan: (plan: ActionPlan) => void;
|
|
setSelectedProvider: (provider: ModelProvider) => void;
|
|
setSelectedModel: (provider: ModelProvider, model: string) => void;
|
|
setAvailableModels: (provider: ModelProvider, models: string[]) => void;
|
|
setApiKey: (provider: ModelProvider, key: string) => void;
|
|
setQwenTokens: (tokens: { accessToken: string; refreshToken?: string; expiresAt?: number }) => void;
|
|
setProcessing: (processing: boolean) => void;
|
|
setError: (error: string | null) => void;
|
|
addToHistory: (prompt: string) => void;
|
|
clearHistory: () => void;
|
|
reset: () => void;
|
|
}
|
|
|
|
const useStore = create<AppState>((set) => ({
|
|
currentPrompt: "",
|
|
enhancedPrompt: null,
|
|
prd: null,
|
|
actionPlan: null,
|
|
selectedProvider: "qwen",
|
|
selectedModels: {
|
|
qwen: "qwen-coder-plus",
|
|
ollama: "gpt-oss:120b",
|
|
zai: "glm-4.7",
|
|
},
|
|
availableModels: {
|
|
qwen: ["qwen-coder-plus", "qwen-coder-turbo", "qwen-coder-lite"],
|
|
ollama: ["gpt-oss:120b", "llama3.1", "gemma3", "deepseek-r1", "qwen3"],
|
|
zai: ["glm-4.7", "glm-4.6", "glm-4.5", "glm-4.5-air", "glm-4-flash", "glm-4-flashx"],
|
|
},
|
|
apiKeys: {
|
|
qwen: "",
|
|
ollama: "",
|
|
zai: "",
|
|
},
|
|
isProcessing: false,
|
|
error: null,
|
|
history: [],
|
|
|
|
setCurrentPrompt: (prompt) => set({ currentPrompt: prompt }),
|
|
setEnhancedPrompt: (enhanced) => set({ enhancedPrompt: enhanced }),
|
|
setPRD: (prd) => set({ prd }),
|
|
setActionPlan: (plan) => set({ actionPlan: plan }),
|
|
setSelectedProvider: (provider) => set({ selectedProvider: provider }),
|
|
setSelectedModel: (provider, model) =>
|
|
set((state) => ({
|
|
selectedModels: { ...state.selectedModels, [provider]: model },
|
|
})),
|
|
setAvailableModels: (provider, models) =>
|
|
set((state) => ({
|
|
availableModels: { ...state.availableModels, [provider]: models },
|
|
})),
|
|
setApiKey: (provider, key) =>
|
|
set((state) => ({
|
|
apiKeys: { ...state.apiKeys, [provider]: key },
|
|
})),
|
|
setQwenTokens: (tokens) => set({ qwenTokens: tokens }),
|
|
setProcessing: (processing) => set({ isProcessing: processing }),
|
|
setError: (error) => set({ error }),
|
|
addToHistory: (prompt) =>
|
|
set((state) => ({
|
|
history: [
|
|
...state.history,
|
|
{
|
|
id: Math.random().toString(36).substr(2, 9),
|
|
prompt,
|
|
timestamp: new Date(),
|
|
},
|
|
],
|
|
})),
|
|
clearHistory: () => set({ history: [] }),
|
|
reset: () =>
|
|
set({
|
|
currentPrompt: "",
|
|
enhancedPrompt: null,
|
|
prd: null,
|
|
actionPlan: null,
|
|
error: null,
|
|
}),
|
|
}));
|
|
|
|
export default useStore;
|