Skip to content

Commit 40238ba

Browse files
author
Dylan Huang
committed
fix
1 parent 1e9f78a commit 40238ba

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

eval_protocol/pytest/default_single_turn_rollout_process.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,9 +24,10 @@ async def process_row(row: EvaluationRow) -> EvaluationRow:
2424

2525
messages_payload = [{"role": m.role, "content": m.content} for m in row.messages]
2626

27-
response = await client.chat.completions.create(
28-
model=config.model, messages=messages_payload, tools=row.tools, **config.input_params
29-
)
27+
create_kwargs = dict(model=config.model, messages=messages_payload, **config.input_params)
28+
if row.tools is not None:
29+
create_kwargs["tools"] = row.tools
30+
response = await client.chat.completions.create(**create_kwargs)
3031
assistant_content = response.choices[0].message.content or ""
3132
tool_calls = response.choices[0].message.tool_calls if response.choices[0].message.tool_calls else None
3233
messages = list(row.messages) + [

0 commit comments

Comments
 (0)