rag-gateway.git

			@@ -10,9 +10,11 @@
			from Log import logger
			from app.api import get_current_user_websocket
			from app.config.config import settings
			from app.config.const import IMAGE_TO_TEXT, DOCUMENT_TO_REPORT, DOCUMENT_TO_CLEANING
			from app.models.agent_model import AgentModel, AgentType
			from app.models.base_model import get_db
			from app.models.user_model import UserModel
			from app.service.v2.api_token import DfTokenDao
			from app.service.dialog import update_session_history
			from app.service.basic import BasicService
			from app.service.difyService import DifyService
			@@ -39,15 +41,18 @@
			ret = {"message": "Agent not found", "type": "close"}
			await websocket.send_json(ret)
			return
			print(1111)
			agent_type = agent.agent_type
			print(agent_type)
			if chat_id == "" or chat_id == "0":
			ret = {"message": "Chat ID not found", "type": "close"}
			await websocket.send_json(ret)
			return

			if agent_type == AgentType.RAGFLOW:
			print(222)
			ragflow_service = RagflowService(settings.fwr_base_url)
			token = get_ragflow_token(db, current_user.id)
			token = await get_ragflow_token(db, current_user.id)
			try:
			async def forward_to_ragflow():
			while True:
			@@ -133,7 +138,7 @@
			pass

			elif agent_type == AgentType.BISHENG:
			token = get_bisheng_token(db, current_user.id)
			token = await get_bisheng_token(db, current_user.id)
			service_uri = f"{settings.sgb_websocket_url}/api/v1/assistant/chat/{agent_id}?t=&chat_id={chat_id}"
			headers = {'cookie': f"access_token_cookie={token};"}

			@@ -310,7 +315,10 @@
			try:
			async def forward_to_dify():
			if agent.type == "imageTalk":
			token = settings.dify_api_token
			token = DfTokenDao(db).get_token_by_id(IMAGE_TO_TEXT)
			if not token:
			await websocket.send_json({"message": "Invalid token", "type": "error"})

			while True:
			image_list = []
			is_image = False
			@@ -334,9 +342,11 @@
			except Exception as e:
			logger.error(e)
			# complete_response = ""

			answer_str = ""
			async for rag_response in dify_service.chat(token, current_user.id, question, upload_file_id,
			conversation_id):
			# print(rag_response)
			try:
			if rag_response[:5] == "data:":
			# 如果是，则截取掉前5个字符，并去除首尾空白符
			@@ -403,8 +413,10 @@
			await websocket.send_json(result)
			print(f"Error process message of ragflow: {e2}")
			elif agent.type == "reportWorkflow":
			print(2323333232)
			token = settings.dify_workflow_clean

			token = DfTokenDao(db).get_token_by_id(DOCUMENT_TO_CLEANING)
			if not token:
			await websocket.send_json({"message": "Invalid token document_to_cleaning", "type": "error"})
			while True:
			receive_message = await websocket.receive_json()
			print(f"Received from client {chat_id}: {receive_message}")
			@@ -426,21 +438,27 @@
			except Exception as e:
			logger.error(e)
			inputs = {
			"input_files": []
			}
			files = []
			for file in upload_files:
			inputs["input_files"].append({
			files.append({
			"type": "document",
			"transfer_method": "local_file",
			"url": "",
			"upload_file_id": file
			})
			if workflow_type == 1:
			inputs["input_files"] = files
			if workflow_type == 2:
			inputs["file_list"] = files
			inputs["Completion_of_main_indicators"] = title
			token = settings.dify_workflow_report
			token = DfTokenDao(db).get_token_by_id(DOCUMENT_TO_REPORT)
			if not token:
			await websocket.send_json(
			{"message": "Invalid token document_to_cleaning", "type": "error"})
			complete_response = ""
			async for rag_response in dify_service.workflow(token, current_user.id, inputs):
			print(rag_response)
			# print(rag_response)
			try:
			if rag_response[:5] == "data:":
			# 如果是，则截取掉前5个字符，并去除首尾空白符