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

add documents to turn

Browse source
This commit is contained in:
Dinesh Yeduguru 2025-01-06 11:40:22 -08:00
parent d0e8e1647b
commit 9efe30c9d3
9 changed files with 887 additions and 381 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -21,6 +21,7 @@ from llama_stack.apis.agents import (
AgentStepResponse,
AgentTool,
AgentTurnCreateRequest,
Document,
Session,
Turn,
)
@ -147,6 +148,7 @@ class MetaReferenceAgentsImpl(Agents):
]
],
tools: Optional[List[AgentTool]] = None,
documents: Optional[List[Document]] = None,
stream: Optional[bool] = False,
) -> AsyncGenerator:
request = AgentTurnCreateRequest(
@ -155,6 +157,7 @@ class MetaReferenceAgentsImpl(Agents):
messages=messages,
stream=True,
tools=tools,
documents=documents,
)
if stream:
return self._create_agent_turn_streaming(request)