From 240aaf7af818d6fdb3472c2127aaabd837023b00 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 11 Apr 2024 09:26:13 -0700 Subject: [PATCH] fix - show api_base in migration script --- cookbook/misc/migrate_proxy_config.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/cookbook/misc/migrate_proxy_config.py b/cookbook/misc/migrate_proxy_config.py index 2b0f1e031..d3f05431d 100644 --- a/cookbook/misc/migrate_proxy_config.py +++ b/cookbook/misc/migrate_proxy_config.py @@ -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: