bvandeusen
359bc5a283
- Migration 0069: new installs default to SigLIP 2 (so400m, 512px, 1152-d drop-in) — UPDATE applies ONLY where no image is embedded yet (fresh install), so an existing library is NOT silently invalidated; it switches deliberately via the dropdown → Re-embed → Retrain. Column server_defaults moved to SigLIP 2. - GET /api/ml/embedder-models: server-authoritative supported list (SigLIP 2 512 recommended / 384 faster / SigLIP 1 384 original) so the UI never free-types. - GpuAgentCard: the two name/version text fields → a single model dropdown; Save sets name+version from the picked option (the current model is always selectable even if off-list). - embedder.py DEFAULT_MODEL_NAME unchanged (stays the baked local-dir SigLIP 1) to avoid a local-dir/weights mismatch; SigLIP 2 loads by HF name, cached on the ml-worker's persistent HF_HOME. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01Ttrj5P7upUTueSfoJcxEqa
114 lines
3.9 KiB
Python
114 lines
3.9 KiB
Python
import pytest
|
|
|
|
from backend.app.celery_app import celery
|
|
from backend.app.models import TagKind
|
|
from backend.app.services.tag_service import TagService
|
|
|
|
pytestmark = pytest.mark.integration
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def eager():
|
|
celery.conf.task_always_eager = True
|
|
yield
|
|
celery.conf.task_always_eager = False
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_and_patch_settings(client):
|
|
resp = await client.get("/api/ml/settings")
|
|
assert resp.status_code == 200
|
|
body = await resp.get_json()
|
|
assert body["head_min_positives"] == 8
|
|
# Retired tagger/suggestion-threshold columns are gone from the payload
|
|
# (Camie retirement #1189/#1199).
|
|
assert "suggestion_threshold_general" not in body
|
|
assert "tagger_store_floor" not in body
|
|
assert "tagger_model_version" not in body
|
|
|
|
resp = await client.patch(
|
|
"/api/ml/settings", json={"head_min_positives": 12}
|
|
)
|
|
assert resp.status_code == 200
|
|
assert (await resp.get_json())["head_min_positives"] == 12
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_embedder_model_default_settable_and_empty_rejected(client):
|
|
# #1203: fresh installs default to SigLIP 2 (migration 0069, no embeddings).
|
|
# The model is operator-settable (a swap) and neither field may be blanked.
|
|
body = await (await client.get("/api/ml/settings")).get_json()
|
|
assert body["embedder_model_name"] == "google/siglip2-so400m-patch16-512"
|
|
|
|
ok = await client.patch("/api/ml/settings", json={
|
|
"embedder_model_name": "google/siglip-so400m-patch14-384",
|
|
"embedder_model_version": "siglip-so400m-patch14-384",
|
|
})
|
|
assert ok.status_code == 200
|
|
out = await ok.get_json()
|
|
assert out["embedder_model_name"] == "google/siglip-so400m-patch14-384"
|
|
|
|
bad = await client.patch("/api/ml/settings", json={"embedder_model_name": " "})
|
|
assert bad.status_code == 400
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_embedder_models_list(client):
|
|
# #1203: the dropdown reads the supported-model list from the server.
|
|
body = await (await client.get("/api/ml/embedder-models")).get_json()
|
|
assert any(
|
|
m["name"] == "google/siglip2-so400m-patch16-512" for m in body["models"]
|
|
)
|
|
assert all({"name", "version", "label"} <= set(m) for m in body["models"])
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_video_settings_default_and_patch(client):
|
|
"""#747: video frame-sampling knobs are exposed + patchable."""
|
|
body = await (await client.get("/api/ml/settings")).get_json()
|
|
assert body["video_frame_interval_seconds"] == pytest.approx(4.0)
|
|
assert body["video_max_frames"] == 64
|
|
|
|
resp = await client.patch(
|
|
"/api/ml/settings",
|
|
json={"video_frame_interval_seconds": 5, "video_max_frames": 40},
|
|
)
|
|
assert resp.status_code == 200
|
|
out = await resp.get_json()
|
|
assert out["video_frame_interval_seconds"] == pytest.approx(5.0)
|
|
assert out["video_max_frames"] == 40
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_video_max_frames_below_one_rejected(client):
|
|
resp = await client.patch(
|
|
"/api/ml/settings", json={"video_max_frames": 0},
|
|
)
|
|
assert resp.status_code == 400
|
|
assert "video_max_frames" in (await resp.get_json())["error"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_backfill_trigger(client):
|
|
r1 = await client.post("/api/ml/backfill")
|
|
assert r1.status_code == 202
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_rename_tag(client, db):
|
|
tag = await TagService(db).find_or_create("oldname", TagKind.character)
|
|
await db.commit()
|
|
resp = await client.patch(f"/api/tags/{tag.id}", json={"name": "New Name"})
|
|
assert resp.status_code == 200
|
|
assert (await resp.get_json())["name"] == "New Name"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_rename_collision_409(client, db):
|
|
svc = TagService(db)
|
|
await svc.find_or_create("Taken", TagKind.character)
|
|
other = await svc.find_or_create("Mover", TagKind.character)
|
|
await db.commit()
|
|
resp = await client.patch(f"/api/tags/{other.id}", json={"name": "Taken"})
|
|
assert resp.status_code == 409
|