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 ...@@ -30,6 +30,7 @@ from core.tools.rule_filter import LufaPartyRuleFilterTool,JpRuleFilterTool,Rule
from core.tools.rules_retrieve_reference import RulesRetrieveReferenceTool from core.tools.rules_retrieve_reference import RulesRetrieveReferenceTool
from core.tools.reflect_retry import ReflectRetryTool from core.tools.reflect_retry import ReflectRetryTool
from core.tools.segment_merger import SegmentMergerTool 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.fact_merger import FactMergerTool
from core.tools.ruleset_router import RulesetRouterTool from core.tools.ruleset_router import RulesetRouterTool
from core.tools.party_role import PartyRoleTool from core.tools.party_role import PartyRoleTool
...@@ -52,6 +53,7 @@ rule_filter_tool = RuleFilterTool() ...@@ -52,6 +53,7 @@ rule_filter_tool = RuleFilterTool()
rules_reference_tool = RulesRetrieveReferenceTool() rules_reference_tool = RulesRetrieveReferenceTool()
reflect_tool = ReflectRetryTool() reflect_tool = ReflectRetryTool()
merger_tool = SegmentMergerTool() merger_tool = SegmentMergerTool()
memory_export_filter = MemoryExportFilter()
fact_merger_tool = FactMergerTool() fact_merger_tool = FactMergerTool()
ruleset_router_tool = RulesetRouterTool() ruleset_router_tool = RulesetRouterTool()
party_role_tool = PartyRoleTool() party_role_tool = PartyRoleTool()
...@@ -606,7 +608,9 @@ def merge_segment_findings(payload: MergerRequest) -> MergerResponse: ...@@ -606,7 +608,9 @@ def merge_segment_findings(payload: MergerRequest) -> MergerResponse:
source_key = payload.finding_key or FINDING_KEY_REVIEW source_key = payload.finding_key or FINDING_KEY_REVIEW
target_segment_id = payload.segment_id - 1 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 = [ unqualified_findings = [
f for f in segment_findings if (f.result or "").strip() == "不合格" 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