diff --git a/servers/fastapi/utils/set_env.py b/servers/fastapi/utils/set_env.py index fbfaf221..836c8caa 100644 --- a/servers/fastapi/utils/set_env.py +++ b/servers/fastapi/utils/set_env.py @@ -18,7 +18,7 @@ def set_ollama_url_env(value): def set_custom_llm_url_env(value): - os.environ["CUSTOM_URL"] = value + os.environ["CUSTOM_LLM_URL"] = value def set_openai_api_key_env(value): @@ -44,9 +44,10 @@ def set_custom_model_env(value): def set_pexels_api_key_env(value): os.environ["PEXELS_API_KEY"] = value + def set_image_provider_env(value): os.environ["IMAGE_PROVIDER"] = value def set_pixabay_api_key_env(value): - os.environ["PIXABAY_API_KEY"] = value \ No newline at end of file + os.environ["PIXABAY_API_KEY"] = value