phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: fix test - handle llm api inconsistency

Browse source
This commit is contained in:
Krrish Dholakia 2025-03-21 10:51:34 -07:00
parent 6c51bdebe7
commit 47a18285f1
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/llm_translation/base_llm_unit_tests.py
View file

@ -1062,6 +1062,7 @@ class BaseAnthropicChatTest(ABC):
**base_completion_call_args, **args, stream=False
)
assert json.loads(built_response.choices[0].message.content) == json.loads(
non_stream_response.choices[0].message.content
assert (
json.loads(built_response.choices[0].message.content).keys()
== json.loads(non_stream_response.choices[0].message.content).keys()
), f"Got={json.loads(built_response.choices[0].message.content)}, Expected={json.loads(non_stream_response.choices[0].message.content)}"