mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 19:24:27 +00:00
fix(main.py): fix passing openrouter specific params (#8184)
* fix(main.py): fix passing openrouter specific params Fixes https://github.com/BerriAI/litellm/issues/8130 * test(test_get_model_info.py): add check for region name w/ cris model Resolves https://github.com/BerriAI/litellm/issues/8115
This commit is contained in:
parent
2b3a93590c
commit
905aee57b1
3 changed files with 39 additions and 1 deletions
|
@ -2605,6 +2605,21 @@ def test_completion_openrouter1():
|
|||
pytest.fail(f"Error occurred: {e}")
|
||||
|
||||
|
||||
def test_completion_openrouter_reasoning_effort():
|
||||
try:
|
||||
litellm.set_verbose = True
|
||||
response = completion(
|
||||
model="openrouter/deepseek/deepseek-r1",
|
||||
messages=messages,
|
||||
include_reasoning=True,
|
||||
max_tokens=5,
|
||||
)
|
||||
# Add any assertions here to check the response
|
||||
print(response)
|
||||
except Exception as e:
|
||||
pytest.fail(f"Error occurred: {e}")
|
||||
|
||||
|
||||
# test_completion_openrouter1()
|
||||
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue