Skip to content

Commit

Permalink
add missing mlx dep, as otherwise tests fail locally
Browse files Browse the repository at this point in the history
  • Loading branch information
sysradium committed Feb 13, 2025
1 parent 149546d commit 8b247fe
Showing 1 changed file with 22 additions and 47 deletions.
69 changes: 22 additions & 47 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,8 @@ name = "smolagents"
version = "1.9.0.dev0"
description = "🤗 smolagents: a barebones library for agents. Agents write python code to call tools or orchestrate other agents."
authors = [
{ name="Aymeric Roucher", email="[email protected]" }, { name="Thomas Wolf"},
{ name = "Aymeric Roucher", email = "[email protected]" },
{ name = "Thomas Wolf" },
]
readme = "README.md"
requires-python = ">=3.10"
Expand All @@ -20,65 +21,38 @@ dependencies = [
"pillow>=11.0.0",
"markdownify>=0.14.1",
"duckduckgo-search>=6.3.7",
"python-dotenv"
"python-dotenv",
]

[project.optional-dependencies]
torch = [
"torch",
"torchvision",
]
audio = [
"soundfile",
"smolagents[torch]",
]
e2b = [
"e2b-code-interpreter>=1.0.3",
"python-dotenv>=1.0.1",
]
gradio = [
"gradio>=5.13.2",
]
litellm = [
"litellm>=1.60.2",
]
mcp = [
"mcpadapt>=0.0.6",
"mcp",
]
mlx-lm = [
"mlx-lm"
]
openai = [
"openai>=1.58.1"
]
torch = ["torch", "torchvision"]
audio = ["soundfile", "smolagents[torch]"]
e2b = ["e2b-code-interpreter>=1.0.3", "python-dotenv>=1.0.1"]
gradio = ["gradio>=5.13.2"]
litellm = ["litellm>=1.60.2"]
mcp = ["mcpadapt>=0.0.6", "mcp"]
mlx-lm = ["mlx-lm"]
openai = ["openai>=1.58.1"]
telemetry = [
"arize-phoenix",
"opentelemetry-sdk",
"opentelemetry-sdk",
"opentelemetry-exporter-otlp",
"openinference-instrumentation-smolagents>=0.1.4"
]
transformers = [
"accelerate",
"transformers>=4.0.0",
"smolagents[torch]",
]
all = [
"smolagents[audio,e2b,gradio,litellm,mcp,openai,telemetry,transformers]",
]
quality = [
"ruff>=0.9.0",
"openinference-instrumentation-smolagents>=0.1.4",
]
transformers = ["accelerate", "transformers>=4.0.0", "smolagents[torch]"]
all = ["smolagents[audio,e2b,gradio,litellm,mcp,openai,telemetry,transformers]"]
quality = ["ruff>=0.9.0"]
test = [
"ipython>=8.31.0", # for interactive environment tests
"ipython>=8.31.0", # for interactive environment tests
"pytest>=8.1.0",
"python-dotenv>=1.0.1", # For test_all_docs
"smolagents[all]",
"rank-bm25", # For test_all_docs
"rank-bm25", # For test_all_docs
]
dev = [
"smolagents[quality,test]",
"sqlalchemy", # for ./examples
"sqlalchemy", # for ./examples
"mlx-lm",
]

[tool.pytest.ini_options]
Expand Down Expand Up @@ -107,4 +81,5 @@ lines-after-imports = 2

[project.scripts]
smolagent = "smolagents.cli:main"
webagent = "smolagents.vision_web_browser:main"
webagent = "smolagents.vision_web_browser:main"

0 comments on commit 8b247fe

Please sign in to comment.