From 4bf4e7954ebe9a497ea7ac62b3810e642d8591a6 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 31 Jan 2026 11:06:22 -0800 Subject: [PATCH] fix gemini files --- litellm/llms/gemini/files/transformation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/llms/gemini/files/transformation.py b/litellm/llms/gemini/files/transformation.py index deb3eeb2481..861f7bd2f2b 100644 --- a/litellm/llms/gemini/files/transformation.py +++ b/litellm/llms/gemini/files/transformation.py @@ -4,7 +4,7 @@ Supports writing files to Google AI Studio Files API. For vertex ai, check out the vertex_ai/files/handler.py file. """ import time -from typing import Any, List, Optional +from typing import Any, List, Literal, Optional import httpx from openai.types.file_deleted import FileDeleted @@ -238,7 +238,7 @@ class GoogleAIStudioFilesHandler(GeminiModelInfo, BaseFilesConfig): # Map Gemini state to OpenAI status gemini_state = response_json.get("state", "STATE_UNSPECIFIED") - status = "uploaded" # Default + status: Literal["uploaded", "processed", "error"] = "uploaded" # Default if gemini_state == "ACTIVE": status = "processed" elif gemini_state == "FAILED":