[tool.poetry]
name = "embedding"
version = "0.1.0"
description = "embedding model serving"
authors = ["nam pham <contact.nampham.work@gmail.com>"]
readme = "README.md"

[tool.poetry.dependencies]
python = "^3.12"
sentence-transformers = {extras = ["onnx"], version = "^3.3.1"}
litserve = "^0.2.5"
torch = {version = "^2.5.1+cpu", source = "pytorch_cpu"}


[[tool.poetry.source]]
name = "pytorch_cpu"
url = "https://download.pytorch.org/whl/cpu"
priority = "explicit"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"