Commit 34f37d76 by ccran

feat: add memory filter to merge;

parent 5a0ce740
No preview for this file type
......@@ -30,6 +30,7 @@ from core.tools.rule_filter import LufaPartyRuleFilterTool,JpRuleFilterTool,Rule
from core.tools.rules_retrieve_reference import RulesRetrieveReferenceTool
from core.tools.reflect_retry import ReflectRetryTool
from core.tools.segment_merger import SegmentMergerTool
from core.tools.memory_export_filter import MemoryExportFilter
from core.tools.fact_merger import FactMergerTool
from core.tools.ruleset_router import RulesetRouterTool
from core.tools.party_role import PartyRoleTool
......@@ -52,6 +53,7 @@ rule_filter_tool = RuleFilterTool()
rules_reference_tool = RulesRetrieveReferenceTool()
reflect_tool = ReflectRetryTool()
merger_tool = SegmentMergerTool()
memory_export_filter = MemoryExportFilter()
fact_merger_tool = FactMergerTool()
ruleset_router_tool = RulesetRouterTool()
party_role_tool = PartyRoleTool()
......@@ -606,7 +608,9 @@ def merge_segment_findings(payload: MergerRequest) -> MergerResponse:
source_key = payload.finding_key or FINDING_KEY_REVIEW
target_segment_id = payload.segment_id - 1
segment_findings = store.get_findings_by_segment(source_key, target_segment_id)
segment_findings = memory_export_filter.filter_findings(
store.get_findings_by_segment(source_key, target_segment_id)
)
unqualified_findings = [
f for f in segment_findings if (f.result or "").strip() == "不合格"
]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment