wehub-resource-sync
23 lines
716 B
Python
23 lines
716 B
Python
from typing import Annotated, List
|
|
from fastapi import APIRouter, Body, HTTPException
|
|
|
|
from utils.available_models import (
|
|
ModelAvailabilityError,
|
|
list_available_openai_compatible_models,
|
|
)
|
|
|
|
OPENAI_ROUTER = APIRouter(prefix="/openai", tags=["OpenAI"])
|
|
|
|
|
|
@OPENAI_ROUTER.post("/models/available", response_model=List[str])
|
|
async def get_available_models(
|
|
url: Annotated[str, Body()],
|
|
api_key: Annotated[str, Body()],
|
|
):
|
|
try:
|
|
return await list_available_openai_compatible_models(url, 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))
|