Fix extending conversation time

Author: kPatch

coffee_ws/src/coffee_voice_agent/scripts/agents/coffee_barista_agent.py

@@ -237,6 +237,15 @@ async def on_user_turn_completed(self, turn_ctx, new_message):
                 content="ADMIN: You've been chatting for 7 minutes - time limit reached. Call the manage_conversation_time tool with action='end' to wrap up gracefully, unless there are special circumstances."
             )
             self.state_manager.seven_minute_warning_sent = True
+            
+        # Extension expired - needs immediate ending
+        if self.state_manager.extension_expired_pending:
+            logger.info("Injecting extension expired admin message via callback")
+            turn_ctx.add_message(
+                role="system",
+                content="ADMIN: Your granted extension time has expired. You must end the conversation gracefully now to help other visitors."
+            )
+            self.state_manager.extension_expired_pending = False
 
     async def start_wake_word_detection(self, room):
         """Start wake word detection in a separate thread"""

coffee_ws/src/coffee_voice_agent/scripts/state/state_manager.py

@@ -63,6 +63,7 @@ def __init__(self, agent=None):
         self.five_minute_warning_sent = False
         self.six_minute_warning_sent = False
         self.seven_minute_warning_sent = False
+        self.extension_expired_pending = False
 
         # Phase 4: Mutual exclusion for virtual request processing
         self.virtual_request_processing_lock = asyncio.Lock()
@@ -123,6 +124,7 @@ async def _enter_new_state(self):
             self.five_minute_warning_sent = False
             self.six_minute_warning_sent = False
             self.seven_minute_warning_sent = False
+            self.extension_expired_pending = False
 
             # Start max conversation timer (absolute limit)
             self.conversation_timer = asyncio.create_task(self._max_conversation_timeout())
@@ -156,6 +158,52 @@ async def _max_conversation_timeout(self):
         except asyncio.CancelledError:
             pass
 
+    async def extend_conversation(self, additional_minutes: int, reason: str):
+        """Extend conversation by additional minutes, cancelling and restarting the timeout timer"""
+        if self.current_state != AgentState.ACTIVE:
+            logger.warning(f"Cannot extend conversation - not in active state (current: {self.current_state})")
+            return
+            
+        logger.info(f"Extending conversation by {additional_minutes} minutes: {reason}")
+        
+        # Cancel existing conversation timer
+        if self.conversation_timer:
+            self.conversation_timer.cancel()
+            self.conversation_timer = None
+            
+        # Calculate new timeout duration (additional time from now)
+        new_timeout_seconds = additional_minutes * 60
+        
+        # Start new conversation timer with extended duration
+        async def extended_timeout():
+            try:
+                await asyncio.sleep(new_timeout_seconds)
+                if self.session and self.current_state == AgentState.ACTIVE:
+                    logger.info(f"Extended conversation time reached - flagging for admin message injection")
+                    
+                    # Set flag for admin message injection on next user turn
+                    self.extension_expired_pending = True
+                    
+                    # Fallback timer - if user doesn't speak within 30 seconds, end with hardcoded message
+                    await asyncio.sleep(30)
+                    if self.session and self.current_state == AgentState.ACTIVE and self.extension_expired_pending:
+                        logger.info("Extension expired and no user response - ending with fallback message")
+                        
+                        # Set ending flag to prevent timer conflicts
+                        self.ending_conversation = True
+                        
+                        # Fallback ending message
+                        timeout_response = "friendly:I've really enjoyed our extended conversation! To help other visitors, I'll need to wrap up here. Say 'hey barista' if you need me again."
+                        emotion, text = self.process_emotional_response(timeout_response)
+                        await self.say_with_emotion(text, emotion)
+                        
+                        await asyncio.sleep(2)
+                        await self.end_conversation()
+            except asyncio.CancelledError:
+                pass
+                
+        self.conversation_timer = asyncio.create_task(extended_timeout())
+
     async def _wait_for_user_response(self):
         """Wait for user response after agent speaks"""
         try:

coffee_ws/src/coffee_voice_agent/scripts/tools/coffee_tools.py

@@ -154,8 +154,14 @@ async def manage_conversation_time_impl(
         result = f"Conversation ending initiated: {reason}"
 
     elif action == "extend":
-        logger.info(f"Conversation extended by {extension_minutes} minutes: {reason}")
-        result = f"Conversation extended by {extension_minutes} minutes: {reason}"
+        # Actually extend the conversation by calling state manager
+        if _agent_instance and hasattr(_agent_instance, 'state_manager') and extension_minutes > 0:
+            await _agent_instance.state_manager.extend_conversation(extension_minutes, reason)
+            logger.info(f"Conversation extended by {extension_minutes} minutes: {reason}")
+            result = f"Conversation extended by {extension_minutes} minutes: {reason}"
+        else:
+            logger.warning(f"Cannot extend conversation - no agent instance or invalid extension minutes: {extension_minutes}")
+            result = f"Extension failed: {reason}"
 
     elif action == "warn":
         logger.info(f"Conversation warning acknowledged: {reason}")