fix: #198

Hk-Gosuto · Feb 20, 2024 · b920027 · b920027
1 parent 6dd0ad6
commit b920027
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/app/api/langchain/tool/agent/agentapi.ts b/app/api/langchain/tool/agent/agentapi.ts
@@ -225,7 +225,8 @@ export class AgentApi {
       const serverConfig = getServerSideConfig();
 
       // const reqBody: RequestBody = await req.json();
-      const isAzure = reqBody.isAzure || serverConfig.isAzure;
+      // ui set azure model provider
+      const isAzure = reqBody.isAzure;
       const authHeaderName = isAzure ? "api-key" : "Authorization";
       const authToken = req.headers.get(authHeaderName) ?? "";
       const token = authToken.trim().replaceAll("Bearer ", "").trim();

diff --git a/app/store/chat.ts b/app/store/chat.ts
@@ -91,6 +91,11 @@ function createEmptySession(): ChatSession {
 }
 
 function getSummarizeModel(currentModel: string) {
+  // if the current model does not exist in the default model
+  // example azure services cannot use SUMMARIZE_MODEL
+  const model = DEFAULT_MODELS.find((m) => m.name === currentModel);
+  console.log("model", model);
+  if (!model) return currentModel;
   // if it is using gpt-* models, force to use 3.5 to summarize
   return currentModel.startsWith("gpt") ? SUMMARIZE_MODEL : currentModel;
 }