fix: check chat_message.graph_data is None

pingcap · Dec 9, 2024 · f588f04 · f588f04
1 parent 1c823d8
commit f588f04
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend/app/rag/chat.py b/backend/app/rag/chat.py
@@ -974,7 +974,7 @@ def get_graph_data_from_chat_message(
     session: Session,
     chat_message: ChatMessage
 ) -> Tuple[list[dict], list[dict]]:
-    if not chat_message.content:
+    if not chat_message.graph_data:
         return [], []
 
     if "relationships" not in chat_message.graph_data: