From 778b41cea52ea39628048ac1cecbc001ffb1f247 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 22 May 2024 12:22:23 -0700 Subject: [PATCH] fix - update migration script to ensure api_key in script --- cookbook/misc/migrate_proxy_config.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/cookbook/misc/migrate_proxy_config.py b/cookbook/misc/migrate_proxy_config.py index f1d736dc8..53551a0ce 100644 --- a/cookbook/misc/migrate_proxy_config.py +++ b/cookbook/misc/migrate_proxy_config.py @@ -54,6 +54,9 @@ def migrate_models(config_file, proxy_base_url): new_value = input(f"Enter value for {value}: ") _in_memory_os_variables[value] = new_value litellm_params[param] = new_value + if "api_key" not in litellm_params: + new_value = input(f"Enter api key for {model_name}: ") + litellm_params["api_key"] = new_value print("\nlitellm_params: ", litellm_params) # Confirm before sending POST request