swarm repositories / source
aboutsummaryrefslogtreecommitdiff
path: root/crates/phone-opus/src/mcp/host/runtime.rs
diff options
context:
space:
mode:
Diffstat (limited to 'crates/phone-opus/src/mcp/host/runtime.rs')
-rw-r--r--crates/phone-opus/src/mcp/host/runtime.rs11
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,