angelo fix tests

Author: AngeloGiacco

src/elevenlabs/conversational_ai/conversation.py

@@ -275,39 +275,80 @@ def __init__(
         self.user_id = user_id
 
 
-class MessageHandler(Protocol):
-    """Protocol defining the interface for message handlers."""
+class SyncMessageHandler(Protocol):
+    """Protocol defining the interface for sync message handlers."""
 
     callback_agent_response: Optional[Callable]
     callback_agent_response_correction: Optional[Callable] 
     callback_user_transcript: Optional[Callable]
     callback_latency_measurement: Optional[Callable]
 
-    def handle_audio_output(self, audio: bytes) -> Union[None, Awaitable[None]]:
+    def handle_audio_output(self, audio: bytes) -> None:
         """Handle audio output."""
         ...
 
-    def handle_agent_response(self, response: str) -> Union[None, Awaitable[None]]:
+    def handle_agent_response(self, response: str) -> None:
         """Handle agent response."""
         ...
 
-    def handle_agent_response_correction(self, original: str, corrected: str) -> Union[None, Awaitable[None]]:
+    def handle_agent_response_correction(self, original: str, corrected: str) -> None:
         """Handle agent response correction."""
         ...
 
-    def handle_user_transcript(self, transcript: str) -> Union[None, Awaitable[None]]:
+    def handle_user_transcript(self, transcript: str) -> None:
         """Handle user transcript."""
         ...
 
-    def handle_interruption(self) -> Union[None, Awaitable[None]]:
+    def handle_interruption(self) -> None:
         """Handle interruption."""
         ...
 
-    def handle_ping(self, event: Dict[str, Any]) -> Union[None, Awaitable[None]]:
+    def handle_ping(self, event: Dict[str, Any]) -> None:
         """Handle ping event."""
         ...
 
-    def handle_latency_measurement(self, latency: int) -> Union[None, Awaitable[None]]:
+    def handle_latency_measurement(self, latency: int) -> None:
+        """Handle latency measurement."""
+        ...
+        
+    def handle_client_tool_call(self, tool_name: str, parameters: Dict[str, Any]) -> None:
+        """Handle client tool call."""
+        ...
+
+
+class AsyncMessageHandler(Protocol):
+    """Protocol defining the interface for async message handlers."""
+    
+    callback_agent_response: Optional[Callable]
+    callback_agent_response_correction: Optional[Callable] 
+    callback_user_transcript: Optional[Callable]
+    callback_latency_measurement: Optional[Callable]
+    
+    async def handle_audio_output(self, audio: bytes) -> None:
+        """Handle audio output."""
+        ...
+    
+    async def handle_agent_response(self, response: str) -> None:
+        """Handle agent response."""
+        ...
+    
+    async def handle_agent_response_correction(self, original: str, corrected: str) -> None:
+        """Handle agent response correction."""
+        ...
+        
+    async def handle_user_transcript(self, transcript: str) -> None:
+        """Handle user transcript."""
+        ...
+        
+    async def handle_interruption(self) -> None:
+        """Handle interruption."""
+        ...
+        
+    async def handle_ping(self, event: Dict[str, Any]) -> None:
+        """Handle ping event."""
+        ...
+        
+    async def handle_latency_measurement(self, latency: int) -> None:
         """Handle latency measurement."""
         ...
 
@@ -340,7 +381,7 @@ def _send_response(self, response: Dict[str, Any]) -> None:
         raise NotImplementedError
 
 
-class BaseConversation:
+class BaseConversation(ABC):
     """Base class for conversation implementations with shared parameters and logic."""
 
     def __init__(
@@ -397,6 +438,11 @@ async def callback(audio: bytes) -> None:
                 await self.end_session()
         return callback
 
+    @abstractmethod
+    def end_session(self) -> Union[None, Awaitable[None]]:
+        """End the conversation session - to be implemented by subclasses."""
+        pass
+    
     def _handle_connection_closed(self) -> Union[None, Awaitable[None]]:
         """Handle WebSocket connection closed - to be implemented by subclasses."""
         raise NotImplementedError
@@ -427,7 +473,7 @@ def _create_initiation_message(self):
             }
         )
 
-    def _handle_message_core(self, message: Dict[str, Any], message_handler: MessageHandler) -> None:
+    def _handle_message_core(self, message: Dict[str, Any], message_handler: SyncMessageHandler) -> None:
         """Core message handling logic shared between sync and async implementations.
         
         Args:
@@ -483,7 +529,7 @@ def _handle_message_core(self, message: Dict[str, Any], message_handler: Message
         else:
             pass  # Ignore all other message types.
 
-    async def _handle_message_core_async(self, message: Dict[str, Any], message_handler: MessageHandler) -> None:
+    async def _handle_message_core_async(self, message: Dict[str, Any], message_handler: AsyncMessageHandler) -> None:
         """Async wrapper for core message handling logic."""
         if message["type"] == "conversation_initiation_metadata":
             event = message["conversation_initiation_metadata_event"]