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

fix the rag query generator types

Browse source
This commit is contained in:
Dinesh Yeduguru 2025-01-07 21:33:25 -08:00
parent efe3189728
commit 82395ba654
4 changed files with 19 additions and 10 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

@ -401,7 +401,9 @@ class ChatAgent(ShieldRunnerMixin):
session_info = await self.storage.get_session_info(session_id)
# if the session has a memory bank id, let the memory tool use it
if session_info.memory_bank_id:
query_args["memory_bank_id"] = session_info.memory_bank_id
if "memory_bank_ids" not in query_args:
query_args["memory_bank_ids"] = []
query_args["memory_bank_ids"].append(session_info.memory_bank_id)
yield AgentTurnResponseStreamChunk(
event=AgentTurnResponseEvent(
payload=AgentTurnResponseStepProgressPayload(