o ưi4@sdZddlZddlmZmZmZmZddlZddlm Z ddl m Z ddl m Z ddlmZmZddlmZdd lmZmZmZmZmZdd lmZd d lmZGd ddeeZdS)zv Supports writing files to Google AI Studio Files API. For vertex ai, check out the vertex_ai/files/handler.py file. N)AnyListLiteralOptional) FileDeleted)verbose_logger)extract_file_data)BaseFilesConfigLiteLLMLoggingObj)GeminiCreateFilesResponseObject)AllMessageValuesCreateFileRequestHttpxBinaryResponseContent%OpenAICreateFileRequestOptionalParamsOpenAIFileObject) LlmProviders)GeminiModelInfoc@s*eZdZddZedefddZ  d2deeefde d e e d eeefd eeefd e e d e e deeeffddZ d3d e e d e e de d ed ede ede fddZde de efddZded ede dedef ddZde ded ed edef ddZde e dejded edef ddZd e d ed edee effd!d"Zdejded edefd#d$Zd e d ed edee effd%d&Zdejded edefd'd(Zd)e e d ed edee effd*d+Z dejded ede efd,d-Z!d ed edee effd.d/Z"dejded ede#fd0d1Z$dS)4GoogleAIStudioFilesHandlercCsdSNselfrr_/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/gemini/files/transformation.py__init__ sz#GoogleAIStudioFilesHandler.__init__returncCstjSr)rZGEMINIrrrrcustom_llm_provider#sz.GoogleAIStudioFilesHandler.custom_llm_providerNheadersmodelmessagesoptional_paramslitellm_paramsapi_keyapi_basec Cs"||}|s td||d<|S)z Validate environment and add Gemini API key to headers. Google AI Studio uses x-goog-api-key header for authentication. z?GEMINI_API_KEY is required for Google AI Studio file operationszx-goog-api-key) get_api_key ValueError) rrrrr r!r"r#Zresolved_api_keyrrrvalidate_environment's z/GoogleAIStudioFilesHandler.validate_environmentstreamc CsNd}||}|s td|p|dp|}|stdd|||} | S)z{ OPTIONAL Get the complete url for the request Some providers need `model` in `api_base` zupload/v1beta/filesapi_base is requiredr"api_key is requiredz {}/{}?key={}) get_api_baser%getr$format) rr#r"rr r!r'ZendpointZ final_api_keyurlrrrget_complete_url<s z+GoogleAIStudioFilesHandler.get_complete_urlcCsgSrr)rrrrrget_supported_openai_paramsYsz6GoogleAIStudioFilesHandler.get_supported_openai_paramsnon_default_params drop_paramscCs|Srr)rr0r rr1rrrmap_openai_params^sz,GoogleAIStudioFilesHandler.map_openai_paramscreate_file_datac Cs|d}|dur tdt|}t|d}ddt||ddd }||d dd |d p7tttii} t|d dd} || d| |dddS)z Transform the OpenAI-style file creation request into Gemini's format Returns: dict: Contains both request data and headers for the two-step upload fileNzFile data is requiredcontentZ resumablestart content_typezapplication/json)zX-Goog-Upload-ProtocolX-Goog-Upload-Commandz#X-Goog-Upload-Header-Content-Lengthz!X-Goog-Upload-Header-Content-Typez Content-TyperZ display_namefilename0zupload, finalize)zContent-LengthzX-Goog-Upload-Offsetr8)rdata)Zinitial_requestZupload_request)r+r%rlenstrupdateinttime) rrr3r r!Z file_dataZextracted_data file_sizerZ initial_dataZupload_headersrrrtransform_create_file_requestgs2  z8GoogleAIStudioFilesHandler.transform_create_file_request raw_response logging_objc Csz3|}tdi|di}t|dt|dttt|dddd|ddd d d d WSt yR}zt d t |t d t |d }~ww)zV Transform Gemini's file upload response into OpenAI-style FileObject r4uri sizeBytes createTimeZ+00:00%Y-%m-%dT%H:%M:%S.%f%z displayName user_datauploadedNidbytesZ created_atr9objectpurposestatusZstatus_detailsz$Error parsing file upload response: r)jsonr r+rr?r@mktimestrptimereplace Exceptionr exceptionr=r%)rrrCrDr! response_jsonZresponse_objecterrrtransform_create_file_responses:  z9GoogleAIStudioFilesHandler.transform_create_file_responsefile_idcCsp|dp|}|std|drd||}|ifS||dp'd}|d}d|||}|ifS) z Get the URL to retrieve a file from Google AI Studio. We expect file_id to be the URI (e.g. https://generativelanguage.googleapis.com/v1beta/files/...) as returned by the upload response. r"r)httpz {}?key={}r#z)https://generativelanguage.googleapis.com/z{}/v1beta/{}?key={})r+r$r% startswithr,r*rstrip)rr]r r!r"r-r#rrrtransform_retrieve_file_requests    z:GoogleAIStudioFilesHandler.transform_retrieve_file_requestc CszS|}|dd}|dkrd}n |dkrd}nd}t|dd t|d d ttt|d d dd|dd dd||dkrOt|dd dWSddWSt yr}zt dt|t dt|d}~ww)zY Transform Gemini's file retrieval response into OpenAI-style FileObject stateZSTATE_UNSPECIFIEDZACTIVE processedZFAILEDerrorrMrErFrrGrHrIrJrKr4rLNrNz&Error parsing file retrieve response: ) rTr+rr?r@rUrVrWr=rXrrYr%)rrCrDr!rZZ gemini_staterSr[rrr transform_retrieve_file_responses@    z;GoogleAIStudioFilesHandler.transform_retrieve_file_responsec Cs||d}|std|dp|}|std|dr*|dd}n |dr1|nd|}|d|}i}||fS) au 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 r#r(r"r)r^z/v1beta/files/)r*r+r%r$r`split) rr]r r!r#r" file_namer-paramsrrrtransform_delete_file_requests z8GoogleAIStudioFilesHandler.transform_delete_file_requestc Csz?|jdkr8d}t|dr0|jr0t|jj}d|vr0|dddd}|ds0d|}t|d d d WStd |j t y^}zt d t|td t|d}~ww)z Transform Gemini's file delete response into OpenAI-style FileDeleted. Google AI Studio returns an empty JSON object {} on successful deletion. deletedrequestz/files/rh?rriTr4)rOrorQzFailed to delete file: z$Error parsing file delete response: N) status_codehasattrrpr=r-rjr`rr%textrXrrY)rrCrDr!r]r-r[rrrtransform_delete_file_response9s(    z9GoogleAIStudioFilesHandler.transform_delete_file_responserRcCtdNz8GoogleAIStudioFilesHandler does not support file listingNotImplementedError)rrRr r!rrrtransform_list_files_request\z7GoogleAIStudioFilesHandler.transform_list_files_requestcCrvrwrxrrCrDr!rrrtransform_list_files_responsedr{z8GoogleAIStudioFilesHandler.transform_list_files_responsecCrvNzBGoogleAIStudioFilesHandler does not support file content retrievalrx)rZfile_content_requestr r!rrrtransform_file_content_requestlr{z9GoogleAIStudioFilesHandler.transform_file_content_requestcCrvr~rxr|rrrtransform_file_content_responsetr{z:GoogleAIStudioFilesHandler.transform_file_content_response)NNr)%__name__ __module__ __qualname__rpropertyrrdictrr=rr rr&boolr.rr/r2r rBhttpxResponser rr\tuplerbrgrmrrurzr}rrrrrrrrs&          8 *   +  * #     r)__doc__r@typingrrrrrZopenai.types.file_deletedrZlitellm._loggingrZ8litellm.litellm_core_utils.prompt_templates.common_utilsrZ*litellm.llms.base_llm.files.transformationr r Zlitellm.types.llms.geminir Zlitellm.types.llms.openair r rrrZlitellm.types.utilsrZ common_utilsrrrrrrs