forked from 0xWheatyz/SPARC
agent-company
b000146585
- Make LLM model configurable via MODEL env var, default anthropic/claude-3.5-sonnet (#12) - Expose SERP cache TTL as SERP_CACHE_TTL_HOURS env var, default 24 hours (#13) - Fix Patent.patent_id type annotation from int to str in types.py (#14) - Replace all print() calls with structured logging in analyzer.py and llm.py (#11) - Add LOG_LEVEL config with basicConfig setup in config.py - Add model and serp_cache_ttl_hours to config.py Closes leeworks-agents/SPARC#11 Closes leeworks-agents/SPARC#12 Closes leeworks-agents/SPARC#13 Closes leeworks-agents/SPARC#14 Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
64 lines
2.4 KiB
Python
64 lines
2.4 KiB
Python
"""Configuration and secrets management.
|
|
|
|
Loads environment variables from .env file for API keys and other secrets.
|
|
"""
|
|
import logging
|
|
import os
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
# Logging configuration
|
|
log_level = os.getenv("LOG_LEVEL", "INFO").upper()
|
|
logging.basicConfig(
|
|
level=getattr(logging, log_level, logging.INFO),
|
|
format="%(asctime)s %(levelname)s %(name)s %(message)s",
|
|
)
|
|
|
|
# SerpAPI key for patent search
|
|
api_key = os.getenv("API_KEY")
|
|
|
|
# OpenRouter API key for LLM analysis
|
|
openrouter_api_key = os.getenv("OPENROUTER_API_KEY")
|
|
|
|
# Database configuration - all messages are stored in the database
|
|
# The database serves as both a persistent store and a cache layer
|
|
database_url = os.getenv("DATABASE_URL", "postgresql://postgres:postgres@localhost:5432/sparc")
|
|
|
|
# Cache configuration
|
|
# When enabled (default), the system checks the database for cached responses
|
|
# before making API calls, saving tokens and reducing latency
|
|
use_cache = os.getenv("USE_CACHE", "true").lower() in ("true", "1", "yes")
|
|
|
|
# Legacy compatibility - USE_DATABASE is deprecated, database is always used
|
|
# This variable is kept for backwards compatibility but has no effect
|
|
use_database = os.getenv("USE_DATABASE", "false").lower() in ("true", "1", "yes")
|
|
|
|
# Patent search configuration
|
|
patent_search_days = int(os.getenv("PATENT_SEARCH_DAYS", "90"))
|
|
patent_thread_workers = int(os.getenv("PATENT_THREAD_WORKERS", "5"))
|
|
|
|
# LLM model to use via OpenRouter (e.g. "anthropic/claude-3.5-sonnet", "openai/gpt-4o")
|
|
model = os.getenv("MODEL", "anthropic/claude-3.5-sonnet")
|
|
|
|
# SERP cache TTL in hours (how long cached search results are considered fresh)
|
|
serp_cache_ttl_hours = int(os.getenv("SERP_CACHE_TTL_HOURS", "24"))
|
|
|
|
# Root path for running behind a reverse proxy (e.g., "/api" when served at /api/)
|
|
# This ensures OpenAPI docs work correctly when accessed via the proxy
|
|
root_path = os.getenv("ROOT_PATH", "")
|
|
|
|
# Application environment: "development", "staging", or "production"
|
|
# Used for safety checks (e.g., refusing default JWT secret in production)
|
|
app_env = os.getenv("APP_ENV", "development")
|
|
|
|
# CORS allowed origins (comma-separated)
|
|
# Defaults to localhost dev origins when unset
|
|
_cors_origins_raw = os.getenv("CORS_ORIGINS", "")
|
|
cors_origins: list[str] = (
|
|
[o.strip() for o in _cors_origins_raw.split(",") if o.strip()]
|
|
if _cors_origins_raw
|
|
else ["http://localhost:3000", "http://localhost:5173"]
|
|
)
|