From 97f9580b1ab8fd814a9ded2a4c4c84442b5c562e Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Thu, 20 Feb 2025 19:49:50 -0800 Subject: [PATCH] rename --- .../providers/inline/agents/meta_reference/agent_instance.py | 4 ++-- llama_stack/providers/inline/agents/meta_reference/agents.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py index d0347e8cd..2344d5a17 100644 --- a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py +++ b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py @@ -247,8 +247,8 @@ class ChatAgent(ShieldRunnerMixin): yield chunk - async def continue_turn(self, request: AgentTurnResumeRequest) -> AsyncGenerator: - with tracing.span("continue_turn") as span: + async def resume_turn(self, request: AgentTurnResumeRequest) -> AsyncGenerator: + with tracing.span("resume_turn") as span: span.set_attribute("agent_id", self.agent_id) span.set_attribute("session_id", request.session_id) span.set_attribute("turn_id", request.turn_id) diff --git a/llama_stack/providers/inline/agents/meta_reference/agents.py b/llama_stack/providers/inline/agents/meta_reference/agents.py index ea438bab8..19b4c0925 100644 --- a/llama_stack/providers/inline/agents/meta_reference/agents.py +++ b/llama_stack/providers/inline/agents/meta_reference/agents.py @@ -195,7 +195,7 @@ class MetaReferenceAgentsImpl(Agents): request: AgentTurnResumeRequest, ) -> AsyncGenerator: agent = await self.get_agent(request.agent_id) - async for event in agent.continue_turn(request): + async for event in agent.resume_turn(request): yield event async def get_agents_turn(self, agent_id: str, session_id: str, turn_id: str) -> Turn: