fix(openai.py): return model name with custom llm provider for openai compatible endpoints

2024-03-12 10:30:10 -07:00 · 2024-03-12 10:30:10 -07:00 · 0033613b9e
commit 0033613b9e
parent 10f5f342bd
4 changed files with 5 additions and 1 deletions
--- a/litellm/main.py
+++ b/litellm/main.py
@ -875,6 +875,7 @@ def completion(
                    custom_prompt_dict=custom_prompt_dict,
                    client=client,  # pass AsyncOpenAI, OpenAI client
                    organization=organization,
+                    custom_llm_provider=custom_llm_provider,
                )
            except Exception as e:
                ## LOGGING - log the original exception returned