65 lines
2.5 KiB
Python
65 lines
2.5 KiB
Python
|
"""Owner-scope tests for the remaining _resolve_model call sites.
|
||
|
|
|
||
|
|
Both the teacher-escalation path and the skill-audit teacher resolution map a
|
||
|
|
model spec to an endpoint (and its decrypted api_key). Like /presets/expand,
|
||
|
|
that lookup must be scoped to the calling user, otherwise it can resolve another
|
||
|
|
owner's ModelEndpoint in a multi-user deployment. See #2283.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import asyncio
|
||
|
|
|
||
|
|
import src.teacher_escalation as teacher_escalation
|
||
|
|
import routes.skills_routes as skills_routes
|
||
|
|
|
||
|
|
|
||
|
|
def test_call_teacher_scopes_model_resolution_to_owner(monkeypatch):
|
||
|
|
seen = {}
|
||
|
|
|
||
|
|
def fake_resolve_model(spec, owner=None):
|
||
|
|
seen["spec"] = spec
|
||
|
|
seen["owner"] = owner
|
||
|
|
return ("http://endpoint.local/v1", "teacher-model", {})
|
||
|
|
|
||
|
|
async def fake_llm_call_async(url, model, messages, **kwargs):
|
||
|
|
return "teacher reply"
|
||
|
|
|
||
|
|
monkeypatch.setattr("src.ai_interaction._resolve_model", fake_resolve_model)
|
||
|
|
monkeypatch.setattr("src.ai_interaction._TEACHER_SYSTEM_PROMPT", "sys", raising=False)
|
||
|
|
monkeypatch.setattr("src.llm_core.llm_call_async", fake_llm_call_async)
|
||
|
|
|
||
|
|
result = asyncio.run(
|
||
|
|
teacher_escalation._call_teacher("teacher-model", "prompt", owner="alice")
|
||
|
|
)
|
||
|
|
|
||
|
|
assert result == "teacher reply"
|
||
|
|
assert seen["owner"] == "alice"
|
||
|
|
assert seen["spec"] == "teacher-model"
|
||
|
|
|
||
|
|
|
||
|
|
def test_audit_teacher_resolution_scoped_to_owner(monkeypatch):
|
||
|
|
seen = {}
|
||
|
|
|
||
|
|
def fake_resolve_endpoint(role, owner=None):
|
||
|
|
return ("http://worker.local/v1", "worker-model", {})
|
||
|
|
|
||
|
|
def fake_get_setting(key, default=None):
|
||
|
|
return {"teacher_enabled": True, "teacher_model": "teacher-model"}.get(key, default)
|
||
|
|
|
||
|
|
def fake_resolve_model(spec, owner=None):
|
||
|
|
seen["spec"] = spec
|
||
|
|
seen["owner"] = owner
|
||
|
|
return ("http://endpoint.local/v1", "teacher-model", {})
|
||
|
|
|
||
|
|
monkeypatch.setattr("src.endpoint_resolver.resolve_endpoint", fake_resolve_endpoint)
|
||
|
|
monkeypatch.setattr("src.settings.get_setting", fake_get_setting)
|
||
|
|
monkeypatch.setattr("src.ai_interaction._resolve_model", fake_resolve_model)
|
||
|
|
# list_model_ids is best-effort; force it to no-op so the worker model passes through.
|
||
|
|
monkeypatch.setattr("src.llm_core.list_model_ids", lambda url, headers=None: [])
|
||
|
|
|
||
|
|
url, model, headers, teacher = skills_routes._resolve_audit_models(owner="alice")
|
||
|
|
|
||
|
|
assert (url, model) == ("http://worker.local/v1", "worker-model")
|
||
|
|
assert teacher == ("http://endpoint.local/v1", "teacher-model", {})
|
||
|
|
assert seen["owner"] == "alice"
|
||
|
|
assert seen["spec"] == "teacher-model"
|