Spaces:
Running
Running
db api online
Browse files- model_service_api.py +11 -1
model_service_api.py
CHANGED
|
@@ -271,14 +271,24 @@ async def get_free_models(
|
|
| 271 |
logger.error(f"Error in get_free_models: {str(e)}")
|
| 272 |
raise HTTPException(status_code=500, detail=str(e))
|
| 273 |
|
| 274 |
-
@app.get("/api/free-models/names", response_model=Dict)
|
| 275 |
async def get_free_model_names(
|
|
|
|
| 276 |
category: Optional[str] = Query(None, description="Filter by category")
|
| 277 |
):
|
| 278 |
"""Get just the names and IDs of free models"""
|
| 279 |
try:
|
| 280 |
free_models = await get_free_models_data(category=category)
|
| 281 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 282 |
model_names = [
|
| 283 |
{
|
| 284 |
"id": model["id"],
|
|
|
|
| 271 |
logger.error(f"Error in get_free_models: {str(e)}")
|
| 272 |
raise HTTPException(status_code=500, detail=str(e))
|
| 273 |
|
| 274 |
+
@app.get("/api/free-models/names/{name}", response_model=Dict)
|
| 275 |
async def get_free_model_names(
|
| 276 |
+
name: bool = False,
|
| 277 |
category: Optional[str] = Query(None, description="Filter by category")
|
| 278 |
):
|
| 279 |
"""Get just the names and IDs of free models"""
|
| 280 |
try:
|
| 281 |
free_models = await get_free_models_data(category=category)
|
| 282 |
|
| 283 |
+
if name:
|
| 284 |
+
return {
|
| 285 |
+
"success": True,
|
| 286 |
+
"message": f"Retrieved {len(free_models)} free model names",
|
| 287 |
+
"count": len(free_models),
|
| 288 |
+
"models": [{"id": model["id"]} for model in free_models],
|
| 289 |
+
"filtered_at": datetime.now().isoformat()
|
| 290 |
+
}
|
| 291 |
+
|
| 292 |
model_names = [
|
| 293 |
{
|
| 294 |
"id": model["id"],
|