hello-agents/Co-creation-projects/Shawnxyxy-HealthRecordAgent/backend/core/config.py

"""
HealthAgent 核心配置模块
"""

from dataclasses import dataclass, field
from typing import Optional
import os
from dotenv import load_dotenv

load_dotenv()

# ========== LLM ==========
@dataclass
class LLMConfig:
    model_name: str = field(
        default_factory=lambda: os.getenv("OPENAI_MODEL_ID", "qwen-turbo")
    )
    api_key: Optional[str] = field(
        default_factory=lambda: os.getenv("OPENAI_API_KEY")
    )
    base_url: Optional[str] = field(
        default_factory=lambda: os.getenv("OPENAI_BASE_URL")
    )
    temperature: float = 0.7
    max_tokens: int = 2048
    timeout: int = 60
# ========== Agent ==========
@dataclass
class AgentConfig:
    max_steps: int = 5
    timeout: int = 300
    history_limit: int = 50
# ========== RAG ==========
@dataclass
class RAGConfig:
    enabled: bool = False
    top_k: int = 5
# ========== App ==========
@dataclass
class AppConfig:
    app_name: str = "HealthRecordAgent"
    debug: bool = False
    log_level: str = "INFO"

# ========== Main ==========
@dataclass
class HealthAgentConfig:
    app: AppConfig = field(default_factory=AppConfig)
    llm: LLMConfig = field(default_factory=LLMConfig)
    agent: AgentConfig = field(default_factory=AgentConfig)
    rag: RAGConfig = field(default_factory=RAGConfig)

# 全局配置
_config = HealthAgentConfig()


def get_config() -> HealthAgentConfig:
    return _config