fix(vertex_ai/files): handle IO[bytes] content in non-JSONL file upload path

This commit is contained in:
Ishaan Jaffer 2026-03-23 09:50:27 -07:00
parent aabc8f2b32
commit 3bde9e1827

View file

@ -1,3 +1,4 @@
import io
import json
import os
import time
@ -292,6 +293,8 @@ class VertexAIFilesConfig(VertexBase, BaseFilesConfig):
return "\n".join(json.dumps(item) for item in vertex_jsonl_content)
elif isinstance(extracted_file_data_content, bytes):
return extracted_file_data_content
elif isinstance(extracted_file_data_content, io.IOBase):
return extracted_file_data_content # type: ignore[return-value]
else:
raise ValueError("Unsupported file content type")