agent_rag/plc_query_tool.py

from plc_query import plclib
from api_tools.access_plc_api import *

def plc_query_helper(query:str, project_id:int):
    """
    PLC点位数据查询工具
    :param query: 需要查询的语句
    :return:
    """
    plclib.helper.load(project_id)  # 演示demo 默认中荷
    tmp_res = plclib.helper.query(query, is_agent=True)
    res_info = tmp_res[0][0]
    res_score = tmp_res[1][0]
    # 数据库字段
    db_codes = [list(item.values())[0] for item in res_info]
    # 开始调用plc
    api_response_info = post_req_2_plc_api(project_id=92, database_codes=db_codes)
    api_response_info = [api_response_info[i] for i in range(len(api_response_info))]  # 暂时不要置信度
    return ';'.join(api_response_info)

if __name__ == '__main__':
    res = plc_query_helper(query="中荷水厂超滤总进水流量")
    print(res)