mirror of
https://github.com/open-webui/computer.git
synced 2026-11-14 18:08:57 +03:00
Timothy Jaeryang Baek
58 lines
3.1 KiB
Python
58 lines
3.1 KiB
Python
"""Centralized environment configuration for cptr.
|
|
|
|
All environment variable reads go here. Import from this module
|
|
instead of reading os.environ directly.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
from pathlib import Path
|
|
|
|
# ── Data directory ──────────────────────────────────────────
|
|
# Where cptr stores its database, config, and user data.
|
|
# Default: ~/.cptr
|
|
DATA_DIR = Path(os.environ.get("CPTR_DATA_DIR", str(Path.home() / ".cptr")))
|
|
CONFIG_FILE = DATA_DIR / "config.toml"
|
|
DB_FILE = DATA_DIR / "app.db"
|
|
|
|
# ── Startup token ───────────────────────────────────────────
|
|
# One-time token for first-time setup. Set by CLI, consumed by app.
|
|
STARTUP_TOKEN: str | None = os.environ.pop("CPTR_STARTUP_TOKEN", None)
|
|
|
|
# ── Chat settings ───────────────────────────────────────────
|
|
CHAT_MAX_ITERATIONS = int(os.environ.get("CHAT_MAX_ITERATIONS", "2048"))
|
|
ENABLE_CHAT_RECONCILE_ON_STARTUP: bool = os.environ.get(
|
|
"ENABLE_CHAT_RECONCILE_ON_STARTUP", "true"
|
|
).lower() in ("true", "1", "yes")
|
|
CHAT_TOOL_MAX_CHARS = int(os.environ.get("CHAT_TOOL_MAX_CHARS", "50000"))
|
|
CHAT_TOOL_COMMAND_MAX_CHARS = int(os.environ.get("CHAT_TOOL_COMMAND_MAX_CHARS", "8000"))
|
|
CHAT_COMPACT_TOKEN_THRESHOLD = int(os.environ.get("CHAT_COMPACT_TOKEN_THRESHOLD", "80000"))
|
|
|
|
# ── Execute timeout ─────────────────────────────────────────
|
|
# Default wait (seconds) for run_command / check_task when the caller
|
|
# doesn't pass an explicit wait value. None = return immediately.
|
|
EXECUTE_TIMEOUT: float | None = None
|
|
_execute_timeout = os.environ.get("CPTR_EXECUTE_TIMEOUT")
|
|
if _execute_timeout is not None:
|
|
EXECUTE_TIMEOUT = float(_execute_timeout)
|
|
|
|
# ── AI stream settings ──────────────────────────────────────
|
|
STREAM_CONNECT_TIMEOUT_SECONDS = float(os.environ.get("CPTR_STREAM_CONNECT_TIMEOUT", "30"))
|
|
STREAM_READ_TIMEOUT_SECONDS = float(os.environ.get("CPTR_STREAM_READ_TIMEOUT", "300"))
|
|
STREAM_WRITE_TIMEOUT_SECONDS = float(os.environ.get("CPTR_STREAM_WRITE_TIMEOUT", "600"))
|
|
|
|
# ── Automation scheduler ────────────────────────────────────
|
|
AUTOMATION_POLL_INTERVAL = int(os.environ.get("AUTOMATION_POLL_INTERVAL", "10"))
|
|
|
|
# ── CORS ────────────────────────────────────────────────────
|
|
# Socket.IO CORS allowed origins.
|
|
# Default → "*" (allow all origins)
|
|
# Comma-separated list → allow specific origins only
|
|
# e.g. "https://example.com,https://app.example.com"
|
|
_cors_raw = os.environ.get("CPTR_CORS_ALLOWED_ORIGINS", "*")
|
|
if _cors_raw.strip() == "*":
|
|
CORS_ALLOWED_ORIGINS = "*"
|
|
else:
|
|
CORS_ALLOWED_ORIGINS = [o.strip() for o in _cors_raw.split(",") if o.strip()] or "*"
|