完全采用交互式使用方式;
This commit is contained in:
parent
ef9a846b51
commit
cb479138af
|
|
@ -1,6 +1,7 @@
|
|||
# core/requirement_analyzer.py - 需求分解、模块分类、函数签名生成
|
||||
import json
|
||||
from typing import List, Optional, Callable
|
||||
|
||||
import config
|
||||
from core.llm_client import LLMClient
|
||||
from database.models import FunctionalRequirement, ChangeHistory
|
||||
|
|
@ -25,6 +26,7 @@ class RequirementAnalyzer:
|
|||
project_id: int,
|
||||
raw_req_id: int,
|
||||
knowledge: str = "",
|
||||
start_index: int = 1,
|
||||
) -> List[FunctionalRequirement]:
|
||||
"""
|
||||
将原始需求文本分解为功能需求列表(含模块分类)。
|
||||
|
|
@ -34,6 +36,7 @@ class RequirementAnalyzer:
|
|||
project_id: 所属项目 ID
|
||||
raw_req_id: 原始需求记录 ID
|
||||
knowledge: 知识库文本(可选)
|
||||
start_index: 功能需求序号起始值(新增需求时传入当前最大序号+1)
|
||||
|
||||
Returns:
|
||||
FunctionalRequirement 对象列表(未持久化,id=None)
|
||||
|
|
@ -54,7 +57,7 @@ class RequirementAnalyzer:
|
|||
raise RuntimeError(f"需求分解失败: {e}")
|
||||
|
||||
reqs = []
|
||||
for i, item in enumerate(items, 1):
|
||||
for i, item in enumerate(items, start_index):
|
||||
req = FunctionalRequirement(
|
||||
project_id = project_id,
|
||||
raw_req_id = raw_req_id,
|
||||
|
|
@ -70,6 +73,41 @@ class RequirementAnalyzer:
|
|||
reqs.append(req)
|
||||
return reqs
|
||||
|
||||
# ══════════════════════════════════════════════════
|
||||
# 新增需求(追加到已有功能需求列表)
|
||||
# ══════════════════════════════════════════════════
|
||||
|
||||
def add_new_requirements(
|
||||
self,
|
||||
new_requirement_text: str,
|
||||
project_id: int,
|
||||
raw_req_id: int,
|
||||
existing_reqs: List[FunctionalRequirement],
|
||||
knowledge: str = "",
|
||||
) -> List[FunctionalRequirement]:
|
||||
"""
|
||||
将新需求文本分解为功能需求,序号接续已有需求,返回新增部分列表(未持久化)。
|
||||
|
||||
Args:
|
||||
new_requirement_text: 新增需求描述文本
|
||||
project_id: 所属项目 ID
|
||||
raw_req_id: 原始需求记录 ID
|
||||
existing_reqs: 已有功能需求列表(用于计算起始序号)
|
||||
knowledge: 知识库文本(可选)
|
||||
|
||||
Returns:
|
||||
新增的 FunctionalRequirement 列表(未持久化)
|
||||
"""
|
||||
start_index = max((r.index_no for r in existing_reqs), default=0) + 1
|
||||
new_reqs = self.decompose(
|
||||
raw_requirement = new_requirement_text,
|
||||
project_id = project_id,
|
||||
raw_req_id = raw_req_id,
|
||||
knowledge = knowledge,
|
||||
start_index = start_index,
|
||||
)
|
||||
return new_reqs
|
||||
|
||||
# ══════════════════════════════════════════════════
|
||||
# 模块分类(独立步骤,可对已有需求列表重新分类)
|
||||
# ══════════════════════════════════════════════════
|
||||
|
|
@ -148,7 +186,6 @@ class RequirementAnalyzer:
|
|||
sig = self.llm.chat_json(prompt)
|
||||
if not isinstance(sig, dict):
|
||||
raise ValueError("LLM 返回结果不是 dict")
|
||||
# 确保 module 字段存在
|
||||
if "module" not in sig:
|
||||
sig["module"] = func_req.module or config.DEFAULT_MODULE
|
||||
return sig
|
||||
|
|
@ -211,7 +248,7 @@ class RequirementAnalyzer:
|
|||
}
|
||||
|
||||
# ══════════════════════════════════════════════════
|
||||
# 记录变更
|
||||
# 变更记录与分析
|
||||
# ══════════════════════════════════════════════════
|
||||
|
||||
def log_change(self, project_id: int, changes: str) -> None:
|
||||
|
|
@ -223,22 +260,32 @@ class RequirementAnalyzer:
|
|||
"""查询项目变更历史"""
|
||||
return db.list_change_history(project_id)
|
||||
|
||||
def analyze_changes(self, old_reqs: List[FunctionalRequirement], new_reqs: List[FunctionalRequirement]) -> List[
|
||||
str]:
|
||||
def analyze_changes(
|
||||
self,
|
||||
old_reqs: List[FunctionalRequirement],
|
||||
new_reqs: List[FunctionalRequirement],
|
||||
) -> List[str]:
|
||||
"""
|
||||
分析需求变更,返回需要变更的代码文件列表。
|
||||
分析需求变更,返回受影响的函数名列表。
|
||||
|
||||
Args:
|
||||
old_reqs: 旧的功能需求列表
|
||||
new_reqs: 新的功能需求列表
|
||||
new_reqs: 变更/新增后的功能需求列表
|
||||
|
||||
Returns:
|
||||
需要变更的文件列表
|
||||
受影响的 function_name 列表
|
||||
"""
|
||||
changed_files = []
|
||||
old_func_names = {req.function_name: req for req in old_reqs}
|
||||
changed_funcs = []
|
||||
old_func_map = {req.function_name: req for req in old_reqs}
|
||||
|
||||
for new_req in new_reqs:
|
||||
old_req = old_func_names.get(new_req.function_name)
|
||||
if not old_req or old_req.description != new_req.description or old_req.module != new_req.module:
|
||||
changed_files.append(new_req.function_name)
|
||||
return changed_files
|
||||
old_req = old_func_map.get(new_req.function_name)
|
||||
# 新增需求 或 描述/模块有变化
|
||||
if (
|
||||
not old_req
|
||||
or old_req.description != new_req.description
|
||||
or old_req.module != new_req.module
|
||||
):
|
||||
changed_funcs.append(new_req.function_name)
|
||||
|
||||
return changed_funcs
|
||||
File diff suppressed because it is too large
Load Diff
Loading…
Reference in New Issue