theroyallab/tabbyAPI
1
0
Fork 0
mirror of https://github.com/theroyallab/tabbyAPI.git synced 2026-04-26 09:18:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

ruff: formatting

Browse Source
This commit is contained in:
AlpinDale
2024-07-26 02:53:14 +00:00
parent 765d3593b3
commit 5adfab1cbd
3 changed files with 52 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
endpoints/OAI/router.py
View File

@@ -15,10 +15,7 @@ from endpoints.OAI.types.chat_completion import (
ChatCompletionRequest,
ChatCompletionResponse,
)
from endpoints.OAI.types.embedding import (
EmbeddingsRequest,
EmbeddingsResponse
)
from endpoints.OAI.types.embedding import EmbeddingsRequest, EmbeddingsResponse
from endpoints.OAI.utils.chat_completion import (
format_prompt_with_template,
generate_chat_completion,
@@ -132,19 +129,19 @@ async def chat_completion_request(
)
return response
# Embeddings endpoint
@router.post(
"/v1/embeddings",
dependencies=[Depends(check_api_key), Depends(check_model_container)],
response_model=EmbeddingsResponse
response_model=EmbeddingsResponse,
)
async def handle_embeddings(request: EmbeddingsRequest):
input = request.input
if not input:
raise JSONResponse(status_code=400,
content={"error": "Missing required argument input"})
raise JSONResponse(
status_code=400, content={"error": "Missing required argument input"}
)
model = request.model if request.model else None
response = await OAIembeddings.embeddings(input, request.encoding_format,
model)
response = await OAIembeddings.embeddings(input, request.encoding_format, model)
return JSONResponse(response)