| | |
|---|
| | | "doc_ids": message.get("doc_ids", []), |
|---|
| | | "role": "user" |
|---|
| | | }) |
|---|
| | | complete_response = "" |
|---|
| | | async for rag_response in ragflow_service.chat(token, chat_id, chat_history): |
|---|
| | | try: |
|---|
| | | if rag_response[:5] == "data:": |
|---|
| | |
|---|
| | | else: |
|---|
| | | # 否则，保持原样 |
|---|
| | | text = rag_response |
|---|
| | | complete_response += text |
|---|
| | | try: |
|---|
| | | json_data = json.loads(text) |
|---|
| | | json_data = json.loads(complete_response) |
|---|
| | | data = json_data.get("data") |
|---|
| | | if data is True: # 完成输出 |
|---|
| | | result = {"message": "", "type": "close"} |
|---|
| | |
|---|
| | | reference = data.get("reference", {}) |
|---|
| | | result = {"message": answer, "type": "message", "reference": reference} |
|---|
| | | await websocket.send_json(result) |
|---|
| | | except json.JSONDecodeError: |
|---|
| | | print(f"Error decode ragflow response: {text}") |
|---|
| | | pass |
|---|
| | | |
|---|
| | | complete_response = "" |
|---|
| | | except json.JSONDecodeError as e: |
|---|
| | | print(f"Error decoding JSON: {e}") |
|---|
| | | print(f"Response text: {text}") |
|---|
| | | except Exception as e2: |
|---|
| | | result = {"message": f"内部错误： {e2}", "type": "close"} |
|---|
| | | await websocket.send_json(result) |
|---|
