Claude
1bd11e6acd
Fixes age validation errors caused by the LLM not knowing the current date.
Changes:
- prompts.py: inject date.today() at the top of both system prompts so the
LLM can accurately calculate a child's age from their date of birth
- llm.py: add optional thinking_budget parameter to complete(); when set,
passes thinking={"type": "enabled", "budget_tokens": N} to litellm and
raises max_tokens to thinking_budget + 4096 (Anthropic models only)
- config.py: add thinking_budget field, read from THINKING_BUDGET env var
- .env.example: document the THINKING_BUDGET option
- core.py: pass thinking_budget through to llm.complete()
- main.py: pass thinking_budget when constructing EmailAgent
- chat_app.py: switch from stream_complete to asyncio.to_thread(complete)
so extended thinking works and so only the reply field is shown to
the parent (not the raw JSON wrapper)
To enable extended thinking set THINKING_BUDGET=8000 in .env.
https://claude.ai/code/session_01SUWzMzFvSfWiHXA2p6rPg9
88 lines
3.2 KiB
Python
88 lines
3.2 KiB
Python
"""Configuration loaded from environment variables."""
|
|
|
|
import os
|
|
from dataclasses import dataclass, field
|
|
from pathlib import Path
|
|
|
|
try:
|
|
from dotenv import load_dotenv
|
|
load_dotenv()
|
|
except ImportError:
|
|
pass # python-dotenv is optional
|
|
|
|
|
|
@dataclass
|
|
class Config:
|
|
# AI model — litellm format, e.g. "anthropic/claude-opus-4-6" or "openai/gpt-4o".
|
|
# The matching API key must be set as an env var (ANTHROPIC_API_KEY, OPENAI_API_KEY, …).
|
|
ai_model: str = "anthropic/claude-opus-4-6"
|
|
|
|
# Email — IMAP (receiving)
|
|
imap_host: str = ""
|
|
imap_port: int = 993
|
|
imap_username: str = ""
|
|
imap_password: str = ""
|
|
imap_use_ssl: bool = True
|
|
|
|
# Email — SMTP (sending)
|
|
smtp_host: str = ""
|
|
smtp_port: int = 587
|
|
smtp_use_tls: bool = True
|
|
|
|
# Registration email address shown to parents
|
|
registration_email: str = ""
|
|
|
|
# Admin notification routing.
|
|
# Each leader receives mail only when a day in their group is booked.
|
|
# For testing, point all three to your own address.
|
|
admin_email_indoor: str = "" # Indoor leader (Andrea Sigrist) — To when indoor booked
|
|
admin_email_outdoor: str = "" # Outdoor leader (Barbara Gross) — To when outdoor booked
|
|
admin_email_cc: str = "" # Always Cc'd (Markus Graf / admin); comma-separated if multiple
|
|
|
|
# Storage
|
|
data_dir: Path = field(default_factory=lambda: Path("data"))
|
|
knowledge_base_dir: Path = field(
|
|
default_factory=lambda: Path(
|
|
"openspec/changes/define-project-scope/content/knowledge-base"
|
|
)
|
|
)
|
|
|
|
# Polling interval in seconds
|
|
poll_interval: int = 60
|
|
|
|
# Extended thinking — Anthropic models only.
|
|
# When set, enables the thinking phase before the LLM replies.
|
|
# Recommended value: 8000 (tokens). Set to None/unset to disable.
|
|
thinking_budget: int | None = None
|
|
|
|
@classmethod
|
|
def from_env(cls) -> "Config":
|
|
return cls(
|
|
ai_model=os.getenv("AI_MODEL", "anthropic/claude-opus-4-6"),
|
|
imap_host=os.getenv("IMAP_HOST", ""),
|
|
imap_port=int(os.getenv("IMAP_PORT", "993")),
|
|
imap_username=os.getenv("IMAP_USERNAME", ""),
|
|
imap_password=os.getenv("IMAP_PASSWORD", ""),
|
|
imap_use_ssl=os.getenv("IMAP_USE_SSL", "true").lower() == "true",
|
|
smtp_host=os.getenv("SMTP_HOST", ""),
|
|
smtp_port=int(os.getenv("SMTP_PORT", "587")),
|
|
smtp_use_tls=os.getenv("SMTP_USE_TLS", "true").lower() == "true",
|
|
registration_email=os.getenv("REGISTRATION_EMAIL", ""),
|
|
admin_email_indoor=os.getenv("ADMIN_EMAIL_INDOOR", ""),
|
|
admin_email_outdoor=os.getenv("ADMIN_EMAIL_OUTDOOR", ""),
|
|
admin_email_cc=os.getenv("ADMIN_EMAIL_CC", ""),
|
|
data_dir=Path(os.getenv("DATA_DIR", "data")),
|
|
knowledge_base_dir=Path(
|
|
os.getenv(
|
|
"KNOWLEDGE_BASE_DIR",
|
|
"openspec/changes/define-project-scope/content/knowledge-base",
|
|
)
|
|
),
|
|
poll_interval=int(os.getenv("POLL_INTERVAL", "60")),
|
|
thinking_budget=(
|
|
int(os.getenv("THINKING_BUDGET"))
|
|
if os.getenv("THINKING_BUDGET")
|
|
else None
|
|
),
|
|
)
|