From 891445783cbbb7d8e9cf49b53b4f2f105313bf14 Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Sat, 2 May 2026 11:31:31 +0530 Subject: [PATCH] Fix code qa --- litellm/llms/vertex_ai/files/transformation.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/litellm/llms/vertex_ai/files/transformation.py b/litellm/llms/vertex_ai/files/transformation.py index 5eda1c836d3..ea4f498283f 100644 --- a/litellm/llms/vertex_ai/files/transformation.py +++ b/litellm/llms/vertex_ai/files/transformation.py @@ -143,12 +143,12 @@ def _get_litellm_batch_custom_id_from_labels(labels: Dict[str, Any]) -> str: return str(labels.get("litellm_custom_id", "unknown")) -def _transform_openai_jsonl_content_to_vertex_ai_jsonl_content( +def _openai_batch_jsonl_entries_to_vertex_wrapped_requests( openai_jsonl_content: List[Dict[str, Any]], map_openai_to_vertex_params: Callable[[Dict[str, Any]], Dict[str, Any]], ) -> List[Dict[str, Any]]: """ - Transforms OpenAI JSONL content to VertexAI JSONL content + Transforms OpenAI JSONL batch entries to Vertex AI JSONL lines. jsonl body for vertex is {"request": } Example Vertex jsonl @@ -371,7 +371,7 @@ class VertexAIFilesConfig(VertexBase, BaseFilesConfig): def _transform_openai_jsonl_content_to_vertex_ai_jsonl_content( self, openai_jsonl_content: List[Dict[str, Any]] ) -> List[Dict[str, Any]]: - return _transform_openai_jsonl_content_to_vertex_ai_jsonl_content( + return _openai_batch_jsonl_entries_to_vertex_wrapped_requests( openai_jsonl_content=openai_jsonl_content, map_openai_to_vertex_params=self._map_openai_to_vertex_params, ) @@ -858,7 +858,7 @@ class VertexAIJsonlFilesTransformation(VertexGeminiConfig): def _transform_openai_jsonl_content_to_vertex_ai_jsonl_content( self, openai_jsonl_content: List[Dict[str, Any]] ) -> List[Dict[str, Any]]: - return _transform_openai_jsonl_content_to_vertex_ai_jsonl_content( + return _openai_batch_jsonl_entries_to_vertex_wrapped_requests( openai_jsonl_content=openai_jsonl_content, map_openai_to_vertex_params=self._map_openai_to_vertex_params, )