rag-gateway.git

			@@ -1,10 +1,11 @@
			import json
			from Log import logger
			from app.config.agent_base_url import DF_CHAT_AGENT
			from app.config.agent_base_url import DF_CHAT_AGENT, RG_CHAT_DIALOG
			from app.config.config import settings
			from app.config.const import message_error, message_event, complex_knowledge_chat
			from app.config.const import message_error, message_event, complex_knowledge_chat, rg_api_token, workflow_finished
			from app.models import ComplexChatSessionDao, ChatData
			from app.service.v2.app_driver.chat_agent import ChatAgent
			from app.service.v2.app_driver.chat_dialog import ChatDialog
			from app.service.v2.chat import get_chat_token


			@@ -77,23 +78,41 @@
			if session.mindmap:
			inputs = {"is_deep": chat_request.get("isDeep", 1)}
			if session.chat_mode == complex_knowledge_chat:
			inputs["query_json"] = json.dumps(
			{"query": chat_request.get("query", ""), "dataset_ids": chat_request.get("knowledgeId", [])})
			try:
			async for ans in chat.chat_completions(url,
			await chat.complex_request_data(message, session.conversation_id,
			str(user_id), files=chat_request.get("files", []), inputs=inputs),
			await chat.get_headers(token)):
			if ans.get("event") == message_error:
			return res
			elif ans.get("event") == message_event:
			mindmap_query += ans.get("answer", "")
			else:
			continue
			token = await get_chat_token(db, rg_api_token)
			# print(token)
			dialog_url = settings.fwr_base_url + RG_CHAT_DIALOG.format(session.chat_id)
			dialog_chat = ChatDialog()
			try:
			async for ans in dialog_chat.chat_completions(dialog_url, await dialog_chat.complex_request_data(f"简要总结：{message}",
			chat_request["knowledgeId"],
			session.conversation_id),
			await dialog_chat.get_headers(token)):
			if ans.get("code", None) == 102:
			return res
			else:
			if isinstance(ans.get("data"), bool) and ans.get("data") is True:
			break
			else:
			data = ans.get("data", {})
			mindmap_query = data.get("answer", "")
			except Exception as e:
			logger.error(e)
			else:
			try:
			async for ans in chat.chat_completions(url,
			await chat.complex_request_data(message, session.conversation_id,
			str(user_id), files=chat_request.get("files", []), inputs=inputs),
			await chat.get_headers(token)):
			if ans.get("event") == message_error:
			return res
			elif ans.get("event") == workflow_finished:
			mindmap_query = ans.get("data", {}).get("outputs", {}).get("answer", "")
			else:
			continue

			except Exception as e:
			logger.error(e)
			return res
			except Exception as e:
			logger.error(e)
			return res
			else:
			mindmap_query = session.content
			# print("-----------------", mindmap_query)
			@@ -107,6 +126,7 @@
			await chat.complex_request_data(mindmap_query, "",
			str(user_id)),
			await chat.get_headers(token)):
			# print(ans)
			if ans.get("event") == message_error:
			return res
			elif ans.get("event") == message_event:
			@@ -195,6 +215,19 @@
			return parent_list[:index]+new_node_list+parent_list[index+1:]


			async def service_message_mindmap_parse(db, message_id, user_id):
			res = {}
			complex_log = ComplexChatSessionDao(db)
			session = await complex_log.get_session_by_id(message_id)

			if session.mindmap:
			try:
			res_str = await mindmap_join_str(session.mindmap)
			res["mindmap"] = res_str
			except Exception as e:
			logger.error(e)
			return res


			if __name__ == '__main__':
			a = '{ "title": "全生命周期管理", "items": [ { "title": "设备规划与采购", "items": [ { "title": "需求分析与选型" ,"items": [{"title": "rererer"}, {"title": "trtrtrtrt"}] }, { "title": "供应商选择与合同管理" } ] }, { "title": "设备安装与调试", "items": [ { "title": "安装规范" }, { "title": "调试测试" } ] }, { "title": "设备使用", "items": [ { "title": "操作培训" }, { "title": "操作规程与记录" } ] }, { "title": "设备维护与维修", "items": [ { "title": "定期维护" }, { "title": "故障诊断" }, { "title": "备件管理" } ] }, { "title": "设备更新与改造", "items": [ { "title": "技术评估" }, { "title": "更新计划" }, { "title": "改造方案" } ] }, { "title": "设备报废", "items": [ { "title": "报废评估" }, { "title": "报废处理" } ] }, { "title": "信息化管理", "items": [ { "title": "设备管理系统" }, { "title": "数据分析" }, { "title": "远程监控" } ] }, { "title": "安全管理", "items": [ { "title": "安全培训" }, { "title": "安全检查" }, { "title": "应急预案" } ] }, { "title": "环境保护", "items": [ { "title": "环保设备" }, { "title": "废物处理" }, { "title": "节能减排" } ] }, { "title": "具体实践案例", "items": [ { "title": "高压开关设备润滑脂选用研究" }, { "title": "环保型 C4 混气 GIS 设备运维技术研究" } ] }, { "title": "总结", "items": [ { "title": "提高运营效率和竞争力" } ] } ]}'