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

Changed order of parameters

Browse source
This commit is contained in:
Sarthak Deshpande 2024-10-18 17:50:14 +05:30
parent 761217dcc4
commit a542509399
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

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

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

@ -155,7 +155,7 @@ class MetaReferenceAgentsImpl(Agents):
return turn return turn
async def get_agents_step( async def get_agents_step(
self, agent_id: str, turn_id: str, session_id: str, step_id: str self, agent_id: str, session_id: str, turn_id: str, step_id: str
) -> AgentStepResponse: ) -> AgentStepResponse:
turn = await self.persistence_store.get(f"session:{agent_id}:{session_id}:{turn_id}") turn = await self.persistence_store.get(f"session:{agent_id}:{session_id}:{turn_id}")
try: try: