Initialize PromptArch: The Prompt Enhancer (Fork of ClavixDev/Clavix)

2025-12-25 16:58:36 +04:00
commit c0a01b5840
35 changed files with 11054 additions and 0 deletions
--- a/lib/store.ts
+++ b/lib/store.ts
@@ -0,0 +1,109 @@
+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;