commit 7e1b37cbe6b5017e6b049de597e4c72ec9ba2399
parent 307b277f1d3e86032d8074ff7ada637549d229c4
Author: triesap <tyson@radroots.org>
Date: Sun, 12 Apr 2026 04:16:19 +0000
control: surface provider runtime truth
Diffstat:
10 files changed, 330 insertions(+), 75 deletions(-)
diff --git a/src/hyf_assist/bridge.mojo b/src/hyf_assist/bridge.mojo
@@ -19,7 +19,13 @@ from hyf_assist.contract import (
assist_bridge_fake_endpoint_prefix,
assist_bridge_runtime_id,
assist_bridge_supported_business_capabilities,
+ provider_runtime_id,
)
+from hyf_provider.config import (
+ default_max_local_provider_config,
+ load_max_local_provider_config,
+)
+from hyf_provider.max_local import max_local_provider_status
from hyf_runtime.config import (
HyfLoadedRuntimeConfig,
assist_bridge_configured,
@@ -240,18 +246,31 @@ def _resolve_real_bridge_status(endpoint: String) raises -> AssistBridgeStatus:
def resolve_assist_bridge_status(
config: HyfLoadedRuntimeConfig,
) -> AssistBridgeStatus:
+ var provider_config = default_max_local_provider_config()
+ try:
+ provider_config = load_max_local_provider_config()
+ except:
+ pass
+
var configured = assist_bridge_configured(config)
var state = "disabled_by_runtime_config"
var reachable = False
- var backend_kind = "fake"
+ var runtime_id = provider_runtime_id()
+ var kind = "provider_runtime"
+ var transport = "in_process"
+ var endpoint = String("")
+ var backend_kind = "max_local"
var provider = ""
- var route = ""
- var model = ""
+ var route = String(provider_config.route)
+ var model = String(provider_config.model)
var supported_capabilities = assist_bridge_supported_business_capabilities()
if assisted_execution_enabled(config):
if configured:
- var endpoint = String(config.effective.assist.endpoint)
+ endpoint = String(config.effective.assist.endpoint)
if _fake_bridge_endpoint_is_reachable(endpoint):
+ runtime_id = assist_bridge_runtime_id()
+ kind = "assist_bridge"
+ transport = String(config.effective.assist.transport)
reachable = True
state = "ready"
backend_kind = "fake"
@@ -259,30 +278,22 @@ def resolve_assist_bridge_status(
route = "assist_bridge.query_rewrite.fake"
model = "fake_query_rewrite_v1"
else:
- try:
- var resolved = _resolve_real_bridge_status(endpoint)
- reachable = resolved.reachable
- state = String(resolved.state)
- backend_kind = String(resolved.backend_kind)
- provider = String(resolved.provider)
- route = String(resolved.route)
- model = String(resolved.model)
- supported_capabilities = copy_string_list(
- resolved.supported_business_capabilities
- )
- except e:
- reachable = False
- state = "unavailable"
- backend_kind = "fake"
+ var resolved = max_local_provider_status(provider_config)
+ reachable = resolved.reachable
+ state = String(resolved.state)
+ backend_kind = String(resolved.backend_kind)
+ provider = String(resolved.provider)
+ endpoint = ""
+ supported_capabilities = assist_bridge_supported_business_capabilities()
else:
state = "unconfigured"
return AssistBridgeStatus(
- id=assist_bridge_runtime_id(),
- kind="assist_bridge",
+ id=runtime_id,
+ kind=kind,
contract_version=assist_bridge_contract_version(),
- transport=String(config.effective.assist.transport),
- endpoint=String(config.effective.assist.endpoint),
+ transport=transport,
+ endpoint=endpoint,
backend_kind=String(backend_kind),
provider=String(provider),
route=String(route),
@@ -301,15 +312,21 @@ def assisted_execution_state_for_capability(
if capability_id != "query_rewrite":
return "deferred"
+ var unavailable_state = "provider_unavailable"
+ var unconfigured_state = "provider_unconfigured"
+ if bridge_status.kind == "assist_bridge":
+ unavailable_state = "bridge_unavailable"
+ unconfigured_state = "bridge_unconfigured"
+
if bridge_status.state == "disabled_by_runtime_config":
return "disabled_by_runtime_config"
if bridge_status.state == "unconfigured":
- return "bridge_unconfigured"
+ return unconfigured_state
if bridge_status.state == "unavailable":
- return "bridge_unavailable"
+ return unavailable_state
if bridge_status.reachable:
return "enabled"
- return "bridge_unavailable"
+ return unavailable_state
def assisted_backend_available_for_capability(
@@ -331,7 +348,8 @@ def serialize_assist_bridge_status_value(
value.set("kind", Value(String(bridge_status.kind)))
value.set("contract_version", Value(bridge_status.contract_version))
value.set("transport", Value(String(bridge_status.transport)))
- value.set("endpoint", Value(String(bridge_status.endpoint)))
+ if bridge_status.endpoint != "":
+ value.set("endpoint", Value(String(bridge_status.endpoint)))
value.set("backend_kind", Value(String(bridge_status.backend_kind)))
if bridge_status.provider != "":
value.set("provider", Value(String(bridge_status.provider)))
diff --git a/src/hyf_assist/contract.mojo b/src/hyf_assist/contract.mojo
@@ -11,6 +11,10 @@ def assist_bridge_runtime_id() -> String:
return "hyf_assistd"
+def provider_runtime_id() -> String:
+ return "hyf_provider_runtime"
+
+
def assist_bridge_supported_business_capabilities() -> List[String]:
var capabilities = List[String]()
capabilities.append("query_rewrite")
diff --git a/src/hyf_core/capabilities/query_rewrite.mojo b/src/hyf_core/capabilities/query_rewrite.mojo
@@ -84,7 +84,9 @@ def _base_source_refs(
def _build_assisted_meta(
- context: RequestContext, result: AssistQueryRewriteResult
+ context: RequestContext,
+ result: AssistQueryRewriteResult,
+ backend: String,
) -> CoreResponseMeta:
var provenance: Optional[ExecutionProvenance] = None
if context.return_provenance:
@@ -98,7 +100,7 @@ def _build_assisted_meta(
return CoreResponseMeta(
execution_mode="assisted",
- backend="assist_bridge",
+ backend=String(backend),
provider=Optional[String](String(result.provider)),
route=Optional[String](String(result.route)),
model=Optional[String](String(result.model)),
@@ -110,7 +112,10 @@ def _build_assisted_meta(
def _build_deterministic_fallback_meta(
- context: RequestContext, analysis: QueryAnalysis, reason: String
+ context: RequestContext,
+ analysis: QueryAnalysis,
+ fallback_kind: String,
+ reason: String,
) -> CoreResponseMeta:
var provenance: Optional[ExecutionProvenance] = None
if context.return_provenance:
@@ -119,7 +124,7 @@ def _build_deterministic_fallback_meta(
signal_tags=query_signal_tags(analysis),
source_refs=_base_source_refs(context, "query_rewrite"),
fallback=ProvenanceFallback(
- fallback_kind="assist_bridge", reason=String(reason)
+ fallback_kind=String(fallback_kind), reason=String(reason)
),
evidence_set_id=None,
)
@@ -173,6 +178,7 @@ def execute_query_rewrite_with_runtime_config(
meta=_build_deterministic_fallback_meta(
context,
fallback_analysis,
+ "provider_runtime",
"disabled_by_runtime_config",
),
)
@@ -182,7 +188,10 @@ def execute_query_rewrite_with_runtime_config(
return successful_capability(
_build_output(fallback_analysis),
meta=_build_deterministic_fallback_meta(
- context, fallback_analysis, "unconfigured"
+ context,
+ fallback_analysis,
+ "provider_runtime",
+ "unconfigured",
),
)
@@ -196,7 +205,11 @@ def execute_query_rewrite_with_runtime_config(
)
return successful_capability(
_build_output(assisted_result.analysis),
- meta=_build_assisted_meta(context, assisted_result),
+ meta=_build_assisted_meta(
+ context,
+ assisted_result,
+ "assist_bridge",
+ ),
)
except e:
var fallback_analysis = analyze_query_text(
@@ -207,6 +220,7 @@ def execute_query_rewrite_with_runtime_config(
meta=_build_deterministic_fallback_meta(
context,
fallback_analysis,
+ "assist_bridge",
"bridge_execution_failed",
),
)
@@ -215,7 +229,10 @@ def execute_query_rewrite_with_runtime_config(
return successful_capability(
_build_output(fallback_analysis),
meta=_build_deterministic_fallback_meta(
- context, fallback_analysis, bridge_status.state
+ context,
+ fallback_analysis,
+ "assist_bridge",
+ bridge_status.state,
),
)
@@ -230,7 +247,11 @@ def execute_query_rewrite_with_runtime_config(
)
return successful_capability(
_build_output(assisted_result.analysis),
- meta=_build_assisted_meta(context, assisted_result),
+ meta=_build_assisted_meta(
+ context,
+ assisted_result,
+ "provider_runtime",
+ ),
)
except e:
var fallback_analysis = analyze_query_text(
@@ -241,7 +262,8 @@ def execute_query_rewrite_with_runtime_config(
meta=_build_deterministic_fallback_meta(
context,
fallback_analysis,
- "bridge_execution_failed",
+ "provider_runtime",
+ "provider_execution_failed",
),
)
except e:
@@ -251,7 +273,10 @@ def execute_query_rewrite_with_runtime_config(
return successful_capability(
_build_output(fallback_analysis),
meta=_build_deterministic_fallback_meta(
- context, fallback_analysis, "unavailable"
+ context,
+ fallback_analysis,
+ "provider_runtime",
+ "unavailable",
),
)
diff --git a/src/hyf_provider/config.mojo b/src/hyf_provider/config.mojo
@@ -13,7 +13,7 @@ struct MaxLocalProviderConfig(Defaultable, Copyable, Movable):
self.base_url = "http://127.0.0.1:8000/v1"
self.health_url = "http://127.0.0.1:8000/health"
self.model = "max-local-query-rewrite"
- self.route = "assist_bridge.query_rewrite.max_local"
+ self.route = "provider_runtime.query_rewrite.max_local"
self.request_timeout_ms = 15_000
diff --git a/src/hyf_stdio/control/capabilities.mojo b/src/hyf_stdio/control/capabilities.mojo
@@ -89,12 +89,15 @@ def build_capabilities_output_with_runtime_context(
capabilities.append(value)
output.set("business_capabilities", capabilities)
- var assisted_backend_capabilities = loads("[]")
- assisted_backend_capabilities.append(
+ var assisted_runtime_capabilities = loads("[]")
+ assisted_runtime_capabilities.append(
serialize_assist_bridge_status_value(assist_bridge)
)
output.set(
- "assisted_backend_capabilities", assisted_backend_capabilities
+ "assisted_runtime_capabilities", assisted_runtime_capabilities.copy()
+ )
+ output.set(
+ "assisted_backend_capabilities", assisted_runtime_capabilities
)
output.set(
"request_context_contract",
diff --git a/src/hyf_stdio/control/status.mojo b/src/hyf_stdio/control/status.mojo
@@ -89,13 +89,19 @@ def build_status_output_with_runtime_context(
var execution_mode_request_behavior = loads("{}")
execution_mode_request_behavior.set("deterministic", Value("execute"))
- var assisted_request_behavior = "bridge_unavailable"
+ var assisted_request_behavior = "provider_unavailable"
+ if assist_bridge.kind == "assist_bridge":
+ assisted_request_behavior = "bridge_unavailable"
if assist_bridge.state == "ready":
assisted_request_behavior = "execute"
elif assist_bridge.state == "disabled_by_runtime_config":
assisted_request_behavior = "disabled_by_runtime_config"
elif assist_bridge.state == "unconfigured":
- assisted_request_behavior = "bridge_unconfigured"
+ assisted_request_behavior = (
+ "bridge_unconfigured"
+ if assist_bridge.kind == "assist_bridge"
+ else "provider_unconfigured"
+ )
execution_mode_request_behavior.set(
"assisted", Value(String(assisted_request_behavior))
)
@@ -116,6 +122,10 @@ def build_status_output_with_runtime_context(
backends.set("assisted_backend", Value(String(assist_bridge.state)))
output.set("backend_reachability", backends)
output.set(
+ "assisted_runtime",
+ serialize_assist_bridge_status_value(assist_bridge),
+ )
+ output.set(
"assist_bridge", serialize_assist_bridge_status_value(assist_bridge)
)
diff --git a/tests/fixtures/v1/scenarios/capabilities_ok.json b/tests/fixtures/v1/scenarios/capabilities_ok.json
@@ -34,11 +34,12 @@
"output.business_capabilities.6.id": "explain_result",
"output.business_capabilities.6.assisted_execution": "deferred",
"output.business_capabilities.6.implementation_status": "implemented",
- "output.assisted_backend_capabilities.0.id": "hyf_assistd",
- "output.assisted_backend_capabilities.0.kind": "assist_bridge",
- "output.assisted_backend_capabilities.0.contract_version": 1,
- "output.assisted_backend_capabilities.0.backend_kind": "fake",
- "output.assisted_backend_capabilities.0.state": "disabled_by_runtime_config",
+ "output.assisted_runtime_capabilities.0.id": "hyf_provider_runtime",
+ "output.assisted_runtime_capabilities.0.kind": "provider_runtime",
+ "output.assisted_runtime_capabilities.0.contract_version": 1,
+ "output.assisted_runtime_capabilities.0.transport": "in_process",
+ "output.assisted_runtime_capabilities.0.backend_kind": "max_local",
+ "output.assisted_runtime_capabilities.0.state": "disabled_by_runtime_config",
"output.request_context_contract.accepted_features": [
"consumer",
"execution_mode_preference",
@@ -59,7 +60,7 @@
"output.request_context_contract.unsupported_field_behavior": "reject"
},
"contains_all": {
- "output.assisted_backend_capabilities.0.supported_business_capabilities": [
+ "output.assisted_runtime_capabilities.0.supported_business_capabilities": [
"query_rewrite"
]
},
diff --git a/tests/fixtures/v1/scenarios/status_ok.json b/tests/fixtures/v1/scenarios/status_ok.json
@@ -35,11 +35,12 @@
"output.execution_mode_request_behavior.assisted": "disabled_by_runtime_config",
"output.backend_reachability.deterministic_backend": "available",
"output.backend_reachability.assisted_backend": "disabled_by_runtime_config",
- "output.assist_bridge.id": "hyf_assistd",
- "output.assist_bridge.kind": "assist_bridge",
- "output.assist_bridge.contract_version": 1,
- "output.assist_bridge.backend_kind": "fake",
- "output.assist_bridge.state": "disabled_by_runtime_config",
+ "output.assisted_runtime.id": "hyf_provider_runtime",
+ "output.assisted_runtime.kind": "provider_runtime",
+ "output.assisted_runtime.contract_version": 1,
+ "output.assisted_runtime.transport": "in_process",
+ "output.assisted_runtime.backend_kind": "max_local",
+ "output.assisted_runtime.state": "disabled_by_runtime_config",
"output.counts.canonical_business_capabilities": 8,
"output.counts.deterministic_registered_business_capabilities": 3,
"output.counts.deterministic_implemented_business_capabilities": 3,
@@ -70,7 +71,7 @@
"semantic_rank",
"explain_result"
],
- "output.assist_bridge.supported_business_capabilities": [
+ "output.assisted_runtime.supported_business_capabilities": [
"query_rewrite"
],
"output.disabled_capabilities": [
diff --git a/tests/test_hyf.mojo b/tests/test_hyf.mojo
@@ -482,7 +482,16 @@ def test_capabilities_output_reflects_registry_truth_for_all_business_capabiliti
assert_equal(
output["assisted_backend_capabilities"][0]["id"].string_value(),
- "hyf_assistd",
+ "hyf_provider_runtime",
+ )
+ assert_equal(
+ output["assisted_backend_capabilities"][0]["kind"].string_value(),
+ "provider_runtime",
+ )
+ assert_equal(
+ output["assisted_backend_capabilities"][0]["transport"]
+ .string_value(),
+ "in_process",
)
assert_equal(
output["assisted_backend_capabilities"][0]["state"].string_value(),
@@ -491,7 +500,7 @@ def test_capabilities_output_reflects_registry_truth_for_all_business_capabiliti
assert_equal(
output["assisted_backend_capabilities"][0]["backend_kind"]
.string_value(),
- "fake",
+ "max_local",
)
@@ -909,7 +918,7 @@ def test_max_local_provider_status_probes_health_without_sidecar() raises:
assert_equal(status.backend_kind, "max_local")
assert_equal(status.provider, "max_local")
assert_equal(
- status.route, "assist_bridge.query_rewrite.max_local"
+ status.route, "provider_runtime.query_rewrite.max_local"
)
assert_equal(status.model, "max-local-query-rewrite")
assert_equal(status.reachable, True)
@@ -1002,7 +1011,7 @@ def test_max_local_query_rewrite_request_is_mojo_owned() raises:
)
assert_equal(result.provider, "max_local")
assert_equal(
- result.route, "assist_bridge.query_rewrite.max_local"
+ result.route, "provider_runtime.query_rewrite.max_local"
)
assert_equal(result.model, "max-local-query-rewrite")
assert_equal(result.schema_version, 1)
diff --git a/tests/test_stdio_contract.mojo b/tests/test_stdio_contract.mojo
@@ -131,7 +131,7 @@ def test_status_reports_repo_local_runtime_truth() raises:
False,
)
assert_equal(
- response["output"]["assist_bridge"]["state"]
+ response["output"]["assisted_runtime"]["state"]
.string_value(),
"disabled_by_runtime_config",
)
@@ -291,14 +291,29 @@ def test_status_loads_valid_runtime_config_truthfully() raises:
response["output"]["execution_mode_request_behavior"][
"assisted"
].string_value(),
- "bridge_unavailable",
+ "provider_unavailable",
)
assert_equal(
- response["output"]["assist_bridge"]["state"]
+ response["output"]["assisted_runtime"]["state"]
.string_value(),
"unavailable",
)
assert_equal(
+ response["output"]["assisted_runtime"]["id"]
+ .string_value(),
+ "hyf_provider_runtime",
+ )
+ assert_equal(
+ response["output"]["assisted_runtime"]["kind"]
+ .string_value(),
+ "provider_runtime",
+ )
+ assert_equal(
+ response["output"]["assisted_runtime"]["transport"]
+ .string_value(),
+ "in_process",
+ )
+ assert_equal(
response["output"]["backend_reachability"][
"assisted_backend"
].string_value(),
@@ -401,7 +416,7 @@ def test_status_reports_invalid_runtime_config_without_crashing() raises:
"disabled_by_runtime_config",
)
assert_equal(
- response["output"]["assist_bridge"]["state"]
+ response["output"]["assisted_runtime"]["state"]
.string_value(),
"disabled_by_runtime_config",
)
@@ -443,7 +458,7 @@ def test_status_reports_invalid_runtime_config_without_crashing() raises:
)
-def test_capabilities_reports_configured_fake_bridge_truthfully() raises:
+def test_capabilities_reports_configured_provider_truthfully() raises:
with TemporaryDirectory() as temp_dir:
var startup_config_path = Path(temp_dir) / "explicit-hyf-config.toml"
startup_config_path.write_text(
@@ -465,25 +480,31 @@ def test_capabilities_reports_configured_fake_bridge_truthfully() raises:
response["output"]["business_capabilities"][0][
"assisted_execution"
].string_value(),
- "bridge_unavailable",
+ "provider_unavailable",
)
assert_equal(
- response["output"]["assisted_backend_capabilities"][0][
+ response["output"]["assisted_runtime_capabilities"][0][
"id"
].string_value(),
- "hyf_assistd",
+ "hyf_provider_runtime",
+ )
+ assert_equal(
+ response["output"]["assisted_runtime_capabilities"][0][
+ "kind"
+ ].string_value(),
+ "provider_runtime",
)
assert_equal(
- response["output"]["assisted_backend_capabilities"][0][
+ response["output"]["assisted_runtime_capabilities"][0][
"state"
].string_value(),
"unavailable",
)
assert_equal(
- response["output"]["assisted_backend_capabilities"][0][
+ response["output"]["assisted_runtime_capabilities"][0][
"backend_kind"
].string_value(),
- "fake",
+ "max_local",
)
@@ -518,13 +539,95 @@ def test_capabilities_reports_ready_fake_bridge_truthfully() raises:
True,
)
assert_equal(
- response["output"]["assisted_backend_capabilities"][0][
+ response["output"]["assisted_runtime_capabilities"][0][
"state"
].string_value(),
"ready",
)
+def test_capabilities_reports_ready_pure_mojo_provider_truthfully() raises:
+ with TemporaryDirectory() as temp_dir:
+ var startup_config_path = Path(temp_dir) / "explicit-hyf-config.toml"
+ startup_config_path.write_text(
+ '[service]\ntransport = "stdio"\n\n'
+ '[runtime]\ndefault_execution_mode = "deterministic"\nallow_assisted = true\n\n'
+ '[assist]\nbridge_enabled = true\ntransport = "stdio"\nendpoint = "hyf-assistd://local"\n'
+ )
+ var health_port = reserve_loopback_port()
+ var health_stub = spawn_max_local_stub(health_port, "health_ok")
+ with ScopedEnvVar(HYF_PATHS_PROFILE_ENV, "repo_local"):
+ with ScopedEnvVar(HYF_PATHS_REPO_LOCAL_ROOT_ENV, temp_dir):
+ with ScopedEnvVar(
+ "HYF_MAX_LOCAL_HEALTH_URL",
+ "http://127.0.0.1:" + String(health_port) + "/health",
+ ):
+ var response = run_stdio_entrypoint(
+ "src/main.mojo",
+ load_scenario_request_json("scenarios/capabilities_ok.json"),
+ "--config",
+ startup_config_path.__fspath__(),
+ )
+
+ assert_true(response["ok"].bool_value())
+ assert_equal(
+ response["output"]["business_capabilities"][0][
+ "assisted_execution"
+ ].string_value(),
+ "enabled",
+ )
+ assert_equal(
+ response["output"]["business_capabilities"][0][
+ "assisted_backend_available"
+ ].bool_value(),
+ True,
+ )
+ assert_equal(
+ response["output"]["assisted_runtime_capabilities"][0][
+ "id"
+ ].string_value(),
+ "hyf_provider_runtime",
+ )
+ assert_equal(
+ response["output"]["assisted_runtime_capabilities"][0][
+ "kind"
+ ].string_value(),
+ "provider_runtime",
+ )
+ assert_equal(
+ response["output"]["assisted_runtime_capabilities"][0][
+ "transport"
+ ].string_value(),
+ "in_process",
+ )
+ assert_equal(
+ response["output"]["assisted_runtime_capabilities"][0][
+ "state"
+ ].string_value(),
+ "ready",
+ )
+ assert_equal(
+ response["output"]["assisted_runtime_capabilities"][0][
+ "backend_kind"
+ ].string_value(),
+ "max_local",
+ )
+ assert_equal(
+ response["output"]["assisted_runtime_capabilities"][0][
+ "provider"
+ ].string_value(),
+ "max_local",
+ )
+ assert_equal(
+ response["output"]["assisted_runtime_capabilities"][0][
+ "route"
+ ].string_value(),
+ "provider_runtime.query_rewrite.max_local",
+ )
+
+ health_stub.wait()
+
+
def test_status_reports_ready_fake_bridge_truthfully() raises:
with TemporaryDirectory() as temp_dir:
var startup_config_path = Path(temp_dir) / "explicit-hyf-config.toml"
@@ -556,7 +659,7 @@ def test_status_reports_ready_fake_bridge_truthfully() raises:
"execute",
)
assert_equal(
- response["output"]["assist_bridge"]["state"]
+ response["output"]["assisted_runtime"]["state"]
.string_value(),
"ready",
)
@@ -568,6 +671,87 @@ def test_status_reports_ready_fake_bridge_truthfully() raises:
)
+def test_status_reports_ready_pure_mojo_provider_truthfully() raises:
+ with TemporaryDirectory() as temp_dir:
+ var startup_config_path = Path(temp_dir) / "explicit-hyf-config.toml"
+ startup_config_path.write_text(
+ '[service]\ntransport = "stdio"\n\n'
+ '[runtime]\ndefault_execution_mode = "deterministic"\nallow_assisted = true\n\n'
+ '[assist]\nbridge_enabled = true\ntransport = "stdio"\nendpoint = "hyf-assistd://local"\n'
+ )
+ var health_port = reserve_loopback_port()
+ var health_stub = spawn_max_local_stub(health_port, "health_ok")
+ with ScopedEnvVar(HYF_PATHS_PROFILE_ENV, "repo_local"):
+ with ScopedEnvVar(HYF_PATHS_REPO_LOCAL_ROOT_ENV, temp_dir):
+ with ScopedEnvVar(
+ "HYF_MAX_LOCAL_HEALTH_URL",
+ "http://127.0.0.1:" + String(health_port) + "/health",
+ ):
+ var response = run_stdio_entrypoint(
+ "src/main.mojo",
+ load_scenario_request_json("scenarios/status_ok.json"),
+ "--config",
+ startup_config_path.__fspath__(),
+ )
+
+ assert_true(response["ok"].bool_value())
+ assert_equal(
+ response["output"]["build_identity"][
+ "assisted_execution_available"
+ ].bool_value(),
+ True,
+ )
+ assert_equal(
+ response["output"]["execution_mode_request_behavior"][
+ "assisted"
+ ].string_value(),
+ "execute",
+ )
+ assert_equal(
+ response["output"]["assisted_runtime"]["id"]
+ .string_value(),
+ "hyf_provider_runtime",
+ )
+ assert_equal(
+ response["output"]["assisted_runtime"]["kind"]
+ .string_value(),
+ "provider_runtime",
+ )
+ assert_equal(
+ response["output"]["assisted_runtime"]["transport"]
+ .string_value(),
+ "in_process",
+ )
+ assert_equal(
+ response["output"]["assisted_runtime"]["state"]
+ .string_value(),
+ "ready",
+ )
+ assert_equal(
+ response["output"]["assisted_runtime"]["backend_kind"]
+ .string_value(),
+ "max_local",
+ )
+ assert_equal(
+ response["output"]["assisted_runtime"]["provider"]
+ .string_value(),
+ "max_local",
+ )
+ assert_equal(
+ response["output"]["assisted_runtime"]["route"]
+ .string_value(),
+ "provider_runtime.query_rewrite.max_local",
+ )
+ assert_equal(
+ response["output"]["backend_reachability"][
+ "assisted_backend"
+ ].string_value(),
+ "ready",
+ )
+
+ health_stub.wait()
+
+
def test_query_rewrite_uses_fake_assist_bridge_when_requested() raises:
with TemporaryDirectory() as temp_dir:
var startup_config_path = Path(temp_dir) / "explicit-hyf-config.toml"
@@ -667,7 +851,7 @@ def test_query_rewrite_uses_pure_mojo_provider_when_requested() raises:
)
assert_equal(
response["meta"]["backend"].string_value(),
- "assist_bridge",
+ "provider_runtime",
)
assert_equal(
response["meta"]["provider"].string_value(),
@@ -675,7 +859,7 @@ def test_query_rewrite_uses_pure_mojo_provider_when_requested() raises:
)
assert_equal(
response["meta"]["route"].string_value(),
- "assist_bridge.query_rewrite.max_local",
+ "provider_runtime.query_rewrite.max_local",
)
assert_equal(
response["meta"]["model"].string_value(),
@@ -752,7 +936,7 @@ def test_query_rewrite_falls_back_deterministically_when_bridge_is_unavailable()
response["meta"]["provenance"]["fallback"][
"fallback_kind"
].string_value(),
- "assist_bridge",
+ "provider_runtime",
)
assert_equal(
response["meta"]["provenance"]["fallback"]["reason"]
