-
Notifications
You must be signed in to change notification settings - Fork 14
Expand file tree
/
Copy pathpyproject.toml
More file actions
114 lines (99 loc) · 2.78 KB
/
pyproject.toml
File metadata and controls
114 lines (99 loc) · 2.78 KB
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "toolbrain"
version = "0.1.4"
description = "A framework for training LLM-powered agents to use tools more effectively using Reinforcement Learning"
readme = "README.md"
requires-python = ">=3.10"
classifiers = [
"Development Status :: 3 - Alpha",
"Intended Audience :: Developers",
"License :: OSI Approved :: MIT License",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Programming Language :: Python :: 3.12",
]
authors = [
{name = "ToolBrain Team", email = "team@toolbrain.ai"}
]
keywords = ["llm", "agents", "reinforcement-learning", "tools", "ai"]
dependencies = [
"smolagents>=0.1.0",
"openai>=1.0.0",
"typing-extensions>=4.0.0",
"python-dotenv>=1.0.0",
"peft>=0.17.0",
"litellm",
"google-generativeai",
"gradio",
"datasets>=3.4.1,<4.0.0",
"langchain>=1.0.0a10",
"langchain-huggingface",
"langgraph",
"langchain_core",
"bitsandbytes"
]
[project.optional-dependencies]
# --- OPTIONAL DEPENDENCIES (Extras) ---
# Extra for core ML libraries
torch = [
"torch",
"accelerate", # Often needed with torch and transformers
]
# Extra for Hugging Face Transformers
transformers = [
"transformers",
"toolbrain[torch]", # This extra depends on the 'torch' extra
]
# 'unsloth' extra for GPU-accelerated training
unsloth = [
"unsloth",
"bitsandbytes",
"xformers",
"toolbrain[transformers]", # Unsloth needs transformers and torch
]
# 'dev' extra for development and testing tools
dev = [
"pytest>=7.0.0",
"black>=22.0.0",
"isort>=5.0.0",
"mypy>=1.0.0",
]
# Basic dependencies for running simple examples and plotting
examples-base = [
"pandas",
"matplotlib",
"seaborn",
"toolbrain[transformers]", # Most examples will need transformers
]
# Full dependencies for running all examples, including advanced ones
examples-full = [
"toolbrain[examples-base]",
"toolbrain[unsloth]", # Add the GPU-specific dependencies
]
# 'all' extra to install everything for a full development environment
all = [
"toolbrain[dev, examples-full]", # 'all' now includes everything
]
[project.urls]
Homepage = "https://github.com/toolbrain/toolbrain"
Documentation = "https://toolbrain.readthedocs.io"
Repository = "https://github.com/toolbrain/toolbrain"
Issues = "https://github.com/toolbrain/toolbrain/issues"
[tool.setuptools.packages.find]
where = ["."]
include = ["toolbrain*"]
[tool.black]
line-length = 88
target-version = ['py310']
[tool.isort]
profile = "black"
line_length = 88
[tool.mypy]
python_version = "3.10"
warn_return_any = true
warn_unused_configs = true
disallow_untyped_defs = true