From 9a07e709ee3e45bed2203a00e3df74ced43b2b2e Mon Sep 17 00:00:00 2001
From: Xi Yan <xiyan@meta.com>
Date: Thu, 20 Feb 2025 19:48:54 -0800
Subject: [PATCH] rename

---
 .../inline/agents/meta_reference/agent_instance.py          | 4 ++--
 .../providers/inline/agents/meta_reference/agents.py        | 6 +++---
 2 files changed, 5 insertions(+), 5 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 b0d822511..d0347e8cd 100644
--- a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py
+++ b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py
@@ -23,7 +23,6 @@ from llama_stack.apis.agents import (
     AgentConfig,
     AgentToolGroup,
     AgentToolGroupWithArgs,
-    AgentTurnContinueRequest,
     AgentTurnCreateRequest,
     AgentTurnResponseEvent,
     AgentTurnResponseEventType,
@@ -34,6 +33,7 @@ from llama_stack.apis.agents import (
     AgentTurnResponseTurnAwaitingInputPayload,
     AgentTurnResponseTurnCompletePayload,
     AgentTurnResponseTurnStartPayload,
+    AgentTurnResumeRequest,
     Attachment,
     Document,
     InferenceStep,
@@ -247,7 +247,7 @@ class ChatAgent(ShieldRunnerMixin):
 
             yield chunk
 
-    async def continue_turn(self, request: AgentTurnContinueRequest) -> AsyncGenerator:
+    async def continue_turn(self, request: AgentTurnResumeRequest) -> AsyncGenerator:
         with tracing.span("continue_turn") as span:
             span.set_attribute("agent_id", self.agent_id)
             span.set_attribute("session_id", request.session_id)
diff --git a/llama_stack/providers/inline/agents/meta_reference/agents.py b/llama_stack/providers/inline/agents/meta_reference/agents.py
index 2b87b4bd6..ea438bab8 100644
--- a/llama_stack/providers/inline/agents/meta_reference/agents.py
+++ b/llama_stack/providers/inline/agents/meta_reference/agents.py
@@ -20,8 +20,8 @@ from llama_stack.apis.agents import (
     AgentSessionCreateResponse,
     AgentStepResponse,
     AgentToolGroup,
-    AgentTurnContinueRequest,
     AgentTurnCreateRequest,
+    AgentTurnResumeRequest,
     Document,
     Session,
     Turn,
@@ -178,7 +178,7 @@ class MetaReferenceAgentsImpl(Agents):
         tool_responses: List[ToolResponseMessage],
         stream: Optional[bool] = False,
     ) -> AsyncGenerator:
-        request = AgentTurnContinueRequest(
+        request = AgentTurnResumeRequest(
             agent_id=agent_id,
             session_id=session_id,
             turn_id=turn_id,
@@ -192,7 +192,7 @@ class MetaReferenceAgentsImpl(Agents):
 
     async def _continue_agent_turn_streaming(
         self,
-        request: AgentTurnContinueRequest,
+        request: AgentTurnResumeRequest,
     ) -> AsyncGenerator:
         agent = await self.get_agent(request.agent_id)
         async for event in agent.continue_turn(request):