fix: function calls (#556)

Author: jpvajda

deepgram/__init__.py

@@ -324,6 +324,7 @@
     ConversationTextResponse,
     UserStartedSpeakingResponse,
     AgentThinkingResponse,
+    FunctionCall,
     FunctionCallRequest,
     AgentStartedSpeakingResponse,
     AgentAudioDoneResponse,

deepgram/client.py

@@ -338,6 +338,7 @@
     ConversationTextResponse,
     UserStartedSpeakingResponse,
     AgentThinkingResponse,
+    FunctionCall,
     FunctionCallRequest,
     AgentStartedSpeakingResponse,
     AgentAudioDoneResponse,

deepgram/clients/__init__.py

@@ -347,6 +347,7 @@
     ConversationTextResponse,
     UserStartedSpeakingResponse,
     AgentThinkingResponse,
+    FunctionCall,
     FunctionCallRequest,
     AgentStartedSpeakingResponse,
     AgentAudioDoneResponse,

deepgram/clients/agent/__init__.py

@@ -22,6 +22,7 @@
     ConversationTextResponse,
     UserStartedSpeakingResponse,
     AgentThinkingResponse,
+    FunctionCall,
     FunctionCallRequest,
     AgentStartedSpeakingResponse,
     AgentAudioDoneResponse,

deepgram/clients/agent/client.py

@@ -21,6 +21,7 @@
     ConversationTextResponse as LatestConversationTextResponse,
     UserStartedSpeakingResponse as LatestUserStartedSpeakingResponse,
     AgentThinkingResponse as LatestAgentThinkingResponse,
+    FunctionCall as LatestFunctionCall,
     FunctionCallRequest as LatestFunctionCallRequest,
     AgentStartedSpeakingResponse as LatestAgentStartedSpeakingResponse,
     AgentAudioDoneResponse as LatestAgentAudioDoneResponse,
@@ -70,6 +71,7 @@
 ConversationTextResponse = LatestConversationTextResponse
 UserStartedSpeakingResponse = LatestUserStartedSpeakingResponse
 AgentThinkingResponse = LatestAgentThinkingResponse
+FunctionCall = LatestFunctionCall
 FunctionCallRequest = LatestFunctionCallRequest
 AgentStartedSpeakingResponse = LatestAgentStartedSpeakingResponse
 AgentAudioDoneResponse = LatestAgentAudioDoneResponse

deepgram/clients/agent/v1/__init__.py

@@ -26,6 +26,7 @@
     ConversationTextResponse,
     UserStartedSpeakingResponse,
     AgentThinkingResponse,
+    FunctionCall,
     FunctionCallRequest,
     AgentStartedSpeakingResponse,
     AgentAudioDoneResponse,

deepgram/clients/agent/v1/websocket/__init__.py

@@ -18,6 +18,7 @@
     ConversationTextResponse,
     UserStartedSpeakingResponse,
     AgentThinkingResponse,
+    FunctionCall,
     FunctionCallRequest,
     AgentStartedSpeakingResponse,
     AgentAudioDoneResponse,

deepgram/clients/agent/v1/websocket/options.py

@@ -448,8 +448,9 @@ class FunctionCallResponse(BaseResponse):
     """
 
     type: str = "FunctionCallResponse"
-    function_call_id: str = field(default="")
-    output: str = field(default="")
+    id: str = field(default="")
+    name: str = field(default="")
+    content: str = field(default="")
 
 
 # Agent Keep Alive

deepgram/clients/agent/v1/websocket/response.py

@@ -68,16 +68,43 @@ class AgentThinkingResponse(BaseResponse):
     content: str
 
 
+@dataclass
+class FunctionCall(BaseResponse):
+    """
+    Individual function call within a FunctionCallRequest.
+    """
+
+    id: str
+    name: str
+    arguments: str
+    client_side: bool
+
+
 @dataclass
 class FunctionCallRequest(BaseResponse):
     """
     The FunctionCallRequest message is used to call a function from the server to the client.
     """
 
     type: str
-    function_name: str
-    function_call_id: str
-    input: str
+    functions: List[FunctionCall]
+
+    def __post_init__(self):
+        """Convert dict functions to FunctionCall objects if needed."""
+        if self.functions:
+            self.functions = [
+                FunctionCall.from_dict(func) if isinstance(func, dict) else func
+                for func in self.functions
+            ]
+
+    def __getitem__(self, key):
+        _dict = self.to_dict()
+        if "functions" in _dict and isinstance(_dict["functions"], list):
+            _dict["functions"] = [
+                FunctionCall.from_dict(func) if isinstance(func, dict) else func
+                for func in _dict["functions"]
+            ]
+        return _dict[key]
 
 
 @dataclass