Add blocking wait for background consult jobs

1 files changed, 10 insertions, 1 deletions

diff --git a/crates/phone-opus/src/mcp/host/runtime.rs b/crates/phone-opus/src/mcp/host/runtime.rs
index 688ffd6..df54844 100644
--- a/crates/phone-opus/src/mcp/host/runtime.rs
+++ b/crates/phone-opus/src/mcp/host/runtime.rs
@@ -25,7 +25,9 @@ use crate::mcp::protocol::{
     FORCE_ROLLOUT_ENV, HOST_STATE_ENV, HostRequestId, HostStateSeed, PROTOCOL_VERSION, SERVER_NAME,
     WORKER_CRASH_ONCE_ENV, WorkerOperation, WorkerSpawnConfig,
 };
-use crate::mcp::service::{consult_job_tool_output, consult_jobs_tool_output};
+use crate::mcp::service::{
+    consult_job_tool_output, consult_jobs_tool_output, consult_wait_tool_output,
+};
 use crate::mcp::telemetry::ServerTelemetry;
 
 pub(crate) fn run_host() -> Result<(), Box<dyn std::error::Error>> {
@@ -371,6 +373,13 @@ impl HostRuntime {
                 FaultStage::Host,
                 &operation,
             ),
+            "consult_wait" => tool_success(
+                consult_wait_tool_output(arguments, generation, FaultStage::Host, &operation)?,
+                presentation,
+                generation,
+                FaultStage::Host,
+                &operation,
+            ),
             "consult_jobs" => tool_success(
                 consult_jobs_tool_output(arguments, generation, FaultStage::Host, &operation)?,
                 presentation,