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/Dockerfile | 8 -------- python/compose.yml | 5 ----- python/fastembed-server.py | 23 ----------------------- 3 files changed, 36 deletions(-) delete mode 100644 python/Dockerfile delete mode 100644 python/compose.yml delete mode 100644 python/fastembed-server.py (limited to 'python') diff --git a/python/Dockerfile b/python/Dockerfile deleted file mode 100644 index f83c1c1b3..000000000 --- a/python/Dockerfile +++ /dev/null @@ -1,8 +0,0 @@ -FROM python:3.9 - -WORKDIR /code -COPY fastembed-server.py /workdir/fastembed-server.py - -RUN pip install --no-cache-dir --upgrade fastembed fastapi uvicorn - -CMD ["python", "/workdir/fastembed-server.py"] diff --git a/python/compose.yml b/python/compose.yml deleted file mode 100644 index d4cb31722..000000000 --- a/python/compose.yml +++ /dev/null @@ -1,5 +0,0 @@ -services: - web: - build: . - ports: - - "11345:11345" 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