diff --git a/llama_stack/apis/agents/agents.py b/llama_stack/apis/agents/agents.py index 7ff276dae..6500cefa0 100644 --- a/llama_stack/apis/agents/agents.py +++ b/llama_stack/apis/agents/agents.py @@ -442,13 +442,13 @@ class Agents(Protocol): async def get_agents_turn( self, agent_id: str, - turn_id: str, - session_id: str + session_id: str, + turn_id: str ) -> Turn: ... @webmethod(route="/agents/step/get") async def get_agents_step( - self, agent_id: str, turn_id: str, step_id: str, session_id: str + self, agent_id: str, session_id: str, turn_id: str, step_id: str ) -> AgentStepResponse: ... @webmethod(route="/agents/session/create") diff --git a/llama_stack/providers/impls/meta_reference/agents/agents.py b/llama_stack/providers/impls/meta_reference/agents/agents.py index 21f5ba3af..dbfa4c526 100644 --- a/llama_stack/providers/impls/meta_reference/agents/agents.py +++ b/llama_stack/providers/impls/meta_reference/agents/agents.py @@ -155,7 +155,7 @@ class MetaReferenceAgentsImpl(Agents): return turn async def get_agents_step( - self, agent_id: str, turn_id: str, session_id: str, step_id: str + self, agent_id: str, session_id: str, turn_id: str, step_id: str ) -> AgentStepResponse: turn = await self.persistence_store.get(f"session:{agent_id}:{session_id}:{turn_id}") try: