import os
from dataclasses import dataclass

@dataclass
class Config:
    # LLM配置（以OpenAI兼容接口为例，可替换为其他LLM）
    LLM_API_KEY: str = os.getenv("LLM_API_KEY", "your-api-key-here")
    LLM_BASE_URL: str = os.getenv("LLM_BASE_URL", "https://api.openai.com/v1")
    LLM_MODEL: str = os.getenv("LLM_MODEL", "gpt-4o")
    LLM_TEMPERATURE: float = 0.2
    LLM_MAX_TOKENS: int = 4096

    # 测试配置
    GENERATED_TESTS_DIR: str = "generated_tests"
    TEST_TIMEOUT: int = 30          # 单个测试超时时间（秒）

    # HTTP测试配置
    HTTP_BASE_URL: str = os.getenv("HTTP_BASE_URL", "http://localhost:8080")
    HTTP_TIMEOUT: int = 10

config = Config()