gddlZddlZddlmZmZmZmZmZmZddl m Z ddl m Z ddl mZddlmZmZmZmZGddeZy) N)AnyDictListOptionalTupleUnion)+_convert_vertex_datetime_to_openai_datetime)_transform_request_body)VertexGeminiConfig)CreateFileRequest FileObject FileTypesPathLikeceZdZdZ ddeedeeeffdZde e ee ffdZ de e ee fdefdZ d e ee fde ee ffd Zdedefd Zd ed e ee fdefdZy)VertexAIFilesTransformationzQ Transforms OpenAI /v1/files/* requests to VertexAI /v1/files/* requests Nopenai_file_contentreturncH| td|j|}|jDcgc](}|jst j |*}}|j |}djd|D}|j|}||fScc}w)zU Transforms OpenAI FileContentRequest to VertexAI FileContentRequest zcontents of file are None c3FK|]}tj|ywN)jsondumps).0items `/var/www/openai/venv/lib/python3.12/site-packages/litellm/llms/vertex_ai/files/transformation.py zfVertexAIFilesTransformation.transform_openai_file_content_to_vertex_ai_file_content..)s( )=DJJt )=s!)openai_jsonl_content) ValueError_get_content_from_openai_file splitlinesstriprloads:_transform_openai_jsonl_content_to_vertex_ai_jsonl_contentjoin_get_gcs_object_name)selfr file_contentlinervertex_jsonl_contentvertex_jsonl_string object_names r7transform_openai_file_content_to_vertex_ai_file_contentzSVertexAIFilesTransformation.transform_openai_file_content_to_vertex_ai_file_contents  &89 999:MN *6)@)@)B )BdjjlDJJt )B    K K$   #ii( )=(  //!50 #K// s BBrc g}|D]j}|jdxsi}t|jdg|jdd|j|did}|jd|il|S) a Transforms OpenAI JSONL content to VertexAI JSONL content jsonl body for vertex is {"request": } Example Vertex jsonl {"request":{"contents": [{"role": "user", "parts": [{"text": "What is the relation between the following video and image samples?"}, {"fileData": {"fileUri": "gs://cloud-samples-data/generative-ai/video/animals.mp4", "mimeType": "video/mp4"}}, {"fileData": {"fileUri": "gs://cloud-samples-data/generative-ai/image/cricket.jpeg", "mimeType": "image/jpeg"}}]}]}} {"request":{"contents": [{"role": "user", "parts": [{"text": "Describe what is happening in this video."}, {"fileData": {"fileUri": "gs://cloud-samples-data/generative-ai/video/another_video.mov", "mimeType": "video/mov"}}]}]}} bodymessagesmodel vertex_aiN)r0r1optional_paramscustom_llm_providerlitellm_paramscached_contentrequest)getr _map_openai_to_vertex_paramsappend)r'rr*_openai_jsonl_contentopenai_request_bodyvertex_request_bodys rr$zVVertexAIFilesTransformation._transform_openai_jsonl_content_to_vertex_ai_jsonl_content1s "%9 !"7";";F"C"Ir "9,00R@)--gr: $ A ABU V$/!# #  ! ' '4G(H I&:$#c|djdijdd}d|vrd|}d|dtj}|S) z Gets a unique GCS object name for the VertexAI batch prediction job named as: litellm-vertex-{model}-{uuid} rr/r1r2zpublishers/google/modelszpublishers/google/models/zlitellm-vertex-files//)r9uuiduuid4)r'r_modelr,s rr&z0VertexAIFilesTransformation._get_gcs_object_nameKsZ&a(,,VR8<0#+I#60 sCx0:$$(c3h$8$4 "4S>2   !#s(^ c3h ##s#J 1 HLSRUX  r?r)rrBtypingrrrrrr#litellm.llms.vertex_ai.common_utilsr ,litellm.llms.vertex_ai.gemini.transformationr @litellm.llms.vertex_ai.gemini.vertex_and_google_ai_studio_geminir litellm.types.llms.openair r rrrror?rrus9 ::QYXT "4T r?