tabby/server/app.py

25 lines
698 B
Python

import uvicorn
from fastapi import FastAPI, Response
from fastapi.responses import JSONResponse
from models import CompletionsRequest, CompletionsResponse
app = FastAPI(
title="TabbyServer",
description="TabbyServer is the backend for tabby, serving code completion requests from code editor / IDE.",
docs_url="/",
)
@app.post("/v1/completions")
async def completions(data: CompletionsRequest) -> CompletionsResponse:
return CompletionsResponse()
@app.post("/v1/completions/{id}/choices/{index}/select")
async def select(id: str, index: int):
return JSONResponse(content=dict(status="ok"))
if __name__ == "__main__":
uvicorn.run("app:app", host="0.0.0.0", port=5000)