summaryrefslogtreecommitdiff
path: root/python
diff options
context:
space:
mode:
authorLain Soykaf <lain@lain.com>2024-05-19 12:47:08 +0400
committerLain Soykaf <lain@lain.com>2024-05-19 12:47:08 +0400
commitdd48810186e3b4ee14e1d3727f37bd470d0711a4 (patch)
treeee7b0480adedac30316b76ff74f099b13c82ca23 /python
parente142ea400a9ed3595f8d432edd90ea26fc7d2eb5 (diff)
downloadpleroma-dd48810186e3b4ee14e1d3727f37bd470d0711a4.tar.gz
pleroma-dd48810186e3b4ee14e1d3727f37bd470d0711a4.zip
B FastembedAPI: Move to more appropriate folder
Diffstat (limited to 'python')
-rw-r--r--python/Dockerfile8
-rw-r--r--python/compose.yml5
-rw-r--r--python/fastembed-server.py23
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)