feat: add secure agent and MCP extensions

backend/apps/modelhub/tests/test_modelhub_api.py

@@ -89,6 +89,45 @@ def test_model_test_runs_through_recorded_invocation(client, db, monkeypatch):
     assert response.json()["output_tokens"] == 4
 
 
+def test_embedding_deployment_can_be_configured_but_uses_knowledge_search_for_test(client, db):
+    login_admin(client)
+    provider_id = create_provider(client).json()["id"]
+    deployment = client.post(
+        "/api/modelhub/deployments",
+        data={
+            "provider_id": provider_id,
+            "display_name": "知识向量模型",
+            "model_name": "embedding-model",
+            "capability": "embedding",
+        },
+        content_type="application/json",
+    )
+
+    assert deployment.status_code == 200
+    assert deployment.json()["capability"] == "embedding"
+
+    response = client.post(
+        f"/api/modelhub/deployments/{deployment.json()['id']}/test",
+        data={"prompt": "测试连接"},
+        content_type="application/json",
+    )
+    assert response.status_code == 400
+    assert "知识检索" in response.json()["detail"]
+
+    rerank = client.post(
+        "/api/modelhub/deployments",
+        data={
+            "provider_id": provider_id,
+            "display_name": "知识重排模型",
+            "model_name": "rerank-model",
+            "capability": "rerank",
+        },
+        content_type="application/json",
+    )
+    assert rerank.status_code == 200
+    assert rerank.json()["capability"] == "rerank"
+
+
 def test_non_ai_admin_cannot_manage_modelhub(client, db):
     user = get_user_model().objects.create_user(username="requester", password="password")
     client.force_login(user)