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:
Krish Dholakia 2025-02-02 22:23:14 -08:00 committed by GitHub
parent 2b3a93590c
commit 905aee57b1
3 changed files with 39 additions and 1 deletions

View file

@ -2206,7 +2206,7 @@ def completion( # type: ignore # noqa: PLR0915
data = {"model": model, "messages": messages, **optional_params}
## COMPLETION CALL
response = openai_chat_completions.completion(
response = openai_like_chat_completion.completion(
model=model,
messages=messages,
headers=headers,
@ -2221,6 +2221,8 @@ def completion( # type: ignore # noqa: PLR0915
acompletion=acompletion,
timeout=timeout, # type: ignore
custom_llm_provider="openrouter",
custom_prompt_dict=custom_prompt_dict,
encoding=encoding,
)
## LOGGING
logging.post_call(