pyproject.toml

[tool.poetry]
name = "llmodel"
version = "0.1.0"
description = ""
authors = ["Ben Van Sleen <benvansleen@gmail.com>"]
readme = "README.md"

[tool.poetry.dependencies]
python = "^3.10"
openai = "^0.27.8"
langchain = "^0.0.206"
numpy = "^1.25.0"
python-dotenv = "^1.0.0"
requests = "^2.31.0"
tiktoken = "^0.4.0"
termcolor = "^2.3.0"
chromadb = "^0.3.26"
ompython = "^3.4.0"
pyzmq = "^25.1.0"
torch = {version = "^2.0.1+cpu", source = "pytorch"}
transformers = "^4.30.2"
openai-function-call = "^0.0.5"


[tool.poetry.group.dev.dependencies]
autopep8 = "^2.0.2"


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

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