phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test(test_custom_callback_input.py): fix custom callback input test

Browse source
This commit is contained in:
Krrish Dholakia 2024-06-07 16:13:30 -07:00
parent f73b6033fd
commit beddb58852
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_custom_callback_input.py
View file

@ -78,7 +78,9 @@ class CompletionCustomHandler(
is LiteLLMCommonStrings.redacted_by_litellm.value
)
else:
assert isinstance(metadata_value["raw_request"], str)
assert "raw_request" not in metadata_value or isinstance(
metadata_value["raw_request"], str
)
except Exception:
print(f"Assertion Error: {traceback.format_exc()}")
self.errors.append(traceback.format_exc())