@ai-sdk/anthropic 2.0.0-alpha.10 → 2.0.0-alpha.12

package/dist/internal/index.mjs

@@ -38,6 +38,29 @@ var webSearchLocationSchema = z2.object({
   country: z2.string(),
   timezone: z2.string().optional()
 });
+var anthropicFilePartProviderOptions = z2.object({
+  /**
+   * Citation configuration for this document.
+   * When enabled, this document will generate citations in the response.
+   */
+  citations: z2.object({
+    /**
+     * Enable citations for this document
+     */
+    enabled: z2.boolean()
+  }).optional(),
+  /**
+   * Custom title for the document.
+   * If not provided, the filename will be used.
+   */
+  title: z2.string().optional(),
+  /**
+   * Context about the document that will be passed to the model
+   * but not used towards cited content.
+   * Useful for storing document metadata as text or stringified JSON.
+   */
+  context: z2.string().optional()
+});
 var anthropicProviderOptions = z2.object({
   /**
   Include reasoning content in requests sent to the model. Defaults to `true`.
@@ -219,7 +242,7 @@ async function convertToAnthropicMessagesPrompt({
   sendReasoning,
   warnings
 }) {
-  var _a, _b, _c;
+  var _a, _b, _c, _d;
   const betas = /* @__PURE__ */ new Set();
   const blocks = groupIntoBlocks(prompt);
   let system = void 0;
@@ -230,6 +253,26 @@ async function convertToAnthropicMessagesPrompt({
     const cacheControlValue = (_a2 = anthropic == null ? void 0 : anthropic.cacheControl) != null ? _a2 : anthropic == null ? void 0 : anthropic.cache_control;
     return cacheControlValue;
   }
+  async function shouldEnableCitations(providerMetadata) {
+    var _a2, _b2;
+    const anthropicOptions = await parseProviderOptions({
+      provider: "anthropic",
+      providerOptions: providerMetadata,
+      schema: anthropicFilePartProviderOptions
+    });
+    return (_b2 = (_a2 = anthropicOptions == null ? void 0 : anthropicOptions.citations) == null ? void 0 : _a2.enabled) != null ? _b2 : false;
+  }
+  async function getDocumentMetadata(providerMetadata) {
+    const anthropicOptions = await parseProviderOptions({
+      provider: "anthropic",
+      providerOptions: providerMetadata,
+      schema: anthropicFilePartProviderOptions
+    });
+    return {
+      title: anthropicOptions == null ? void 0 : anthropicOptions.title,
+      context: anthropicOptions == null ? void 0 : anthropicOptions.context
+    };
+  }
   for (let i = 0; i < blocks.length; i++) {
     const block = blocks[i];
     const isLastBlock = i === blocks.length - 1;
@@ -283,6 +326,12 @@ async function convertToAnthropicMessagesPrompt({
                       });
                     } else if (part.mediaType === "application/pdf") {
                       betas.add("pdfs-2024-09-25");
+                      const enableCitations = await shouldEnableCitations(
+                        part.providerOptions
+                      );
+                      const metadata = await getDocumentMetadata(
+                        part.providerOptions
+                      );
                       anthropicContent.push({
                         type: "document",
                         source: part.data instanceof URL ? {
@@ -293,6 +342,11 @@ async function convertToAnthropicMessagesPrompt({
                           media_type: "application/pdf",
                           data: convertToBase64(part.data)
                         },
+                        title: (_b = metadata.title) != null ? _b : part.filename,
+                        ...metadata.context && { context: metadata.context },
+                        ...enableCitations && {
+                          citations: { enabled: true }
+                        },
                         cache_control: cacheControl
                       });
                     } else {
@@ -310,7 +364,7 @@ async function convertToAnthropicMessagesPrompt({
               for (let i2 = 0; i2 < content.length; i2++) {
                 const part = content[i2];
                 const isLastPart = i2 === content.length - 1;
-                const cacheControl = (_b = getCacheControl(part.providerOptions)) != null ? _b : isLastPart ? getCacheControl(message.providerOptions) : void 0;
+                const cacheControl = (_c = getCacheControl(part.providerOptions)) != null ? _c : isLastPart ? getCacheControl(message.providerOptions) : void 0;
                 const toolResultContent = part.content != null ? part.content.map((part2) => {
                   var _a2;
                   switch (part2.type) {
@@ -360,7 +414,7 @@ async function convertToAnthropicMessagesPrompt({
           for (let k = 0; k < content.length; k++) {
             const part = content[k];
             const isLastContentPart = k === content.length - 1;
-            const cacheControl = (_c = getCacheControl(part.providerOptions)) != null ? _c : isLastContentPart ? getCacheControl(message.providerOptions) : void 0;
+            const cacheControl = (_d = getCacheControl(part.providerOptions)) != null ? _d : isLastContentPart ? getCacheControl(message.providerOptions) : void 0;
             switch (part.type) {
               case "text": {
                 anthropicContent.push({
@@ -509,6 +563,40 @@ function mapAnthropicStopReason({
 }
 
 // src/anthropic-messages-language-model.ts
+function processPageLocationCitation(citation, citationDocuments, generateId2, onSource) {
+  if (citation.type === "page_location") {
+    const source = createCitationSource(
+      citation,
+      citationDocuments,
+      generateId2
+    );
+    if (source) {
+      onSource(source);
+    }
+  }
+}
+function createCitationSource(citation, citationDocuments, generateId2) {
+  var _a;
+  const documentInfo = citationDocuments[citation.document_index];
+  if (!documentInfo) {
+    return null;
+  }
+  return {
+    type: "source",
+    sourceType: "document",
+    id: generateId2(),
+    mediaType: documentInfo.mediaType,
+    title: (_a = citation.document_title) != null ? _a : documentInfo.title,
+    filename: documentInfo.filename,
+    providerMetadata: {
+      anthropic: {
+        citedText: citation.cited_text,
+        startPageNumber: citation.start_page_number,
+        endPageNumber: citation.end_page_number
+      }
+    }
+  };
+}
 var AnthropicMessagesLanguageModel = class {
   constructor(modelId, config) {
     this.specificationVersion = "v2";
@@ -706,9 +794,29 @@ var AnthropicMessagesLanguageModel = class {
     var _a, _b, _c;
     return (_c = (_b = (_a = this.config).transformRequestBody) == null ? void 0 : _b.call(_a, args)) != null ? _c : args;
   }
+  extractCitationDocuments(prompt) {
+    const isCitationEnabled = (part) => {
+      var _a, _b;
+      const anthropic = (_a = part.providerOptions) == null ? void 0 : _a.anthropic;
+      const citationsConfig = anthropic == null ? void 0 : anthropic.citations;
+      return (_b = citationsConfig == null ? void 0 : citationsConfig.enabled) != null ? _b : false;
+    };
+    return prompt.filter((message) => message.role === "user").flatMap((message) => message.content).filter(
+      (part) => part.type === "file" && part.mediaType === "application/pdf" && isCitationEnabled(part)
+    ).map((part) => {
+      var _a;
+      const filePart = part;
+      return {
+        title: (_a = filePart.filename) != null ? _a : "Untitled Document",
+        filename: filePart.filename,
+        mediaType: filePart.mediaType
+      };
+    });
+  }
   async doGenerate(options) {
     var _a, _b, _c, _d, _e;
     const { args, warnings, betas, jsonResponseTool } = await this.getArgs(options);
+    const citationDocuments = this.extractCitationDocuments(options.prompt);
     const {
       responseHeaders,
       value: response,
@@ -730,6 +838,16 @@ var AnthropicMessagesLanguageModel = class {
         case "text": {
           if (jsonResponseTool == null) {
             content.push({ type: "text", text: part.text });
+            if (part.citations) {
+              for (const citation of part.citations) {
+                processPageLocationCitation(
+                  citation,
+                  citationDocuments,
+                  this.generateId,
+                  (source) => content.push(source)
+                );
+              }
+            }
           }
           break;
         }
@@ -836,6 +954,7 @@ var AnthropicMessagesLanguageModel = class {
   }
   async doStream(options) {
     const { args, warnings, betas, jsonResponseTool } = await this.getArgs(options);
+    const citationDocuments = this.extractCitationDocuments(options.prompt);
     const body = { ...args, stream: true };
     const { responseHeaders, value: response } = await postJsonToApi({
       url: this.buildRequestUrl(true),
@@ -867,6 +986,9 @@ var AnthropicMessagesLanguageModel = class {
           },
           transform(chunk, controller) {
             var _a, _b, _c, _d, _e, _f, _g;
+            if (options.includeRawChunks) {
+              controller.enqueue({ type: "raw", rawValue: chunk.rawValue });
+            }
             if (!chunk.success) {
               controller.enqueue({ type: "error", error: chunk.error });
               return;
@@ -1020,6 +1142,13 @@ var AnthropicMessagesLanguageModel = class {
                     return;
                   }
                   case "citations_delta": {
+                    const citation = value.delta.citation;
+                    processPageLocationCitation(
+                      citation,
+                      citationDocuments,
+                      generateId2,
+                      (source) => controller.enqueue(source)
+                    );
                     return;
                   }
                   default: {
@@ -1088,7 +1217,26 @@ var anthropicMessagesResponseSchema = z3.object({
     z3.discriminatedUnion("type", [
       z3.object({
         type: z3.literal("text"),
-        text: z3.string()
+        text: z3.string(),
+        citations: z3.array(
+          z3.discriminatedUnion("type", [
+            z3.object({
+              type: z3.literal("web_search_result_location"),
+              cited_text: z3.string(),
+              url: z3.string(),
+              title: z3.string(),
+              encrypted_index: z3.string()
+            }),
+            z3.object({
+              type: z3.literal("page_location"),
+              cited_text: z3.string(),
+              document_index: z3.number(),
+              document_title: z3.string().nullable(),
+              start_page_number: z3.number(),
+              end_page_number: z3.number()
+            })
+          ])
+        ).optional()
       }),
       z3.object({
         type: z3.literal("thinking"),
@@ -1227,13 +1375,23 @@ var anthropicMessagesChunkSchema = z3.discriminatedUnion("type", [
       }),
       z3.object({
         type: z3.literal("citations_delta"),
-        citation: z3.object({
-          type: z3.literal("web_search_result_location"),
-          cited_text: z3.string(),
-          url: z3.string(),
-          title: z3.string(),
-          encrypted_index: z3.string()
-        })
+        citation: z3.discriminatedUnion("type", [
+          z3.object({
+            type: z3.literal("web_search_result_location"),
+            cited_text: z3.string(),
+            url: z3.string(),
+            title: z3.string(),
+            encrypted_index: z3.string()
+          }),
+          z3.object({
+            type: z3.literal("page_location"),
+            cited_text: z3.string(),
+            document_index: z3.number(),
+            document_title: z3.string().nullable(),
+            start_page_number: z3.number(),
+            end_page_number: z3.number()
+          })
+        ])
       })
     ])
   }),