mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-08-05 18:22:41 +00:00
rebase eval context extraction with query_memory tool
This commit is contained in:
parent
b7e59ba002
commit
bae2896528
1 changed files with 6 additions and 2 deletions
|
@ -146,8 +146,12 @@ class MetaReferenceEvalImpl(
|
||||||
# check if there's a memory retrieval step and extract the context
|
# check if there's a memory retrieval step and extract the context
|
||||||
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.memory_retrieval.value:
|
if step.step_type == StepType.tool_execution.value:
|
||||||
memory_rag_context = " ".join(x.text for x in step.inserted_context)
|
for i, tool_response in enumerate(step.tool_responses):
|
||||||
|
if tool_response.tool_name == "query_memory":
|
||||||
|
memory_rag_context = " ".join(
|
||||||
|
x.text for x in tool_response.content
|
||||||
|
)
|
||||||
|
|
||||||
agent_generation = {}
|
agent_generation = {}
|
||||||
agent_generation[ColumnName.generated_answer.value] = (
|
agent_generation[ColumnName.generated_answer.value] = (
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue