support pdf based prompting in anthropic

Portkey-AI · Dec 19, 2024 · 9db1cee · 9db1cee
1 parent f921bcb
commit 9db1cee
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/providers/anthropic/chatComplete.ts b/src/providers/anthropic/chatComplete.ts
@@ -1,4 +1,4 @@
-import { ANTHROPIC } from '../../globals';
+import { ANTHROPIC, fileExtensionMimeTypeMap } from '../../globals';
 import {
   Params,
   Message,
@@ -146,7 +146,10 @@ export const AnthropicChatCompleteConfig: ProviderConfig = {
                     if (mediaTypeParts.length === 2 && base64Image) {
                       const mediaType = mediaTypeParts[1];
                       transformedMessage.content.push({
-                        type: 'image',
+                        type:
+                          mediaType === fileExtensionMimeTypeMap.pdf
+                            ? 'document'
+                            : 'image',
                         source: {
                           type: 'base64',
                           media_type: mediaType,