mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 19:24:27 +00:00
test_nova_optional_params_tool_choice
This commit is contained in:
parent
4a4328b5bb
commit
ef6bf02ac4
1 changed files with 59 additions and 56 deletions
|
@ -2430,7 +2430,9 @@ def test_bedrock_process_empty_text_blocks():
|
||||||
assert modified_message["content"][0]["text"] == "Please continue."
|
assert modified_message["content"][0]["text"] == "Please continue."
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def test_nova_optional_params_tool_choice():
|
def test_nova_optional_params_tool_choice():
|
||||||
|
try:
|
||||||
litellm.drop_params = True
|
litellm.drop_params = True
|
||||||
litellm.set_verbose = True
|
litellm.set_verbose = True
|
||||||
litellm.completion(
|
litellm.completion(
|
||||||
|
@ -2486,7 +2488,8 @@ def test_nova_optional_params_tool_choice():
|
||||||
],
|
],
|
||||||
tool_choice={"type": "function", "function": {"name": "GameDefinition"}},
|
tool_choice={"type": "function", "function": {"name": "GameDefinition"}},
|
||||||
)
|
)
|
||||||
|
except litellm.APIConnectionError:
|
||||||
|
pass
|
||||||
|
|
||||||
class TestBedrockEmbedding(BaseLLMEmbeddingTest):
|
class TestBedrockEmbedding(BaseLLMEmbeddingTest):
|
||||||
def get_base_embedding_call_args(self) -> dict:
|
def get_base_embedding_call_args(self) -> dict:
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue