Yuge Zhang
105 lines
1.7 KiB
TOML
105 lines
1.7 KiB
TOML
[project]
|
|
name = "agentlightning"
|
|
version = "0.2.0"
|
|
description = "Agent Lightning is the absolute trainer to light up AI agents."
|
|
readme = "README.md"
|
|
requires-python = ">=3.10"
|
|
dependencies = [
|
|
"graphviz",
|
|
"psutil",
|
|
"setproctitle",
|
|
"flask",
|
|
"agentops<=0.4.21",
|
|
"httpdbg",
|
|
"uvicorn",
|
|
"fastapi",
|
|
"aiohttp",
|
|
"opentelemetry-api>=1.35",
|
|
"opentelemetry-sdk>=1.35",
|
|
"opentelemetry-exporter-otlp>=1.35",
|
|
"litellm[proxy]",
|
|
"pydantic>=2.11",
|
|
"openai",
|
|
"rich",
|
|
]
|
|
|
|
[project.scripts]
|
|
agl = "agentlightning.cli:main"
|
|
|
|
[project.optional-dependencies]
|
|
dev = [
|
|
"flake8",
|
|
"pytest",
|
|
"hatch",
|
|
"pytest-asyncio",
|
|
"pre-commit",
|
|
"pytest-rerunfailures",
|
|
"black",
|
|
"isort",
|
|
"pyright",
|
|
"mkdocs",
|
|
"mkdocs-material",
|
|
"mkdocstrings[python]",
|
|
"mike",
|
|
"mkdocs-git-revision-date-localized-plugin",
|
|
"mkdocs-git-authors-plugin",
|
|
"mkdocs-macros-plugin",
|
|
]
|
|
experiment = [
|
|
"random-word",
|
|
]
|
|
agent = [
|
|
"autogen-agentchat",
|
|
"autogen-ext[openai]",
|
|
"mcp",
|
|
"openai-agents",
|
|
"langgraph",
|
|
"langchain[openai]",
|
|
"langchain-community",
|
|
"langchain-text-splitters",
|
|
"sqlparse",
|
|
"nltk",
|
|
"uv",
|
|
"anthropic",
|
|
]
|
|
trl = [
|
|
"unsloth",
|
|
"unsloth_zoo",
|
|
"bitsandbytes",
|
|
"peft",
|
|
"datasets",
|
|
"transformers",
|
|
"trl",
|
|
"kernels",
|
|
]
|
|
|
|
[build-system]
|
|
requires = ["hatchling"]
|
|
build-backend = "hatchling.build"
|
|
|
|
[tool.hatch.build.targets.wheel]
|
|
packages = ["agentlightning"]
|
|
include = ["**/*.yaml", "**/*.yml"]
|
|
|
|
[tool.pytest.ini_options]
|
|
testpaths = ["tests"]
|
|
|
|
[tool.black]
|
|
line-length = 120
|
|
target-version = ['py312']
|
|
include = '\.pyi?$'
|
|
extend-exclude = '''
|
|
/(
|
|
_version\.py
|
|
)
|
|
'''
|
|
|
|
[tool.isort]
|
|
profile = "black"
|
|
line_length = 120
|
|
known_first_party = ["agentlightning"]
|
|
extend_skip_glob = [
|
|
"data/**",
|
|
"_version.py",
|
|
]
|