From b3b1ff6882e81fb811f5bb8ea0f9912d06a01d32 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 20 Nov 2024 12:07:28 -0800 Subject: [PATCH] fix AnthropicConfig test --- .../llm_translation/test_anthropic_completion.py | 16 ++++------------ 1 file changed, 4 insertions(+), 12 deletions(-) diff --git a/tests/llm_translation/test_anthropic_completion.py b/tests/llm_translation/test_anthropic_completion.py index 8a788e0fb..9781297fe 100644 --- a/tests/llm_translation/test_anthropic_completion.py +++ b/tests/llm_translation/test_anthropic_completion.py @@ -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