zyc
b178d24e73
Some checks failed
Build and Deploy Log Center / build-and-deploy (push) Failing after 5m9s
101 lines
3.1 KiB
Python
101 lines
3.1 KiB
Python
"""
|
|
Bug 数据模型
|
|
"""
|
|
from datetime import datetime
|
|
from typing import Optional, Any
|
|
from enum import Enum
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class BugStatus(str, Enum):
|
|
"""Bug 状态"""
|
|
NEW = "NEW"
|
|
VERIFYING = "VERIFYING"
|
|
CANNOT_REPRODUCE = "CANNOT_REPRODUCE"
|
|
PENDING_FIX = "PENDING_FIX"
|
|
FIXING = "FIXING"
|
|
FIXED = "FIXED"
|
|
FIX_FAILED = "FIX_FAILED"
|
|
VERIFIED = "VERIFIED"
|
|
DEPLOYED = "DEPLOYED"
|
|
|
|
|
|
class ErrorInfo(BaseModel):
|
|
"""错误信息"""
|
|
type: str = Field(description="异常类型")
|
|
message: str = Field(description="错误消息")
|
|
file_path: Optional[str] = Field(default=None, description="文件路径")
|
|
line_number: Optional[int] = Field(default=None, description="行号")
|
|
stack_trace: Optional[list[str]] = Field(default=None, description="堆栈信息")
|
|
|
|
|
|
class Bug(BaseModel):
|
|
"""Bug 数据模型"""
|
|
id: int = Field(description="Bug ID")
|
|
project_id: str = Field(description="项目ID")
|
|
environment: str = Field(default="production", description="环境")
|
|
level: str = Field(default="ERROR", description="日志级别")
|
|
error: ErrorInfo = Field(description="错误信息")
|
|
context: Optional[dict[str, Any]] = Field(default=None, description="上下文")
|
|
status: BugStatus = Field(default=BugStatus.NEW, description="状态")
|
|
retry_count: int = Field(default=0, description="重试次数")
|
|
severity: Optional[int] = Field(default=None, description="严重等级 1-10")
|
|
created_at: Optional[datetime] = Field(default=None, description="创建时间")
|
|
|
|
def format_for_prompt(self) -> str:
|
|
"""格式化为 Prompt 内容"""
|
|
lines = [
|
|
f"## Bug #{self.id}: {self.error.type}",
|
|
f"- 文件: {self.error.file_path or 'unknown'}",
|
|
f"- 行号: {self.error.line_number or 'unknown'}",
|
|
f"- 错误: {self.error.message}",
|
|
]
|
|
|
|
if self.error.stack_trace:
|
|
lines.append("- 堆栈:")
|
|
lines.append("```")
|
|
lines.extend(self.error.stack_trace[:10]) # 最多10行
|
|
lines.append("```")
|
|
|
|
if self.context:
|
|
lines.append(f"- 上下文: {self.context}")
|
|
|
|
return "\n".join(lines)
|
|
|
|
|
|
class FixResult(BaseModel):
|
|
"""修复结果"""
|
|
bug_id: int
|
|
success: bool
|
|
message: str
|
|
modified_files: list[str] = Field(default_factory=list)
|
|
diff: Optional[str] = None
|
|
|
|
|
|
class BatchFixResult(BaseModel):
|
|
"""批量修复结果"""
|
|
project_id: str
|
|
total: int
|
|
success_count: int
|
|
failed_count: int
|
|
results: list[FixResult] = Field(default_factory=list)
|
|
|
|
|
|
class RepairReport(BaseModel):
|
|
"""修复报告（一次批量修复 = 一条报告）"""
|
|
error_log_ids: list[int]
|
|
status: BugStatus
|
|
project_id: str
|
|
ai_analysis: str
|
|
fix_plan: str
|
|
code_diff: str
|
|
modified_files: list[str]
|
|
test_output: str
|
|
test_passed: bool
|
|
repair_round: int = 1
|
|
failure_reason: Optional[str] = None
|
|
pr_url: Optional[str] = None
|
|
pr_number: Optional[int] = None
|
|
branch_name: Optional[str] = None
|
|
|