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

(refactor) anthropic - move _process_response in transformation.py (#6834)

Browse source 
* move _process_response in transformation

* fix AnthropicConfig test
This commit is contained in:
Ishaan Jaff 2024-11-20 17:24:19 -08:00 • committed by GitHub
parent b11bc0374e
commit 434b1d3d86
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 188 additions and 178 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