@lobehub/chat 1.75.3 → 1.75.5

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

@@ -1,7 +1,7 @@
-// import { AgentRuntimeErrorType } from '../error';
 import type { ChatModelCard } from '@/types/llm';
 
-import { ModelProvider } from '../types';
+import { AgentRuntimeErrorType } from '../error';
+import { ChatCompletionErrorPayload, ModelProvider } from '../types';
 import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
 
 export interface InfiniAIModelCard {
@@ -10,10 +10,42 @@ export interface InfiniAIModelCard {
 
 export const LobeInfiniAI = LobeOpenAICompatibleFactory({
   baseURL: 'https://cloud.infini-ai.com/maas/v1',
+  chatCompletion: {
+    handleError(error): Omit<ChatCompletionErrorPayload, 'provider'> | undefined {
+      let errorResponse: Response | undefined;
+      if (error instanceof Response) {
+        errorResponse = error;
+      } else if ('status' in (error as any)) {
+        errorResponse = error as Response;
+      }
+      if (errorResponse) {
+        if (errorResponse.status === 401) {
+          return {
+            error,
+            errorType: AgentRuntimeErrorType.InvalidProviderAPIKey,
+          };
+        }
+
+        if (errorResponse.status === 429) {
+          return {
+            error,
+            errorType: AgentRuntimeErrorType.QuotaLimitReached,
+          };
+        }
+      }
+      return {
+        error,
+      };
+    },
+  },
+  debug: {
+    chatCompletion: () => process.env.DEBUG_INFINIAI_CHAT_COMPLETION === '1',
+  },
   models: async ({ client }) => {
     const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
 
     const reasoningKeywords = ['deepseek-r1', 'qwq'];
+    const visionKeywords = ['qwen2.5-vl'];
 
     const modelsPage = (await client.models.list()) as any;
     const modelList: InfiniAIModelCard[] = modelsPage.data;
@@ -34,7 +66,10 @@ export const LobeInfiniAI = LobeOpenAICompatibleFactory({
             reasoningKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
             knownModel?.abilities?.reasoning ||
             false,
-          vision: knownModel?.abilities?.vision || false,
+          vision:
+            visionKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
+            knownModel?.abilities?.vision ||
+            false,
         };
       })
       .filter(Boolean) as ChatModelCard[];

package/src/utils/format.ts

@@ -117,7 +117,7 @@ export const formatPrice = (price: number, fractionDigits: number = 2) => {
 };
 
 export const formatPriceByCurrency = (price?: number, currency?: ModelPriceCurrency) => {
-  if (!price) return '-';
+  if (!price && price !== 0) return '-';
 
   if (currency === 'CNY') {
     return formatPrice(price / USD_TO_CNY);

package/src/utils/parseModels.test.ts

@@ -86,6 +86,20 @@ describe('parseModelString', () => {
       });
     });
 
+    it('token and image output', () => {
+      const result = parseModelString('gemini-2.0-flash-exp-image-generation=Gemini 2.0 Flash (Image Generation) Experimental<32768:imageOutput>');
+
+      expect(result.add[0]).toEqual({
+        displayName: 'Gemini 2.0 Flash (Image Generation) Experimental',
+        abilities: {
+          imageOutput: true,
+        },
+        id: 'gemini-2.0-flash-exp-image-generation',
+        contextWindowTokens: 32_768,
+        type: 'chat',
+      });
+    });
+
     it('multi models', () => {
       const result = parseModelString(
         'gemini-1.5-flash-latest=Gemini 1.5 Flash<16000:vision>,gpt-4-all=ChatGPT Plus<128000:fc:vision:file>',

package/src/utils/parseModels.ts

@@ -84,6 +84,10 @@ export const parseModelString = (modelString: string = '', withDeploymentName =
             model.abilities!.search = true;
             break;
           }
+          case 'imageOutput': {
+            model.abilities!.imageOutput = true;
+            break;
+          }
           default: {
             console.warn(`Unknown capability: ${capability}`);
           }