phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-06 10:42:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

reuse some variables

Browse source
This commit is contained in:
Ashwin Bharambe 2025-01-21 21:19:19 -08:00
parent 89f51a86dd
commit 5297aef9e8
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/providers/inline/agents/meta_reference/agent_instance.py
View file

@ -396,7 +396,7 @@ class ChatAgent(ShieldRunnerMixin):
session_id, documents, input_messages, tool_defs session_id, documents, input_messages, tool_defs
) )
if "builtin::memory" in toolgroups and len(input_messages) > 0: if MEMORY_GROUP in toolgroups and len(input_messages) > 0:
with tracing.span(MEMORY_QUERY_TOOL) as span: with tracing.span(MEMORY_QUERY_TOOL) as span:
step_id = str(uuid.uuid4()) step_id = str(uuid.uuid4())
yield AgentTurnResponseStreamChunk( yield AgentTurnResponseStreamChunk(
@ -408,7 +408,7 @@ class ChatAgent(ShieldRunnerMixin):
) )
) )
args = toolgroup_args.get("builtin::memory", {}) args = toolgroup_args.get(MEMORY_GROUP, {})
vector_db_ids = args.get("vector_db_ids", []) vector_db_ids = args.get("vector_db_ids", [])
session_info = await self.storage.get_session_info(session_id) session_info = await self.storage.get_session_info(session_id)
@ -475,7 +475,7 @@ class ChatAgent(ShieldRunnerMixin):
"input", [m.model_dump_json() for m in input_messages] "input", [m.model_dump_json() for m in input_messages]
) )
span.set_attribute("output", retrieved_context) span.set_attribute("output", retrieved_context)
span.set_attribute("tool_name", "builtin::memory") span.set_attribute("tool_name", MEMORY_QUERY_TOOL)
if retrieved_context: if retrieved_context:
last_message = input_messages[-1] last_message = input_messages[-1]
last_message.context = retrieved_context last_message.context = retrieved_context