pyproject.toml

Rama main · réplica de lectura del repositorio original.

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

[project]
name = "mem0ai"
version = "2.2.1"
description = "Long-term memory for AI Agents"
authors = [
    { name = "Mem0", email = "support@mem0.ai" }
]
readme = "README.md"
license = "Apache-2.0"
license-files = ["LICENSE"]
requires-python = ">=3.10,<4.0"
dependencies = [
    "qdrant-client>=1.12.0",
    "pydantic>=2.7.3",
    "openai>=1.90.0",
    "httpx>=0.28.0",
    "posthog>=7.14.0",
    "pytz>=2024.1",
    "sqlalchemy>=2.0.31",
    "protobuf>=5.29.6,<7.0.0",
]

[project.optional-dependencies]
nlp = [
    "spacy>=3.7.0",
]
vector-stores = [
    "vecs>=0.4.0",
    "chromadb>=0.4.24",
    "cassandra-driver>=3.29.0",
    "weaviate-client>=4.15.4,<5.0.0",
    "pinecone<=7.3.0",
    "pinecone-text>=0.10.0",
    "faiss-cpu>=1.7.4",
    "upstash-vector>=0.6.0",
    "azure-search-documents>=11.4.0b8",
    "psycopg>=3.2.8",
    "psycopg-pool>=3.2.6,<4.0.0",
    "pymongo>=4.13.2",
    "pymochow>=2.2.9",
    "pymysql>=1.1.0",
    "dbutils>=3.0.3",
    "valkey>=6.0.0",
    "databricks-sdk>=0.63.0",
    "azure-identity>=1.24.0",
    "redis>=5.0.0,<6.0.0",
    "redisvl>=0.1.0,<1.0.0",
    "elasticsearch>=8.0.0,<9.0.0",
    "pymilvus>=2.4.0,<2.6.0",
    "langchain-aws>=0.2.23,<0.3.0",
    "oracledb>=2.2.0",
]
llms = [
    "groq>=0.3.0",
    "together>=0.2.10",
    "litellm>=1.83.7,<1.98.0",
    "openai>=1.90.0",
    "ollama>=0.3.0",
    "vertexai>=0.1.0",
    "google-generativeai>=0.3.0",
    "google-genai>=1.0.0",
]
extras = [
    "boto3>=1.34.0",
    "langchain>=0.3.30,<1.0.0",
    "langchain-community>=0.3.27,<1.0.0",
    "langchain-core>=0.3.85,<1.0.0",
    "sentence-transformers>=5.2.0",
    "transformers>=5.3.0",
    "elasticsearch>=8.0.0,<9.0.0",
    "opensearch-py>=2.0.0",
    "fastembed>=0.3.1",
]
test = [
    "pytest>=8.2.2",
    "pytest-mock>=3.14.0",
    "pytest-asyncio>=0.23.7",
]
dev = [
    "ruff==0.16.0",
    "isort>=5.13.2",
    "pre-commit>=3.5.0",
]

[tool.pytest.ini_options]
pythonpath = ["."]

[tool.hatch.build]
include = [
    "mem0/**/*.py",
    "mem0/memory/oss_notices_config.json",
]
exclude = [
    "**/*",
    "!mem0/**/*.py",
    "!mem0/memory/oss_notices_config.json",
]

[tool.hatch.build.targets.wheel]
packages = ["mem0"]
only-include = ["mem0"]

[tool.hatch.envs.dev_py_3_10]
python = "3.10"
features = [
  "test",
  "vector-stores",
  "llms",
  "extras",
  "dev",
]

[tool.hatch.envs.dev_py_3_11]
python = "3.11"
features = [
  "test",
  "vector-stores",
  "llms",
  "extras",
  "dev",
]

[tool.hatch.envs.dev_py_3_12]
python = "3.12"
features = [
  "test",
  "vector-stores",
  "llms",
  "extras",
  "dev",
]

[tool.hatch.envs.default]
features = [
  "dev",
]

[tool.hatch.envs.default.scripts]
format = [
    "ruff format",
]
format-check = [
    "ruff format --check",
]
lint = [
    "ruff check",
]
lint-fix = [
    "ruff check --fix",
]
test = [
    "pytest tests/ {args}",
]

[tool.ruff]
line-length = 120

[tool.ruff.lint]
select = ["E4", "E7", "E9", "F"]

[tool.ruff.lint.isort]
known-first-party = ["mem0", "mem0_cli"]

[tool.isort]
profile = "black"
known_first_party = ["mem0", "mem0_cli"]
# isort scope kept aligned with [tool.ruff.lint.isort] above.
# black-equivalent profile here matches the formatter behaviour ruff applies.
# Plugin-version bumps need a touch here to fire required CI checks (path-filter trap).
# Last touched: plugin v0.1.3