Upload backend/services/v4_complement_client.py with huggingface_hub
Browse files
backend/services/v4_complement_client.py
CHANGED
|
@@ -90,36 +90,44 @@ async def get_models_status() -> Dict[str, Any]:
|
|
| 90 |
|
| 91 |
|
| 92 |
async def get_models_list_summary() -> Dict[str, Any]:
|
| 93 |
-
"""Model catalog from v4 —
|
|
|
|
|
|
|
|
|
|
| 94 |
list_result = await _request(
|
| 95 |
"GET",
|
| 96 |
"/api/models/list",
|
| 97 |
-
params={"limit":
|
| 98 |
cache_key="v4:models_list_summary",
|
| 99 |
)
|
| 100 |
if list_result.get("success"):
|
| 101 |
payload = list_result.get("data") or {}
|
| 102 |
models = payload.get("models") or []
|
|
|
|
|
|
|
|
|
|
| 103 |
return {
|
| 104 |
"success": True,
|
| 105 |
"total": payload.get("total_models") or payload.get("total") or len(models),
|
| 106 |
-
"loaded":
|
| 107 |
-
"hf_mode":
|
|
|
|
| 108 |
"items": models[:12],
|
|
|
|
| 109 |
"source": "v4_complement",
|
| 110 |
}
|
| 111 |
|
| 112 |
-
status = await get_models_status()
|
| 113 |
if not status.get("success"):
|
| 114 |
return status
|
| 115 |
-
|
| 116 |
-
items = payload.get("items") or payload.get("models", {}).get("items", [])
|
| 117 |
return {
|
| 118 |
"success": True,
|
| 119 |
-
"total":
|
| 120 |
-
"loaded":
|
| 121 |
-
"hf_mode":
|
|
|
|
| 122 |
"items": items[:12],
|
|
|
|
| 123 |
"source": "v4_complement",
|
| 124 |
}
|
| 125 |
|
|
|
|
| 90 |
|
| 91 |
|
| 92 |
async def get_models_list_summary() -> Dict[str, Any]:
|
| 93 |
+
"""Model catalog from v4 — status for loaded/hf_mode, list for catalog items."""
|
| 94 |
+
status = await get_models_status()
|
| 95 |
+
status_payload = (status.get("data") or {}) if status.get("success") else {}
|
| 96 |
+
|
| 97 |
list_result = await _request(
|
| 98 |
"GET",
|
| 99 |
"/api/models/list",
|
| 100 |
+
params={"limit": 12},
|
| 101 |
cache_key="v4:models_list_summary",
|
| 102 |
)
|
| 103 |
if list_result.get("success"):
|
| 104 |
payload = list_result.get("data") or {}
|
| 105 |
models = payload.get("models") or []
|
| 106 |
+
model_info = payload.get("model_info") or {}
|
| 107 |
+
loaded_from_list = sum(1 for m in models if m.get("loaded"))
|
| 108 |
+
loaded = status_payload.get("models_loaded", loaded_from_list)
|
| 109 |
return {
|
| 110 |
"success": True,
|
| 111 |
"total": payload.get("total_models") or payload.get("total") or len(models),
|
| 112 |
+
"loaded": loaded,
|
| 113 |
+
"hf_mode": status_payload.get("hf_mode") or model_info.get("hf_mode"),
|
| 114 |
+
"inference_api_mode": status_payload.get("inference_api_mode"),
|
| 115 |
"items": models[:12],
|
| 116 |
+
"loaded_models": status_payload.get("loaded_models") or [],
|
| 117 |
"source": "v4_complement",
|
| 118 |
}
|
| 119 |
|
|
|
|
| 120 |
if not status.get("success"):
|
| 121 |
return status
|
| 122 |
+
items = status_payload.get("items") or status_payload.get("models", {}).get("items", [])
|
|
|
|
| 123 |
return {
|
| 124 |
"success": True,
|
| 125 |
+
"total": status_payload.get("models_total") or len(items),
|
| 126 |
+
"loaded": status_payload.get("models_loaded", 0),
|
| 127 |
+
"hf_mode": status_payload.get("hf_mode"),
|
| 128 |
+
"inference_api_mode": status_payload.get("inference_api_mode"),
|
| 129 |
"items": items[:12],
|
| 130 |
+
"loaded_models": status_payload.get("loaded_models") or [],
|
| 131 |
"source": "v4_complement",
|
| 132 |
}
|
| 133 |
|