2026-03-04 18:09:45 +00:00
|
|
|
|
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
|
2026-03-09 02:47:52 +00:00
|
|
|
|
UNIT_KEYWORDS: tuple = ("units", "functions", "apis")
|
2026-03-04 18:09:45 +00:00
|
|
|
|
|
|
|
|
|
|
config = Config()
|
