BEM-AI/pyproject.toml at master · pnnl/BEM-AI · GitHub

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
[project]
name = "automa_ai"
version = "0.6.8"
description = "PNNL Auto Multi-Agent AI: Dynamic multi-agent system for building applications"
requires-python = ">=3.12"
authors = [{name = "Weili Xu", email = "weili.xu@pnnl.gov"}, {name = "Hanlong, Wan", email = "hanlong.wan@pnnl.gov"}]
license = "BSD-2-Clause"
readme = "README.md"
dependencies = [
    "a2a-sdk>=1.0.2",
    "asyncclick>=8.3.0.7",
    "boto3>=1.41.4",
    "bs4>=0.0.2",
    "chromadb>=1.5.8",
    "deepeval>=3.9.9",
    "google-adk>=1.31.1",
    "google-genai>=1.73.1",
    "httpx>=0.28.1",
    "jwcrypto>=1.5.6",
    "langchain>=1.2.15",
    "langchain-anthropic>=1.4.1",
    "langchain-aws>=1.4.5",
    "langchain-chroma>=1.1.0",
    "langchain-community>=0.4.1",
    "langchain-google-genai>=4.2.2",
    "langchain-huggingface>=1.2.2",
    "langchain-mcp-adapters>=0.2.2",
    "langchain-ollama>=1.1.0",
    "langchain-openai>=1.1.7",
    "langgraph>=1.0.7",
    "langgraph-checkpoint-redis>=0.4.1",
    "litellm>=1.79.0",
    "networkx>=3.5",
    "pandas>=2.3.3",
    "pandas-stubs==2.3.3.251219",
    "pydantic>=2.12.3",
    "pyjwt>=2.11.0",
    "pyyaml>=6.0.2",
    "redis>=6.4.0",
    "streamlit>=1.54.0",
    "uvicorn>=0.40.0",
    "multiprocessing-logging>=0.3.4",
    "langgraph-checkpoint-aws>=1.0.7",
]
packages = [{include = "automa_ai", from = "."}]

# --- Optional dependencies for examples ---
[project.optional-dependencies]
web = [
    "trafilatura>=1.12.2",
    "duckduckgo-search>=7.2.1",
    "rank-bm25>=0.2.2",
]
rerank = ["sentence-transformers>=3.4.1"]
sim_bem_network = ["openstudio>=3.10.0", "streamlit>=1.51.0"]
sim_chat_demo = ["streamlit>=1.51.0"]
sim_chat_stream_demo = ["streamlit>=1.51.0"]
eplus_mcp_demo = ["eppy>=0.5.63", "matplotlib>=3.10.7", "plotly>=6.4.0"]

[tool.uv]
dev-dependencies = [
    "notebook>=7.4.7",
    "pytest>=8.4.2",
    "pytest-asyncio>=1.2.0",
    "black>=25.9.0",
]

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project.entry-points."automa_ai.memory_stores"]
default_sqlite = "automa_ai.memory.sqlite_memory_store:SQLiteMemoryStore"
default_chroma = "automa_ai.memory.chroma_memory_store:ChromaVectorMemoryStore"

[project.entry-points."automa_ai.token_usage_stores"]
sqlite = "automa_ai.token_management.store:SQLiteTokenUsageStore"

[project.entry-points."automa_ai.tools"]
web_search = "automa_ai.tools.web_search:build_web_search_tool"
run_command = "automa_ai.tools.run_command:build_run_command_tool"
run_python = "automa_ai.tools.run_python:build_run_python_tool"
yaml_agent = "automa_ai.tools.yaml_agent:build_yaml_agent_tool"