Skip to content

test(litellm): make tool-call argument log tests independent of test order #7461

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
ankit2235 wants to merge 1 commit into from
+4 −4
Closed

test(litellm): make tool-call argument log tests independent of test order #7461

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions tests/unittests/models/test_litellm.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10762,7 +10762,7 @@ def test_repaired_arguments_log_warning_without_raw_payload(self, caplog):

def test_repaired_arguments_log_debug_with_raw_payload(self, caplog):
"""Repaired non-strict JSON arguments emit warning with payload at debug level."""
with caplog.at_level(logging.DEBUG):
with caplog.at_level(logging.DEBUG, logger="google_adk"):
_parse_tool_call_arguments("{a: 1}", function_name="test_fn")
assert "test_fn" in caplog.text
assert "{a: 1}" in caplog.text
Expand All @@ -10780,7 +10780,7 @@ def test_repaired_arguments_without_function_name(self, caplog):
assert "{a: 1}" not in caplog.text

caplog.clear()
with caplog.at_level(logging.DEBUG):
with caplog.at_level(logging.DEBUG, logger="google_adk"):
_parse_tool_call_arguments("{a: 1}")
warning_records = [
r for r in caplog.records if r.levelno == logging.WARNING
Expand All @@ -10794,7 +10794,7 @@ def test_literal_eval_layer_logs_debug_not_warning(self, caplog):
_parse_tool_call_arguments("{'a': 1}", function_name="test_fn")
assert not any(r.levelno == logging.WARNING for r in caplog.records)

with caplog.at_level(logging.DEBUG):
with caplog.at_level(logging.DEBUG, logger="google_adk"):
_parse_tool_call_arguments("{'a': 1}", function_name="test_fn")
assert not any(r.levelno == logging.WARNING for r in caplog.records)
debug_records = [
Expand All @@ -10808,7 +10808,7 @@ def test_literal_eval_layer_logs_debug_not_warning(self, caplog):

def test_literal_eval_without_function_name_logs_debug(self, caplog):
"""Python dict literal without function_name logs at debug."""
with caplog.at_level(logging.DEBUG):
with caplog.at_level(logging.DEBUG, logger="google_adk"):
_parse_tool_call_arguments("{'a': 1}")
assert not any(r.levelno == logging.WARNING for r in caplog.records)
debug_records = [
Expand Down
Loading