models - anthropic - document - plain text (#141)

Author: pgrayy

src/strands/models/anthropic.py

@@ -97,13 +97,16 @@ def _format_request_message_content(self, content: ContentBlock) -> dict[str, An
             Anthropic formatted content block.
         """
         if "document" in content:
+            mime_type = mimetypes.types_map.get(f".{content['document']['format']}", "application/octet-stream")
             return {
                 "source": {
-                    "data": base64.b64encode(content["document"]["source"]["bytes"]).decode("utf-8"),
-                    "media_type": mimetypes.types_map.get(
-                        f".{content['document']['format']}", "application/octet-stream"
+                    "data": (
+                        content["document"]["source"]["bytes"].decode("utf-8")
+                        if mime_type == "text/plain"
+                        else base64.b64encode(content["document"]["source"]["bytes"]).decode("utf-8")
                     ),
-                    "type": "base64",
+                    "media_type": mime_type,
+                    "type": "text" if mime_type == "text/plain" else "base64",
                 },
                 "title": content["document"]["name"],
                 "type": "document",

tests/strands/models/test_anthropic.py

@@ -102,19 +102,46 @@ def test_format_request_with_system_prompt(model, messages, model_id, max_tokens
     assert tru_request == exp_request
 
 
-def test_format_request_with_document(model, model_id, max_tokens):
+@pytest.mark.parametrize(
+    ("content", "formatted_content"),
+    [
+        # PDF
+        (
+            {
+                "document": {"format": "pdf", "name": "test doc", "source": {"bytes": b"pdf"}},
+            },
+            {
+                "source": {
+                    "data": "cGRm",
+                    "media_type": "application/pdf",
+                    "type": "base64",
+                },
+                "title": "test doc",
+                "type": "document",
+            },
+        ),
+        # Plain text
+        (
+            {
+                "document": {"format": "txt", "name": "test doc", "source": {"bytes": b"txt"}},
+            },
+            {
+                "source": {
+                    "data": "txt",
+                    "media_type": "text/plain",
+                    "type": "text",
+                },
+                "title": "test doc",
+                "type": "document",
+            },
+        ),
+    ],
+)
+def test_format_request_with_document(content, formatted_content, model, model_id, max_tokens):
     messages = [
         {
             "role": "user",
-            "content": [
-                {
-                    "document": {
-                        "format": "pdf",
-                        "name": "test-doc",
-                        "source": {"bytes": b"base64encodeddoc"},
-                    },
-                },
-            ],
+            "content": [content],
         },
     ]
 
@@ -124,17 +151,7 @@ def test_format_request_with_document(model, model_id, max_tokens):
         "messages": [
             {
                 "role": "user",
-                "content": [
-                    {
-                        "source": {
-                            "data": "YmFzZTY0ZW5jb2RlZGRvYw==",
-                            "media_type": "application/pdf",
-                            "type": "base64",
-                        },
-                        "title": "test-doc",
-                        "type": "document",
-                    },
-                ],
+                "content": [formatted_content],
             },
         ],
         "model": model_id,