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

fix(main.py): pass in openrouter as custom provider for openai client call

Browse source 
Fixes https://github.com/BerriAI/litellm/issues/4414
This commit is contained in:
Krrish Dholakia 2024-06-28 21:26:22 -07:00
parent d0c89ddbe3
commit d10912beeb
2 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/llms/openai.py
View file

@ -678,17 +678,17 @@ class OpenAIChatCompletion(BaseLLM):
if headers:
optional_params["extra_headers"] = headers
if model is None or messages is None:
raise OpenAIError(status_code=422, message=f"Missing model or messages")
raise OpenAIError(status_code=422, message="Missing model or messages")
if not isinstance(timeout, float) and not isinstance(
timeout, httpx.Timeout
):
raise OpenAIError(
status_code=422,
message=f"Timeout needs to be a float or httpx.Timeout",
message="Timeout needs to be a float or httpx.Timeout",
)
if custom_llm_provider != "openai":
if custom_llm_provider is not None and custom_llm_provider != "openai":
model_response.model = f"{custom_llm_provider}/{model}"
# process all OpenAI compatible provider logic here
if custom_llm_provider == "mistral":