phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-12 13:00:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

rename

Browse source
This commit is contained in:
Xi Yan 2025-02-20 19:48:54 -08:00
parent 6d08a935ba
commit 9a07e709ee
2 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/inline/agents/meta_reference/agent_instance.py
View file

@ -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)

6
llama_stack/providers/inline/agents/meta_reference/agents.py
View file

@ -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):