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

package/dist/providers/anthropic/index.mjs

@@ -702,14 +702,45 @@ var AnthropicAdapter = class {
         ) : {},
         required: action.parameters ? Object.entries(action.parameters).filter(([, param]) => param.required).map(([key]) => key) : []
       }
-    }));
+    })) || [];
+    const webSearchConfig = request.webSearch ?? this.config.webSearch;
+    let serverToolConfiguration;
+    if (webSearchConfig) {
+      tools.push({
+        type: "web_search_20260209",
+        name: "web_search",
+        allowed_callers: ["direct"]
+      });
+      const wsConfig = typeof webSearchConfig === "object" ? webSearchConfig : {};
+      const webSearchToolConfig = {};
+      if (wsConfig.maxUses !== void 0) {
+        webSearchToolConfig.max_uses = wsConfig.maxUses;
+      }
+      if (wsConfig.allowedDomains && wsConfig.allowedDomains.length > 0) {
+        webSearchToolConfig.allowed_domains = wsConfig.allowedDomains;
+      }
+      if (wsConfig.blockedDomains && wsConfig.blockedDomains.length > 0) {
+        webSearchToolConfig.blocked_domains = wsConfig.blockedDomains;
+      }
+      if (wsConfig.userLocation) {
+        webSearchToolConfig.user_location = wsConfig.userLocation;
+      }
+      if (Object.keys(webSearchToolConfig).length > 0) {
+        serverToolConfiguration = {
+          web_search: webSearchToolConfig
+        };
+      }
+    }
     const options = {
       model: request.config?.model || this.model,
       max_tokens: request.config?.maxTokens || this.config.maxTokens || 4096,
       system: systemMessage,
       messages,
-      tools: tools?.length ? tools : void 0
+      tools: tools.length ? tools : void 0
     };
+    if (serverToolConfiguration) {
+      options.server_tool_configuration = serverToolConfiguration;
+    }
     if (this.config.thinking?.type === "enabled") {
       options.thinking = {
         type: "enabled",
@@ -765,6 +796,8 @@ var AnthropicAdapter = class {
       const stream = await client.messages.stream(options);
       let currentToolUse = null;
       let isInThinkingBlock = false;
+      const collectedCitations = [];
+      let citationIndex = 0;
       let usage;
       for await (const event of stream) {
         if (request.signal?.aborted) {
@@ -793,11 +826,30 @@ var AnthropicAdapter = class {
                 name: event.content_block.name,
                 input: ""
               };
-              yield {
-                type: "action:start",
-                id: currentToolUse.id,
-                name: currentToolUse.name
-              };
+              if (currentToolUse.name !== "web_search") {
+                yield {
+                  type: "action:start",
+                  id: currentToolUse.id,
+                  name: currentToolUse.name
+                };
+              }
+            } else if (event.content_block.type === "web_search_tool_result") {
+              const webSearchResult = event.content_block;
+              if (webSearchResult.content && Array.isArray(webSearchResult.content)) {
+                for (const result of webSearchResult.content) {
+                  if (result.type === "web_search_result" && result.url) {
+                    citationIndex++;
+                    const domain = extractDomain(result.url);
+                    collectedCitations.push({
+                      index: citationIndex,
+                      url: result.url,
+                      title: result.title || domain,
+                      domain,
+                      favicon: domain ? `https://www.google.com/s2/favicons?domain=${domain}&sz=32` : void 0
+                    });
+                  }
+                }
+              }
             } else if (event.content_block.type === "thinking") {
               isInThinkingBlock = true;
               yield { type: "thinking:start" };
@@ -806,6 +858,20 @@ var AnthropicAdapter = class {
           case "content_block_delta":
             if (event.delta.type === "text_delta") {
               yield { type: "message:delta", content: event.delta.text };
+            } else if (event.delta.type === "citations_delta") {
+              const citationsDelta = event.delta;
+              if (citationsDelta.citation?.url) {
+                citationIndex++;
+                const domain = extractDomain(citationsDelta.citation.url);
+                collectedCitations.push({
+                  index: citationIndex,
+                  url: citationsDelta.citation.url,
+                  title: citationsDelta.citation.title || domain,
+                  citedText: citationsDelta.citation.cited_text,
+                  domain,
+                  favicon: domain ? `https://www.google.com/s2/favicons?domain=${domain}&sz=32` : void 0
+                });
+              }
             } else if (event.delta.type === "thinking_delta") {
               yield { type: "thinking:delta", content: event.delta.thinking };
             } else if (event.delta.type === "input_json_delta" && currentToolUse) {
@@ -814,11 +880,18 @@ var AnthropicAdapter = class {
             break;
           case "content_block_stop":
             if (currentToolUse) {
-              yield {
-                type: "action:args",
-                id: currentToolUse.id,
-                args: currentToolUse.input
-              };
+              if (currentToolUse.name !== "web_search") {
+                yield {
+                  type: "action:args",
+                  id: currentToolUse.id,
+                  args: currentToolUse.input
+                };
+                yield {
+                  type: "action:end",
+                  id: currentToolUse.id,
+                  name: currentToolUse.name
+                };
+              }
               currentToolUse = null;
             }
             if (isInThinkingBlock) {
@@ -830,6 +903,10 @@ var AnthropicAdapter = class {
             break;
         }
       }
+      if (collectedCitations.length > 0) {
+        const uniqueCitations = deduplicateCitations(collectedCitations);
+        yield { type: "citation", citations: uniqueCitations };
+      }
       yield { type: "message:end" };
       yield { type: "done", usage };
     } catch (error) {
@@ -841,6 +918,25 @@ var AnthropicAdapter = 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 createAnthropicAdapter(config) {
   return new AnthropicAdapter(config);
 }