@yourgpt/llm-sdk 2.0.1 → 2.0.2-beta.2

package/dist/providers/openrouter/index.js

@@ -1,166 +1,6 @@
 'use strict';
 
 // src/providers/openrouter/provider.ts
-var OPENROUTER_MODELS = {
-  // OpenAI
-  "openai/gpt-4o": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 128e3
-  },
-  "openai/gpt-4o-mini": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 128e3
-  },
-  "openai/gpt-4-turbo": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 128e3
-  },
-  "openai/o1": {
-    vision: true,
-    tools: false,
-    jsonMode: false,
-    maxTokens: 128e3
-  },
-  "openai/o1-mini": {
-    vision: true,
-    tools: false,
-    jsonMode: false,
-    maxTokens: 128e3
-  },
-  // Anthropic
-  "anthropic/claude-3.5-sonnet": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 2e5
-  },
-  "anthropic/claude-3.5-sonnet-20241022": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 2e5
-  },
-  "anthropic/claude-3-opus": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 2e5
-  },
-  "anthropic/claude-3-sonnet": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 2e5
-  },
-  "anthropic/claude-3-haiku": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 2e5
-  },
-  // Google
-  "google/gemini-pro-1.5": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 1e6
-  },
-  "google/gemini-flash-1.5": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 1e6
-  },
-  "google/gemini-2.0-flash-exp": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 1e6
-  },
-  // Meta Llama
-  "meta-llama/llama-3.1-405b-instruct": {
-    vision: false,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 131072
-  },
-  "meta-llama/llama-3.1-70b-instruct": {
-    vision: false,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 131072
-  },
-  "meta-llama/llama-3.1-8b-instruct": {
-    vision: false,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 131072
-  },
-  "meta-llama/llama-3.2-90b-vision-instruct": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 131072
-  },
-  // Mistral
-  "mistralai/mistral-large": {
-    vision: false,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 128e3
-  },
-  "mistralai/mixtral-8x7b-instruct": {
-    vision: false,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 32768
-  },
-  "mistralai/mistral-nemo": {
-    vision: false,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 128e3
-  },
-  // DeepSeek
-  "deepseek/deepseek-chat": {
-    vision: false,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 64e3
-  },
-  "deepseek/deepseek-coder": {
-    vision: false,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 64e3
-  },
-  // Qwen
-  "qwen/qwen-2.5-72b-instruct": {
-    vision: false,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 131072
-  },
-  "qwen/qwen-2-vl-72b-instruct": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 32768
-  },
-  // OpenRouter Auto (magic model)
-  "openrouter/auto": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 128e3
-  }
-};
 var DEFAULT_MODEL_CONFIG = {
   vision: true,
   tools: true,
@@ -189,7 +29,7 @@ function openrouter(modelId, options = {}) {
     }
     return client;
   }
-  const modelConfig = OPENROUTER_MODELS[modelId] ?? DEFAULT_MODEL_CONFIG;
+  const modelConfig = DEFAULT_MODEL_CONFIG;
   return {
     provider: "openrouter",
     modelId,
@@ -388,6 +228,29 @@ function formatMessagesForOpenRouter(messages) {
     }
   });
 }
+async function fetchOpenRouterModels(apiKey) {
+  const headers = {
+    "Content-Type": "application/json"
+  };
+  if (apiKey) {
+    headers["Authorization"] = `Bearer ${apiKey}`;
+  }
+  const response = await fetch("https://openrouter.ai/api/v1/models", {
+    headers
+  });
+  if (!response.ok) {
+    throw new Error(`Failed to fetch models: ${response.statusText}`);
+  }
+  const data = await response.json();
+  return data.data || [];
+}
+async function searchOpenRouterModels(query, apiKey) {
+  const models = await fetchOpenRouterModels(apiKey);
+  const lowerQuery = query.toLowerCase();
+  return models.filter(
+    (model) => model.id.toLowerCase().includes(lowerQuery) || model.name.toLowerCase().includes(lowerQuery)
+  );
+}
 
 // src/core/utils.ts
 function generateId(prefix = "id") {
@@ -586,29 +449,60 @@ var OpenAIAdapter = class {
         request.systemPrompt
       );
     }
