File size: 3,525 Bytes
ff0e97f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
"""
Configuration for LangGraph agents.
Loads environment variables and manages settings.
"""
import os
from dotenv import load_dotenv
from typing import Optional
load_dotenv()
class AgentConfig:
"""Configuration for bird classification agents."""
# Modal Bird classifier (Phase 1)
MODAL_MCP_URL: str = os.getenv("MODAL_MCP_URL", "")
BIRD_CLASSIFIER_API_KEY: str = os.getenv("BIRD_CLASSIFIER_API_KEY", "")
# eBird Server (Phase 2 - deprecated, replaced by Nuthatch)
EBIRD_API_KEY: str = os.getenv("EBIRD_API_KEY", "") # For Cornell eBird API calls
EBIRD_BASE_URL: str = os.getenv("EBIRD_BASE_URL", "https://api.ebird.org/v2") # Cornell eBird API endpoint
EBIRD_MCP_URL: str = os.getenv("EBIRD_MCP_URL", "http://localhost:8000/mcp") # MCP server endpoint
EBIRD_USE_STDIO: bool = os.getenv("EBIRD_USE_STDIO", "false").lower() == "true"
EBIRD_MCP_AUTH_KEY: Optional[str] = os.getenv("EBIRD_MCP_AUTH_KEY") # For HTTP transport auth (legacy: MCP_API_KEY)
# Nuthatch Server (Phase 2.5 - species reference database)
NUTHATCH_API_KEY: str = os.getenv("NUTHATCH_API_KEY", "") # For Nuthatch API calls
NUTHATCH_BASE_URL: str = os.getenv("NUTHATCH_BASE_URL", "https://nuthatch.lastelm.software/v2") # Nuthatch API endpoint
NUTHATCH_MCP_URL: str = os.getenv("NUTHATCH_MCP_URL", "http://localhost:8001/mcp") # MCP server endpoint
NUTHATCH_USE_STDIO: bool = os.getenv("NUTHATCH_USE_STDIO", "true").lower() == "true"
NUTHATCH_MCP_AUTH_KEY: Optional[str] = os.getenv("NUTHATCH_MCP_AUTH_KEY") # For HTTP transport
# LLM Settings
OPENAI_API_KEY: str = os.getenv("OPENAI_API_KEY", "") # For dev/testing with backend-provided key
DEFAULT_OPENAI_MODEL: str = os.getenv("DEFAULT_OPENAI_MODEL", "gpt-4o-mini")
DEFAULT_HF_MODEL: str = os.getenv("DEFAULT_HF_MODEL", "Qwen/Qwen2.5-Coder-32B-Instruct")
# Anthropic Settings
ANTHROPIC_API_KEY: str = os.getenv("ANTHROPIC_API_KEY", "")
DEFAULT_ANTHROPIC_MODEL: str = os.getenv("DEFAULT_ANTHROPIC_MODEL", "claude-sonnet-4-5-20250929")
ANTHROPIC_TEMPERATURE: float = float(os.getenv("ANTHROPIC_TEMPERATURE", "0.0"))
# Provider-specific temperature settings
OPENAI_TEMPERATURE: float = float(os.getenv("OPENAI_TEMPERATURE", "0.0"))
HF_TEMPERATURE: float = float(os.getenv("HF_TEMPERATURE", "0.1"))
# Agent Settings
MAX_ITERATIONS: int = int(os.getenv("AGENT_MAX_ITERATIONS", "10"))
TIMEOUT: int = int(os.getenv("AGENT_TIMEOUT", "120"))
@classmethod
def print_config(cls) -> None:
"""Print current configuration (hiding sensitive data)."""
print("\n" + "="*70)
print("Agent Configuration")
print("="*70)
print(f"[MODAL URL]: {cls.MODAL_MCP_URL}")
print(f"[MODAL API KEY]: {cls.BIRD_CLASSIFIER_API_KEY[:20]}..." if cls.BIRD_CLASSIFIER_API_KEY else "[MODAL API KEY]: Not set")
print(f"[NUTHATCH STDIO]: {cls.NUTHATCH_USE_STDIO}")
print(f"[NUTHATCH URL]: {cls.NUTHATCH_MCP_URL}")
print(f"[NUTHATCH API KEY]: {'✅ Set' if cls.NUTHATCH_API_KEY else '❌ Not set'}")
print(f"[OPENAI MODEL]: {cls.DEFAULT_OPENAI_MODEL}")
print(f"[OPENAI TEMP]: {cls.OPENAI_TEMPERATURE}")
print(f"[HF MODEL]: {cls.DEFAULT_HF_MODEL}")
print(f"[HF TEMP]: {cls.HF_TEMPERATURE}")
print(f"[ANTHROPIC_MODEL]: {cls.DEFAULT_ANTHROPIC_MODEL}")
print(f"[ANTHROPIC_TEMP]: {cls.ANTHROPIC_TEMPERATURE}")
print("="*70 + "\n")
|