fix: Replace done_callback with coroutine chain for judge tracking

[jsonbailey]

Summary

• Replaces add_done_callback in ManagedModel._track_judge_results with a proper _run_and_track coroutine wrapped in asyncio.create_task, so that awaiting response.evaluations guarantees both evaluation and tracker.track_judge_result() calls complete in sequence
• The old callback approach ran outside the asyncio scheduler and gave no ordering guarantees; the new approach makes the chain explicit and testable
• Adds test_managed_model.py with 5 tests covering: return-before-resolve, collect results, tracking fires inside the awaited chain, failed results skip tracking, noop evaluator returns empty list

Test plan

• All 138 existing tests pass
• 5 new test_managed_model.py tests added and passing
• e2e validation via hello-python-ai agent

🤖 Generated with Claude Code

---

Note

Medium Risk
Changes async task orchestration for judge evaluation/tracking, which can affect ordering, exception propagation, and background task behavior. New tests reduce regression risk but subtle event-loop edge cases remain possible.

Overview
Fixes the judge-evaluation tracking chain so that awaiting ModelResponse.evaluations now guarantees judge results are tracked in-sequence with evaluation completion.

ManagedModel._track_judge_results replaces the prior add_done_callback approach with an awaited coroutine wrapper (_run_and_track) scheduled via asyncio.create_task, adds warning logs for failed judge results and tracking exceptions, and introduces a focused test_managed_model.py suite covering non-blocking invoke(), result collection, tracking timing, and failure/noop behavior.

^{Reviewed by Cursor Bugbot for commit 7ae2024. Bugbot is set up for automated code reviews on this repo. Configure here.}

[cursor]

Cursor Bugbot has reviewed your changes and found 1 potential issue.

^{❌ Bugbot Autofix is OFF. To automatically fix reported issues with cloud agents, have a team admin enable autofix in the Cursor dashboard.}

^{Reviewed by Cursor Bugbot for commit 2f0fd53. Configure here.}