d2e85a8ffa
- Added Codex authentication endpoints and logic for OAuth flow. - Updated Docker configuration to expose port 1455 for Codex callback. - Enhanced user configuration model to include Codex-related fields. - Integrated Codex selection into the LLM provider UI. - Implemented token management and refresh logic for Codex. - Added utility functions for handling Codex OAuth tokens and state management.
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
|