@ai-sdk/anthropic 2.0.0-alpha.11 → 2.0.0-alpha.13

package/dist/internal/index.mjs

@@ -237,12 +237,28 @@ import {
   UnsupportedFunctionalityError as UnsupportedFunctionalityError2
 } from "@ai-sdk/provider";
 import { convertToBase64, parseProviderOptions } from "@ai-sdk/provider-utils";
+function convertToString(data) {
+  if (typeof data === "string") {
+    return Buffer.from(data, "base64").toString("utf-8");
+  }
+  if (data instanceof Uint8Array) {
+    return new TextDecoder().decode(data);
+  }
+  if (data instanceof URL) {
+    throw new UnsupportedFunctionalityError2({
+      functionality: "URL-based text documents are not supported for citations"
+    });
+  }
+  throw new UnsupportedFunctionalityError2({
+    functionality: `unsupported data type for text documents: ${typeof data}`
+  });
+}
 async function convertToAnthropicMessagesPrompt({
   prompt,
   sendReasoning,
   warnings
 }) {
-  var _a, _b, _c, _d;
+  var _a, _b, _c, _d, _e;
   const betas = /* @__PURE__ */ new Set();
   const blocks = groupIntoBlocks(prompt);
   let system = void 0;
@@ -349,6 +365,30 @@ async function convertToAnthropicMessagesPrompt({
                         },
                         cache_control: cacheControl
                       });
+                    } else if (part.mediaType === "text/plain") {
+                      const enableCitations = await shouldEnableCitations(
+                        part.providerOptions
+                      );
+                      const metadata = await getDocumentMetadata(
+                        part.providerOptions
+                      );
+                      anthropicContent.push({
+                        type: "document",
+                        source: part.data instanceof URL ? {
+                          type: "url",
+                          url: part.data.toString()
+                        } : {
+                          type: "text",
+                          media_type: "text/plain",
+                          data: convertToString(part.data)
+                        },
+                        title: (_c = metadata.title) != null ? _c : part.filename,
+                        ...metadata.context && { context: metadata.context },
+                        ...enableCitations && {
+                          citations: { enabled: true }
+                        },
+                        cache_control: cacheControl
+                      });
                     } else {
                       throw new UnsupportedFunctionalityError2({
                         functionality: `media type: ${part.mediaType}`
@@ -364,7 +404,7 @@ async function convertToAnthropicMessagesPrompt({
               for (let i2 = 0; i2 < content.length; i2++) {
                 const part = content[i2];
                 const isLastPart = i2 === content.length - 1;
-                const cacheControl = (_c = getCacheControl(part.providerOptions)) != null ? _c : isLastPart ? getCacheControl(message.providerOptions) : void 0;
+                const cacheControl = (_d = getCacheControl(part.providerOptions)) != null ? _d : isLastPart ? getCacheControl(message.providerOptions) : void 0;
                 const toolResultContent = part.content != null ? part.content.map((part2) => {
                   var _a2;
                   switch (part2.type) {
@@ -414,7 +454,7 @@ async function convertToAnthropicMessagesPrompt({
           for (let k = 0; k < content.length; k++) {
             const part = content[k];
             const isLastContentPart = k === content.length - 1;
-            const cacheControl = (_d = getCacheControl(part.providerOptions)) != null ? _d : isLastContentPart ? getCacheControl(message.providerOptions) : void 0;
+            const cacheControl = (_e = getCacheControl(part.providerOptions)) != null ? _e : isLastContentPart ? getCacheControl(message.providerOptions) : void 0;
             switch (part.type) {
               case "text": {
                 anthropicContent.push({
@@ -563,8 +603,42 @@ function mapAnthropicStopReason({
 }
 
 // src/anthropic-messages-language-model.ts
-function processPageLocationCitation(citation, citationDocuments, generateId2, onSource) {
-  if (citation.type === "page_location") {
+var citationSchemas = {
+  webSearchResult: z3.object({
+    type: z3.literal("web_search_result_location"),
+    cited_text: z3.string(),
+    url: z3.string(),
+    title: z3.string(),
+    encrypted_index: z3.string()
+  }),
+  pageLocation: 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()
+  }),
+  charLocation: z3.object({
+    type: z3.literal("char_location"),
+    cited_text: z3.string(),
+    document_index: z3.number(),
+    document_title: z3.string().nullable(),
+    start_char_index: z3.number(),
+    end_char_index: z3.number()
+  })
+};
+var citationSchema = z3.discriminatedUnion("type", [
+  citationSchemas.webSearchResult,
+  citationSchemas.pageLocation,
+  citationSchemas.charLocation
+]);
+var documentCitationSchema = z3.discriminatedUnion("type", [
+  citationSchemas.pageLocation,
+  citationSchemas.charLocation
+]);
+function processCitation(citation, citationDocuments, generateId2, onSource) {
+  if (citation.type === "page_location" || citation.type === "char_location") {
     const source = createCitationSource(
       citation,
       citationDocuments,
@@ -581,6 +655,15 @@ function createCitationSource(citation, citationDocuments, generateId2) {
   if (!documentInfo) {
     return null;
   }
+  const providerMetadata = citation.type === "page_location" ? {
+    citedText: citation.cited_text,
+    startPageNumber: citation.start_page_number,
+    endPageNumber: citation.end_page_number
+  } : {
+    citedText: citation.cited_text,
+    startCharIndex: citation.start_char_index,
+    endCharIndex: citation.end_char_index
+  };
   return {
     type: "source",
     sourceType: "document",
@@ -589,11 +672,7 @@ function createCitationSource(citation, citationDocuments, generateId2) {
     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
-      }
+      anthropic: providerMetadata
     }
   };
 }
@@ -795,15 +874,19 @@ var AnthropicMessagesLanguageModel = class {
     return (_c = (_b = (_a = this.config).transformRequestBody) == null ? void 0 : _b.call(_a, args)) != null ? _c : args;
   }
   extractCitationDocuments(prompt) {
-    const isCitationEnabled = (part) => {
+    const isCitationPart = (part) => {
       var _a, _b;
+      if (part.type !== "file") {
+        return false;
+      }
+      if (part.mediaType !== "application/pdf" && part.mediaType !== "text/plain") {
+        return false;
+      }
       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) => {
+    return prompt.filter((message) => message.role === "user").flatMap((message) => message.content).filter(isCitationPart).map((part) => {
       var _a;
       const filePart = part;
       return {
@@ -840,7 +923,7 @@ var AnthropicMessagesLanguageModel = class {
             content.push({ type: "text", text: part.text });
             if (part.citations) {
               for (const citation of part.citations) {
-                processPageLocationCitation(
+                processCitation(
                   citation,
                   citationDocuments,
                   this.generateId,
@@ -986,6 +1069,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;
@@ -1140,7 +1226,7 @@ var AnthropicMessagesLanguageModel = class {
                   }
                   case "citations_delta": {
                     const citation = value.delta.citation;
-                    processPageLocationCitation(
+                    processCitation(
                       citation,
                       citationDocuments,
                       generateId2,
@@ -1215,25 +1301,7 @@ var anthropicMessagesResponseSchema = z3.object({
       z3.object({
         type: z3.literal("text"),
         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()
+        citations: z3.array(citationSchema).optional()
       }),
       z3.object({
         type: z3.literal("thinking"),
@@ -1372,23 +1440,7 @@ var anthropicMessagesChunkSchema = z3.discriminatedUnion("type", [
       }),
       z3.object({
         type: z3.literal("citations_delta"),
-        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()
-          })
-        ])
+        citation: citationSchema
       })
     ])
   }),