scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-28 06:23:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 94

test(logging): type the input-path regression test

Browse source
This commit is contained in:
mateo-berri 2026-10-20 18:59:53 -07:00
parent f0d3436496
commit c7b235bee6
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/unit/litellm_core_utils/test_litellm_logging.py
View file

@ -407,14 +407,16 @@ class _InputEventRecorder(CustomLogger):
self._label = label
self.events = events
def log_pre_api_call(self, model, messages, kwargs) -> None:
def log_pre_api_call(self, model: object, messages: object, kwargs: object) -> None:
self.events.append(f"{self._label} pre")
def log_post_api_call(self, kwargs, response_obj, start_time, end_time) -> None:
def log_post_api_call(self, kwargs: object, response_obj: object, start_time: object, end_time: object) -> None:
self.events.append(f"{self._label} post")
def test_input_events_reach_the_replacement_and_not_the_global_callback_it_replaced(logging_obj, monkeypatch):
def test_input_events_reach_the_replacement_and_not_the_global_callback_it_replaced(
logging_obj: LitellmLogging, monkeypatch: pytest.MonkeyPatch
) -> None:
events: Final[list[str]] = []
monkeypatch.setattr(litellm, "input_callback", [_InputEventRecorder("global", events)])
logging_obj.replace_global_callback(_InputEventRecorder("replacement", events))