杜宇轩
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
import os
|
|
|
|
try:
|
|
# Optional: load a local .env file if python-dotenv is installed
|
|
from dotenv import load_dotenv
|
|
load_dotenv()
|
|
|
|
except ImportError:
|
|
pass
|
|
|
|
|
|
|
|
# ===== Model backend selection =====
|
|
MODEL_PROVIDER = os.getenv("MODEL_PROVIDER", "openai").strip().lower()
|
|
MODEL_ID = os.getenv("MODEL_ID", "gpt-4o-mini").strip()
|
|
|
|
MAX_TOKENS = int(os.getenv("MAX_TOKENS", 1024))
|
|
TEMPERATURE = float(os.getenv("TEMPERATURE", 0.7))
|
|
|
|
|
|
# ===== Provider credentials / endpoints =====
|
|
|
|
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "").strip()
|
|
OPENAI_API_BASE_URL = os.getenv("OPENAI_API_BASE", "https://api.openai.com/v1").strip()
|
|
|
|
ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY", "").strip()
|
|
|
|
OLLAMA_HOST = os.getenv("OLLAMA_HOST", "http://localhost:11434").strip()
|
|
|
|
AWS_REGION = os.getenv("AWS_REGION", "us-east-1").strip()
|
|
|
|
|
|
# ===== Conversation / history =====
|
|
CONV_MAX_BYTES = int(os.getenv("CONV_MAX_BYTES", 1024 * 1024)) # 1MB
|
|
|
|
|
|
# ===== History backend: "json" (one file per session) or "sqlite" (single SQLite database) =====
|
|
SESSION_BACKEND = os.getenv("SESSION_BACKEND", "json").strip().lower()
|
|
SESSION_DIR = os.getenv("SESSION_DIR", "./sessions").strip()
|
|
SQLITE_PATH = os.getenv("SQLITE_PATH", "./sessions/agent.db").strip() |