phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

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 69c095b768
commit 240aaf7af8
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") model_name = model.get("model_name")
print("\nAdding model: ", model_name) print("\nAdding model: ", model_name)
litellm_params = model.get("litellm_params", {}) 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 "" litellm_model_name = litellm_params.get("model", "") or ""
if "vertex_ai/" in litellm_model_name: if "vertex_ai/" in litellm_model_name: