From a1b7795af6095cd821a3bc279ec2fcc9a03b3b98 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期二, 29 十月 2024 21:30:22 +0800 Subject: [PATCH] 代码优化 --- app/api/chat.py | 12 ++++++------ 1 files changed, 6 insertions(+), 6 deletions(-) diff --git a/app/api/chat.py b/app/api/chat.py index 07a05da..7bb7d8a 100644 --- a/app/api/chat.py +++ b/app/api/chat.py @@ -38,7 +38,7 @@ return if agent_type == AgentType.RAGFLOW: - ragflow_service = RagflowService(settings.ragflow_base_url) + ragflow_service = RagflowService(settings.fwr_base_url) token = get_ragflow_token(db, current_user.id) try: async def forward_to_ragflow(): @@ -65,7 +65,6 @@ }) async for rag_response in ragflow_service.chat(token, chat_id, chat_history): try: - print(f"Received from ragflow: {rag_response}") if rag_response[:5] == "data:": # 濡傛灉鏄紝鍒欐埅鍙栨帀鍓�5涓瓧绗︼紝骞跺幓闄ら灏剧┖鐧界 text = rag_response[5:].strip() @@ -83,10 +82,11 @@ else: # 姝e父杈撳嚭 answer = data.get("answer", "") result = {"message": answer, "type": "message"} + await websocket.send_json(result) except json.JSONDecodeError: - result = {"message": text, "type": "message"} - await websocket.send_json(result) - print(f"Forwarded to client {chat_id}: {result}") + print(f"Error decode ragflow response: {text}") + pass + except Exception as e: result = {"message": f"鍐呴儴閿欒锛� {e}", "type": "close"} await websocket.send_json(result) @@ -102,7 +102,7 @@ elif agent_type == AgentType.BISHENG: token = get_bisheng_token(db, current_user.id) - service_uri = f"{settings.bisheng_websocket_url}/api/v1/assistant/chat/{agent_id}?t=&chat_id={chat_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};"} async with websockets.connect(service_uri, extra_headers=headers) as service_websocket: -- Gitblit v1.8.0