changes per comments

Author: narengogi

src/providers/anthropic/chatComplete.ts

@@ -64,26 +64,26 @@ interface AnthropicTextContentItem {
 }
 
 interface AnthropicUrlPdfContentItem {
-  type: 'document';
+  type: string;
   source: {
-    type: 'url';
+    type: string;
     url: string;
   };
 }
 
 interface AnthropicBase64PdfContentItem {
-  type: 'document';
+  type: string;
   source: {
-    type: 'base64';
+    type: string;
     data: string;
     media_type: string;
   };
 }
 
 interface AnthropicPlainTextContentItem {
-  type: 'document';
+  type: string;
   source: {
-    type: 'text';
+    type: string;
     data: string;
     media_type: string;
   };
@@ -213,25 +213,14 @@ const transformAndAppendFileContentItem = (
   } else if (item.file?.file_data) {
     const contentType =
       mimeType === fileExtensionMimeTypeMap.txt ? 'text' : 'base64';
-    if (contentType === 'text') {
-      transformedMessage.content.push({
-        type: 'document',
-        source: {
-          type: contentType,
-          data: item.file.file_data,
-          media_type: mimeType,
-        },
-      });
-    } else {
-      transformedMessage.content.push({
-        type: 'document',
-        source: {
-          type: contentType,
-          data: item.file.file_data,
-          media_type: mimeType,
-        },
-      });
-    }
+    transformedMessage.content.push({
+      type: 'document',
+      source: {
+        type: contentType,
+        data: item.file.file_data,
+        media_type: mimeType,
+      },
+    });
   }
 };
 

src/providers/bedrock/chatComplete.ts

@@ -188,7 +188,7 @@ const getMessageContent = (message: Message) => {
           out.push({
             document: {
               format: fileFormat,
-              name: crypto.randomUUID(),
+              name: item.file.file_name || crypto.randomUUID(),
               source: {
                 s3Location: {
                   uri: item.file.file_url,
@@ -200,7 +200,7 @@ const getMessageContent = (message: Message) => {
           out.push({
             document: {
               format: fileFormat,
-              name: crypto.randomUUID(),
+              name: item.file.file_name || crypto.randomUUID(),
               source: {
                 bytes: item.file.file_data,
               },