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