diff --git a/docker-compose.yaml b/docker-compose.yaml new file mode 100644 index 0000000..fdfe9da --- /dev/null +++ b/docker-compose.yaml @@ -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 diff --git a/pyproject.toml b/pyproject.toml index 06e3004..c0c0591 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -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" @@ -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" @@ -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