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

fix(main.py): revert model alias map change

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-20 21:07:52 -08:00
parent ab2986290e
commit b991a35ebe
2 changed files with 4 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/main.py
View file

@ -356,7 +356,10 @@ def completion(
if model_list is not None: if model_list is not None:
deployments = [m["litellm_params"] for m in model_list if m["model_name"] == model] deployments = [m["litellm_params"] for m in model_list if m["model_name"] == model]
return batch_completion_models(deployments=deployments, **args) return batch_completion_models(deployments=deployments, **args)
if litellm.model_alias_map and model in litellm.model_alias_map:
model = litellm.model_alias_map[
model
] # update the model to the actual value if an alias has been passed in
model_response = ModelResponse() model_response = ModelResponse()
if kwargs.get('azure', False) == True: # don't remove flag check, to remain backwards compatible for repos like Codium if kwargs.get('azure', False) == True: # don't remove flag check, to remain backwards compatible for repos like Codium

5
litellm/utils.py
View file

@ -2246,11 +2246,6 @@ def get_llm_provider(model: str, custom_llm_provider: Optional[str] = None, api_
dynamic_api_key = None dynamic_api_key = None
# check if llm provider provided # check if llm provider provided
if litellm.model_alias_map and model in litellm.model_alias_map:
model = litellm.model_alias_map[
model
] # update the model to the actual value if an alias has been passed in
if custom_llm_provider: if custom_llm_provider:
return model, custom_llm_provider, dynamic_api_key, api_base return model, custom_llm_provider, dynamic_api_key, api_base