pyproject.toml 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125
  1. [tool.poetry]
  2. name = "embedchain"
  3. version = "0.0.45"
  4. description = "embedchain is a framework to easily create LLM powered bots over any dataset"
  5. authors = ["Taranjeet Singh"]
  6. license = "Apache License"
  7. readme = "README.md"
  8. [build-system]
  9. requires = ["setuptools", "wheel"]
  10. build-backend = "setuptools.build_meta"
  11. [tool.ruff]
  12. select = ["E", "F"]
  13. ignore = []
  14. fixable = ["ALL"]
  15. unfixable = []
  16. exclude = [
  17. ".bzr",
  18. ".direnv",
  19. ".eggs",
  20. ".git",
  21. ".git-rewrite",
  22. ".hg",
  23. ".mypy_cache",
  24. ".nox",
  25. ".pants.d",
  26. ".pytype",
  27. ".ruff_cache",
  28. ".svn",
  29. ".tox",
  30. ".venv",
  31. "__pypackages__",
  32. "_build",
  33. "buck-out",
  34. "build",
  35. "dist",
  36. "node_modules",
  37. "venv",
  38. ]
  39. line-length = 120
  40. dummy-variable-rgx = "^(_+|(_+[a-zA-Z0-9_]*[a-zA-Z0-9]+?))$"
  41. target-version = "py38"
  42. [tool.ruff.mccabe]
  43. max-complexity = 10
  44. # Ignore `E402` (import violations) in all `__init__.py` files, and in `path/to/file.py`.
  45. [tool.ruff.per-file-ignores]
  46. "embedchain/__init__.py" = ["E401"]
  47. [tool.black]
  48. line-length = 120
  49. target-version = ["py38", "py39", "py310", "py311"]
  50. include = '\.pyi?$'
  51. exclude = '''
  52. /(
  53. \.eggs
  54. | \.git
  55. | \.hg
  56. | \.mypy_cache
  57. | \.nox
  58. | \.pants.d
  59. | \.pytype
  60. | \.ruff_cache
  61. | \.svn
  62. | \.tox
  63. | \.venv
  64. | __pypackages__
  65. | _build
  66. | buck-out
  67. | build
  68. | dist
  69. | node_modules
  70. | venv
  71. )/
  72. '''
  73. [tool.black.format]
  74. color = true
  75. [tool.poetry.dependencies]
  76. python = ">=3.9,<3.9.7 || >3.9.7,<4.0"
  77. python-dotenv = "^1.0.0"
  78. langchain = "^0.0.237"
  79. requests = "^2.31.0"
  80. openai = "^0.27.5"
  81. chromadb ="^0.4.2"
  82. youtube-transcript-api = "^0.6.1"
  83. beautifulsoup4 = "^4.12.2"
  84. pypdf = "^3.11.0"
  85. pytube = "^15.0.0"
  86. llama-index = { version = "^0.7.21", optional = true }
  87. sentence-transformers = { version = "^2.2.2", optional = true }
  88. torch = { version = ">=2.0.0, !=2.0.1", optional = true }
  89. # Torch 2.0.1 is not compatible with poetry (https://github.com/pytorch/pytorch/issues/100974)
  90. gpt4all = { version = "^1.0.8", optional = true }
  91. elasticsearch = { version = "^8.9.0", optional = true }
  92. flask = "^2.3.3"
  93. twilio = "^8.5.0"
  94. [tool.poetry.group.dev.dependencies]
  95. black = "^23.3.0"
  96. pre-commit = "^3.2.2"
  97. ruff = "^0.0.220"
  98. pytest = "^7.3.1"
  99. pytest-mock = "^3.10.0"
  100. pytest-env = "^0.8.1"
  101. click = "^8.1.3"
  102. isort = "^5.12.0"
  103. [tool.poetry.extras]
  104. streamlit = ["streamlit"]
  105. community = ["llama-index"]
  106. opensource = ["sentence-transformers", "torch", "gpt4all"]
  107. elasticsearch = ["elasticsearch"]
  108. [tool.poetry.group.docs.dependencies]
  109. [tool.poetry.scripts]