Skip to content

Commit

Permalink
add qdrant docker-compose
Browse files Browse the repository at this point in the history
  • Loading branch information
TerboucheHacene committed Aug 27, 2023
1 parent fa227be commit 376cb63
Show file tree
Hide file tree
Showing 2 changed files with 20 additions and 2 deletions.
12 changes: 12 additions & 0 deletions docker-compose.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
version: '3.8'

services:
qdrant:
image: qdrant/qdrant:v1.1.2
ports:
- "6333:6333"
- "6334:6334"
ulimits: # Only required for tests, as there are a lot of collections created
nofile:
soft: 65535
hard: 65535
10 changes: 8 additions & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -8,16 +8,18 @@ readme = "README.md"
packages = [ { include = "scholar_sense", from = "src" } ]

[tool.poetry.dependencies]
python = ">=3.9,<3.9.7 || >3.9.7,<4.0"
python = ">=3.9,<3.9.7 || >3.9.7,<3.12"
arxiv = "^1.4.8"
pandas = "^2.0.3"
tqdm = "^4.65.0"
nltk = "^3.8.1"
docarray = {extras = ["annlite"], version = "^0.36.0"}
docarray = {extras = ["qdrant"], version = "^0.37.0"}
torch = { version="^1.13.0", extras=["cuda116"] }
torchvision = { version="^0.14", extras=["cuda116"] }
sentence-transformers = "^2.2.2"
streamlit = "^1.25.0"
openai = "^0.27.8"
typer = {extras = ["all"], version = "^0.9.0"}

[tool.poe.tasks]
force-cuda113 = "python -m pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html"
Expand All @@ -41,6 +43,7 @@ flake8 = "^5.0"
isort = "^5.10"
mypy = ">=0.991"
pre-commit = "^2.16"

[tool.poetry.group.tests.dependencies]
pytest = "^7.1"
pytest-cov = "^3.0"
Expand All @@ -53,6 +56,9 @@ myst-nb = "^0.16"
sphinx-autoapi = ">=1.8"
sphinx-rtd-theme = ">=1.0"

[tool.poetry.scripts]
scholarsense = "scholar_sense.main:app"

[tool.black]
target-version = ['py39']
line_length = 89
Expand Down

0 comments on commit 376cb63

Please sign in to comment.