增加模型选择

2023-06-16 10:41:32 +08:00
parent 8c8d5bf4e1
commit 7c51471d91
6 changed files with 32 additions and 11 deletions
--- a/src/hooks/useTranslate.ts
+++ b/src/hooks/useTranslate.ts
@@ -13,6 +13,7 @@ import {
 import {
  LANGUAGE_DEFAULT,
  LANGUAGES_MAP,
+  MODEL_DEFAULT,
  PROMPT_DEFAULTS,
  PROMPT_TYPE_TRANSLATE,
  SUMMARIZE_LANGUAGE_DEFAULT,
@@ -84,7 +85,7 @@ const useTranslate = () => {
          type: 'chatComplete',
          serverUrl: envData.serverUrl,
          data: {
-            model: 'gpt-3.5-turbo',
+            model: envData.model??MODEL_DEFAULT,
            messages: [
              {
                role: 'user',
@@ -137,7 +138,7 @@ const useTranslate = () => {
        type: 'chatComplete',
        serverUrl: envData.serverUrl,
        data: {
-          model: 'gpt-3.5-turbo',
+          model: envData.model??MODEL_DEFAULT,
          messages: [
            {
              role: 'user',