fix - update migration script to ensure api_key in script

This commit is contained in:
Ishaan Jaff 2024-05-22 12:22:23 -07:00
parent 0623eb0274
commit 778b41cea5

View file

@ -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