逻辑还要处理

src/pages/ai-chat/store/ai-store.ts

@@ -41,6 +41,9 @@ export type AiStore = {
   setType?: (type: string) => void;
   formData: any;
   setFormData: (data: any) => void;
+  // Context Data
+  data: any;
+  setData: (data: any) => void;
   title: string; // AI Module的标题
   setTitle: (title: string) => void;
   messages: { role: string; content: string }[];
@@ -73,6 +76,8 @@ export const useAiStore = create<AiStore>((set, get, store) => {
     setFormData: (data) => set({ formData: data }),
     title: '',
     setTitle: (title) => set({ title }),
+    data: {},
+    setData: (data) => set({ data }),
     messages: [],
     setMessage: (messages) => set({ messages }),
     getPrompt: async (key) => {