Skip to content

Commit 975bc76

Browse files
author
Dylan Huang
committed
fix
1 parent 5155849 commit 975bc76

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

eval_protocol/pytest/default_agent_rollout_processor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,11 +8,11 @@
88
from openai.types.chat import ChatCompletionContentPartTextParam, ChatCompletionMessage, ChatCompletionToolParam
99
from openai.types.chat.chat_completion_message_param import ChatCompletionMessageParam
1010

11+
from eval_protocol.dataset_logger import default_logger
1112
from eval_protocol.mcp.execution.policy import LiteLLMPolicy
1213
from eval_protocol.mcp.mcp_multi_client import MCPMultiClient
1314
from eval_protocol.models import EvaluationRow, Message
1415
from eval_protocol.pytest.types import Dataset, RolloutProcessorConfig
15-
from eval_protocol.dataset_logger import default_logger
1616

1717

1818
class Agent:
@@ -106,7 +106,7 @@ def _get_content_from_tool_result(self, tool_result: CallToolResult) -> List[Tex
106106
return [TextContent(text=json.dumps(tool_result.structuredContent), type="text")]
107107
if not all(isinstance(content, TextContent) for content in tool_result.content):
108108
raise NotImplementedError("Non-text content is not supported yet")
109-
return tool_result.content[0].text
109+
return tool_result.content
110110

111111

112112
async def default_agent_rollout_processor(

0 commit comments

Comments
 (0)