@lobehub/chat 1.51.6 → 1.51.8

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

@@ -1,6 +1,12 @@
 import { ChatStreamPayload, ModelProvider, OpenAIChatMessage } from '../types';
 import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
 
+import { LOBE_DEFAULT_MODEL_LIST } from '@/config/aiModels';
+
+export interface OpenAIModelCard {
+  id: string;
+}
+
 export const pruneReasoningPayload = (payload: ChatStreamPayload) => {
   // TODO: 临时写法，后续要重构成 model card 展示配置
   const disableStreamModels = new Set([
@@ -49,5 +55,36 @@ export const LobeOpenAI = LobeOpenAICompatibleFactory({
   debug: {
     chatCompletion: () => process.env.DEBUG_OPENAI_CHAT_COMPLETION === '1',
   },
+  models: {
+    transformModel: (m) => {
+      const functionCallKeywords = [
+        'gpt-4',
+        'gpt-3.5',
+        'o3-mini',
+      ];
+
+      const visionKeywords = [
+        'gpt-4o',
+        'vision',
+      ];
+
+      const reasoningKeywords = [
+        'o1',
+        'o3',
+      ];
+
+      const model = m as unknown as OpenAIModelCard;
+
+      return {
+        contextWindowTokens: LOBE_DEFAULT_MODEL_LIST.find((m) => model.id === m.id)?.contextWindowTokens ?? undefined,
+        displayName: LOBE_DEFAULT_MODEL_LIST.find((m) => model.id === m.id)?.displayName ?? undefined,
+        enabled: LOBE_DEFAULT_MODEL_LIST.find((m) => model.id === m.id)?.enabled || false,
+        functionCall: functionCallKeywords.some(keyword => model.id.toLowerCase().includes(keyword)) && !model.id.toLowerCase().includes('audio'),
+        id: model.id,
+        reasoning: reasoningKeywords.some(keyword => model.id.toLowerCase().includes(keyword)),
+        vision: visionKeywords.some(keyword => model.id.toLowerCase().includes(keyword)) && !model.id.toLowerCase().includes('audio'),
+      };
+    },
+  },
   provider: ModelProvider.OpenAI,
 });