Update agent.py

agent.py

@@ -1,6 +1,7 @@
 import logging
 import asyncio
 import json
+import uuid
 import os
 from datetime import datetime
 from zoneinfo import ZoneInfo
@@ -737,6 +738,7 @@ async def generate_and_save_summary(db: Database, chat_ctx: llm.ChatContext, con
     logger.info("Generating conversation summary...")
     
     transcript = ""
+    messages_to_save = []
     
     # Try to extract messages from chat context
     try:
@@ -751,11 +753,54 @@ async def generate_and_save_summary(db: Database, chat_ctx: llm.ChatContext, con
             if isinstance(item, llm.ChatMessage):
                 role = item.role
                 content = item.content
+                
+                # Format content for string manipulation
+                content_str = content
                 if isinstance(content, list):
-                    content = " ".join([str(c) for c in content])
+                    content_str = " ".join([str(c) for c in content])
+                
+                if isinstance(content_str, str):
+                    transcript += f"{role}: {content_str}\n"
+
+                # Prepare for DB
+                msg_data = {
+                    "role": role,
+                    "content": content_str,
+                    "tool_name": None,
+                    "tool_args": None
+                }
                 
-                if isinstance(content, str):
-                    transcript += f"{role}: {content}\n"
+                # Attempt to extract tool info safely
+                if hasattr(item, 'tool_calls') and item.tool_calls:
+                     try:
+                         tc = item.tool_calls[0]
+                         # Handle both object and dict (depending on underlying library version)
+                         if isinstance(tc, dict):
+                             msg_data["tool_name"] = tc.get('function', {}).get('name')
+                             msg_data["tool_args"] = tc.get('function', {}).get('arguments')
+                         else:
+                             # accessing attributes of ToolCall object
+                             fn = getattr(tc, 'function', None)
+                             if fn:
+                                 msg_data["tool_name"] = getattr(fn, 'name', None)
+                                 msg_data["tool_args"] = getattr(fn, 'arguments', None)
+                     except Exception:
+                         pass # Ignore tool extraction errors
+                
+                if role == "tool":
+                    msg_data["tool_name"] = getattr(item, 'name', getattr(item, 'tool_call_id', None))
+
+                messages_to_save.append(msg_data)
+        
+        # Save transcript to DB
+        if messages_to_save:
+            try:
+                # Generate a session ID for this conversation batch
+                session_id = str(uuid.uuid4())
+                db.save_chat_transcript(session_id, contact_number, messages_to_save)
+            except Exception as e:
+                logger.error(f"Failed to save chat transcript to DB: {e}")
+
     except Exception as e:
         logger.error(f"Error extracting transcript: {e}")
     
@@ -1153,7 +1198,7 @@ async def entrypoint(ctx: JobContext):
         await asyncio.sleep(1)
 
     contact_number = user_context.get("contact_number")
-    if contact_number:
+    if contact_number and not assistant.summary_generated:
          logger.info("Disconnect summary generation (backup)...")
          duration = (datetime.now() - start_time).total_seconds()
          user_name = user_context.get("user_name", "the patient")