tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-18 04:28:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix all optional param tests

Browse Source
This commit is contained in:
Ishaan Jaff
2024-08-07 17:52:40 -07:00
parent 6b03998e32
commit 4e90a25417
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_optional_params.py
+4 -4
View File
@@ -301,7 +301,7 @@ def test_dynamic_drop_params(drop_params):
optional_params = litellm.utils.get_optional_params(
model="command-r",
custom_llm_provider="cohere",
response_format="json",
response_format={"type": "json"},
drop_params=drop_params,
)
else:
@@ -309,7 +309,7 @@ def test_dynamic_drop_params(drop_params):
optional_params = litellm.utils.get_optional_params(
model="command-r",
custom_llm_provider="cohere",
response_format="json",
response_format={"type": "json"},
drop_params=drop_params,
)
pytest.fail("Expected to fail")
@@ -389,7 +389,7 @@ def test_dynamic_drop_additional_params(drop_params):
optional_params = litellm.utils.get_optional_params(
model="command-r",
custom_llm_provider="cohere",
response_format="json",
response_format={"type": "json"},
additional_drop_params=["response_format"],
)
else:
@@ -397,7 +397,7 @@ def test_dynamic_drop_additional_params(drop_params):
optional_params = litellm.utils.get_optional_params(
model="command-r",
custom_llm_provider="cohere",
response_format="json",
response_format={"type": "json"},
)
pytest.fail("Expected to fail")
except Exception as e: