| | |
|---|
| | | }) |
|---|
| | | 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() |
|---|
| | |
|---|
| | | result = {"message": "", "type": "close"} |
|---|
| | | elif data is None: # 发生错误 |
|---|
| | | answer = json_data.get("retmsg", json_data.get("retcode")) |
|---|
| | | result = {"message": "内部错误：" + answer, "type": "stream"} |
|---|
| | | result = {"message": "内部错误：" + answer, "type": "message"} |
|---|
| | | else: # 正常输出 |
|---|
| | | answer = data.get("answer", "") |
|---|
| | | result = {"message": answer, "type": "stream"} |
|---|
| | | result = {"message": answer, "type": "message"} |
|---|
| | | await websocket.send_json(result) |
|---|
| | | except json.JSONDecodeError: |
|---|
| | | result = {"message": text, "type": "stream"} |
|---|
| | | 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) |
|---|
