💄 style: add support function call for 360AI (lobehub#4099)

bentwnghk · Sep 24, 2024 · 536696b · 536696b
1 parent db1f81c
commit 536696b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/config/modelProviders/ai360.ts b/src/config/modelProviders/ai360.ts
@@ -8,7 +8,6 @@ const Ai360: ModelProviderCard = {
         '360GPT2 Pro 是 360 公司推出的高级自然语言处理模型，具备卓越的文本生成和理解能力，尤其在生成与创作领域表现出色，能够处理复杂的语言转换和角色演绎任务。',
       displayName: '360GPT2 Pro',
       enabled: true,
-      functionCall: false,
       id: '360gpt2-pro',
       maxOutput: 7000,
       pricing: {
@@ -22,6 +21,8 @@ const Ai360: ModelProviderCard = {
       description:
         '360GPT Pro 作为 360 AI 模型系列的重要成员，以高效的文本处理能力满足多样化的自然语言应用场景，支持长文本理解和多轮对话等功能。',
       displayName: '360GPT Pro',
+      enabled: true,
+      functionCall: true,
       id: '360gpt-pro',
       maxOutput: 7000,
       pricing: {

diff --git a/src/libs/agent-runtime/ai360/index.ts b/src/libs/agent-runtime/ai360/index.ts
@@ -3,6 +3,14 @@ import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
 
 export const LobeAi360AI = LobeOpenAICompatibleFactory({
   baseURL: 'https://ai.360.cn/v1',
+  chatCompletion: {
+    handlePayload: (payload) => {
+      return {
+        ...payload,
+        stream: !payload.tools,
+      } as any;
+    },
+  },
   debug: {
     chatCompletion: () => process.env.DEBUG_AI360_CHAT_COMPLETION === '1',
   },