From dd48810186e3b4ee14e1d3727f37bd470d0711a4 Mon Sep 17 00:00:00 2001 From: Lain Soykaf Date: Sun, 19 May 2024 12:47:08 +0400 Subject: B FastembedAPI: Move to more appropriate folder --- python/fastembed-server.py | 23 ----------------------- 1 file changed, 23 deletions(-) delete mode 100644 python/fastembed-server.py (limited to 'python/fastembed-server.py') diff --git a/python/fastembed-server.py b/python/fastembed-server.py deleted file mode 100644 index dd4a7a9c8..000000000 --- a/python/fastembed-server.py +++ /dev/null @@ -1,23 +0,0 @@ -from fastembed import TextEmbedding -from fastapi import FastAPI -from pydantic import BaseModel - -models = {} - -app = FastAPI() - -class EmbeddingRequest(BaseModel): - model: str - input: str - -@app.post("/v1/embeddings") -def embeddings(request: EmbeddingRequest): - model = models.get(request.model) or TextEmbedding(request.model) - models[request.model] = model - embeddings = next(model.embed(request.input)).tolist() - return {"data": [{"embedding": embeddings}]} - -if __name__ == "__main__": - import uvicorn - - uvicorn.run(app, host="0.0.0.0", port=11345) -- cgit v1.2.3