Update session_rag.py

session_rag.py

@@ -1,4 +1,3 @@
-
 # session_rag.py
 from __future__ import annotations
 import logging, hashlib
@@ -59,6 +58,7 @@ class SessionRAG:
       - register_artifacts(arts)
       - retrieve(query, k=5)
       - get_latest_csv_columns()
+      - get_csv_summaries()
       - clear()
     """
     def __init__(self, model_name: str = "all-MiniLM-L6-v2"):
@@ -133,14 +133,31 @@ class SessionRAG:
         top_idx = np.argsort(-sims)[: min(k, len(self.texts))]
         return [self.texts[i] for i in top_idx]
 
-    # ---------- helpers for structured Qs ----------
+    # ---------- helpers for structured data ----------
     def get_latest_csv_columns(self) -> List[str]:
-        # scan artifacts in reverse insertion order
         for a in reversed(self.artifacts):
             if a.get("kind") == "csv" and a.get("columns"):
                 return list(map(str, a["columns"]))
         return []
 
+    def get_csv_summaries(self) -> List[Dict[str, Any]]:
+        """
+        Return a list of dicts with keys:
+          - file (str)
+          - digest (str)
+          - summary (dict)
+        newest-first
+        """
+        out: List[Dict[str, Any]] = []
+        for a in reversed(self.artifacts):
+            if a.get("kind") == "csv_summary":
+                out.append({
+                    "file": a.get("name"),
+                    "digest": a.get("digest"),
+                    "summary": a.get("summary"),
+                })
+        return out
+
     def clear(self) -> None:
         self.texts = []
         self.embeddings = None