Skip to content

Commit 41c74b6

Browse files
author
Dylan Huang
committed
fix defualt_agent_rollout_processor.py
1 parent adc73df commit 41c74b6

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

eval_protocol/pytest/default_agent_rollout_processor.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -41,9 +41,9 @@ async def _get_tools(self) -> Optional[List[ChatCompletionToolParam]]:
4141
def messages(self) -> list[Message]:
4242
return self.evaluation_row.messages
4343

44-
def append_message_and_log(self, message: Message, logger: DatasetLogger):
44+
def append_message_and_log(self, message: Message):
4545
self.messages.append(message)
46-
logger.log(self.evaluation_row)
46+
self.logger.log(self.evaluation_row)
4747

4848
async def call_agent(self) -> str:
4949
"""
@@ -78,8 +78,7 @@ async def call_agent(self) -> str:
7878
ChatCompletionContentPartTextParam(text=content.text, type="text") for content in content
7979
],
8080
tool_call_id=tool_call_id,
81-
),
82-
logger=self.logger,
81+
)
8382
)
8483
return await self.call_agent()
8584
return message.content

0 commit comments

Comments
 (0)