From 22915223d4828837775f6dd41c433e5b4c38cb3d Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Tue, 12 Aug 2025 18:53:34 +0800 Subject: [PATCH] Fix: citation issue. (#9424) ### What problem does this PR solve? #8474 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- api/apps/conversation_app.py | 11 ++--------- 1 file changed, 2 insertions(+), 9 deletions(-) diff --git a/api/apps/conversation_app.py b/api/apps/conversation_app.py index 5e5b44a47..802ac50ca 100644 --- a/api/apps/conversation_app.py +++ b/api/apps/conversation_app.py @@ -67,7 +67,7 @@ def set_conversation(): if not e: return get_data_error_result(message="Dialog not found") conv = {"id": conv_id, "dialog_id": req["dialog_id"], "name": name, "message": [{"role": "assistant", "content": dia.prompt_config["prologue"]}],"user_id": current_user.id, - "reference":[{}],} + "reference":[],} ConversationService.save(**conv) return get_json_result(data=conv) except Exception as e: @@ -187,14 +187,7 @@ def completion(): if not conv.reference: conv.reference = [] - else: - for ref in conv.reference: - if isinstance(ref, list): - continue - ref["chunks"] = chunks_format(ref) - - if not conv.reference: - conv.reference = [] + conv.reference = [r for r in conv.reference if r] conv.reference.append({"chunks": [], "doc_aggs": []}) def stream():