pyproject.toml

[project]
name = "veadk-python"
version = "0.5.26"
description = "Volcengine agent development kit, integrations with Volcengine cloud services."
readme = "README.md"
requires-python = ">=3.10"
license = { file = "LICENSE" }
authors = [
    {name = "Yaozheng Fang", email = "fangyozheng@gmail.com"},
    {name = "Guodong Li", email = "cu.eric.lee@gmail.com"},
    {name = "Zhi Han", email = "sliverydayday@gmail.com"},
    {name = "Meng Wang", email = "mengwangwm@gmail.com"}
]
dependencies = [
    "pydantic-settings==2.10.1", # Config management
    "a2a-sdk>=0.3.7", # For Google Agent2Agent protocol
    "deprecated==1.2.18",
    "google-adk>=1.19.0", # For basic agent architecture
    "litellm>=1.74.3", # For model inference
    "loguru==0.7.3", # For better logging
    "opentelemetry-exporter-otlp==1.37.0",
    "opentelemetry-instrumentation-logging>=0.56b0",
    "wrapt==1.17.2", # For patching built-in functions
    "volcengine-python-sdk>=5.0.1", # For Volcengine API
    "volcengine>=1.0.193", # For Volcengine sign
    "agent-pilot-sdk==0.1.2", # Prompt optimization by Volcengine AgentPilot/PromptPilot toolkits
    "fastmcp>=2.12.3", # For running MCP
    "trustedmcp==0.0.5", # For running TrustedMCP
    "mcp==1.23.0", # There might be some async generator problem logs in mcp 1.25.0
    "cookiecutter==2.6.0", # For cloud deploy
    "omegaconf==2.3.0", # For agent builder
    "psycopg2-binary==2.9.10", # For PostgreSQL database (short term memory)
    "asyncpg>=0.29.0", # For async PostgreSQL database (short term memory)
    "pymysql==1.1.1", # For MySQL database (short term memory)
    "aiomysql==0.3.2", # For async MySQL database (short term memory)
    "filetype==1.2.0",
    "vikingdb-python-sdk>=0.1.3", # For Viking DB
    "agentkit-sdk-python>=0.2.0",
    "python-frontmatter==1.1.0",
    "tos>=2.8.4",   # For TOS storage and Viking DB
]

[project.scripts]
veadk = "veadk.cli.cli:veadk"

[project.optional-dependencies]
extensions = [
    "redis>=5.0",                                           # For Redis database
    "cozeloop>=0.1.21",                                     # For Cozeloop Prompt manager
    "llama-index==0.14.0",                                  # For KnowledgeBase and LongTermMemory
    "llama-index-embeddings-openai-like==0.2.2",            # For Embeddings
    "llama-index-llms-openai-like==0.5.1",                  # For KnowledgeBase and LongTermMemory
    "llama-index-vector-stores-redis>=0.6.1",               # For Redis database
    "llama-index-vector-stores-opensearch==0.6.1",          # For Opensearch database
    "opensearch-py==2.8.0",
]
database = [
    "redis>=5.0",                   # For Redis database
    "pymysql>=1.1.1",               # For MySQL database
    "volcengine>=1.0.193",          # For Viking DB
    "mem0ai==0.1.118",              # For mem0
]
speech = []
eval = [
    "prometheus-client>=0.22.1",    # For exporting data to Prometheus pushgateway
    "deepeval>=3.2.6",              # For DeepEval-based evaluation
    "google-adk[eval]",             # For Google ADK-based evaluation
]
cli = []
dev = [
    "pre-commit>=4.2.0",            # Format checking
    "pytest>=8.4.1",
    "pytest-asyncio>=1.0.0",
    "pytest-xdist>=3.8.0",
]

[dependency-groups]
dev = [
    "pre-commit>=4.2.0",            # Format checking
    "pytest>=8.4.1",
    "pytest-asyncio>=1.0.0",
    "pytest-xdist>=3.8.0",
]


[tool.setuptools.packages.find]
include = ["veadk*"]
exclude = ["assets*", "ide*", "tests*"]

[tool.setuptools]
include-package-data = true

[tool.setuptools.package-data]
"veadk" = ["**/*"]

[tool.ruff]
exclude = [
    "veadk/integrations/ve_faas/template/*",
    "veadk/integrations/ve_faas/web_template/*"
]