prompt-analyser/prompt_analyzer/config.py

30 lines
851 B
Python

"""Configuration for the Prompt Analyzer."""
import os
from pathlib import Path
from dotenv import load_dotenv
# Load .env from the project root (two levels up from this file)
_project_root = Path(__file__).resolve().parent.parent
load_dotenv(_project_root / ".env")
# Anthropic
ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY")
ANTHROPIC_MODEL = os.getenv("ANTHROPIC_MODEL", "claude-sonnet-4-20250514")
# LLM parameters
LLM_MAX_TOKENS = int(os.getenv("LLM_MAX_TOKENS", "4096"))
LLM_TEMPERATURE = float(os.getenv("LLM_TEMPERATURE", "0.3"))
# Context store
CONTEXT_STORE_DIR = os.getenv(
"CONTEXT_STORE_DIR",
os.path.join(os.path.dirname(os.path.dirname(__file__)), "context_store"),
)
# Analytics DB
ANALYTICS_DB_PATH = os.getenv(
"ANALYTICS_DB_PATH",
os.path.join(os.path.dirname(os.path.dirname(__file__)), "analytics.db"),
)