update handling tables

rag_pipeline.py

@@ -301,8 +301,8 @@ def process_question_with_sequential_citations(agent, question: str, chunks_dire
         for source_id in unique_ids:  # This preserves the order
             for element in all_chunks_data:
                 if element.get("type") == 'TableElement':
-                    if element.get("element_id") == source_id:
-                        cited_elements_ordered.append(element)
+                    if element.get("elements",{}).get("element_id") == source_id:
+                        cited_elements_ordered.append(element.get("elements",{}))
                         break
                 else:
                     if "elements" in element: