diff --git a/src/pipecat/frames/frames.py b/src/pipecat/frames/frames.py
index 39e25fc11..11cd17801 100644
--- a/src/pipecat/frames/frames.py
+++ b/src/pipecat/frames/frames.py
@@ -367,7 +367,7 @@ class LLMSetToolsFrame(DataFrame):
 class LLMSetToolChoiceFrame(DataFrame):
     """A frame containing a tool choice for an LLM to use for function calling."""
 
-    tool_choice: Literal["none", "auto", "required"] | str
+    tool_choice: Literal["none", "auto", "required"] | dict
 
 
 @dataclass
diff --git a/src/pipecat/processors/aggregators/llm_response.py b/src/pipecat/processors/aggregators/llm_response.py
index c0ef65f12..af8bf1a2e 100644
--- a/src/pipecat/processors/aggregators/llm_response.py
+++ b/src/pipecat/processors/aggregators/llm_response.py
@@ -210,7 +210,7 @@ class LLMContextResponseAggregator(BaseLLMResponseAggregator):
     def set_tools(self, tools: List):
         self._context.set_tools(tools)
 
-    def set_tool_choice(self, tool_choice: Literal["none", "auto", "required"] | str):
+    def set_tool_choice(self, tool_choice: Literal["none", "auto", "required"] | dict):
         self._context.set_tool_choice(tool_choice)
 
     def reset(self):