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.
78 lines
2.2 KiB
Python
78 lines
2.2 KiB
Python
from typing import Optional
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class UserConfig(BaseModel):
|
|
LLM: Optional[str] = None
|
|
|
|
# OpenAI
|
|
OPENAI_API_KEY: Optional[str] = None
|
|
OPENAI_MODEL: Optional[str] = None
|
|
|
|
# Google
|
|
GOOGLE_API_KEY: Optional[str] = None
|
|
GOOGLE_MODEL: Optional[str] = None
|
|
|
|
# Vertex AI
|
|
VERTEX_API_KEY: Optional[str] = None
|
|
VERTEX_MODEL: Optional[str] = None
|
|
VERTEX_PROJECT: Optional[str] = None
|
|
VERTEX_LOCATION: Optional[str] = None
|
|
VERTEX_BASE_URL: Optional[str] = None
|
|
|
|
# Azure OpenAI
|
|
AZURE_OPENAI_API_KEY: Optional[str] = None
|
|
AZURE_OPENAI_MODEL: Optional[str] = None
|
|
AZURE_OPENAI_ENDPOINT: Optional[str] = None
|
|
AZURE_OPENAI_BASE_URL: Optional[str] = None
|
|
AZURE_OPENAI_API_VERSION: Optional[str] = None
|
|
AZURE_OPENAI_DEPLOYMENT: Optional[str] = None
|
|
|
|
# Anthropic
|
|
ANTHROPIC_API_KEY: Optional[str] = None
|
|
ANTHROPIC_MODEL: Optional[str] = None
|
|
|
|
# Ollama
|
|
OLLAMA_URL: Optional[str] = None
|
|
OLLAMA_MODEL: Optional[str] = None
|
|
|
|
# Custom LLM
|
|
CUSTOM_LLM_URL: Optional[str] = None
|
|
CUSTOM_LLM_API_KEY: Optional[str] = None
|
|
CUSTOM_MODEL: Optional[str] = None
|
|
|
|
# Image Provider
|
|
DISABLE_IMAGE_GENERATION: Optional[bool] = None
|
|
IMAGE_PROVIDER: Optional[str] = None
|
|
PEXELS_API_KEY: Optional[str] = None
|
|
PIXABAY_API_KEY: Optional[str] = None
|
|
|
|
# ComfyUI
|
|
COMFYUI_URL: Optional[str] = None
|
|
COMFYUI_WORKFLOW: Optional[str] = None
|
|
|
|
# Open WebUI Image Provider
|
|
OPEN_WEBUI_IMAGE_URL: Optional[str] = None
|
|
OPEN_WEBUI_IMAGE_API_KEY: Optional[str] = None
|
|
|
|
# Dalle 3 Quality
|
|
DALL_E_3_QUALITY: Optional[str] = None
|
|
# Gpt Image 1.5 Quality
|
|
GPT_IMAGE_1_5_QUALITY: Optional[str] = None
|
|
|
|
# Reasoning
|
|
DISABLE_THINKING: Optional[bool] = None
|
|
EXTENDED_REASONING: Optional[bool] = None
|
|
|
|
# Web Search
|
|
WEB_GROUNDING: Optional[bool] = None
|
|
|
|
# Codex OAuth (ChatGPT)
|
|
CODEX_MODEL: Optional[str] = None
|
|
CODEX_ACCESS_TOKEN: Optional[str] = None
|
|
CODEX_REFRESH_TOKEN: Optional[str] = None
|
|
CODEX_TOKEN_EXPIRES: Optional[str] = None
|
|
CODEX_ACCOUNT_ID: Optional[str] = None
|
|
CODEX_USERNAME: Optional[str] = None
|
|
CODEX_EMAIL: Optional[str] = None
|
|
CODEX_IS_PRO: Optional[bool] = None
|