10 changed files with 245 additions and 113 deletions
@ -0,0 +1,96 @@ |
|||||
|
#对llm返回的指令进行校验 |
||||
|
import re |
||||
|
class CommandVerify: |
||||
|
def __init__(self): |
||||
|
pass |
||||
|
|
||||
|
#验证节点指令的结构完整性--主要是判断JSON元素是否完整 |
||||
|
def verify_node_cmds(self,node_cmds): |
||||
|
''' |
||||
|
验证节点指令的合规性，持续维护 |
||||
|
:param node_cmds: |
||||
|
:param node: |
||||
|
:return: Flase 存在问题， True 合规 |
||||
|
''' |
||||
|
strerror = "" |
||||
|
for node_json in node_cmds: |
||||
|
if "action" not in node_json: |
||||
|
self.logger.error(f"缺少action节点：{node_json}") |
||||
|
strerror = {"节点指令错误":f"{node_json}缺少action节点，不符合格式要求!"} |
||||
|
break |
||||
|
action = node_json["action"] |
||||
|
if action == "add_node": |
||||
|
if "parent" not in node_json or "status" not in node_json or "nodes" not in node_json: |
||||
|
strerror = {"节点指令错误": f"{node_json}不符合格式要求,缺少节点！"} |
||||
|
break |
||||
|
elif action == "update_status": |
||||
|
if "status" not in node_json or "node" not in node_json: |
||||
|
strerror = {"节点指令错误": f"{node_json}不符合格式要求,缺少节点！"} |
||||
|
break |
||||
|
elif action =="no_instruction" or action=="no_create": |
||||
|
if "nodes" not in node_json: |
||||
|
strerror = {"节点指令错误": f"{node_json}不符合格式要求,缺少节点！"} |
||||
|
break |
||||
|
else: |
||||
|
strerror = {"节点指令错误": f"{node_json}不可识别的action值！"} |
||||
|
break |
||||
|
if not strerror: |
||||
|
return True,strerror |
||||
|
return False,strerror |
||||
|
|
||||
|
# 验证节点数据的合规性 |
||||
|
def verify_node_data(self,node_cmds): |
||||
|
add_nodes = [] |
||||
|
no_instr_nodes = [] |
||||
|
for node_cmd in node_cmds: |
||||
|
do_type = node_cmd["action"] |
||||
|
if do_type == "add_node": |
||||
|
nodes = node_cmd["nodes"].split(",") |
||||
|
add_nodes.extend(nodes) |
||||
|
elif do_type == "update_status": |
||||
|
pass #修改节点的完成情况，若有指令已处理修改为未完成 |
||||
|
elif do_type == "no_instruction": |
||||
|
nodes = node_cmd["nodes"].split(",") |
||||
|
no_instr_nodes.extend(nodes) |
||||
|
else: |
||||
|
print("遇到未知节点处理类型") |
||||
|
#核对指令是否有缺失 |
||||
|
had_inst_nodes = self._difference_a_simple(add_nodes,no_instr_nodes) #在新增节点，但不在没有指令列表，就是应该要有指令的节点数据 |
||||
|
no_add_nodes = self._difference_a_simple(no_instr_nodes,add_nodes) #在未新增指令的节点，但不在新增节点，就是没有add的节点，需要新增 |
||||
|
return had_inst_nodes,no_add_nodes |
||||
|
|
||||
|
|
||||
|
|
||||
|
#--------------辅助函数----------------- |
||||
|
def get_path_from_command(self,command): |
||||
|
pass |
||||
|
|
||||
|
def _difference_a(list_a: list, list_b: list) -> list: |
||||
|
"""获取 list_a 中存在但 list_b 中不存在的元素（去重版）""" |
||||
|
set_b = set(list_b) |
||||
|
return [x for x in list_a if x not in set_b] |
||||
|
|
||||
|
def _difference_b(list_a: list, list_b: list) -> list: |
||||
|
"""获取 list_b 中存在但 list_a 中不存在的元素（去重版）""" |
||||
|
set_a = set(list_a) |
||||
|
return [x for x in list_b if x not in set_a] |
||||
|
|
||||
|
def _difference_a_keep_duplicates(list_a: list, list_b: list) -> list: |
||||
|
"""获取 list_a 中存在但 list_b 中不存在的元素（保留所有重复项和顺序）""" |
||||
|
set_b = set(list_b) |
||||
|
return [x for x in list_a if x not in set_b] |
||||
|
|
||||
|
def _difference_b_keep_duplicates(list_a: list, list_b: list) -> list: |
||||
|
"""获取 list_b 中存在但 list_a 中不存在的元素（保留所有重复项和顺序）""" |
||||
|
set_a = set(list_a) |
||||
|
return [x for x in list_b if x not in set_a] |
||||
|
|
||||
|
def _difference_a_simple(list_a: list, list_b: list) -> list: |
||||
|
"""集合差集：list_a - list_b""" |
||||
|
return list(set(list_a) - set(list_b)) |
||||
|
|
||||
|
def _difference_b_simple(list_a: list, list_b: list) -> list: |
||||
|
"""集合差集：list_b - list_a""" |
||||
|
return list(set(list_b) - set(list_a)) |
||||
|
|
||||
|
g_CV = CommandVerify() |
||||
Loading…
Reference in new issue