phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

allow custom provider via model name

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-19 16:34:52 -07:00
parent 245b00f5f5
commit 6cee3b22ab
11 changed files with 15 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/main.py
View file

@ -94,6 +94,9 @@ def completion(
model_response = ModelResponse()
if azure: # this flag is deprecated, remove once notebooks are also updated.
custom_llm_provider = "azure"
elif model.split("/", 1)[0] in litellm.provider_list: # allow custom provider to be passed in via the model name "azure/chatgpt-test"
custom_llm_provider = model.split("/", 1)[0]
model = model.split("/", 1)[1]
args = locals()
# check if user passed in any of the OpenAI optional params
optional_params = get_optional_params(