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

fix AnthropicConfig test

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-20 12:07:28 -08:00
parent f121b8f630
commit b3b1ff6882
1 changed files with 4 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/llm_translation/test_anthropic_completion.py
View file

@ -712,9 +712,7 @@ def test_convert_tool_response_to_message_with_values():
)
]
message = AnthropicChatCompletion._convert_tool_response_to_message(
tool_calls=tool_calls
)
message = AnthropicConfig._convert_tool_response_to_message(tool_calls=tool_calls)
assert message is not None
assert message.content == '{"name": "John", "age": 30}'
@ -739,9 +737,7 @@ def test_convert_tool_response_to_message_without_values():
)
]
message = AnthropicChatCompletion._convert_tool_response_to_message(
tool_calls=tool_calls
)
message = AnthropicConfig._convert_tool_response_to_message(tool_calls=tool_calls)
assert message is not None
assert message.content == '{"name": "John", "age": 30}'
@ -760,9 +756,7 @@ def test_convert_tool_response_to_message_invalid_json():
)
]
message = AnthropicChatCompletion._convert_tool_response_to_message(
tool_calls=tool_calls
)
message = AnthropicConfig._convert_tool_response_to_message(tool_calls=tool_calls)
assert message is not None
assert message.content == "invalid json"
@ -779,8 +773,6 @@ def test_convert_tool_response_to_message_no_arguments():
)
]
message = AnthropicChatCompletion._convert_tool_response_to_message(
tool_calls=tool_calls
)
message = AnthropicConfig._convert_tool_response_to_message(tool_calls=tool_calls)
assert message is None