diff --git a/llm_gemini.py b/llm_gemini.py index 69a49f0..49a69b3 100644 --- a/llm_gemini.py +++ b/llm_gemini.py @@ -60,6 +60,9 @@ class _SharedGemini: can_stream = True attachment_types = ( + # Text + "text/plain", + "text/csv", # PDF "application/pdf", # Images @@ -335,4 +338,4 @@ def embed_batch(self, items): ) response.raise_for_status() - return [item["values"] for item in response.json()["embeddings"]] \ No newline at end of file + return [item["values"] for item in response.json()["embeddings"]]