summaryrefslogtreecommitdiff
path: root/python/fastembed-server.py
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/fastembed-server.py
parente142ea400a9ed3595f8d432edd90ea26fc7d2eb5 (diff)
downloadpleroma-dd48810186e3b4ee14e1d3727f37bd470d0711a4.tar.gz
pleroma-dd48810186e3b4ee14e1d3727f37bd470d0711a4.zip
B FastembedAPI: Move to more appropriate folder
Diffstat (limited to 'python/fastembed-server.py')
-rw-r--r--python/fastembed-server.py23
1 files changed, 0 insertions, 23 deletions
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)