tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-06-18 05:28:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix: lint error

Browse Source
This commit is contained in:
Sameer Kankute
2025-12-15 12:00:04 +05:30
parent 858879919c
commit d619199cee
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/llms/vertex_ai/ocr/deepseek_transformation.py
+2 -2
View File
@@ -109,7 +109,7 @@ class VertexAIDeepSeekOCRConfig(BaseOCRConfig):
# Get API base URL
if api_base is None:
api_base = f"https://aiplatform.googleapis.com"
api_base = "https://aiplatform.googleapis.com"
# Ensure no trailing slash
api_base = api_base.rstrip("/")
@@ -195,7 +195,7 @@ class VertexAIDeepSeekOCRConfig(BaseOCRConfig):
data.update(chat_completion_params)
verbose_logger.debug(f"Vertex AI DeepSeek OCR: Transformed request to chat completion format")
verbose_logger.debug("Vertex AI DeepSeek OCR: Transformed request to chat completion format")
return OCRRequestData(data=data, files=None)
tests/ocr_tests/test_ocr_vertex_ai.py
+2 -2
View File
@@ -85,7 +85,7 @@ class TestVertexAIDeepSeekOCR(BaseOCRTest):
"""
load_vertex_ai_credentials()
return {
"model": "vertex_ai/deepseek-ai/deepseek-ocr-maas",
"model": "vertex_ai/deepseek-ocr-maas",
"vertex_location": "us-central1",
}
@@ -99,7 +99,7 @@ def test_vertex_ai_ocr_routing():
from litellm.llms.vertex_ai.ocr.transformation import VertexAIOCRConfig
# Test DeepSeek OCR routing
deepseek_config = get_vertex_ai_ocr_config("vertex_ai/deepseek-ai/deepseek-ocr-maas")
deepseek_config = get_vertex_ai_ocr_config("vertex_ai/deepseek-ocr-maas")
assert isinstance(deepseek_config, VertexAIDeepSeekOCRConfig), \
"DeepSeek model should route to VertexAIDeepSeekOCRConfig"