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

rename

Browse source
This commit is contained in:
Xi Yan 2025-02-20 19:49:50 -08:00
parent 9a07e709ee
commit 97f9580b1a
2 changed files with 3 additions and 3 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

@ -247,8 +247,8 @@ class ChatAgent(ShieldRunnerMixin):
yield chunk yield chunk
async def continue_turn(self, request: AgentTurnResumeRequest) -> AsyncGenerator: async def resume_turn(self, request: AgentTurnResumeRequest) -> AsyncGenerator:
with tracing.span("continue_turn") as span: with tracing.span("resume_turn") as span:
span.set_attribute("agent_id", self.agent_id) span.set_attribute("agent_id", self.agent_id)
span.set_attribute("session_id", request.session_id) span.set_attribute("session_id", request.session_id)
span.set_attribute("turn_id", request.turn_id) span.set_attribute("turn_id", request.turn_id)

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

@ -195,7 +195,7 @@ class MetaReferenceAgentsImpl(Agents):
request: AgentTurnResumeRequest, request: AgentTurnResumeRequest,
) -> AsyncGenerator: ) -> AsyncGenerator:
agent = await self.get_agent(request.agent_id) 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 yield event
async def get_agents_turn(self, agent_id: str, session_id: str, turn_id: str) -> Turn: async def get_agents_turn(self, agent_id: str, session_id: str, turn_id: str) -> Turn: