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

fix - show api_base in migration script

Browse source
This commit is contained in:
Ishaan Jaff 2024-04-11 09:26:13 -07:00
parent cd5bf9e319
commit 1dc0d0fa33
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
cookbook/misc/migrate_proxy_config.py
View file

@ -29,6 +29,8 @@ def migrate_models(config_file, proxy_base_url):
model_name = model.get("model_name")
print("\nAdding model: ", model_name)
litellm_params = model.get("litellm_params", {})
api_base = litellm_params.get("api_base", "")
print("api_base on config.yaml: ", api_base)
litellm_model_name = litellm_params.get("model", "") or ""
if "vertex_ai/" in litellm_model_name: