[eric] 9router: never orphan-sweep custom nodes off an empty provider list (corrupt-settings boot mass-reap guard)

This commit is contained in:
ciregenz
2026-07-08 17:33:15 -07:00
parent f669fc16df
commit e6c781a380
2 changed files with 121 additions and 0 deletions
+6
View File
@@ -252,6 +252,12 @@ async def sync_custom_providers(providers: list) -> None:
logger.warning(f"9Router custom connection {prefix} sync failed: {e}")
# Drop managed nodes no longer in settings; DELETE cascades to connections.
# NEVER sweep on an EMPTY list: a corrupt/defaulted settings load at boot would hand us [] and mass-reap every custom connection (the accidental-disconnect class). Cost of the guard: deleting your LAST custom provider leaves one stale node in 9Router, invisible to the picker (models come from settings) and overwritten on the next add.
if not seen_prefixes:
remaining = [p for p in managed_by_prefix if p != NINE_ROUTER_OPENAI_KEYED_PREFIX]
if remaining:
logger.info(f"9Router: skipping orphan sweep (empty provider list, {len(remaining)} managed nodes kept)")
return
for prefix, node in managed_by_prefix.items():
# cp-openai wears the same managed suffix but belongs to sync_openai_compat_node; reaping it here killed every gpt-*-api request with "No credentials".
if prefix in seen_prefixes or prefix == NINE_ROUTER_OPENAI_KEYED_PREFIX:
+115
View File
@@ -0,0 +1,115 @@
"""Guards against the accidental-disconnect class in the 9Router mirror sync: the orphan sweep
must never mass-reap managed nodes off an EMPTY provider list (a corrupt/defaulted settings load
at boot hands it []), while a NON-empty list still reaps exactly the removed provider and never
touches cp-openai (the Jul-3 'No credentials' regression)."""
import asyncio
import json
from typing import Dict, List, Optional
import backend.apps.nine_router.sync_custom as sc
class FakeResponse:
def __init__(self, status_code: int, payload: Optional[Dict] = None):
self.status_code = status_code
self.text = json.dumps(payload or {})
self.p_payload = payload or {}
def json(self) -> Dict:
return self.p_payload
class FakeAsyncClient:
"""Records every HTTP call; serves the provider-nodes list from the harness."""
def __init__(self, harness: "Harness", **kwargs):
self.harness = harness
async def __aenter__(self) -> "FakeAsyncClient":
return self
async def __aexit__(self, *exc) -> None:
return None
async def get(self, url: str, **kw) -> FakeResponse:
self.harness.calls.append(("GET", url))
if url.endswith("/provider-nodes"):
return FakeResponse(200, {"nodes": self.harness.nodes})
if url.endswith("/providers"):
return FakeResponse(200, {"connections": []})
return FakeResponse(404)
async def post(self, url: str, json: Optional[Dict] = None, **kw) -> FakeResponse:
self.harness.calls.append(("POST", url))
if url.endswith("/provider-nodes"):
return FakeResponse(200, {"node": {"id": "new-node-id"}})
return FakeResponse(200, {})
async def put(self, url: str, json: Optional[Dict] = None, **kw) -> FakeResponse:
self.harness.calls.append(("PUT", url))
return FakeResponse(200, {})
async def patch(self, url: str, json: Optional[Dict] = None, **kw) -> FakeResponse:
self.harness.calls.append(("PATCH", url))
return FakeResponse(200, {})
async def delete(self, url: str, **kw) -> FakeResponse:
self.harness.calls.append(("DELETE", url))
return FakeResponse(200, {})
class Harness:
def __init__(self, nodes: List[Dict]):
self.nodes = nodes
self.calls: List = []
def deletes(self) -> List[str]:
return [u for m, u in self.calls if m == "DELETE"]
class FakeNr:
def __init__(self, harness: Harness):
self.harness = harness
h = self
class P_Httpx:
def AsyncClient(self, **kwargs) -> FakeAsyncClient:
return FakeAsyncClient(h.harness)
self.httpx = P_Httpx()
def is_running(self) -> bool:
return True
def managed(prefix: str) -> Dict:
return {"id": f"id-{prefix}", "prefix": prefix, "name": f"{prefix}{sc.NINE_ROUTER_CUSTOM_NAME_SUFFIX}"}
def setup(monkeypatch, nodes: List[Dict]) -> Harness:
harness = Harness(nodes)
monkeypatch.setattr(sc, "nr", lambda: FakeNr(harness))
monkeypatch.setattr(sc, "cli_auth_headers", lambda: {})
monkeypatch.setattr(sc, "find_keyed_connection", p_no_connection)
return harness
async def p_no_connection(node_id: str, name: str) -> None:
return None
def test_empty_list_never_sweeps(monkeypatch):
"""Corrupt/defaulted settings at boot pass []; every managed node must survive."""
harness = setup(monkeypatch, [managed("cp-ollama"), managed("cp-together"), managed(sc.NINE_ROUTER_OPENAI_KEYED_PREFIX)])
asyncio.run(sc.sync_custom_providers([]))
assert harness.deletes() == []
def test_nonempty_list_reaps_only_removed(monkeypatch):
"""A real one-provider settings list still reaps the genuinely removed node, keeps the kept one, and never touches cp-openai."""
harness = setup(monkeypatch, [managed("cp-ollama"), managed("cp-together"), managed(sc.NINE_ROUTER_OPENAI_KEYED_PREFIX)])
asyncio.run(sc.sync_custom_providers([{"name": "ollama", "base_url": "http://localhost:11434/v1", "api_key": "k"}]))
deleted_ids = [u.split("/")[-1] for u in harness.deletes()]
assert deleted_ids == ["id-cp-together"]
assert f"id-{sc.NINE_ROUTER_OPENAI_KEYED_PREFIX}" not in deleted_ids