Improve agent retries and tool progress

2026-08-23 22:22:32 -05:00
parent 7c106eecd5
commit f4dc6231d6
14 changed files with 913 additions and 141 deletions
@@ -81,6 +81,7 @@ pub(crate) enum ProviderRunProjection {
 #[derive(Clone, Debug, PartialEq)]
 pub(crate) enum ProviderRunBlock {
    Tools(PendingToolBatch),
+    ReadyToCallModel,
    AwaitingDriver {
        work_id: ExternalWorkId,
        stop_reason: StopReason,
@@ -383,7 +384,7 @@ impl ProviderRunCoordinator {
                    }
                    if batch.is_complete() {
                        self.run.commit_tool_batch(&batch.work_id)?;
-                        continue;
+                        return Ok(ProviderRunBlock::ReadyToCallModel);
                    }
                    return Ok(ProviderRunBlock::Tools(batch));
                }