21 lines
696 B
Python
21 lines
696 B
Python
|
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()
|