🐛 fix: fix tencent cloud api issue (#6058)

hedeqiang · web-flow · commit 025d0bcd62bf · 2025-02-12T22:27:52.000+08:00
* fix: tencentcloud

* fix: tencentcloud
diff --git a/.env.example b/.env.example
@@ -122,6 +122,11 @@ OPENAI_API_KEY=sk-xxxxxxxxx
 
 # SILICONCLOUD_API_KEY=sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
 
+
+### TencentCloud AI  ####
+
+# TENCENT_CLOUD_API_KEY=sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
+
 ########################################
 ############ Market Service ############
 ########################################
diff --git a/Dockerfile b/Dockerfile
@@ -222,7 +222,9 @@ ENV \
     # 01.AI
     ZEROONE_API_KEY="" ZEROONE_MODEL_LIST="" \
     # Zhipu
-    ZHIPU_API_KEY="" ZHIPU_MODEL_LIST=""
+    ZHIPU_API_KEY="" ZHIPU_MODEL_LIST="" \
+    # Tencent Cloud
+    TENCENT_CLOUD_API_KEY="" TENCENT_CLOUD_MODEL_LIST=""
 
 USER nextjs
 
diff --git a/Dockerfile.database b/Dockerfile.database
@@ -259,7 +259,9 @@ ENV \
     # 01.AI
     ZEROONE_API_KEY="" ZEROONE_MODEL_LIST="" \
     # Zhipu
-    ZHIPU_API_KEY="" ZHIPU_MODEL_LIST=""
+    ZHIPU_API_KEY="" ZHIPU_MODEL_LIST="" \
+    # Tencent Cloud
+    TENCENT_CLOUD_API_KEY="" TENCENT_CLOUD_MODEL_LIST=""
 
 USER nextjs
 
diff --git a/src/config/llm.ts b/src/config/llm.ts
@@ -125,6 +125,9 @@ export const getLLMConfig = () => {
 
       ENABLED_DOUBAO: z.boolean(),
       DOUBAO_API_KEY: z.string().optional(),
+
+      ENABLED_TENCENT_CLOUD: z.boolean(),
+      TENCENT_CLOUD_API_KEY: z.string().optional(),
     },
     runtimeEnv: {
       API_KEY_SELECT_MODE: process.env.API_KEY_SELECT_MODE,
@@ -248,6 +251,9 @@ export const getLLMConfig = () => {
 
       ENABLED_DOUBAO: !!process.env.DOUBAO_API_KEY,
       DOUBAO_API_KEY: process.env.DOUBAO_API_KEY,
+
+      ENABLED_TENCENT_CLOUD: !!process.env.TENCENT_CLOUD_API_KEY,
+      TENCENT_CLOUD_API_KEY: process.env.TENCENT_CLOUD_API_KEY,
     },
   });
 };
diff --git a/src/server/modules/AgentRuntime/index.ts b/src/server/modules/AgentRuntime/index.ts
@@ -100,6 +100,14 @@ const getLlmOptionsFromPayload = (provider: string, payload: JWTPayload) => {
 
       return { apiKey };
     }
+    
+    case ModelProvider.TencentCloud: {
+      const { TENCENT_CLOUD_API_KEY } = llmConfig;
+
+      const apiKey = apiKeyManager.pick(payload?.apiKey || TENCENT_CLOUD_API_KEY);
+
+      return { apiKey };
+    }
   }
 };
 

Original file line number	Diff line number	Diff line change
`@@ -100,6 +100,14 @@ const getLlmOptionsFromPayload = (provider: string, payload: JWTPayload) => {`
`100`	`100`
`101`	`101`	`return { apiKey };`
`102`	`102`	`}`
	`103`	`+`
	`104`	`+ case ModelProvider.TencentCloud: {`
	`105`	`+ const { TENCENT_CLOUD_API_KEY } = llmConfig;`
	`106`	`+`
	`107`	`+ const apiKey = apiKeyManager.pick(payload?.apiKey \|\| TENCENT_CLOUD_API_KEY);`
	`108`	`+`
	`109`	`+ return { apiKey };`
	`110`	`+ }`
`103`	`111`	`}`
`104`	`112`	`};`
`105`	`113`