From 78dd4b2b2fa5233d9a10a1179d72bd7705982cd0 Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Fri, 30 Jan 2026 16:06:07 +0530 Subject: [PATCH] feat: add /delete endpoint support for gemini --- litellm/files/main.py | 13 +-- litellm/llms/gemini/files/transformation.py | 91 ++++++++++++++++++++- 2 files changed, 94 insertions(+), 10 deletions(-) diff --git a/litellm/files/main.py b/litellm/files/main.py index 913ec84626..93a10dac7a 100644 --- a/litellm/files/main.py +++ b/litellm/files/main.py @@ -9,6 +9,7 @@ import asyncio import contextvars import os import time +import uuid from functools import partial from typing import Any, Coroutine, Dict, Literal, Optional, Union, cast @@ -61,7 +62,7 @@ async def acreate_file( file: FileTypes, purpose: Literal["assistants", "batch", "fine-tune"], expires_after: Optional[FileExpiresAfter] = None, - custom_llm_provider: Literal["openai", "azure", "vertex_ai", "bedrock", "hosted_vllm", "manus"] = "openai", + custom_llm_provider: Literal["openai", "azure", "gemini", "vertex_ai", "bedrock", "hosted_vllm", "manus"] = "openai", extra_headers: Optional[Dict[str, str]] = None, extra_body: Optional[Dict[str, str]] = None, **kwargs, @@ -106,7 +107,7 @@ def create_file( file: FileTypes, purpose: Literal["assistants", "batch", "fine-tune"], expires_after: Optional[FileExpiresAfter] = None, - custom_llm_provider: Optional[Literal["openai", "azure", "vertex_ai", "bedrock", "hosted_vllm", "manus"]] = None, + custom_llm_provider: Optional[Literal["openai", "azure", "gemini", "vertex_ai", "bedrock", "hosted_vllm", "manus"]] = None, extra_headers: Optional[Dict[str, str]] = None, extra_body: Optional[Dict[str, str]] = None, **kwargs, @@ -294,7 +295,7 @@ def create_file( @client async def afile_retrieve( file_id: str, - custom_llm_provider: Literal["openai", "azure", "hosted_vllm", "manus"] = "openai", + custom_llm_provider: Literal["openai", "azure", "gemini", "hosted_vllm", "manus"] = "openai", extra_headers: Optional[Dict[str, str]] = None, extra_body: Optional[Dict[str, str]] = None, **kwargs, @@ -493,7 +494,7 @@ def file_retrieve( @client async def afile_delete( file_id: str, - custom_llm_provider: Literal["openai", "azure", "manus"] = "openai", + custom_llm_provider: Literal["openai", "azure", "gemini", "manus"] = "openai", extra_headers: Optional[Dict[str, str]] = None, extra_body: Optional[Dict[str, str]] = None, **kwargs, @@ -537,7 +538,7 @@ async def afile_delete( def file_delete( file_id: str, model: Optional[str] = None, - custom_llm_provider: Union[Literal["openai", "azure", "manus"], str] = "openai", + custom_llm_provider: Union[Literal["openai", "azure", "gemini", "manus"], str] = "openai", extra_headers: Optional[Dict[str, str]] = None, extra_body: Optional[Dict[str, str]] = None, **kwargs, @@ -680,7 +681,7 @@ def file_delete( ) else: raise litellm.exceptions.BadRequestError( - message="LiteLLM doesn't support {} for 'file_delete'. Only 'openai', 'azure', and 'manus' are supported.".format( + message="LiteLLM doesn't support {} for 'file_delete'. Only 'openai', 'azure', 'gemini', and 'manus' are supported.".format( custom_llm_provider ), model="n/a", diff --git a/litellm/llms/gemini/files/transformation.py b/litellm/llms/gemini/files/transformation.py index d9ebf69a97..499744840f 100644 --- a/litellm/llms/gemini/files/transformation.py +++ b/litellm/llms/gemini/files/transformation.py @@ -35,6 +35,26 @@ class GoogleAIStudioFilesHandler(GeminiModelInfo, BaseFilesConfig): def custom_llm_provider(self) -> LlmProviders: return LlmProviders.GEMINI + def validate_environment( + self, + api_key: Optional[str], + headers: dict, + model: str, + messages: list, + optional_params: dict, + litellm_params: dict, + ) -> dict: + """ + Validate environment and add Gemini API key to headers. + Google AI Studio uses x-goog-api-key header for authentication. + """ + api_key = self.get_api_key(api_key) + if not api_key: + raise ValueError("GEMINI_API_KEY is required for Google AI Studio file operations") + + headers["x-goog-api-key"] = api_key + return headers + def get_complete_url( self, api_base: Optional[str], @@ -56,10 +76,12 @@ class GoogleAIStudioFilesHandler(GeminiModelInfo, BaseFilesConfig): if not api_base: raise ValueError("api_base is required") - if not api_key: + # Get API key from multiple sources + final_api_key = api_key or litellm_params.get("api_key") or self.get_api_key() + if not final_api_key: raise ValueError("api_key is required") - url = "{}/{}?key={}".format(api_base, endpoint, api_key) + url = "{}/{}?key={}".format(api_base, endpoint, final_api_key) return url def get_supported_openai_params( @@ -196,7 +218,41 @@ class GoogleAIStudioFilesHandler(GeminiModelInfo, BaseFilesConfig): optional_params: dict, litellm_params: dict, ) -> tuple[str, dict]: - raise NotImplementedError("GoogleAIStudioFilesHandler does not support file deletion") + """ + Transform delete file request for Google AI Studio. + + Args: + file_id: The file URI (e.g., "files/abc123" or full URI) + optional_params: Optional parameters + litellm_params: LiteLLM parameters containing api_key + + Returns: + tuple[str, dict]: (url, params) for the DELETE request + """ + api_base = self.get_api_base(litellm_params.get("api_base")) + if not api_base: + raise ValueError("api_base is required") + + # Get API key from multiple sources (same pattern as get_complete_url) + api_key = litellm_params.get("api_key") or self.get_api_key() + if not api_key: + raise ValueError("api_key is required") + + # Extract file name from URI if full URI is provided + # file_id could be "files/abc123" or "https://generativelanguage.googleapis.com/v1beta/files/abc123" + if file_id.startswith("http"): + # Extract the file path from full URI + file_name = file_id.split("/v1beta/")[-1] + else: + file_name = file_id + + # Construct the delete URL + url = f"{api_base}/v1beta/{file_name}" + + # Add API key as header (Google AI Studio uses x-goog-api-key header) + params = {} + + return url, params def transform_delete_file_response( self, @@ -204,7 +260,34 @@ class GoogleAIStudioFilesHandler(GeminiModelInfo, BaseFilesConfig): logging_obj: LiteLLMLoggingObj, litellm_params: dict, ) -> FileDeleted: - raise NotImplementedError("GoogleAIStudioFilesHandler does not support file deletion") + """ + Transform Gemini's file delete response into OpenAI-style FileDeleted. + + Google AI Studio returns an empty JSON object {} on successful deletion. + """ + try: + # Google AI Studio returns {} on successful deletion + if raw_response.status_code == 200: + # Extract file ID from the request URL if possible + file_id = "deleted" + if hasattr(raw_response, "request") and raw_response.request: + url = str(raw_response.request.url) + if "/files/" in url: + file_id = url.split("/files/")[-1].split("?")[0] + # Add the files/ prefix if not present + if not file_id.startswith("files/"): + file_id = f"files/{file_id}" + + return FileDeleted( + id=file_id, + deleted=True, + object="file" + ) + else: + raise ValueError(f"Failed to delete file: {raw_response.text}") + except Exception as e: + verbose_logger.exception(f"Error parsing file delete response: {str(e)}") + raise ValueError(f"Error parsing file delete response: {str(e)}") def transform_list_files_request( self,