fix(ai): fix error when chat without params

mayneyao · Nov 26, 2024 · fa03eb8 · fa03eb8
1 parent bd31071
commit fa03eb8
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/electron/server/server.ts b/electron/server/server.ts
@@ -58,7 +58,7 @@ export function startServer({ dist, port }: { dist: string, port: number }) {
             request: c.req,
             respondWith: (response: Response) => response,
         } as unknown as FetchEvent, {
-            getDataspace: (space) => getOrSetDataSpace(space)
+            getDataspace: (space) => space ? getOrSetDataSpace(space) : Promise.resolve(null)
         });
         return response;
     });

diff --git a/worker/service-worker/ai/index.ts b/worker/service-worker/ai/index.ts
@@ -6,7 +6,7 @@ import { handleWebLLM } from "./webllm"
 
 export const pathname = "/api/chat"
 export default async function handle(event: FetchEvent, ctx?: {
-  getDataspace: (space: string) => Promise<DataSpace>
+  getDataspace: (space: string) => Promise<DataSpace | null>
 }) {
   const data = (await event.request.json()) as IData
   const { type } = data

diff --git a/worker/service-worker/ai/openai.ts b/worker/service-worker/ai/openai.ts
@@ -15,7 +15,7 @@ import { IData } from "./interface";
 export async function handleOpenAI(
   data: IData,
   ctx?: {
-    getDataspace: (space: string) => Promise<DataSpace>
+    getDataspace: (space: string) => Promise<DataSpace | null>
   }
 ) {
   // only use functions on desktop app
@@ -57,7 +57,7 @@ export async function handleOpenAI(
     return new Response('No user message found', { status: 400 });
   }
 
-  const dataspace = await ctx?.getDataspace(space)
+  const dataspace = space && await ctx?.getDataspace(space)
 
   if (dataspace) {
     const chat = await getChatById(id, dataspace);