Update retrieval_engine.py

retrieval_engine.py

@@ -55,11 +55,20 @@ class RetrievalEngine:
                 rows.append(
                     {
                         "text": item.get("text", ""),
-                        "topic": item.get("topic", item.get("section", "general")) or "general",
-                        "source": item.get("source", "local_corpus"),
+                        "topic": (
+                            item.get("topic")
+                            or item.get("topic_guess")
+                            or item.get("section")
+                            or "general"
+                        ),
+                        "source": (
+                            item.get("source")
+                            or item.get("source_name")
+                            or item.get("source_file")
+                            or "local_corpus"
+                        ),
                     }
                 )
-        return rows
 
     def _topic_bonus(self, desired_topic: str, row_topic: str, intent: str) -> float:
         desired_topic = (desired_topic or "").lower()