diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/Dockerfile | 8 | ||||
| -rw-r--r-- | python/compose.yml | 5 | ||||
| -rw-r--r-- | python/fastembed-server.py | 23 |
3 files changed, 0 insertions, 36 deletions
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) |
