diff --git a/llama_stack/providers/inline/agents/meta_reference/agents.py b/llama_stack/providers/inline/agents/meta_reference/agents.py
index 87d73bf41..c208388be 100644
--- a/llama_stack/providers/inline/agents/meta_reference/agents.py
+++ b/llama_stack/providers/inline/agents/meta_reference/agents.py
@@ -175,7 +175,7 @@ class MetaReferenceAgentsImpl(Agents):
         session_id: str,
         turn_id: str,
         tool_response_messages: List[ToolResponseMessage],
-    ) -> Union[Turn, AsyncIterator[AgentTurnResponseStreamChunk]]:
+    ) -> AsyncGenerator:
         pass
 
     async def get_agents_turn(self, agent_id: str, session_id: str, turn_id: str) -> Turn: