Piggyback txt

Author: kevinkassimo

src/google/adk/models/lite_llm.py

@@ -80,7 +80,7 @@
 }
 
 _SUPPORTED_FILE_CONTENT_MIME_TYPES = set(
-    ["application/pdf", "application/json"]
+    ["application/pdf", "application/json", "text/plain"]
 )
 
 

tests/unittests/models/test_litellm.py

@@ -90,6 +90,7 @@
 FILE_URI_TEST_CASES = [
     pytest.param("gs://bucket/document.pdf", "application/pdf", id="pdf"),
     pytest.param("gs://bucket/data.json", "application/json", id="json"),
+    pytest.param("gs://bucket/data.txt", "text/plain", id="txt"),
 ]
 
 STREAMING_MODEL_RESPONSE = [
@@ -1328,6 +1329,17 @@ def test_get_content_json():
   assert content[0]["file"]["format"] == "application/json"
 
 
+def test_get_content_txt():
+  parts = [types.Part.from_bytes(data=b"hello world", mime_type="text/plain")]
+  content = _get_content(parts)
+  assert content[0]["type"] == "file"
+  assert (
+      content[0]["file"]["file_data"]
+      == "data:text/plain;base64,aGVsbG8gd29ybGQ="
+  )
+  assert content[0]["file"]["format"] == "text/plain"
+
+
 @pytest.mark.parametrize("file_uri,mime_type", FILE_URI_TEST_CASES)
 def test_get_content_file_uri(file_uri, mime_type):
   parts = [types.Part.from_uri(file_uri=file_uri, mime_type=mime_type)]