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

Add a simple agents test case

Browse source
This commit is contained in:
Ashwin Bharambe 2024-10-09 21:52:49 -07:00
parent 2d94ca71a9
commit ef4b74c935
6 changed files with 128 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

1
llama_stack/providers/impls/meta_reference/agents/__init__.py
View file

@ -21,6 +21,7 @@ async def get_provider_impl(
deps[Api.inference],
deps[Api.memory],
deps[Api.safety],
deps[Api.memory_banks],
)
await impl.initialize()
return impl

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

@ -137,3 +137,25 @@ class MetaReferenceAgentsImpl(Agents):
agent = await self.get_agent(request.agent_id)
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_step(
self, agent_id: str, turn_id: str, step_id: str
) -> AgentStepResponse:
raise NotImplementedError()
async def get_agents_session(
self,
agent_id: str,
session_id: str,
turn_ids: Optional[List[str]] = None,
) -> Session:
raise NotImplementedError()
async def delete_agents_session(self, agent_id: str, session_id: str) -> None:
raise NotImplementedError()
async def delete_agents(self, agent_id: str) -> None:
raise NotImplementedError()