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

Added implementation for get_agents_step and get_agents_turn

Browse source
This commit is contained in:
Sarthak Deshpande 2024-10-18 17:21:33 +05:30
parent c2e8011175
commit 761217dcc4
2 changed files with 37 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llama_stack/apis/agents/agents.py
View file

@ -443,11 +443,12 @@ class Agents(Protocol):
self,
agent_id: str,
turn_id: str,
session_id: str
) -> Turn: ...
@webmethod(route="/agents/step/get")
async def get_agents_step(
self, agent_id: str, turn_id: str, step_id: str
self, agent_id: str, turn_id: str, step_id: str, session_id: str
) -> AgentStepResponse: ...
@webmethod(route="/agents/session/create")

39
llama_stack/providers/impls/meta_reference/agents/agents.py
View file

@ -138,13 +138,44 @@ class MetaReferenceAgentsImpl(Agents):
async for event in agent.create_and_execute_turn(request):
yield event
async def get_agents_turn(self, agent_id: str, turn_id: str) -> Turn:
raise NotImplementedError()
async def get_agents_turn(self, agent_id: str, session_id: str, turn_id: str) -> Turn:
turn = await self.persistence_store.get(f"session:{agent_id}:{session_id}:{turn_id}")
try:
turn = json.loads(turn)
except json.JSONDecodeError as e:
raise ValueError(
f"Could not JSON decode turn for {turn_id}"
) from e
try:
turn = Turn(**turn)
except Exception as e:
raise ValueError(
f"Could not validate(?) Turns for {turn_id}"
) from e
return turn
async def get_agents_step(
self, agent_id: str, turn_id: str, step_id: str
self, agent_id: str, turn_id: str, session_id: str, step_id: str
) -> AgentStepResponse:
raise NotImplementedError()
turn = await self.persistence_store.get(f"session:{agent_id}:{session_id}:{turn_id}")
try:
turn = json.loads(turn)
except json.JSONDecodeError as e:
raise ValueError(
f"Could not JSON decode turn for {turn_id}"
) from e
try:
turn = Turn(**turn)
except Exception as e:
raise ValueError(
f"Could not validate(?) Turns for {turn_id}"
) from e
steps = turn.steps
for step in steps:
if step.step_id == step_id:
return AgentStepResponse(step=step)
raise ValueError("Provided step_id could not be found")
async def get_agents_session(
self,