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

clean up azure implementation

Browse source
This commit is contained in:
ishaan-jaff 2023-09-05 15:31:57 -07:00
parent 463e57d0a8
commit 2250d1375e
5 changed files with 61 additions and 90 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/main.py
View file

@ -115,10 +115,9 @@ def completion(
model
] # update the model to the actual value if an alias has been passed in
model_response = ModelResponse()
if azure: # this flag is deprecated, remove once notebooks are also updated.
custom_llm_provider = "azure"
if deployment_id:
if deployment_id != None:
model=deployment_id
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"