00be7ee946
- Added Codex-related fields to UserConfig and LLMConfig for managing OAuth tokens and model settings. - Implemented Codex authentication endpoints and integrated them into the application. - Enhanced LLMClient to support Codex as a valid provider, including token management and refresh logic. - Updated UI components to include Codex configuration options and authentication status handling. - Refactored environment variable utilities to accommodate Codex-specific settings.
57 lines
1.5 KiB
Python
57 lines
1.5 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
|
|
|
|
# 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
|
|
|
|
# Dalle 3 Quality
|
|
DALL_E_3_QUALITY: Optional[str] = None
|
|
# Gpt Image 1.5 Quality
|
|
GPT_IMAGE_1_5_QUALITY: Optional[str] = None
|
|
|
|
# Reasoning
|
|
TOOL_CALLS: Optional[bool] = None
|
|
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
|