debug model addition to registry

app/routers/models.py

@@ -9,22 +9,23 @@ from ..core.model_registry import ModelSpec, get_model_spec, list_models
 router = APIRouter(prefix="/v1", tags=["models"])
 
 
-def _serialize_model(spec: ModelSpec) -> dict:
+def _serialize_model(spec: ModelSpec, include_metadata: bool = False) -> dict:
     payload = {
         "id": spec.name,
         "object": "model",
         "owned_by": "owner",
         "permission": [],
     }
-    metadata = spec.metadata.to_dict() if spec.metadata else {"description": "No additional details provided."}
-    metadata.setdefault("huggingface_repo", spec.hf_repo)
-    if spec.max_context_tokens is not None:
-        metadata.setdefault("max_context_tokens", spec.max_context_tokens)
-    if spec.dtype:
-        metadata.setdefault("dtype", spec.dtype)
-    if spec.device:
-        metadata.setdefault("default_device", spec.device)
-    payload["metadata"] = metadata
+    if include_metadata:
+        metadata = spec.metadata.to_dict() if spec.metadata else {"description": "No additional details provided."}
+        metadata.setdefault("huggingface_repo", spec.hf_repo)
+        if spec.max_context_tokens is not None:
+            metadata.setdefault("max_context_tokens", spec.max_context_tokens)
+        if spec.dtype:
+            metadata.setdefault("dtype", spec.dtype)
+        if spec.device:
+            metadata.setdefault("default_device", spec.device)
+        payload["metadata"] = metadata
     return payload
 
 
@@ -40,4 +41,4 @@ def retrieve_model(model_id: str) -> dict:
         spec = get_model_spec(model_id)
     except KeyError:
         raise model_not_found(model_id)
-    return _serialize_model(spec)
+    return _serialize_model(spec, include_metadata=True)

tests/test_openai_compat.py

@@ -260,6 +260,6 @@ def test_model_detail_serialization(monkeypatch: pytest.MonkeyPatch) -> None:
     monkeypatch.setattr(models, "get_model_spec", lambda _: spec)
 
     listing = models.list_available_models()
-    assert listing["data"][0]["metadata"]["description"] == "Example"
+    assert "metadata" not in listing["data"][0]
     detail = models.retrieve_model("example")
     assert detail["metadata"]["huggingface_repo"] == "example/repo"