[Bugfix] Restore Omni health route ownership (#6723)

Signed-off-by: qujing226 <qujing226@gmail.com>
Signed-off-by: Eugene Zhao <113281545+qujing226@users.noreply.github.com>
This commit is contained in:
Eugene Zhao authored and GitHub committed 2026-09-03 00:54:36 -07:00
1 parent 711daacabf
commit ec22c7ac06
2 files changed
+7 -6

No files matched your search

@@ -351,7 +351,7 @@ def test_router_openapi_paths_cover_http_manifest() -> None:
async def test_api_server_assembly_replaces_upstream_routes_and_mounts_omni_router(monkeypatch) -> None:
"""Lock worker assembly: override, mount, storage, handlers, full census.
Fails if assembly stops replacing upstream chat/batch/models/profiler,
Fails if assembly stops replacing upstream chat/batch/models/health/profiler,
deletes unrelated upstream routes, forgets Omni/profiler mounts, skips
storage start, or drops Omni ``EngineDeadError`` / ``EngineGenerateError``
handlers.
@@ -373,6 +373,10 @@ async def test_api_server_assembly_replaces_upstream_routes_and_mounts_omni_rout
async def upstream_models():
return {"owner": "upstream"}
@app.get("/health")
async def upstream_health():
return {"owner": "upstream"}
@app.post("/start_profile")
async def upstream_start_profile():
return {"owner": "upstream"}
@@ -454,6 +458,7 @@ async def test_api_server_assembly_replaces_upstream_routes_and_mounts_omni_rout
is api_server.create_batch_chat_completion
)
assert _single_http_route(routes, "GET", "/v1/models").endpoint is api_server.show_available_models
assert _single_http_route(routes, "GET", "/health").endpoint is api_server.health
assert _single_http_route(routes, "POST", "/start_profile").endpoint is api_server.start_profile
assert _single_http_route(routes, "POST", "/stop_profile").endpoint is api_server.stop_profile