chore: optimize the code

JMcn · Jul 5, 2024 · 14bc1b6 · 14bc1b6
1 parent 183ad2a
commit 14bc1b6
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/app/client/api.ts b/app/client/api.ts
@@ -162,7 +162,7 @@ export function getHeaders() {
     Accept: "application/json",
   };
   const modelConfig = useChatStore.getState().currentSession().mask.modelConfig;
-  const isGoogle = modelConfig.providerName == ServiceProvider.Azure;
+  const isGoogle = modelConfig.providerName == ServiceProvider.Google;
   const isAzure = modelConfig.providerName === ServiceProvider.Azure;
   const isAnthropic = modelConfig.providerName === ServiceProvider.Anthropic;
   const authHeader = isAzure

diff --git a/app/client/platforms/openai.ts b/app/client/platforms/openai.ts
@@ -142,22 +142,21 @@ export class ChatGPTApi implements LLMApi {
 
     try {
       let chatPath = "";
-      if (modelConfig.providerName == ServiceProvider.Azure) {
+      if (modelConfig.providerName === ServiceProvider.Azure) {
         // find model, and get displayName as deployName
         const { models: configModels, customModels: configCustomModels } =
           useAppConfig.getState();
         const { defaultModel, customModels: accessCustomModels } =
           useAccessStore.getState();
-
         const models = collectModelsWithDefaultModel(
           configModels,
           [configCustomModels, accessCustomModels].join(","),
           defaultModel,
         );
         const model = models.find(
           (model) =>
-            model.name == modelConfig.model &&
-            model?.provider?.providerName == ServiceProvider.Azure,
+            model.name === modelConfig.model &&
+            model?.provider?.providerName === ServiceProvider.Azure,
         );
         chatPath = this.path(
           Azure.ChatPath(

diff --git a/app/constant.ts b/app/constant.ts
@@ -153,6 +153,7 @@ const openaiModels = [
   "gpt-4o-2024-05-13",
   "gpt-4-vision-preview",
   "gpt-4-turbo-2024-04-09",
+  "gpt-4-1106-preview",
 ];
 
 const googleModels = [