Skip to content

Commit d76d967

Browse files
committed
remove pydantic usage tracking for now
1 parent 0859289 commit d76d967

File tree

2 files changed

+8
-7
lines changed

2 files changed

+8
-7
lines changed

eval_protocol/models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -545,7 +545,7 @@ class ExecutionMetadata(BaseModel):
545545
)
546546

547547
run_id: Optional[str] = Field(
548-
None,
548+
default=None,
549549
description=("The ID of the run that this row belongs to."),
550550
)
551551

eval_protocol/pytest/default_pydantic_ai_rollout_processor.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -91,12 +91,13 @@ async def process_row(row: EvaluationRow) -> EvaluationRow:
9191
)
9292
row.messages = await self.convert_pyd_message_to_ep_message(response.all_messages())
9393

94-
usage_info = response.usage()
95-
row.execution_metadata.usage = CompletionUsage(
96-
prompt_tokens=usage_info.request_tokens or 0,
97-
completion_tokens=usage_info.response_tokens or 0,
98-
total_tokens=usage_info.total_tokens or 0,
99-
)
94+
# TODO: pydantic ai accumulates usage info across all models in multi-agent setup, so this simple tracking doesn't work for cost. to discuss with @dphuang2 when he's back.
95+
# usage_info = response.usage()
96+
# row.execution_metadata.usage = CompletionUsage(
97+
# prompt_tokens=usage_info.request_tokens or 0,
98+
# completion_tokens=usage_info.response_tokens or 0,
99+
# total_tokens=usage_info.total_tokens or 0,
100+
# )
100101

101102
return row
102103

0 commit comments

Comments
 (0)