@lobehub/chat 1.47.12 → 1.47.14

package/src/libs/agent-runtime/zhipu/index.ts

@@ -3,6 +3,15 @@ import OpenAI from 'openai';
 import { ChatStreamPayload, ModelProvider } from '../types';
 import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
 
+import { LOBE_DEFAULT_MODEL_LIST } from '@/config/aiModels';
+import type { ChatModelCard } from '@/types/llm';
+
+export interface ZhipuModelCard {
+  description: string;
+  modelCode: string;
+  modelName: string;
+}
+
 export const LobeZhipuAI = LobeOpenAICompatibleFactory({
   baseURL: 'https://open.bigmodel.cn/api/paas/v4',
   chatCompletion: {
@@ -25,8 +34,34 @@ export const LobeZhipuAI = LobeOpenAICompatibleFactory({
             }),
       }) as OpenAI.ChatCompletionCreateParamsStreaming,
   },
+  constructorOptions: {
+    defaultHeaders: {
+      'Bigmodel-Organization': 'lobehub',
+      'Bigmodel-project': 'lobechat',
+    },
+  },
   debug: {
     chatCompletion: () => process.env.DEBUG_ZHIPU_CHAT_COMPLETION === '1',
   },
+  models: async ({ client }) => {
+    // ref: https://open.bigmodel.cn/console/modelcenter/square
+    client.baseURL = 'https://open.bigmodel.cn/api/fine-tuning/model_center/list?pageSize=100&pageNum=1';
+
+    const modelsPage = await client.models.list() as any;
+    const modelList: ZhipuModelCard[] = modelsPage.body.rows;
+
+    return modelList
+      .map((model) => {
+        return {
+          description: model.description,
+          displayName: model.modelName,
+          enabled: LOBE_DEFAULT_MODEL_LIST.find((m) => model.modelCode.endsWith(m.id))?.enabled || false,
+          functionCall: model.modelCode.toLowerCase().includes('glm-4') && !model.modelCode.toLowerCase().includes('glm-4v'),
+          id: model.modelCode,
+          vision: model.modelCode.toLowerCase().includes('glm-4v'),
+        };
+      })
+      .filter(Boolean) as ChatModelCard[];
+  },
   provider: ModelProvider.ZhiPu,
 });

package/src/server/globalConfig/genServerAiProviderConfig.ts

@@ -25,7 +25,10 @@ export const genServerAiProvidersConfig = (specificConfig: Record<any, any>) =>
       const defaultChatModels = providerCard.filter((c) => c.type === 'chat');
 
       config[provider] = {
-        enabled: llmConfig[providerConfig.enabledKey || `ENABLED_${providerUpperCase}`],
+        enabled:
+          providerConfig.enabled ||
+          llmConfig[providerConfig.enabledKey || `ENABLED_${providerUpperCase}`],
+
         enabledModels: extractEnabledModels(
           providerModelList,
           providerConfig.withDeploymentName || false,

package/src/server/globalConfig/index.ts

@@ -29,6 +29,9 @@ export const getServerGlobalConfig = () => {
         enabledKey: 'ENABLED_GITEE_AI',
         modelListKey: 'GITEE_AI_MODEL_LIST',
       },
+      /* ↓ cloud slot ↓ */
+
+      /* ↑ cloud slot ↑ */
       ollama: {
         fetchOnClient: !process.env.OLLAMA_PROXY_URL,
       },

package/src/services/chat.ts

@@ -294,9 +294,13 @@ class ChatService {
       onFinish: options?.onFinish,
       onMessageHandle: options?.onMessageHandle,
       signal,
-      // use smoothing when enable client fetch
-      // https://github.com/lobehub/lobe-chat/issues/3800
-      smoothing: providerConfig?.smoothing || enableFetchOnClient,
+      smoothing:
+        providerConfig?.settings?.smoothing ||
+        // @deprecated in V2
+        providerConfig?.smoothing ||
+        // use smoothing when enable client fetch
+        // https://github.com/lobehub/lobe-chat/issues/3800
+        enableFetchOnClient,
     });
   };
 

package/src/types/aiProvider.ts

@@ -85,6 +85,7 @@ export interface AiProviderSettings {
    */
   sdkType?: AiProviderSDKType;
 
+  showAddNewModel?: boolean;
   /**
    * whether show api key in the provider config
    * so provider like ollama don't need api key field