根据离心式海水泵近一年（2023年1月至2023年12月）的备件消耗统计数据显示，系统运行期间未记录到任何故障事件（total_events=0），相关备件消耗数据及故障频率数据均为空。初步判断可能为数据采集系统未启用、设备运行周期未覆盖统计时段或设备处于低负荷运行状态。建议进一步核查数据采集机制完整性，并结合设备实际运行工况开展专项分析。
+ +| 备件名称 | +库存数量 | +安全库存阈值 | +库存状态 | +
|---|---|---|---|
| 泵轴密封圈 | +0 | +5 | +告警 | +
| 叶轮组件 | +2 | +3 | +预警 | +
| 轴承组（6308型号） | +8 | +10 | +正常 | +
| 故障类型 | +发生次数 | +占比（%） | +
|---|---|---|
| 无记录 | +0 | +0 | +
注：当前统计时段内未发生可记录故障事件，建议结合设备巡检日志开展隐性故障排查。
+ +| 统计时段 | +泵轴密封圈消耗量 | +叶轮组件消耗量 | +轴承组消耗量 | +
|---|---|---|---|
| 2023Q1 | +0 | +0 | +0 | +
| 2023Q2 | +0 | +0 | +0 | +
| 2023Q3 | +0 | +0 | +0 | +
| 2023Q4 | +0 | +0 | +0 | +
趋势说明：因无实际消耗数据，无法绘制趋势曲线。建议结合设备运行日志补充关键工况参数（如启停频次、海水含沙量等）开展关联分析。
+ +| 预测维度 | +预测周期 | +预测结论 | +
|---|---|---|
| 备件需求量 | +2024年全年 | +基于当前零消耗数据，预测需求量为0（需结合设备启用计划修正） | +
| 故障概率 | +2024年Q1-Q4 | +因无历史故障数据，无法进行概率建模，建议开展设备健康度评估 | +
结论：当前数据缺失可能导致保障策略失准，建议尽快启动设备运行数据采集与历史故障档案重建工作。
\ No newline at end of file diff --git a/Prompt.py b/Prompt.py new file mode 100644 index 0000000..66ca160 --- /dev/null +++ b/Prompt.py @@ -0,0 +1,66 @@ +from openai import OpenAI +query_type_prompt = """ +你是一个专业的装备维修领域用户查询判断助手，能够根据以下规则准确判断用户问题属于哪种类型。 + +### 判断规则 +- **简单问答类型（simple）**：问题属于日常生活常识、通用知识、非技术性提问，**不涉及**装备、设备、机械、电子、维修、故障诊断、零部件、操作流程等专业领域内容。例如问候、自我介绍、天气、时间等。 +- **复杂问答类型（complex）**：问题**涉及**装备、设备、机械系统（如发动机、液压系统）、电子装置、维修流程、故障排查、零部件组成、技术参数、操作规范等专业领域，即使只是询问定义、组成或功能，也视为 complex。 + +### 输出要求 +- 仅输出 JSON 格式：{"query_type":"simple"} 或 {"query_type":"complex"} +- 不得包含任何其他文字、解释或格式。 + +### 示例 +输入：今天心情怎么样 +输出：{"query_type":"simple"} + +输入：请问你是谁 +输出：{"query_type":"simple"} + +输入：发动机是什么 +输出：{"query_type":"complex"} + +输入：缸盖组包含哪些组件 +输出：{"query_type":"complex"} + +输入：手机屏幕碎了怎么办 +输出：{"query_type":"complex"} + +输入：水的沸点是多少 +输出：{"query_type":"simple"} + +输入：如何更换坦克履带 +输出：{"query_type":"complex"} + +### 用户查询： +text +""" +def open_api_chat(query: str, model: str = None): + """ + 同步调用Chat API + + Args: + query: 查询文本 + model: 模型名称（可选，默认使用环境变量OPENAI_MODEL） + """ + if model is None: + model = "Qwen3.5-35B-A3B" + response = OpenAI( + api_key = "none", + base_url = "http://192.168.0.46:59800/v1" + ).chat.completions.create( + model=model, + messages=[ + {"role": "user", "content": query}, + ], + temperature=0.6, + stream=False, + response_format = {"type": "json_object"}, + + ) + return response.choices[0].message.content +# querycontent = "发动机异响" +# query = query_type_prompt.replace("text",querycontent) + +# res = open_api_chat(query=query) +# print(res) diff --git a/README.md b/README.md new file mode 100644 index 0000000..e69de29 diff --git a/agent_create_request.py b/agent_create_request.py new file mode 100644 index 0000000..bc657d4 --- /dev/null +++ b/agent_create_request.py @@ -0,0 +1,361 @@ +# create_agents.py +import requests +import urllib3 +from uuid import uuid4 + +urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning) + +BASE_URL = "http://192.168.0.46:22000" +AGENT_API = f"{BASE_URL}/api/v1/agents/" +HEADERS = { + "Accept": "application/json", + "Content-Type": "application/json", +} + +TIMEOUT = 10 +OUTPUT_FILE = "agent_ids.txt" + +AGENT_META = { + str(uuid4()): { + "icon_index": 1, + "name": "修理工程管理", + "description": "对舰船修理工程实施全过程管理，包括任务分解、进度控制、工序协调和资源调配。", + "prompt": "助理员您好，我是修理工程管理助手，我已经绑定了技术资料、通用知识、故障知识、法规标准知识库。我可以为您提供舰船设备的标准修理内容范围、作业工序清单以及安全施工要求，请直接告诉我您需要查询的设备名称。", + "default_query": [ + {"title": "查询设备标准 →", "content": "查询设备标准"}, + {"title": "查看作业工序 →", "content": "查看作业工序"} + ], + }, + str(uuid4()): { + "icon_index": 2, + "name": "器材统计与分析", + "description": "对舰船维修保障过程中涉及的器材进行统计分析与趋势研判。", + "prompt": "助理员您好，我是器材统计与分析助手。我已经绑定了技术资料、通用知识、故障知识、法规标准知识库。为您提供多维度的器材消耗统计与库存分析服务。请告诉我您需要统计的具体业务场景。", + "default_query": [ + {"title": "统计第一季度器材消耗 →", "content": "统计第一季度器材消耗"}, + {"title": "分析当前库存周转率 →", "content": "分析当前库存周转率"}, + {"title": "查看特定型号历史记录 →", "content": "查看特定型号历史记录"} + ], + }, + str(uuid4()): { + "icon_index": 3, + "name": "修理法规知识查询", + "description": "提供舰船修理相关法规、标准和制度的查询与解读。", + "prompt": "您好！我是修理法规知识查询助手。我已经绑定了技术资料、通用知识、故障知识、法规标准知识库。请告诉我需要查询的主题。", + "default_query": [ + {"title": "查询焊接工艺规程 →", "content": "查询焊接工艺规程"}, + {"title": "查看设备拆装安全规程 →", "content": "查看设备拆装安全规程"}, + {"title": "搜索环保作业规定 →", "content": "搜索环保作业规定"} + ], + }, + str(uuid4()): { + "icon_index": 4, + "name": "船舰知识查询", + "description": "提供舰船结构、系统组成与功能原理等知识。", + "prompt": "您好！我是舰船知识查询助手。我已经绑定了技术资料、通用知识、故障知识、法规标准知识库。请告诉我需要了解的主题。", + "default_query": [ + {"title": "了解舰船电力系统组成 →", "content": "了解舰船电力系统组成"}, + {"title": "查询舰船推进装置类型 →", "content": "查询舰船推进装置类型"}, + {"title": "搜索舰船雷达系统原理 →", "content": "搜索舰船雷达系统原理"} + ], + }, + str(uuid4()): { + "icon_index": 5, + "name": "故障排查与修理", + "description": "辅助进行舰船设备故障分析、原因定位与修理方案制定。", + "prompt": "您好！我是智能故障排查与修理助手。我已经绑定了技术资料、通用知识、故障知识、法规标准知识库。提供舷号、设备名称和故障现象，我可以协助您进行复杂故障的研判诊断。", + "default_query": [ + {"title": "多模态输出与校验 →", "content": "多模态输出与校验"}, + {"title": "诊断研判与可能性排序 →", "content": "诊断研判与可能性排序"}, + {"title": "GT-25000燃气轮机的日常维护保养方案 →", "content": "GT-25000燃气轮机的日常维护保养方案"} + ], + }, + str(uuid4()): { + "icon_index": 6, + "name": "舰船机电知识", + "description": "提供舰船动力、电气、液压等机电系统相关知识。", + "prompt": "机电长您好，我是舰船机电知识查询助手。我已经绑定了技术资料、通用知识、故障知识、法规标准知识库。请告诉我需了解的主题。", + "default_query": [ + {"title": "了解主推进系统构成 →", "content": "了解主推进系统构成"}, + {"title": "查看舰艇液压机参数 →", "content": "查看舰艇液压机参数"}, + {"title": "查看舰机液压系统参数 →", "content": "查看舰机液压系统参数"} + ], + }, + str(uuid4()): { + "icon_index": 7, + "name": "现场维修作业向导", + "description": "为一线维修人员提供现场操作流程与安全指导。", + "prompt": "您好！我是现场维修作业向导。我已经绑定了技术资料、通用知识、故障知识、法规标准知识库。您可以询问具体的维修步骤、拆装工艺或查询安全操作规程。", + "default_query": [ + {"title": "离心泵汽蚀的典型特征 →", "content": "离心泵汽蚀的典型特征"}, + {"title": "海水泵汽蚀的维修步骤 →", "content": "海水泵汽蚀的维修步骤"}, + {"title": "GT-25000燃气轮机的日常维护步骤 →", "content": "GT-25000燃气轮机的日常维护步骤"} + ], + }, + str(uuid4()): { + "icon_index": 8, + "name": "舰船技术知识百科", + "description": "综合性舰船技术知识库。", + "prompt": "您好，我是舰船技术知识百科助手。我已经绑定了技术资料、通用知识、故障知识、法规标准知识库。请告诉我您需要查询的技术主题。", + "default_query": [ + {"title": "了解船体高强钢性能 →", "content": "了解船体高强钢性能"}, + {"title": "查询焊接工艺评定标准 →", "content": "查询焊接工艺评定标准"}, + {"title": "查看防腐涂装技术规范 →", "content": "查看防腐涂装技术规范"} + ], + }, +} + + +def build_agent_json(agent_id: str) -> dict: + meta = AGENT_META[agent_id] + return { + "id": agent_id, + "name": meta["name"], + "description": meta["description"], + "prompt": meta["prompt"], + "background": "string", + "access_control": {}, + "default_query": meta["default_query"], + } + + +def create_agent(agent_id: str) -> bool: + resp = requests.post( + AGENT_API, + headers=HEADERS, + json=build_agent_json(agent_id), + verify=False, + timeout=TIMEOUT, + ) + print(resp.json()) + print(f"[CREATE] {agent_id} -> {resp.status_code}") + + return resp.status_code in (200, 201, 409) + + +def main(): + created_ids = [] + + for agent_id in AGENT_META.keys(): + if create_agent(agent_id): + created_ids.append(agent_id) + + with open(OUTPUT_FILE, "w", encoding="utf-8") as f: + for aid in created_ids: + f.write(aid + "\n") + + print(f"\n✅ 已写入 {len(created_ids)} 个 Agent UUID 到 {OUTPUT_FILE}") + + +if __name__ == "__main__": + main() + + + + +# import requests +# import urllib3 +# from uuid import uuid4 +# from typing import List, Tuple + +# # ========================= +# # 0. 关闭 HTTPS 警告（开发环境） +# # ========================= +# urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning) + +# # ========================= +# # 1. 基础配置 +# # ========================= + +# BASE_URL = "https://devb.zdht.hjkl01.cn:65432" + +# AGENT_API = f"{BASE_URL}/api/v1/agents/" +# ROLE_BIND_API = f"{BASE_URL}/api/v1/agent-roles/association" + +# HEADERS = { +# "Accept": "application/json", +# "Content-Type": "application/json", +# } + +# TIMEOUT = 10 + +# # ========================= +# # 2. 角色 ID（数据库真实存在） +# # ========================= + +# AGENT_ROLE_1 = "2" # 助理员助手 (1-4) +# AGENT_ROLE_2 = "3" # 机电长助手 (5-6) +# AGENT_ROLE_3 = "4" # 维修员助手 (7-8) + +# ROLE_NAME_MAP = { +# AGENT_ROLE_1: "助理员助手", +# AGENT_ROLE_2: "机电长助手", +# AGENT_ROLE_3: "维修员助手", +# } + +# # ========================= +# # 3. Agent 元数据 +# # ========================= + +# AGENT_META = { +# str(uuid4()): { +# "icon_index": 1, +# "name": "修理工程管理", +# "description": "对舰船修理工程实施全过程管理，包括任务分解、进度控制、工序协调和资源调配。", +# "prompt": "您好！我是一名舰船修理工程管理专家。", +# }, +# str(uuid4()): { +# "icon_index": 2, +# "name": "器材统计与分析", +# "description": "对舰船维修保障过程中涉及的器材进行统计分析与趋势研判。", +# "prompt": "您好！我是一名舰船器材统计与分析专家。", +# }, +# str(uuid4()): { +# "icon_index": 3, +# "name": "修理法规知识查询", +# "description": "提供舰船修理相关法规、标准和制度的查询与解读。", +# "prompt": "您好！我是一名舰船修理法规专家。", +# }, +# str(uuid4()): { +# "icon_index": 4, +# "name": "船舰知识查询", +# "description": "提供舰船结构、系统组成与功能原理等知识。", +# "prompt": "您好！我是一名舰船专业知识顾问。", +# }, +# str(uuid4()): { +# "icon_index": 5, +# "name": "故障排查与修理", +# "description": "辅助进行舰船设备故障分析、原因定位与修理方案制定。", +# "prompt": "您好！我是一名舰船设备故障诊断专家。", +# }, +# str(uuid4()): { +# "icon_index": 6, +# "name": "舰船机电知识", +# "description": "提供舰船动力、电气、液压等机电系统相关知识。", +# "prompt": "您好！我是一名舰船机电系统专家。", +# }, +# str(uuid4()): { +# "icon_index": 7, +# "name": "现场维修作业向导", +# "description": "为一线维修人员提供现场操作流程与安全指导。", +# "prompt": "您好！我是一名现场维修作业向导。", +# }, +# str(uuid4()): { +# "icon_index": 8, +# "name": "舰船技术知识百科", +# "description": "综合性舰船技术知识库。", +# "prompt": "您好！我是一部舰船技术知识百科。", +# }, +# } + +# # ========================= +# # 4. 构建 Agent JSON +# # ========================= + +# def build_agent_json(agent_id: str) -> dict: +# meta = AGENT_META[agent_id] +# return { +# "id": agent_id, +# "name": meta["name"], +# "description": meta["description"], +# "prompt": meta["prompt"], +# "background": "string", +# "access_control": {}, +# } + +# # ========================= +# # 5. 创建 Agent +# # ========================= + +# def create_agent(agent_id: str) -> bool: +# resp = requests.post( +# AGENT_API, +# headers=HEADERS, +# json=build_agent_json(agent_id), +# verify=False, +# timeout=TIMEOUT, +# ) + +# print(f"[CREATE] Agent {agent_id} -> {resp.status_code}") + +# if resp.status_code in (200, 201, 409): +# return True + +# print(" ↳ 返回值:", resp.text) +# return False + +# # ========================= +# # 6. 绑定 Agent 到角色 +# # ========================= + +# def bind_agent_to_role(agent_id: str, role_id: str) -> bool: +# payload = [ +# { +# "agent_id": agent_id, +# "agent_role_id": role_id, +# "is_active": True, +# } +# ] + +# resp = requests.post( +# ROLE_BIND_API, +# headers=HEADERS, +# json=payload, +# verify=False, +# timeout=TIMEOUT, +# ) + +# print( +# f"[BIND] Agent {agent_id} -> Role {role_id} " +# f"({ROLE_NAME_MAP.get(role_id)}) => {resp.status_code}" +# ) + +# if resp.status_code in (200, 201): +# return True + +# print(" ↳ 返回值:", resp.text) +# return False + +# # ========================= +# # 7. 主流程（先创建 → 再统一绑定） +# # ========================= + +# def main(): +# pending_bindings: List[Tuple[str, str]] = [] + +# print("=== 开始创建 Agent ===") + +# for agent_id, meta in AGENT_META.items(): +# try: +# if not create_agent(agent_id): +# continue + +# idx = meta["icon_index"] + +# if 1 <= idx <= 4: +# role_id = AGENT_ROLE_1 +# elif 5 <= idx <= 6: +# role_id = AGENT_ROLE_2 +# elif 7 <= idx <= 8: +# role_id = AGENT_ROLE_3 +# else: +# continue + +# pending_bindings.append((agent_id, role_id)) + +# except Exception as e: +# print(f"[ERROR] Agent {agent_id} -> {e}") + +# print("\n=== 开始统一绑定角色 ===") + +# for agent_id, role_id in pending_bindings: +# bind_agent_to_role(agent_id, role_id) + +# print("\n=== 全部完成 ===") + +# # ========================= +# # 8. 入口 +# # ========================= + +# if __name__ == "__main__": +# main() diff --git a/agent_ids.txt b/agent_ids.txt new file mode 100644 index 0000000..c2e48f8 --- /dev/null +++ b/agent_ids.txt @@ -0,0 +1,8 @@ +17cb4f33-f84d-473a-9299-d8138e8ea9d6 +a8352abb-900d-468a-97b1-ac93f012ccb2 +f6363563-fd44-4c11-aa24-d2d97abbaa96 +2fc6a72e-69a9-497e-b7f4-12cd6833c781 +16c1320c-444a-4fb7-9e2e-34748f16d361 +c742915b-4bf1-47e3-b64f-903a4ce60de8 +6089a40f-a1b3-4514-a229-25bc8f42789e +9cc84831-747e-4cdf-b941-736dcdc9514c diff --git a/agent_role_request.py b/agent_role_request.py new file mode 100644 index 0000000..a70cc36 --- /dev/null +++ b/agent_role_request.py @@ -0,0 +1,156 @@ +# # bind_agents.py +# import requests +# import urllib3 + +# urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning) + +# BASE_URL = "https://devb.zdht.hjkl01.cn:65432" +# ROLE_BIND_API = f"{BASE_URL}/api/v1/agent-roles/association" + +# HEADERS = { +# "Accept": "application/json", +# "Content-Type": "application/json", +# } + +# TIMEOUT = 10 +# INPUT_FILE = "agent_ids.txt" + +# # 角色 ID +# AGENT_ROLE_0 = "1" # 1-8 +# AGENT_ROLE_1 = "2" # 1-4 +# AGENT_ROLE_2 = "3" # 5-6 +# AGENT_ROLE_3 = "4" # 7-8 + + +# def calc_role_id_by_index(index: int) -> str: +# if 1 <= index <= 4: +# return AGENT_ROLE_1 +# elif 5 <= index <= 6: +# return AGENT_ROLE_2 +# elif 7 <= index <= 8: +# return AGENT_ROLE_3 +# else: +# raise ValueError("非法 icon_index") + + +# def bind_agent(agent_id: str, role_id: str): +# payload = [ +# { +# "agent_id": agent_id, +# "agent_role_id": role_id, +# "is_active": True, +# } +# ] + +# resp = requests.post( +# ROLE_BIND_API, +# headers=HEADERS, +# json=payload, +# verify=False, +# timeout=TIMEOUT, +# ) + +# print(f"[BIND] {agent_id} -> role {role_id} => {resp.status_code}") +# if resp.status_code not in (200, 201): +# print(" ↳", resp.text) + + +# def main(): +# with open(INPUT_FILE, "r", encoding="utf-8") as f: +# agent_ids = [line.strip() for line in f if line.strip()] + +# print(f"=== 读取到 {len(agent_ids)} 个 Agent ===") + +# for idx, agent_id in enumerate(agent_ids, start=1): +# role_id = calc_role_id_by_index(idx) +# bind_agent(agent_id, role_id) + +# print("\n✅ 绑定完成") + + +# if __name__ == "__main__": +# main() + + + + + + + +import requests +import urllib3 + +urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning) + +BASE_URL = "http://192.168.0.46:22000" +ROLE_BIND_API = f"{BASE_URL}/api/v1/agent-roles/association" +HEADERS = { + "Accept": "application/json", + "Content-Type": "application/json", +} + +TIMEOUT = 10 +INPUT_FILE = "agent_ids.txt" + +# 角色 ID +AGENT_ROLE_0 = "1" # 所有 Agent 必绑 +AGENT_ROLE_1 = "2" # 1-4 +AGENT_ROLE_2 = "3" # 5-6 +AGENT_ROLE_3 = "4" # 7-8 + + +def calc_role_id_by_index(index: int) -> str: + if 1 <= index <= 4: + return AGENT_ROLE_1 + elif 5 <= index <= 6: + return AGENT_ROLE_2 + elif 7 <= index <= 8: + return AGENT_ROLE_3 + else: + raise ValueError("非法 index") + + +def bind_agent(agent_id: str, role_id: str): + payload = [ + { + "agent_id": agent_id, + "agent_role_id": role_id, + "is_active": True, + } + ] + + resp = requests.post( + ROLE_BIND_API, + headers=HEADERS, + json=payload, + verify=False, + timeout=TIMEOUT, + ) + + print(f"[BIND] {agent_id} -> role {role_id} => {resp.status_code}") + if resp.status_code not in (200, 201): + print(" ↳", resp.text) + + +def main(): + with open(INPUT_FILE, "r", encoding="utf-8") as f: + agent_ids = [line.strip() for line in f if line.strip()] + + print(f"=== 读取到 {len(agent_ids)} 个 Agent ===") + + for idx, agent_id in enumerate(agent_ids, start=1): + # 1️⃣ 所有 Agent 统一绑定角色 1 + bind_agent(agent_id, AGENT_ROLE_0) + + # 2️⃣ 再绑定分组角色 + group_role = calc_role_id_by_index(idx) + + # 防止重复绑定 + if group_role != AGENT_ROLE_0: + bind_agent(agent_id, group_role) + + print("\n✅ 绑定完成（角色 1 + 分组角色）") + + +if __name__ == "__main__": + main() diff --git a/app.py b/app.py new file mode 100644 index 0000000..102fe31 --- /dev/null +++ b/app.py @@ -0,0 +1,1127 @@ +"""app.py +FastAPI 主 Agent 接口 - 修复重复执行问题 +提供主脑 Agent 的 HTTP API 服务，支持流式输出 +集成 LangGraph Checkpointer 实现状态持久化 +""" +import logging +from time import sleep + +from dotenv import load_dotenv +from fastapi import FastAPI, HTTPException +from fastapi.responses import StreamingResponse, FileResponse +from fastapi.middleware.cors import CORSMiddleware +from fastapi.staticfiles import StaticFiles +from pydantic import BaseModel +from typing import Optional, Dict, Any, AsyncGenerator,List +import uvicorn +import json +import os +import uuid +from main_agent import create_main_agent +from utils.function_tracker import ( + register_event_callback, + create_stream_event_handler, + unregister_event_callback, + clear_current_stream_handler +) +import asyncio +import importlib + +from workflow_registry import WORKFLOW_CONFIG, VALID_ROUTE_FLAGS +from main_agent import extract_conversation_title +from workflows.history_manager import filter_image_urls, preprocess_from_request +from config import set_request_user_config +from checkpointer_config import ( + CheckpointerManager, + checkpointer_manager +) + +# 创建 FastAPI 应用 +app = FastAPI(max_request_size=1024 * 1024 * 10) + + +load_dotenv() + +logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s') +logger = logging.getLogger(__name__) + +app.add_middleware( + CORSMiddleware, + allow_origins=["*"], + allow_credentials=True, + allow_methods=["*"], + allow_headers=["*"], +) + +# app.mount("/picture", StaticFiles(directory="/app/picture"), name="pictures") + + +@app.on_event("startup") +async def startup_event(): + """应用启动时初始化""" + await checkpointer_manager.setup() + # 初始化智能体使用统计表 + from tools.agent_usage_statistics import init_agent_usage_table + await init_agent_usage_table() + logger.info("应用启动完成，PostgreSQL Checkpointer 已配置") + + +# =============== PDF 文件目录配置 =============== +# 创建 PDF 文件目录（用于存储生成的 PDF 等文件） +PDF_DIR = os.path.join(os.path.dirname(__file__), "created_pdf") +# 创建 word 文件目录（用于存储生成的 word 等文件） +WORD_DIR = os.path.join(os.path.dirname(__file__), "created_word") +os.makedirs(PDF_DIR, exist_ok=True) + + + +# =============== 请求模型 =============== + + +class AgentRequest(BaseModel): + """Agent 请求模型""" + query: Optional[Any] = None # 用户文本输入 + file_text: Optional[str] = None # 文件文本输入 + image: Optional[str] = None # 图片路径 + audio: Optional[str] = None # 音频路径（支持 audio 或 asr） + + # 路由标志（可选，如果提供则跳过任务分类） + route_flag: Optional[str] = None # 路由标志 + history_message: Optional[Any] = None # 历史信息（支持字符串或列表格式） + + # 接收参数 + top_k: Optional[Any] = None # rag 前top_k + rag_prompt: Optional[Any] = None # rag 提示词模板（支持字符串、列表、字典等多种类型） + image_kb_id: Optional[str] = None # 图库 知识库id + image_file_id: Optional[str] = None # 图库 文件id + text_kb_id: Optional[str] = None # 检索 知识库id + text_file_id: Optional[str] = None # 检索 文件id + + # Socket.IO 格式所需字段 + chat_id: Optional[str] = None # 聊天ID + message_id: Optional[str] = None # 消息ID + + # 知识库配置 + x_user_id: Optional[str] = "1" # 用户ID + x_user_name: Optional[str] = "testuser" # 用户名称 + x_role: Optional[str] = "admin" # 角色 + + +class AgentResponse(BaseModel): + """Agent 响应模型""" + success: bool + message: str + data: Optional[Dict[str, Any]] = None + error: Optional[str] = None + +class FeedbackClassifyRequest(BaseModel): + """反馈分类请求模型""" + feedback_text: str + + + +# =============== 辅助函数 =============== +def process_rag_prompt(rag_prompt: Any) -> str: + """ + 处理 rag_prompt 参数，将各种类型转换为字符串 + + Args: + rag_prompt: 可以是字符串、列表、字典等任意类型 + + Returns: + 处理后的字符串 + """ + if rag_prompt is None: + return "" + + # 如果是字符串，直接返回 + if isinstance(rag_prompt, str): + return rag_prompt.strip() + + # 如果是列表，转换为多行字符串 + if isinstance(rag_prompt, list): + # 将列表中的每个元素转换为字符串，并用换行符连接 + return "\n".join(str(item) for item in rag_prompt if item) + + # 如果是字典，转换为 JSON 字符串 + if isinstance(rag_prompt, dict): + try: + return json.dumps(rag_prompt, ensure_ascii=False, indent=2) + except Exception as e: + logger.warning(f"rag_prompt 字典转换失败: {e}") + return str(rag_prompt) + + # 其他类型，直接转换为字符串 + return str(rag_prompt).strip() + + +def format_stream_event(event_type: str, + data: Dict[str, Any], + chat_id: str = "", + message_id: str = "", + type: str = "chat:completion", + ) -> str: + """格式化流式事件输出为 Socket.IO 格式""" + # 生成 message_id（如果没有提供） + if not message_id: + message_id = str(uuid.uuid4()) + + # 根据事件类型构建不同的数据结构 + if event_type == "stream_content": + # stream_content 事件使用 chat:completion 格式 + content = data.get("content", "") + event_data = { + "chat_id": chat_id, + "message_id": message_id, + "data": { + "type": "chat:completion", + "data": { + "content": content + } + } + } + elif event_type == "function_end": + # function_end 事件 + event_data = { + "chat_id": chat_id, + "message_id": message_id, + "data": { + "type": "chat:completion", + "data": { + "stepsBlueprint": data + } + } + } + elif event_type == "execution_complete": + # execution_complete 事件 + + rag_result = data.get("rag_result", {}) + graph_result = data.get("graph_result", []) + final_result = data.get("final_result", {}) + title = data.get("title", "") + + # 判断是否有有效的 RAG 结果（非空 dict） + has_rag = isinstance(rag_result, dict) and bool(rag_result) + + # 判断是否有有效的图谱结果（非空 list，且至少一个元素的 nodes 非空） + has_graph = ( + isinstance(graph_result, list) + and len(graph_result) > 0 + and any(isinstance(item, dict) and item.get("nodes") for item in graph_result) + ) + + + # 只有当 RAG 或图谱有有效内容时，才构建 sourceCitation + source_citation = None + if has_rag or has_graph: + source_citation = {} + if has_rag: + source_citation.update(rag_result) + if has_graph: + source_citation["xxxx.graph"] = graph_result + + # 构建 event_data（title 由调用方异步计算后传入） + if final_result: + content = final_result.get("content", "") + event_data = { + "chat_id": chat_id, + "message_id": message_id, + "data": { + "type": "chat:completion", + "data": { + "content": content, + }, + "actions": final_result.get("actions", ""), + "suggestedReplies": final_result.get("suggestedReplies", "") + } + } + else: + event_data = { + "chat_id": chat_id, + "message_id": message_id, + "data": { + "type": "chat:completion", + "data": { + "data": {}, + "done": True, + "title": title or "新对话", + } + } + } + + # 仅在有引用来源时添加 sourceCitation 字段 + if source_citation is not None: + event_data["data"]["data"]["sourceCitation"] = source_citation + + else: + # 其他事件 + event_data = { + "chat_id": chat_id, + "message_id": message_id, + "type": event_type, + "data": { + "type": event_type, + "data": "" + } + } + + # Socket.IO 格式：["chat-events", {...}] + socket_io_message = ["chat-events", event_data] + return f"{json.dumps(socket_io_message, ensure_ascii=False)}\n" + + + + +async def execute_workflow_chain( + route_flag: str, + route_params: Dict[str, Any], + chat_id: str = "", + message_id: str = "", + checkpointer=None +) -> Dict[str, Any]: + """ + 执行工作流 + + Args: + route_flag: 路由标志 + route_params: 路由参数 + chat_id: 聊天会话 ID + message_id: 消息 ID + checkpointer: LangGraph checkpointer 实例 + + Returns: + 工作流执行结果 + """ + + print("子工作流标志：", route_flag) + print("子工作流参数：", route_params) + + if not route_flag: + return {} + + extracted_text = route_params.get("extracted_text", "") + combined_query = route_params.get("combined_query", "") + history_message = route_params.get("history_message", "") + + workflow_info = WORKFLOW_CONFIG.get(route_flag) + + if not workflow_info: + logger.warning(f"未知的路由标志: {route_flag}，使用默认的qa工作流") + workflow_info = WORKFLOW_CONFIG.get("qa", WORKFLOW_CONFIG.get(list(WORKFLOW_CONFIG.keys())[0])) + + try: + module = importlib.import_module(workflow_info["module"]) + workflow_func = getattr(module, workflow_info["function"]) + except (ImportError, AttributeError) as e: + print(e) + logger.error(f"无法导入工作流 {route_flag}: {e}") + return {} + + print("zigongggggggggggggggggggggggg") + + try: + import inspect + sig = inspect.signature(workflow_func) + params = list(sig.parameters.keys()) + + kwargs = { + "extracted_text": extracted_text, + "combined_query": combined_query, + "history_message": history_message, + "route_flag": route_flag, + } + + if "chat_id" in params: + kwargs["chat_id"] = chat_id + if "message_id" in params: + kwargs["message_id"] = message_id + if "checkpointer" in params: + kwargs["checkpointer"] = checkpointer + + result = await workflow_func(**kwargs) + except Exception as e: + print(e) + logger.error(f"工作流执行失败: {e}") + return {} + + return result or {} + + +async def stream_main_agent_execution( + raw_input: Dict[str, Any], + route_flag: str = "", + rag_prompt: str = "", + chat_id: str = "", + message_id: str = "" +) -> AsyncGenerator[str, None]: + """ + 流式执行主Agent并输出函数调用信息（使用装饰器事件） + + Args: + raw_input: 原始输入数据 + route_flag: 路由标志 + rag_prompt: RAG 提示词 + chat_id: 聊天会话 ID + message_id: 消息 ID + + Yields: + 格式化的流式事件字符串 + """ + if not chat_id: + chat_id = str(uuid.uuid4()) + if not message_id: + message_id = str(uuid.uuid4()) + + stream_handler, event_queue = create_stream_event_handler() + register_event_callback(stream_handler) + + try: + execution_done = asyncio.Event() + execution_error = None + final_result = None + checkpointer = None + + async def run_agent(): + nonlocal execution_error, final_result, checkpointer + route_flag_value: str = "" + try: + from checkpointer_config import checkpointer_manager + checkpointer = await checkpointer_manager.get_async_checkpointer() + + main_agent = create_main_agent() + + initial_route_flag: str = route_flag if (route_flag and route_flag in VALID_ROUTE_FLAGS) else "" + + initial_state = { + "raw_input": raw_input, + "input_type": "", + "has_query":False, + "has_image": False, + "has_audio": False, + "has_file_text": False, + "history_message":raw_input.get("history_message", ""), + "extracted_text": "", + "image_description": "", + "asr_text": "", + "file_text": raw_input.get("file_text", ""), + "combined_query": "", + "task_type": "", + "task_classification_result": None, + "workflow_result": None, + "final_response": "", + "error_message": "", + "route_flag": initial_route_flag, + "route_params": {} + } + + main_result = await main_agent.ainvoke(initial_state) + + route_flag_value = (main_result.get("route_flag")or main_result.get("task_type")or "") + + route_params = main_result.get("route_params") or {"combined_query": main_result.get("combined_query", "") + or main_result.get("extracted_text", ""),"extracted_text": main_result.get("extracted_text", "")} + + if "file_text" in raw_input: + route_params["file_text"] = raw_input["file_text"] + + if "history_message" not in route_params and "history_message" in raw_input: + route_params["history_message"] = filter_image_urls(raw_input["history_message"]) + + print("历史记录",route_params["history_message"]) + print("用户问题", route_params["combined_query"]) + print("ttttttt", route_params["extracted_text"]) + + sub_result = await execute_workflow_chain( + route_flag=route_flag_value, + route_params=route_params, + chat_id=chat_id, + message_id=message_id, + checkpointer=checkpointer + ) + + processed_content = "" + if sub_result.get("response"): + processed_content = sub_result.get("response", "") + + else: + logger.warning("子agent没有返回response") + processed_content = "" + + final_result = { + "content": processed_content, + "actions": sub_result.get("actions", []), + "result_tag": sub_result.get("result_tag", ""), + "suggestedReplies": sub_result.get("suggestedReplies", []) + } + except Exception as e: + execution_error = e + finally: + execution_done.set() + await event_queue.put({"type": "execution_complete"}) + + agent_task = asyncio.create_task(run_agent()) + + execution_complete_received = False + rag_result = [] + graph_result = [] + while True: + try: + event = await asyncio.wait_for(event_queue.get(), timeout=0.1) + + if event.get("type") == "execution_complete": + execution_complete_received = True + continue + + event_type = event.get("type") + title = event.get("title", "") + details = event.get("details", "") + result = event.get("result") + + if event_type == "function_execution": + logger.info(f"function_execution event - title: {title}, has_result: {result is not None}, result_type: {type(result)}") + if title == "知识库搜索工具" and result and isinstance(result, dict): + print("===============================", result) + rag_result_raw = result.get("sourceCitation", {}) + rag_result = {} + for doc_name, chunks in rag_result_raw.items(): + if isinstance(chunks, list): + rag_result[doc_name] = [ + {k: v for k, v in item.items() if k != "text"} + if isinstance(item, dict) else item + for item in chunks + ] + else: + rag_result[doc_name] = chunks + elif title == "图谱检索工具" and result and isinstance(result, dict): + print("===============================", result) + graph_result = result.get("xxxx.graph", []) + else: + yield format_stream_event("function_end", { + "title": title, + "details": details + }, chat_id=chat_id, message_id=message_id) + + elif event_type == "function_error": + yield format_stream_event("function_error", { + "title": title, + "error": details, + "details": details + }, chat_id=chat_id, message_id=message_id) + + elif event_type == "stream_content": + content = event.get("content", "") + + yield format_stream_event("stream_content", { + "content": content + }, chat_id=chat_id, message_id=message_id) + + + except asyncio.TimeoutError: + if execution_complete_received: + break + if execution_done.is_set() and event_queue.empty(): + break + continue + + await agent_task + + if execution_error: + yield format_stream_event("error", { + "error": str(execution_error) + }, chat_id=chat_id, message_id=message_id) + else: + title = await extract_conversation_title(final_result.get("content", "") if final_result else "") + yield format_stream_event("execution_complete", { + "final_result": final_result + }, chat_id=chat_id, message_id=message_id) + + yield format_stream_event("execution_complete", { + "rag_result": rag_result, + "graph_result": graph_result, + "title": title + }, chat_id=chat_id, message_id=message_id) + + except Exception as e: + yield format_stream_event("error", { + "error": str(e) + }, chat_id=chat_id, message_id=message_id) + finally: + unregister_event_callback(stream_handler) + clear_current_stream_handler() + +# =============== API 接口 =============== +@app.post("/api/agent/stream_process") +async def stream_process_request(request: AgentRequest): + """ + 流式处理 Agent 请求，实时输出节点执行信息 + 不保存历史，调用端会自己保存历史对话信息 + 每次请求时创建新的主Agent实例 + """ + + print(f"🔍 [调试] 接收到的原始请求: {request.model_dump_json(indent=2)}") + + print(f"[DEBUG] 收到请求 chat_id={request.chat_id!r}, message_id={request.message_id!r}, route_flag={request.route_flag!r}, query={str(request.query)[:100]}") + # 1. 构建当前请求的输入数据 + raw_input = {} + if request.query: + if isinstance(request.query, str): + raw_input["query"] = request.query + elif isinstance(request.query, list): + for item in request.query: + if item.get("type") == "text": + raw_input["query"] = item.get("text") + print(f"当前查询: {raw_input['query']}") + if request.image: + raw_input["image"] = request.image + if request.audio: + # audio 字段支持 audio 或 asr，统一使用 audio 键 + raw_input["audio"] = request.audio + # [新增] 提取 file_text + if request.file_text: + raw_input["file_text"] = request.file_text[:5000] + + # 历史消息预处理：解析、移除当前用户消息、序列化 + if request.history_message: + raw_input["history_message"] = preprocess_from_request(request.history_message, exclude_last=True) + + # 2. 验证至少有一个输入 + if not raw_input: + raise HTTPException( + status_code=400, + detail="至少需要提供一个输入：query、image 或 audio" + ) + + # 3. 获取路由标志（如果提供，将跳过任务分类步骤） + # 始终传递 route_flag，默认值为空字符串 + route_flag = request.route_flag or "" + print("打印agent对应的route_flag：") + print(route_flag) + rag_prompt = process_rag_prompt(request.rag_prompt) if request.rag_prompt else "" + chat_id = request.chat_id or "" + message_id = request.message_id or "" + + # 记录智能体使用情况 + if route_flag and chat_id: + from tools.agent_usage_statistics import record_agent_usage + asyncio.create_task(record_agent_usage( + chat_id=chat_id, + route_flag=route_flag, + message_id=message_id + )) + logger.info(f"记录智能体使用: chat_id={chat_id}, route_flag={route_flag}") + + + + # 4. 设置当前请求的用户配置和知识库参数（线程安全，支持多用户并发） + # 如果请求中提供了这些参数，则使用请求中的值；否则使用默认值 + set_request_user_config( + x_user_id=request.x_user_id, + x_user_name=request.x_user_name, + x_role=request.x_role, + text_kb_id=request.text_kb_id, + text_file_id=request.text_file_id + ) + + #5. 执行主Agent + async def event_generator(): + async for event in stream_main_agent_execution( + raw_input=raw_input, + route_flag=route_flag, + rag_prompt=rag_prompt, + chat_id=chat_id, + message_id=message_id + ): + yield event + + return StreamingResponse(event_generator(), media_type="text/event-stream") + + +@app.get("/api/download/{file_name}") +async def download_file(file_name: str): + """ + 文件下载接口 + + Args: + file_name: 文件名 + + Returns: + 文件下载响应 + """ + pdf_path = os.path.join(PDF_DIR, file_name) + word_path = os.path.join(WORD_DIR, file_name) + file_path = pdf_path if os.path.exists(pdf_path) else word_path + print(f"下载文件：{file_path}") + # 检查文件是否存在 + ext = os.path.splitext(file_name)[1].lower() + media_type = { + ".pdf": "application/pdf", + ".docx": "application/vnd.openxmlformats-officedocument.wordprocessingml.document", + ".doc": "application/msword", + }.get(ext, "application/octet-stream") + + # 返回文件 + return FileResponse( + path=file_path, + filename=file_name, + media_type=media_type + ) + + +@app.get("/api/health") +async def health_check(): + """健康检查接口""" + return { + "status": "healthy", + "message": "服务正常运行，每次请求时创建新的Agent实例" + } + + +@app.get("/api/v1/statistics/fault-type-statistics") +async def fault_type_statistics(ship_number: Optional[str] = None): + """ + 故障类型统计接口 + 统计生成报告里面的故障类型，按舷号分类并计算百分比 + + Args: + ship_number: 舷号过滤（可选，如果提供则只统计该舷号的报告） + + Returns: + 统计结果，包含 faultTypeData 字段 + """ + print("故障类型统计街路口") + from tools.fault_statistics import statistics_fault_types + """ + result = await statistics_fault_types( + ship_number_filter=ship_number + ) + + if not result.get("success", False): + raise HTTPException( + status_code=500, + detail=result.get("error", "统计失败") + ) + """ + result = { + "success": True, + "faultTypeData": [ + { "value": 35, "name": "作战系统", "itemStyle": { "color": "#00C2FF" } }, + { "value": 25, "name": "电力系统", "itemStyle": { "color": "#FF4500" } }, + { "value": 20, "name": "推进系统", "itemStyle": { "color": "#9400D3" } }, + { "value": 10, "name": "航空保障系统", "itemStyle": { "color": "#228B22" } }, + { "value": 5, "name": "船舶保障系统", "itemStyle": { "color": "#FF1493" } }, + { "value": 5, "name": "其他", "itemStyle": { "color": "#2C2C2C" } } + ] + } + + + return result + + +@app.get("/api/v1/statistics/fault-frequency-statistics") +async def fault_frequency_statistics( + top_n: Optional[int] = 6, + time_range: Optional[str] = "全部" +): + """ + 故障发生次数统计接口 + 统计生成报告里面的故障发生次数，按频率排序 + + Args: + top_n: 返回前 N 个高频故障（默认为 10） + time_range: 时间范围（可选，默认为"全部"） + - "全部": 统计所有报告 + - "近七天": 统计最近7天的报告 + - "近一个月": 统计最近30天的报告 + - "近一年": 统计最近365天的报告 + + Returns: + 统计结果，包含 highFreqFaults 字段 + """ + print("故障发生次数接口") + from datetime import datetime, timedelta + + start_date = None + end_date = None + + result = { + "success": True, + "highFreqFaults": [ + { "rank": 1, "name": "燃气轮机燃油增压泵组泵体旋塞处泄漏", "system": "推进系统", "count": 135 }, + { "rank": 2, "name": "气动蝶阀控制箱无法上电工作", "system": "推进系统", "count": 103 }, + { "rank": 3, "name": "组合报警灯柱不能正常工作", "system": "船舶保障系统", "count": 86 }, + { "rank": 4, "name": "岸电联接箱指示灯不亮", "system": "电力系统", "count": 77 }, + { "rank": 5, "name": "主机冷却水泵密封泄漏", "system": "航空保障系统", "count": 76 }, + { "rank": 6, "name": "加电后屏幕和故障指示灯没显示", "system": "作战系统", "count": 65 } + ] + } + + if time_range == "近七天": + end_date = datetime.now().strftime("%Y-%m-%d") + start_date = (datetime.now() - timedelta(days=7)).strftime("%Y-%m-%d") + result = { + "success": True, + "highFreqFaults": [ + { "rank": 1, "name": "气动蝶阀控制箱无法上电工作", "system": "推进系统", "count": 10 }, + { "rank": 2, "name": "组合报警灯柱不能正常工作", "system": "船舶保障系统", "count": 6 }, + { "rank": 3, "name": "岸电联接箱指示灯不亮", "system": "电力系统", "count": 3 }, + { "rank": 4, "name": "主机冷却水泵密封泄漏", "system": "航空保障系统", "count": 3 }, + { "rank": 5, "name": "加电后屏幕和故障指示灯没显示", "system": "作战系统", "count": 1 } + ] + } + + elif time_range == "近一个月": + end_date = datetime.now().strftime("%Y-%m-%d") + start_date = (datetime.now() - timedelta(days=30)).strftime("%Y-%m-%d") + + result = { + "success": True, + "highFreqFaults": [ + { "rank": 1, "name": "燃气轮机燃油增压泵组泵体旋塞处泄漏", "system": "推进系统", "count": 38 }, + { "rank": 2, "name": "气动蝶阀控制箱无法上电工作", "system": "推进系统", "count": 20 }, + { "rank": 3, "name": "组合报警灯柱不能正常工作", "system": "船舶保障系统", "count": 14 }, + { "rank": 4, "name": "岸电联接箱指示灯不亮", "system": "电力系统", "count": 13 }, + { "rank": 5, "name": "主机冷却水泵密封泄漏", "system": "航空保障系统", "count": 10 }, + { "rank": 6, "name": "加电后屏幕和故障指示灯没显示", "system": "作战系统", "count": 8 } + ] + } + + elif time_range == "近一年": + end_date = datetime.now().strftime("%Y-%m-%d") + start_date = (datetime.now() - timedelta(days=365)).strftime("%Y-%m-%d") + result = { + "success": True, + "highFreqFaults": [ + { "rank": 1, "name": "燃气轮机燃油增压泵组泵体旋塞处泄漏", "system": "推进系统", "count": 68 }, + { "rank": 2, "name": "气动蝶阀控制箱无法上电工作", "system": "推进系统", "count": 74 }, + { "rank": 3, "name": "组合报警灯柱不能正常工作", "system": "船舶保障系统", "count": 73 }, + { "rank": 4, "name": "岸电联接箱指示灯不亮", "system": "电力系统", "count": 63 }, + { "rank": 5, "name": "主机冷却水泵密封泄漏", "system": "航空保障系统", "count": 56 }, + { "rank": 6, "name": "加电后屏幕和故障指示灯没显示", "system": "作战系统", "count": 46 } + ] + } + + + """ + if time_range == "近七天": + end_date = datetime.now().strftime("%Y-%m-%d") + start_date = (datetime.now() - timedelta(days=7)).strftime("%Y-%m-%d") + elif time_range == "近一个月": + end_date = datetime.now().strftime("%Y-%m-%d") + start_date = (datetime.now() - timedelta(days=30)).strftime("%Y-%m-%d") + elif time_range == "近一年": + end_date = datetime.now().strftime("%Y-%m-%d") + start_date = (datetime.now() - timedelta(days=365)).strftime("%Y-%m-%d") + + from tools.fault_statistics import statistics_fault_frequency + + result = await statistics_fault_frequency( + top_n=top_n, + start_date=start_date, + end_date=end_date + ) + + if not result.get("success", False): + raise HTTPException( + status_code=500, + detail=result.get("error", "统计失败") + ) + """ + return result + + +@app.get("/api/v1/statistics/spare-parts-statistics") +async def spare_parts_statistics( + ship_number: Optional[str] = None, + time_range: Optional[str] = "全部", + top_n: Optional[int] = 6 +): + """ + 备件消耗统计接口 + 统计生成报告里面的备件消耗情况 + + Args: + ship_number: 舷号过滤（可选） + time_range: 时间范围（可选，默认为"全部"） + - "全部": 统计所有报告 + - "近七天": 统计最近7天的报告 + - "近一个月": 统计最近30天的报告 + - "近一年": 统计最近365天的报告 + top_n: 返回前 N 个备件（默认为 5） + + Returns: + 统计结果，包含 sparePartData 字段 + """ + print("备件消耗统计接口") + from datetime import datetime, timedelta + + start_date = None + end_date = None + + result = { + "success": True, + "sparePartData": { + "categories": ["螺旋式熔断体", "断路器", "机械密封", "门铰链固定端", "减振器"], + "system": ["电力系统", "船舶保障系统", "推进系统", "作战系统", "推进系统"], + "counter": [86, 76, 43, 42, 33] + } + } + + if time_range == "近七天": + end_date = datetime.now().strftime("%Y-%m-%d") + start_date = (datetime.now() - timedelta(days=7)).strftime("%Y-%m-%d") + result = { + "success": True, + "sparePartData": { + "categories": ["螺旋式熔断体", "断路器", "机械密封", "门铰链固定端"], + "system": ["电力系统", "船舶保障系统", "推进系统", "作战系统"], + "counter": [13, 9, 8, 5] + } + } + elif time_range == "近一个月": + end_date = datetime.now().strftime("%Y-%m-%d") + start_date = (datetime.now() - timedelta(days=30)).strftime("%Y-%m-%d") + + result = { + "success": True, + "sparePartData": { + "categories": ["螺旋式熔断体", "断路器", "机械密封", "门铰链固定端", "减振器"], + "system": ["电力系统", "船舶保障系统", "推进系统", "作战系统", "推进系统"], + "counter": [32, 26, 20, 14, 10] + } + } + elif time_range == "近一年": + end_date = datetime.now().strftime("%Y-%m-%d") + start_date = (datetime.now() - timedelta(days=365)).strftime("%Y-%m-%d") + + result = { + "success": True, + "sparePartData": { + "categories": ["螺旋式熔断体", "断路器", "机械密封", "门铰链固定端", "减振器"], + "system": ["电力系统", "船舶保障系统", "推进系统", "作战系统", "推进系统"], + "counter": [64, 63, 23, 20, 19] + } + } + + + + + """ + if time_range == "近七天": + end_date = datetime.now().strftime("%Y-%m-%d") + start_date = (datetime.now() - timedelta(days=7)).strftime("%Y-%m-%d") + elif time_range == "近一个月": + end_date = datetime.now().strftime("%Y-%m-%d") + start_date = (datetime.now() - timedelta(days=30)).strftime("%Y-%m-%d") + elif time_range == "近一年": + end_date = datetime.now().strftime("%Y-%m-%d") + start_date = (datetime.now() - timedelta(days=365)).strftime("%Y-%m-%d") + + from tools.fault_statistics import statistics_spare_parts + + result = await statistics_spare_parts( + ship_number_filter=ship_number, + start_date=start_date, + end_date=end_date, + top_n=top_n + ) + + if not result.get("success", False): + raise HTTPException( + status_code=500, + detail=result.get("error", "统计失败") + ) + """ + return result + + +@app.get("/api/v1/statistics/ship-numbers") +async def get_all_ship_numbers(): + """ + 获取所有舷号接口 + 从知识库树形结构中提取所有可用的舷号列表 + + Returns: + { + "success": true, + "data": { + "ship_numbers": ["101", "111", "122", ...], + "total": 10 + } + } + """ + import httpx + import re + from config import KB_TREE_CONFIG + + KB_TREE_URL = KB_TREE_CONFIG['url'] + KB_TREE_HEADERS = { + "accept": "application/json", + "x-user-id": "1", + "x-user-name": "testuser", + "x-role": "admin", + } + + try: + async with httpx.AsyncClient(timeout=30.0, verify=False) as client: + params = {"kb_id": 2} + response = await client.get(KB_TREE_URL, headers=KB_TREE_HEADERS, params=params) + + if response.status_code != 200: + raise HTTPException( + status_code=response.status_code, + detail=f"知识库API请求失败: {response.status_code}" + ) + + result = response.json() + + if isinstance(result, dict): + if "code" in result and result.get("code") != 200: + raise HTTPException( + status_code=500, + detail=f"知识库API返回错误: {result}" + ) + kb_data = result.get("data") or result.get("list") or {} + elif isinstance(result, list): + kb_data = result[0] if result else {} + else: + kb_data = {} + + children = kb_data.get("children", []) + + if not children: + return { + "success": True, + "data": { + "ship_numbers": [], + "total": 0 + } + } + + ship_numbers_set = set() + + for kb in children: + kb_name = kb.get("name") or "" + kb_numbers = re.findall(r"\d+", kb_name) + for num in kb_numbers: + ship_numbers_set.add(num) + + ship_numbers = sorted(list(ship_numbers_set), key=lambda x: int(x)) + + return { + "success": True, + "data": { + "ship_numbers": ship_numbers, + "total": len(ship_numbers) + } + } + + except HTTPException: + raise + except Exception as e: + logger.error(f"获取舷号列表失败: {str(e)}") + raise HTTPException( + status_code=500, + detail=f"获取舷号列表失败: {str(e)}" + ) + +@app.get("/api/v1/statistics/agent-usage-statistics") +async def agent_usage_statistics(): + """ + 智能体使用统计接口 + 统计智能体的调用次数，按类型分类 + + Returns: + { + "success": true, + "data": { + "total": "8,794", + "types": [ + { "name": "操作使用", "count": "2,966", "icon": "repair", "color": "#00C2FF" }, + { "name": "故障排查与修理", "count": "3,620", "icon": "detection", "color": "#FFB800" }, + { "name": "舰船百科", "count": "2,208", "icon": "doc", "color": "#9153FF" } + ], + "ratio": [ + { "name": "操作使用", "value": 34, "itemStyle": { "color": "#00C2FF" } }, + { "name": "故障排查与修理", "value": 41, "itemStyle": { "color": "#FFB800" } }, + { "name": "舰船百科", "value": 25, "itemStyle": { "color": "#9153FF" } } + ] + } + } + """ + print("智能体调用统计接口") + from tools.agent_usage_statistics import get_agent_usage_statistics + + result = await get_agent_usage_statistics() + + if not result.get("success", False): + raise HTTPException( + status_code=500, + detail="统计失败" + ) + + return result + + +@app.post("/api/v1/feedback/classify") +async def feedback_classify(request: FeedbackClassifyRequest): + """ + 反馈内容分类接口 + + 功能： + 1. 从反馈文本中提取设备名称 + 2. 根据设备名称反推系统名称 + 3. 对反馈内容进行分类 + + Args: + request: 包含 feedback_text 字段的请求体 + + Returns: + { + "success": true, + "system": "系统名称", + "category": "分类结果" + } + + 分类类别： + - 信息过时：反映文档或手册中的信息已经过时，不再适用 + - 步骤错误：反映操作步骤或流程存在错误 + - 步骤缺失：反映缺少必要的操作步骤或流程说明 + - 备件信息有误：反映备件型号、规格、数量等信息有误 + - 安全提示不足：反映缺少必要的安全警示或注意事项 + - 图示不清：反映图片、示意图不清晰或难以理解 + - 其他：不属于以上类别 + """ + from tools.fault_statistics import analyze_feedback_and_classify + + result = await analyze_feedback_and_classify(request.feedback_text) + + if not result.get("success", False): + raise HTTPException( + status_code=500, + detail=result.get("error", "分类失败") + ) + + return result + + + +@app.get("/") +async def root(): + """根路径""" + return { + "message": "Agent API 服务", + "docs": "/docs" + } + + +# =============== 主函数 =============== +if __name__ == "__main__": + uvicorn.run( + "app:app", + host="0.0.0.0", + port=9088 + ) + + + + diff --git a/audio/test.wav b/audio/test.wav new file mode 100644 index 0000000000000000000000000000000000000000..02b336c2689f2e265e9bd5517613e79aac50334e GIT binary patch literal 13615182 zcmeF)3HVOc`!D{zQPF@TDv3xUL^5V9bH->;B%+j%kg1H32BI<)6-5~uBtwNH%2X&B zQ)WU$sEEDKYd?O@x8FJ6GhF|3UH^0a|JUQW)@Sd%_8RYX5AXY4>v`N>w^prMU#t%9 zs(xq92RnAlpFIeIFvt+}pXcUKh9FarJ*Z!|X|pRl)@@j;QQfAsnq_rgzx5LjmMLGN zT-g$(N|p@Tw(izC&`SDWngVGGq$!Z5K$-$+3ZyBJra+nkX$qt%kfuPI0%;1QDUhZ> zngVGGq$!Z5K$-$+3ZyBJra+nkX$qt%kfuPI0%;1QDUhZ>ngVGGq$!Z5K$-$+3ZyBJ zra+nkX$qt%kfuPI0%;1QDUhZ>ngVGGq$!Z5K$-$+3ZyBJra+nkX$qt%kfuPI0%;1Q zDUhZ>ngVGGq$!Z5K$-$+3ZyBJra+nkX$qt%kfuPI0%;1QDUhZ>ngVGGq$!Z5K$-$+ z3ZyBJra+nkX$qt%kfuPI0%;1QDUhZ>ngVGGq$!Z5K$-$+3ZyBJra+nkX$qt%kfuPI z0%;1QDUhZ>ngVGGq$!Z5K$-$+3ZyBJra+nkX$qt%kfuPI0%;1QDUhZ>ngVGGq$!Z5 zK$-$+3ZyBJra+nkX$qt%kfuPI0%;1QDUhZ>ngVGGq$!Z5K$-$+3ZyBJra+nkX$qt% zkfuPI0%;1QDUhZ>ngVGGq$!Z5K$-$+3ZyBJra+nkX$qt%kfuPI0%;1QDUhZ>ngVGG zq$!Z5K$-$+3ZyBJra+nkX$qt%kfuPI0%;1QDUhZ>ngVGGq$!Z5K$-$+3ZyBJra+nk zX$qt%kfuPI0%;1QDUhZ>ngVGGq$!Z5K$-$+3ZyBJra+nkX$qt%kfuPI0%;1QDezAU zg#Uc_$Nm2({~tyEU%QRtAV|{LfB)_C+5BY8p-=KR`Ip@P>-p?`>eavf^Y8jd@}gA# zXL^nOJ6ri|PiOo6=W|l#?DN_3|KGa)Yo7m^`v2=$oT1d2TL11oY2j@C|GKqwiT@iE z_~-g0_h