Skip to content

Commit 58c7540

Browse files
author
Dylan Huang
committed
fix cleanup
1 parent a46c4b4 commit 58c7540

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

eval_protocol/pytest/default_agent_rollout_processor.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,5 +105,6 @@ async def default_agent_rollout_processor(
105105
await agent.setup()
106106
await agent.call_agent()
107107
dataset.append(EvaluationRow(messages=agent.messages, ground_truth=row.ground_truth))
108-
await agent.mcp_client.cleanup()
108+
if agent.mcp_client:
109+
await agent.mcp_client.cleanup()
109110
return dataset

0 commit comments

Comments
 (0)