Initial commit

src/api/routers/model.py

@@ -0,0 +1,41 @@
+from typing import Annotated
+
+from fastapi import APIRouter, Depends, HTTPException, Path
+
+from api.auth import api_key_auth
+from api.models import SUPPORTED_BEDROCK_MODELS
+from api.schema import Models, Model
+
+router = APIRouter()
+
+router = APIRouter(
+    prefix="/models",
+    tags=["items"],
+    dependencies=[Depends(api_key_auth)],
+    # responses={404: {"description": "Not found"}},
+)
+
+
+async def validate_model_id(model_id: str):
+    if model_id not in SUPPORTED_BEDROCK_MODELS.keys():
+        raise HTTPException(status_code=400, detail="Unsupported Model Id")
+
+
+@router.get("/", response_model=Models)
+async def list_models():
+    model_list = [Model(id=model_id) for model_id in SUPPORTED_BEDROCK_MODELS.keys()]
+    return Models(data=model_list)
+
+
+@router.get(
+    "/{model_id}",
+    response_model=Model,
+)
+async def get_model(
+    model_id: Annotated[
+        str,
+        Path(description="Model ID", example="anthropic.claude-3-sonnet-20240229-v1:0"),
+    ]
+):
+    await validate_model_id(model_id)
+    return Model(id=model_id)