Soumik555 commited on
Commit
96cbae9
·
1 Parent(s): 672fb1a

db api online

Browse files
Files changed (1) hide show
  1. 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"],