Skip to content

Commit 5329944

Browse files
committed
fix mypy
1 parent 9c2c534 commit 5329944

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/agents/extensions/models/litellm_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -262,10 +262,10 @@ async def stream_response(
262262
),
263263
model=self.model,
264264
)
265-
cost = litellm.completion_cost(completion_response=mock_response)
265+
cost = litellm.completion_cost(completion_response=mock_response) # type: ignore[attr-defined]
266266
# Attach cost as a custom attribute on the Response object so
267267
# run.py can access it when creating the Usage object.
268-
final_response._litellm_cost = cost
268+
final_response._litellm_cost = cost # type: ignore[attr-defined]
269269
except Exception:
270270
# If cost calculation fails (e.g., unknown model), continue
271271
# without cost.

0 commit comments

Comments
 (0)