wehub-resource-sync
17 lines
646 B
Python
17 lines
646 B
Python
from typing import Annotated, List
|
|
from fastapi import APIRouter, Body, HTTPException
|
|
|
|
from utils.available_models import ModelAvailabilityError, list_available_google_models
|
|
|
|
GOOGLE_ROUTER = APIRouter(prefix="/google", tags=["Google"])
|
|
|
|
|
|
@GOOGLE_ROUTER.post("/models/available", response_model=List[str])
|
|
async def get_available_models(api_key: Annotated[str, Body(embed=True)]):
|
|
try:
|
|
return await list_available_google_models(api_key)
|
|
except ModelAvailabilityError as e:
|
|
raise HTTPException(status_code=e.status_code, detail=str(e))
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|