mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-09 03:18:44 +00:00
fix(ocr): log resolved api_base and headers on Rust path
This commit is contained in:
parent
b126fbb439
commit
b87dcb2c97
1 changed files with 15 additions and 2 deletions
|
|
@ -344,6 +344,19 @@ def ocr(
|
|||
from litellm.secret_managers.main import get_secret_str
|
||||
|
||||
resolved_api_key = api_key or get_secret_str("MISTRAL_API_KEY")
|
||||
resolved_headers = ocr_provider_config.validate_environment(
|
||||
headers={},
|
||||
model=model,
|
||||
api_key=resolved_api_key,
|
||||
api_base=api_base,
|
||||
litellm_params=dict(litellm_params),
|
||||
)
|
||||
resolved_complete_url = ocr_provider_config.get_complete_url(
|
||||
api_base=api_base,
|
||||
model=model,
|
||||
optional_params=optional_params,
|
||||
litellm_params=dict(litellm_params),
|
||||
)
|
||||
litellm_logging_obj.pre_call(
|
||||
input="OCR document processing",
|
||||
api_key=resolved_api_key,
|
||||
|
|
@ -353,8 +366,8 @@ def ocr(
|
|||
"document": document,
|
||||
**optional_params,
|
||||
},
|
||||
"api_base": api_base,
|
||||
"headers": {},
|
||||
"api_base": resolved_complete_url,
|
||||
"headers": resolved_headers,
|
||||
},
|
||||
)
|
||||
return OCRResponse(
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue