use 0613

anc95 · Jun 14, 2023 · 09b5ed8 · 09b5ed8
1 parent a38096a
commit 09b5ed8
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/src/common/api/openai.ts b/src/common/api/openai.ts
@@ -116,9 +116,11 @@ export const useQueryOpenAIPrompt = () => {
       settings.model === 'gpt-4'
 
     const commonOption = {
-      max_tokens: 4000 - prompt.replace(/[\u4e00-\u9fa5]/g, 'aa').length,
+      // max_tokens: 4000 - prompt.replace(/[\u4e00-\u9fa5]/g, 'aa').length,
       stream: true,
-      model: isWritelyService ? defaultSetting.model : settings.model,
+      model: ensureUsing0613Model(
+        isWritelyService ? defaultSetting.model : settings.model
+      ),
       temperature: parseFloat(settings.temperature),
     }
 
@@ -245,3 +247,12 @@ export const useModels = () => {
     ]
   }, [])
 }
+
+// TODO: remove later
+const ensureUsing0613Model = (model: string) => {
+  if (model.startsWith('gpt-3.5')) {
+    return model + '-0613'
+  }
+
+  return model
+}