Skip to content

Commit 8a5130a

Browse files
committed
fix tests
1 parent 391d8b7 commit 8a5130a

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

eval_protocol/pytest/default_single_turn_rollout_process.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -166,13 +166,17 @@ async def process_row(row: EvaluationRow) -> EvaluationRow:
166166
row.execution_metadata.tool_call_count = (
167167
len(converted_tool_calls) if converted_tool_calls is not None else 0
168168
)
169-
row.execution_metadata.usage = (
170-
CompletionUsage( # Note: LiteLLM sets usage dynamically via setattr(), not as a typed field
171-
prompt_tokens=response.usage.prompt_tokens, # pyright: ignore[reportAttributeAccessIssue]
172-
completion_tokens=response.usage.completion_tokens, # pyright: ignore[reportAttributeAccessIssue]
173-
total_tokens=response.usage.total_tokens, # pyright: ignore[reportAttributeAccessIssue]
169+
usage = getattr(response, "usage", None)
170+
if usage:
171+
row.execution_metadata.usage = (
172+
CompletionUsage( # Note: LiteLLM sets usage dynamically via setattr(), not as a typed field
173+
prompt_tokens=getattr(usage, "prompt_tokens", 0),
174+
completion_tokens=getattr(usage, "completion_tokens", 0),
175+
total_tokens=getattr(usage, "total_tokens", 0),
176+
)
174177
)
175-
)
178+
else:
179+
row.execution_metadata.usage = None
176180

177181
row.messages = messages
178182

0 commit comments

Comments
 (0)