-    const tools = request.actions?.length ? formatTools(request.actions) : void 0;
+    const tools = request.actions?.length ? formatTools(request.actions) : [];
+    const webSearchConfig = request.webSearch ?? this.config.webSearch;
+    if (webSearchConfig) {
+      const webSearchTool = {
+        type: "web_search_preview"
+      };
+      const wsConfig = typeof webSearchConfig === "object" ? webSearchConfig : {};
+      if (wsConfig.userLocation) {
+        webSearchTool.search_context_size = "medium";
+      }
+      tools.push(webSearchTool);
+    }
     const messageId = generateMessageId();
     yield { type: "message:start", id: messageId };
     try {
       const stream = await client.chat.completions.create({
         model: request.config?.model || this.model,
         messages,
-        tools,
+        tools: tools.length > 0 ? tools : void 0,
         temperature: request.config?.temperature ?? this.config.temperature,
         max_tokens: request.config?.maxTokens ?? this.config.maxTokens,
         stream: true,
         stream_options: { include_usage: true }
       });
       let currentToolCall = null;
+      const collectedCitations = [];
+      let citationIndex = 0;
       let usage;
       for await (const chunk of stream) {
         if (request.signal?.aborted) {
           break;
         }
         const delta = chunk.choices[0]?.delta;
+        const choice = chunk.choices[0];
         if (delta?.content) {
           yield { type: "message:delta", content: delta.content };
         }
+        const annotations = delta?.annotations;
+        if (annotations && annotations.length > 0) {
+          for (const annotation of annotations) {
+            if (annotation.type === "url_citation" && annotation.url_citation?.url) {
+              citationIndex++;
+              const url = annotation.url_citation.url;
+              const domain = extractDomain(url);
+              collectedCitations.push({
+                index: citationIndex,
+                url,
+                title: annotation.url_citation.title || domain,
+                domain,
+                favicon: domain ? `https://www.google.com/s2/favicons?domain=${domain}&sz=32` : void 0
+              });
+            }
+          }
+        }
         if (delta?.tool_calls) {
           for (const toolCall of delta.tool_calls) {
             if (toolCall.id) {
@@ -641,7 +535,7 @@ var OpenAIAdapter = class {
             total_tokens: chunk.usage.total_tokens
           };
         }
-        if (chunk.choices[0]?.finish_reason) {
+        if (choice?.finish_reason) {
           if (currentToolCall) {
             yield {
               type: "action:args",
@@ -651,6 +545,10 @@ var OpenAIAdapter = class {
           }
         }
       }
+      if (collectedCitations.length > 0) {
+        const uniqueCitations = deduplicateCitations(collectedCitations);
+        yield { type: "citation", citations: uniqueCitations };
+      }
       yield { type: "message:end" };
       yield { type: "done", usage };
     } catch (error) {
@@ -662,6 +560,25 @@ var OpenAIAdapter = class {
     }
   }
 };
+function extractDomain(url) {
+  try {
+    const parsed = new URL(url);
+    return parsed.hostname;
+  } catch {
+    return "";
+  }
+}
+function deduplicateCitations(citations) {
+  const seen = /* @__PURE__ */ new Map();
+  let index = 0;
+  for (const citation of citations) {
+    if (!seen.has(citation.url)) {
+      index++;
+      seen.set(citation.url, { ...citation, index });
+    }
+  }
+  return Array.from(seen.values());
+}
 function createOpenAIAdapter(config) {
   return new OpenAIAdapter(config);
 }
@@ -683,111 +600,6 @@ function createCallableProvider(providerFn, properties) {
 }
 
 // src/providers/openrouter/index.ts
-var OPENROUTER_MODELS2 = {
-  // OpenAI
-  "openai/gpt-4o": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 128e3
-  },
-  "openai/gpt-4o-mini": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 128e3
-  },
-  "openai/gpt-4-turbo": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 128e3
-  },
-  "openai/o1": {
-    vision: true,
-    tools: false,
-    jsonMode: false,
-    maxTokens: 128e3
-  },
-  // Anthropic
-  "anthropic/claude-3.5-sonnet": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 2e5
-  },
-  "anthropic/claude-3.5-sonnet-20241022": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 2e5
-  },
-  "anthropic/claude-3-opus": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 2e5
-  },
-  "anthropic/claude-3-haiku": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 2e5
-  },
-  // Google
-  "google/gemini-pro-1.5": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 1e6
-  },
-  "google/gemini-flash-1.5": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 1e6
-  },
-  // Meta Llama
-  "meta-llama/llama-3.1-405b-instruct": {
-    vision: false,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 131072
-  },
-  "meta-llama/llama-3.1-70b-instruct": {
-    vision: false,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 131072
-  },
-  // Mistral
-  "mistralai/mistral-large": {
-    vision: false,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 128e3
-  },
-  "mistralai/mixtral-8x7b-instruct": {
-    vision: false,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 32768
-  },
-  // DeepSeek
-  "deepseek/deepseek-chat": {
-    vision: false,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 64e3
-  },
-  // OpenRouter Auto (magic model - picks best for your prompt)
-  "openrouter/auto": {
-    vision: true,
-    tools: true,
-    jsonMode: true,
-    maxTokens: 128e3
-  }
-};
 var DEFAULT_CAPABILITIES = {
   vision: true,
   tools: true,
@@ -805,24 +617,24 @@ function createOpenRouter(config = {}) {
     });
   };
   const getCapabilities = (modelId) => {
-    const model = OPENROUTER_MODELS2[modelId] ?? DEFAULT_CAPABILITIES;
     return {
-      supportsVision: model.vision,
-      supportsTools: model.tools,
+      supportsVision: DEFAULT_CAPABILITIES.vision,
+      supportsTools: DEFAULT_CAPABILITIES.tools,
       supportsThinking: false,
       supportsStreaming: true,
       supportsPDF: false,
       supportsAudio: false,
       supportsVideo: false,
-      maxTokens: model.maxTokens,
-      supportedImageTypes: model.vision ? ["image/png", "image/jpeg", "image/gif", "image/webp"] : [],
-      supportsJsonMode: model.jsonMode,
+      maxTokens: DEFAULT_CAPABILITIES.maxTokens,
+      supportedImageTypes: ["image/png", "image/jpeg", "image/gif", "image/webp"] ,
+      supportsJsonMode: DEFAULT_CAPABILITIES.jsonMode,
       supportsSystemMessages: true
     };
   };
   return createCallableProvider(providerFn, {
     name: "openrouter",
-    supportedModels: Object.keys(OPENROUTER_MODELS2),
+    supportedModels: [],
+    // Use fetchOpenRouterModels() to get live model list
     getCapabilities
   });
 }
@@ -831,6 +643,8 @@ var createOpenRouterProvider = createOpenRouter;
 exports.createOpenRouter = createOpenRouter;
 exports.createOpenRouterModel = openrouter;
 exports.createOpenRouterProvider = createOpenRouterProvider;
+exports.fetchOpenRouterModels = fetchOpenRouterModels;
 exports.openrouter = openrouter;
+exports.searchOpenRouterModels = searchOpenRouterModels;
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map