AVB/bedrock-access-gateway
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix api path

Browse Source
This commit is contained in:
Aiden Dai
2024-04-03 20:29:50 +08:00
parent cee335486a
commit 912c19fcff
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/api/routers/embeddings.py
View File

@@ -15,7 +15,7 @@ router = APIRouter(
) )
@router.post("/", response_model=EmbeddingsResponse) @router.post("", response_model=EmbeddingsResponse)
async def embeddings( async def embeddings(
embeddings_request: Annotated[ embeddings_request: Annotated[
EmbeddingsRequest, EmbeddingsRequest,

2
src/api/routers/model.py
View File

@@ -20,7 +20,7 @@ async def validate_model_id(model_id: str):
raise HTTPException(status_code=500, detail="Unsupported Model Id") raise HTTPException(status_code=500, detail="Unsupported Model Id")
@router.get("/", response_model=Models) @router.get("", response_model=Models)
async def list_models(): async def list_models():
model_list = [Model(id=model_id) for model_id in model_list = [Model(id=model_id) for model_id in
(SUPPORTED_BEDROCK_MODELS | SUPPORTED_BEDROCK_EMBEDDING_MODELS).keys()] (SUPPORTED_BEDROCK_MODELS | SUPPORTED_BEDROCK_EMBEDDING_MODELS).keys()]