b20199a4e3
- Added environment variables for Vertex AI and Azure OpenAI configurations in docker-compose and user configuration models. - Updated the application logic to handle Vertex and Azure as new LLM providers, including validation and API key management. - Enhanced the UI components to support model selection and API key input for Vertex and Azure. - Updated relevant utility functions and constants to accommodate the new providers. - Ensured proper error handling for configuration requirements specific to Vertex and Azure.
187 lines
3.8 KiB
Python
187 lines
3.8 KiB
Python
import os
|
|
|
|
|
|
def set_temp_directory_env(value):
|
|
os.environ["TEMP_DIRECTORY"] = value
|
|
|
|
|
|
def set_user_config_path_env(value):
|
|
os.environ["USER_CONFIG_PATH"] = value
|
|
|
|
|
|
def set_llm_provider_env(value):
|
|
os.environ["LLM"] = value
|
|
|
|
|
|
def set_ollama_url_env(value):
|
|
os.environ["OLLAMA_URL"] = value
|
|
|
|
|
|
def set_custom_llm_url_env(value):
|
|
os.environ["CUSTOM_LLM_URL"] = value
|
|
|
|
|
|
def set_openai_api_key_env(value):
|
|
os.environ["OPENAI_API_KEY"] = value
|
|
|
|
|
|
def set_openai_model_env(value):
|
|
os.environ["OPENAI_MODEL"] = value
|
|
|
|
|
|
def set_google_api_key_env(value):
|
|
os.environ["GOOGLE_API_KEY"] = value
|
|
|
|
|
|
def set_google_model_env(value):
|
|
os.environ["GOOGLE_MODEL"] = value
|
|
|
|
|
|
def set_vertex_api_key_env(value):
|
|
os.environ["VERTEX_API_KEY"] = value
|
|
|
|
|
|
def set_vertex_model_env(value):
|
|
os.environ["VERTEX_MODEL"] = value
|
|
|
|
|
|
def set_vertex_project_env(value):
|
|
os.environ["VERTEX_PROJECT"] = value
|
|
|
|
|
|
def set_vertex_location_env(value):
|
|
os.environ["VERTEX_LOCATION"] = value
|
|
|
|
|
|
def set_vertex_base_url_env(value):
|
|
os.environ["VERTEX_BASE_URL"] = value
|
|
|
|
|
|
def set_azure_openai_api_key_env(value):
|
|
os.environ["AZURE_OPENAI_API_KEY"] = value
|
|
|
|
|
|
def set_azure_openai_model_env(value):
|
|
os.environ["AZURE_OPENAI_MODEL"] = value
|
|
|
|
|
|
def set_azure_openai_endpoint_env(value):
|
|
os.environ["AZURE_OPENAI_ENDPOINT"] = value
|
|
|
|
|
|
def set_azure_openai_base_url_env(value):
|
|
os.environ["AZURE_OPENAI_BASE_URL"] = value
|
|
|
|
|
|
def set_azure_openai_api_version_env(value):
|
|
os.environ["AZURE_OPENAI_API_VERSION"] = value
|
|
|
|
|
|
def set_azure_openai_deployment_env(value):
|
|
os.environ["AZURE_OPENAI_DEPLOYMENT"] = value
|
|
|
|
|
|
def set_anthropic_api_key_env(value):
|
|
os.environ["ANTHROPIC_API_KEY"] = value
|
|
|
|
|
|
def set_anthropic_model_env(value):
|
|
os.environ["ANTHROPIC_MODEL"] = value
|
|
|
|
|
|
def set_custom_llm_api_key_env(value):
|
|
os.environ["CUSTOM_LLM_API_KEY"] = value
|
|
|
|
|
|
def set_ollama_model_env(value):
|
|
os.environ["OLLAMA_MODEL"] = value
|
|
|
|
|
|
def set_custom_model_env(value):
|
|
os.environ["CUSTOM_MODEL"] = 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
|
|
|
|
|
|
def set_disable_image_generation_env(value):
|
|
os.environ["DISABLE_IMAGE_GENERATION"] = value
|
|
|
|
|
|
def set_disable_thinking_env(value):
|
|
os.environ["DISABLE_THINKING"] = value
|
|
|
|
|
|
def set_extended_reasoning_env(value):
|
|
os.environ["EXTENDED_REASONING"] = value
|
|
|
|
|
|
def set_web_grounding_env(value):
|
|
os.environ["WEB_GROUNDING"] = value
|
|
|
|
|
|
def set_comfyui_url_env(value):
|
|
os.environ["COMFYUI_URL"] = value
|
|
|
|
|
|
def set_comfyui_workflow_env(value):
|
|
os.environ["COMFYUI_WORKFLOW"] = value
|
|
|
|
|
|
def set_dall_e_3_quality_env(value):
|
|
os.environ["DALL_E_3_QUALITY"] = value
|
|
|
|
|
|
def set_gpt_image_1_5_quality_env(value):
|
|
os.environ["GPT_IMAGE_1_5_QUALITY"] = value
|
|
|
|
|
|
# Codex OAuth
|
|
def set_codex_access_token_env(value: str):
|
|
os.environ["CODEX_ACCESS_TOKEN"] = value
|
|
|
|
|
|
def set_codex_refresh_token_env(value: str):
|
|
os.environ["CODEX_REFRESH_TOKEN"] = value
|
|
|
|
|
|
def set_codex_token_expires_env(value: str):
|
|
os.environ["CODEX_TOKEN_EXPIRES"] = value
|
|
|
|
|
|
def set_codex_account_id_env(value: str):
|
|
os.environ["CODEX_ACCOUNT_ID"] = value
|
|
|
|
|
|
def set_codex_username_env(value: str):
|
|
os.environ["CODEX_USERNAME"] = value
|
|
|
|
|
|
def set_codex_email_env(value: str):
|
|
os.environ["CODEX_EMAIL"] = value
|
|
|
|
|
|
def set_codex_is_pro_env(value: str):
|
|
os.environ["CODEX_IS_PRO"] = value
|
|
|
|
|
|
def set_codex_model_env(value: str):
|
|
os.environ["CODEX_MODEL"] = value
|
|
|
|
|
|
# Open WebUI Image Provider
|
|
def set_open_webui_image_url_env(value: str):
|
|
os.environ["OPEN_WEBUI_IMAGE_URL"] = value
|
|
|
|
|
|
def set_open_webui_image_api_key_env(value: str):
|
|
os.environ["OPEN_WEBUI_IMAGE_API_KEY"] = value
|