kor

1
[tool.poetry]
2
name = "kor"
3
version = "1.0.1"
4
description = "Extract information with LLMs from text"
5
authors = ["Eugene Yurtsev <eyurtsev@gmail.com>"]
6
license = "MIT"
7
readme = "README.md"
8
repository = "https://www.github.com/eyurtsev/kor"
9

10
[tool.poetry.dependencies]
11
python = "^3.8.1"
12
pandas = "^1.5.3"
13
markdownify = {version = "^0.11.6", optional = false}
14
pydantic = "<3"
15
langchain-core = "^0.1"
16
langchain = "^0.1"
17

18
[tool.poetry.group.dev.dependencies]
19
jupyterlab = "^3"
20

21
[tool.poetry.group.test.dependencies]
22
pytest = "^7.2.1"
23
black = { version="^23.1.0", extras=["jupyter"] }
24
poethepoet = "^0.18.1"
25
ruff = "^0.1.4"
26
pytest-cov = "^4.0.0"
27
langchain-openai = "^0.0.5"
28

29
[tool.poetry.group.docs.dependencies]
30
nbsphinx = "^0.8.9"
31
sphinx = "^4.5.0"
32
sphinx-autobuild = "^2021.3.14"
33
sphinx_book_theme = "^0.3.3"
34
sphinx_rtd_theme = "^1.0.0"
35
sphinx-typlog-theme = "^0.8.0"
36
sphinx-panels = "^0.6.0"
37
myst-nb = "^0.17.1"
38
linkchecker = "^10.2.1"
39
toml = "^0.10.2"
40
sphinx-copybutton = "^0.5.1"
41

42
[tool.poetry.group.typing.dependencies]
43
mypy = "^0.991"
44

45
[tool.poetry.group.types.dependencies]
46
types-toml = "^0.10.8.5"
47

48

49
[tool.poetry.extras]
50
html = ["markdownify"]
51

52
[tool.poe.tasks]
53
black = "black"
54
ruff = "ruff"
55
pytest.cmd = "py.test --capture=no --durations=5 -W error::RuntimeWarning --cov --cov-config=.coveragerc --cov-report xml --cov-report term-missing:skip-covered"
56
mypy = "mypy . --pretty --show-error-codes"
57
fix = { shell = "poe black . && poe ruff --fix ." }
58
fix_docs = "black docs"
59
# Using a --preview feature. Commit your code prior to use this fix.
60
fix_strings = "black kor --preview"
61
test = { shell = "poe black . --check --diff && poe ruff . && poe pytest && poe mypy" }
62
# Use to auto-generate docs
63
apidoc = "sphinx-apidoc -o docs/source/generated kor"
64
pyintegration = { shell = "py.test --capture=no ./tests/integration/*.py" }
65
integration = { shell = "poe black . --check --diff && poe ruff . && poe pyintegration && poe mypy" }
66

67
[tool.ruff]
68
select = [
69
  "E",  # pycodestyle
70
  "F",  # pyflakes
71
  "I",  # isort
72
]
73

74
# Same as Black.
75
line-length = 88
76

77
[tool.mypy]
78
disallow_untyped_defs = "True"
79
ignore_missing_imports = "True"
80

81
[tool.coverage.run]
82
omit = [
83
    "tests/*",
84
]
85

86

87
[build-system]
88
requires = ["poetry-core>=1.0.0"]
89
build-backend = "poetry.core.masonry.api"
90