-
Notifications
You must be signed in to change notification settings - Fork 15
/
pyproject.toml
79 lines (69 loc) · 1.91 KB
/
pyproject.toml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
[tool.poetry]
name = "sotopia"
version = "0.0.8"
description = "A platform for simulating and evaluating social interaction."
authors = ["Hao Zhu <prokilchu@gmail.com>"]
license = "MIT License"
readme = "README.md"
packages = [{include = "sotopia"}]
[tool.poetry.dependencies]
python = ">=3.10, <3.13"
pandas = "^2.1.1"
lxml = "^4.9.3"
openai = "^1.11.0"
langchain = "0.1.5"
rich = "^13.6.0"
PettingZoo = "1.24.0"
redis-om = "^0.2.1"
pandas-stubs = "*"
types-tqdm = "*"
gin-config = "^0.5.0"
absl-py = "^2.0.0"
together = "^0.2.4"
pydantic = "1.10.12"
beartype = "^0.14.0"
langchain-openai = "^0.0.5"
litellm = "~1.23.12"
# dependency versions for extras
fastapi = { version = "^0.109.2", optional = true }
tabulate = { version = "^0.9.0", optional = true }
anthropic = { version = "^0.26.0", optional = true }
xmltodict = { version = "^0.13.0", optional = true }
groq = { version = "^0.4.2", optional = true }
cohere = { version = "^5.1.8", optional = true }
google-generativeai = { version = "^0.5.4", optional = true }
transformers = { version = "^4.41.0", optional = true }
datasets = { version = "^2.19.0", optional = true }
scipy = { version = "^1.13.1", optional = true }
torch = { version = "^2.3.0", optional = true }
[tool.poetry.group.dev.dependencies]
pre-commit = "*"
nbmake = "*"
types-setuptools = "*"
ruff = "*"
[tool.poetry.group.test.dependencies]
pytest = "*"
pytest-asyncio = "*"
mypy = "^1.8.0"
[tool.poetry.extras]
anthropic = ["anthropic"]
groq = ["groq"]
cohere = ["cohere"]
google-generativeai = ["google-generativeai"]
examples = ["transformers", "datasets", "scipy", "torch"]
chat = ["fastapi"]
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
[tool.mypy]
mypy_path = "stubs"
strict = true
plugins = [
"pydantic.mypy"
]
[[tool.mypy.overrides]]
module = "transformers.*"
ignore_missing_imports = true
[tool.pytest.ini_options]
testpaths = ["tests"]
python_files = "test_*.py"