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

comments

Browse source
This commit is contained in:
Xi Yan 2025-01-15 12:52:25 -08:00
parent bae2896528
commit 26bcb9603e
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
llama_stack/providers/inline/eval/meta_reference/eval.py
View file

@ -16,6 +16,9 @@ from llama_stack.apis.scoring import Scoring
from llama_stack.distribution.datatypes import Api from llama_stack.distribution.datatypes import Api
from llama_stack.providers.datatypes import EvalTasksProtocolPrivate from llama_stack.providers.datatypes import EvalTasksProtocolPrivate
from llama_stack.providers.inline.agents.meta_reference.agent_instance import (
MEMORY_QUERY_TOOL,
)
from llama_stack.providers.utils.common.data_schema_validator import ( from llama_stack.providers.utils.common.data_schema_validator import (
ColumnName, ColumnName,
get_valid_schemas, get_valid_schemas,
@ -147,8 +150,8 @@ class MetaReferenceEvalImpl(
memory_rag_context = None memory_rag_context = None
for step in final_event.turn.steps: for step in final_event.turn.steps:
if step.step_type == StepType.tool_execution.value: if step.step_type == StepType.tool_execution.value:
for i, tool_response in enumerate(step.tool_responses): for tool_response in step.tool_responses:
if tool_response.tool_name == "query_memory": if tool_response.tool_name == MEMORY_QUERY_TOOL:
memory_rag_context = " ".join( memory_rag_context = " ".join(
x.text for x in tool_response.content x.text for x in tool_response.content
) )