scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-18 04:44:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 125

test: count only OCR calls in the OCR matrix recorder

Browse source
This commit is contained in:
Yuneng Jiang 2026-10-10 11:47:01 -07:00
parent 1556841251
commit 067f77eb0d
No known key found for this signature in database
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/integration/sdk/test_ocr_matrix_sdk.py
View file

@ -36,6 +36,7 @@ DOCUMENTS: Final = {
"/documents/scan.png": (PNG_BYTES, "image/png"),
}
DI_ANALYZE_PATH: Final = "/documentintelligence/documentModels/prebuilt-layout"
OCR_CALL_TYPES: Final = frozenset({"ocr", "aocr"})
_JSON: Final = TypeAdapter(dict[str, JsonValue])
_STRING_KEYED: Final = TypeAdapter(dict[str, object])
_FLOAT: Final = TypeAdapter(float)
@ -104,7 +105,8 @@ class RecordingLogger(CustomLogger):
def _record(self, kwargs: Mapping[str, object], response_obj: object) -> None:
payload: Final = _STRING_KEYED.validate_python(kwargs.get("standard_logging_object"))
self.calls = (*self.calls, LoggedCall(payload, response_obj))
if payload.get("call_type") in OCR_CALL_TYPES:
self.calls = (*self.calls, LoggedCall(payload, response_obj))
def log_success_event(
self, kwargs: Mapping[str, object], response_obj: object, start_time: datetime, end_time: datetime