Really-amin commited on
Commit
77c93b0
·
verified ·
1 Parent(s): 0b79f88

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 — uses /api/models/list?limit=N for total_models count."""
 
 
 
94
  list_result = await _request(
95
  "GET",
96
  "/api/models/list",
97
- params={"limit": 8},
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": sum(1 for m in models if m.get("loaded")),
107
- "hf_mode": payload.get("hf_mode") or (payload.get("model_info") or {}).get("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
- payload = status.get("data") or {}
116
- items = payload.get("items") or payload.get("models", {}).get("items", [])
117
  return {
118
  "success": True,
119
- "total": payload.get("models_total") or len(items),
120
- "loaded": payload.get("models_loaded", 0),
121
- "hf_mode": payload.get("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