scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

Fix code qa

Browse source
This commit is contained in:
Sameer Kankute 2026-05-02 11:31:31 +05:30
parent 60ca4424a6
commit 891445783c
No known key found for this signature in database
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/llms/vertex_ai/files/transformation.py
View file

@ -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": <request_body>}
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,
)