Improve scripts

view_summary.py

@@ -4,12 +4,17 @@
 import asyncio
 
 from google.cloud.firestore_v1.async_client import AsyncClient
+from rich.console import Console
+from rich.markdown import Markdown
+from rich.panel import Panel
 
 from adk_firestore_sessionmanager import FirestoreSessionService
 
 APP_NAME = "test_agent"
 USER_ID = "dev_user"
 
+console = Console()
+
 
 async def main() -> None:
     db = AsyncClient()
@@ -20,7 +25,7 @@ async def main() -> None:
     )
 
     if not resp.sessions:
-        print("No sessions found.")
+        console.print("[dim]No sessions found.[/]")
         return
 
     for s in resp.sessions:
@@ -29,12 +34,10 @@ async def main() -> None:
         data = snap.to_dict() or {}
         summary = data.get("conversation_summary")
 
-        print(f"Session: {s.id}")
         if summary:
-            print(f"Summary:\n{summary}")
+            console.print(Panel(Markdown(summary), title=f"Session {s.id}", border_style="cyan"))
         else:
-            print("No summary yet.")
-        print()
+            console.print(Panel("[dim]No summary yet.[/]", title=f"Session {s.id}", border_style="yellow"))
 
 
 if __name__ == "__main__":