@lobehub/chat 1.68.7 → 1.68.8

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

@@ -2,7 +2,12 @@ import type { ChatModelCard } from '@/types/llm';
 
 import { ModelProvider } from '../types';
 import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
-import { OpenRouterModelCard } from './type';
+import { OpenRouterModelCard, OpenRouterModelExtraInfo } from './type';
+
+const formatPrice = (price: string) => {
+  if (price === '-1') return undefined;
+  return Number((Number(price) * 1e6).toPrecision(5));
+};
 
 export const LobeOpenRouterAI = LobeOpenAICompatibleFactory({
   baseURL: 'https://openrouter.ai/api/v1',
@@ -28,8 +33,6 @@ export const LobeOpenRouterAI = LobeOpenAICompatibleFactory({
   models: async ({ client }) => {
     const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
 
-    const visionKeywords = ['qwen/qvq', 'vision'];
-
     const reasoningKeywords = [
       'deepseek/deepseek-r1',
       'openai/o1',
@@ -42,11 +45,21 @@ export const LobeOpenRouterAI = LobeOpenAICompatibleFactory({
     const modelsPage = (await client.models.list()) as any;
     const modelList: OpenRouterModelCard[] = modelsPage.data;
 
+    const response = await fetch('https://openrouter.ai/api/frontend/models');
+    const modelsExtraInfo: OpenRouterModelExtraInfo[] = [];
+    if (response.ok) {
+      const data = await response.json();
+      modelsExtraInfo.push(...data['data']);
+    }
+
     return modelList
       .map((model) => {
         const knownModel = LOBE_DEFAULT_MODEL_LIST.find(
           (m) => model.id.toLowerCase() === m.id.toLowerCase(),
         );
+        const extraInfo = modelsExtraInfo.find(
+          (m) => m.slug.toLowerCase() === model.id.toLowerCase(),
+        );
 
         return {
           contextWindowTokens: model.context_length,
@@ -56,6 +69,7 @@ export const LobeOpenRouterAI = LobeOpenAICompatibleFactory({
           functionCall:
             model.description.includes('function calling') ||
             model.description.includes('tools') ||
+            extraInfo?.endpoint?.supports_tool_parameters ||
             knownModel?.abilities?.functionCall ||
             false,
           id: model.id,
@@ -63,16 +77,18 @@ export const LobeOpenRouterAI = LobeOpenAICompatibleFactory({
             typeof model.top_provider.max_completion_tokens === 'number'
               ? model.top_provider.max_completion_tokens
               : undefined,
+          pricing: {
+            input: formatPrice(model.pricing.prompt),
+            output: formatPrice(model.pricing.completion),
+          },
           reasoning:
             reasoningKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
+            extraInfo?.endpoint?.supports_reasoning ||
             knownModel?.abilities?.reasoning ||
             false,
+          releasedAt: new Date(model.created * 1000).toISOString().split('T')[0],
           vision:
-            model.description.includes('vision') ||
-            model.description.includes('multimodal') ||
-            visionKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
-            knownModel?.abilities?.vision ||
-            false,
+            model.architecture.modality.includes('image') || knownModel?.abilities?.vision || false,
         };
       })
       .filter(Boolean) as ChatModelCard[];

package/src/libs/agent-runtime/openrouter/type.ts

@@ -19,6 +19,7 @@ interface ModelTopProvider {
 export interface OpenRouterModelCard {
   architecture: ModelArchitecture;
   context_length: number;
+  created: number;
   description: string;
   id: string;
   name: string;
@@ -26,3 +27,13 @@ export interface OpenRouterModelCard {
   pricing: ModelPricing;
   top_provider: ModelTopProvider;
 }
+
+interface OpenRouterModelEndpoint {
+  supports_reasoning?: boolean;
+  supports_tool_parameters?: boolean;
+}
+
+export interface OpenRouterModelExtraInfo {
+  endpoint?: OpenRouterModelEndpoint;
+  slug: string;
+}

package/src/types/llm.ts

@@ -49,7 +49,7 @@ export interface ChatModelCard {
   reasoning?: boolean;
 
   /**
-   * whether model is legacy (deprecated but not removed yet)
+   * the date when model is released
    */
   releasedAt?: string;