deepagents-graph-memory 0.1.1__cp313-cp313-win_amd64.whl

deepagents_graph_memory/__init__.py

@@ -0,0 +1,32 @@
+# - Gives apps one place to import the main graph features.
+# - Tests: test_vgs.py checks the public graph setup helpers;
+#        test_optional_vgs_dependency.py checks what happens without LadybugDB.
+
+"""Graph-backed memory backend for LangChain Deep Agents."""
+
+
+# start delvewheel patch
+def _delvewheel_patch_1_13_1():
+    import os
+    if os.path.isdir(libs_dir := os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir, 'deepagents_graph_memory.libs'))):
+        os.add_dll_directory(libs_dir)
+
+
+_delvewheel_patch_1_13_1()
+del _delvewheel_patch_1_13_1
+# end delvewheel patch
+
+from deepagents_graph_memory.backend import GraphMemoryBackend
+from deepagents_graph_memory.paths import make_graph_subject
+from deepagents_graph_memory.tools import graph_memory_tools
+from deepagents_graph_memory.vgs import VFS_TOOL_NAMES, graph_context_middleware, register_vgs_harness_profile, vgs_harness_profile
+
+__all__ = [
+    "GraphMemoryBackend",
+    "VFS_TOOL_NAMES",
+    "graph_context_middleware",
+    "graph_memory_tools",
+    "make_graph_subject",
+    "register_vgs_harness_profile",
+    "vgs_harness_profile",
+]

deepagents_graph_memory/_runtime.py

@@ -0,0 +1,67 @@
+"""Load the bundled runtime under Ladybug's canonical native module name."""
+
+from __future__ import annotations
+
+import ctypes
+import sys
+from functools import cache
+from importlib.abc import MetaPathFinder
+from importlib.util import spec_from_file_location
+from pathlib import Path
+from typing import Any
+
+from deepagents_graph_memory.errors import GraphMemoryConfigurationError
+
+_VENDOR = Path(__file__).resolve().parent / "_vendor"
+
+
+class _BundledLadybug(MetaPathFinder):
+    def find_spec(self, fullname: str, path: Any = None, target: Any = None) -> Any:
+        if fullname == "ladybug":
+            return spec_from_file_location(fullname, _VENDOR / "ladybug" / "__init__.py")
+        return None
+
+
+def load_ladybug() -> Any:
+    """Use one module identity, including when standalone Ladybug was imported first."""
+    if not (_VENDOR / "ladybug").is_dir():
+        # Source/editable development uses the explicitly installed test dependency.
+        import ladybug
+
+        return ladybug
+    finder = _BundledLadybug()
+    sys.meta_path.insert(0, finder)
+    try:
+        import ladybug as module
+    finally:
+        sys.meta_path.remove(finder)
+    if module.Database.get_version() != "0.20.3":
+        msg = "Graph memory bundles LadybugDB 0.20.3; another Ladybug version is already loaded. Use a separate process."
+        raise ImportError(msg)
+    return module
+
+
+@cache
+def _preload_windows_fts(path: Path) -> Any:
+    # Ladybug uses LoadLibraryW, which ignores Python's added DLL directories.
+    # ctypes uses LoadLibraryEx with secure search flags; retain its handle.
+    return ctypes.CDLL(str(path))
+
+
+def fts_load_query() -> str:
+    """Load the wheel's FTS binary directly, without using a user extension cache."""
+    name = "libfts.dll" if sys.platform == "win32" else "libfts.so"
+    path = _VENDOR / name
+    if not (_VENDOR / "ladybug").is_dir():
+        return "LOAD fts;"
+    if not path.is_file():
+        msg = "Bundled LadybugDB search extension is missing; reinstall deepagents-graph-memory."
+        raise GraphMemoryConfigurationError(msg)
+    if sys.platform == "win32":
+        try:
+            _preload_windows_fts(path)
+        except OSError as exc:
+            msg = "Bundled LadybugDB search dependencies could not load; reinstall deepagents-graph-memory."
+            raise GraphMemoryConfigurationError(msg) from exc
+    escaped = path.as_posix().replace("\\", "\\\\").replace("'", "\\'")
+    return f"LOAD EXTENSION '{escaped}';"

deepagents_graph_memory/_vendor/__init__.py

@@ -0,0 +1 @@
+"""Unmodified, pinned LadybugDB runtime; populated during wheel builds."""

deepagents_graph_memory/_vendor/ladybug/__init__.py

@@ -0,0 +1,106 @@
+# ruff: noqa
+"""
+# Lbug Python API bindings.
+
+This package provides a Python API for Lbug graph database management system.
+
+To install the package, run:
+```
+python3 -m pip install ladybug
+```
+
+Example usage:
+```python
+import ladybug as lb
+
+db = lb.Database("./test")
+conn = lb.Connection(db)
+
+# Define the schema
+conn.execute("CREATE NODE TABLE User(name STRING, age INT64, PRIMARY KEY (name))")
+conn.execute("CREATE NODE TABLE City(name STRING, population INT64, PRIMARY KEY (name))")
+conn.execute("CREATE REL TABLE Follows(FROM User TO User, since INT64)")
+conn.execute("CREATE REL TABLE LivesIn(FROM User TO City)")
+
+# Load some data
+conn.execute('COPY User FROM "user.csv"')
+conn.execute('COPY City FROM "city.csv"')
+conn.execute('COPY Follows FROM "follows.csv"')
+conn.execute('COPY LivesIn FROM "lives-in.csv"')
+
+# Query the data
+results = conn.execute("MATCH (u:User) RETURN u.name, u.age;")
+while results.has_next():
+    print(results.get_next())
+```
+
+The dataset used in this example can be found [here](https://github.com/LadybugDB/ladybug/tree/master/dataset/demo-db/csv).
+
+"""
+
+from __future__ import annotations
+
+
+# start delvewheel patch
+def _delvewheel_patch_1_13_1():
+    import os
+    if os.path.isdir(libs_dir := os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir, 'ladybug.libs'))):
+        os.add_dll_directory(libs_dir)
+
+
+_delvewheel_patch_1_13_1()
+del _delvewheel_patch_1_13_1
+# end delvewheel patch
+
+from pathlib import Path
+
+# In local dev/test runs the optional pybind extension is built under build/ladybug
+# while the package sources live in src_py. Extend the package path so
+# `from . import _lbug` can discover the built extension without installation.
+_pkg_dir = Path(__file__).resolve().parent
+_repo_build_pkg_dir = _pkg_dir.parent / "build" / "ladybug"
+if _repo_build_pkg_dir.is_dir():
+    __path__.append(str(_repo_build_pkg_dir))
+
+from ._backend import get_capi_module, get_pybind_module  # noqa: E402
+
+from .async_connection import AsyncConnection  # noqa: E402
+from .connection import Connection  # noqa: E402
+from .database import Database  # noqa: E402
+from .prepared_statement import PreparedStatement  # noqa: E402
+from .query_result import ArrowQueryResult, CSRResult, QueryResult  # noqa: E402
+from .types import ArrowRelTableLayout, Type  # noqa: E402
+
+_VERSION_INFO: tuple[str, int] | None = None
+
+
+def _get_version_info() -> tuple[str, int]:
+    global _VERSION_INFO
+    if _VERSION_INFO is None:
+        _VERSION_INFO = (Database.get_version(), Database.get_storage_version())
+    return _VERSION_INFO
+
+
+def __getattr__(name: str) -> str | int:
+    if name == "version" or name == "__version__":
+        return _get_version_info()[0]
+    if name == "storage_version":
+        return _get_version_info()[1]
+    msg = f"module {__name__!r} has no attribute {name!r}"
+    raise AttributeError(msg)
+
+
+__all__ = [
+    "AsyncConnection",
+    "ArrowQueryResult",
+    "ArrowRelTableLayout",
+    "Connection",
+    "CSRResult",
+    "Database",
+    "PreparedStatement",
+    "QueryResult",
+    "Type",
+    "__version__",  # noqa: F822
+    "storage_version",  # noqa: F822
+    "version",  # noqa: F822
+]

deepagents_graph_memory/_vendor/ladybug/_backend.py

@@ -0,0 +1,43 @@
+from __future__ import annotations
+
+import os
+import sys
+from importlib import import_module
+from typing import Any
+
+_CAPI_MODULE: Any | None = None
+_PYBIND_MODULE: Any | None = None
+_PYBIND_IMPORT_ATTEMPTED = False
+
+
+def _import_pybind_module() -> Any:
+    if sys.platform != "linux":
+        return import_module("._lbug", __package__)
+
+    original_dlopen_flags = sys.getdlopenflags()
+    try:
+        # Keep pybind's symbols visible to any transitive native extensions
+        # without affecting the process-wide import path for the C-API backend.
+        sys.setdlopenflags(os.RTLD_GLOBAL | os.RTLD_LAZY)
+        return import_module("._lbug", __package__)
+    finally:
+        sys.setdlopenflags(original_dlopen_flags)
+
+
+def get_capi_module() -> Any:
+    global _CAPI_MODULE
+    if _CAPI_MODULE is None:
+        _CAPI_MODULE = import_module("._lbug_capi", __package__)
+    return _CAPI_MODULE
+
+
+def get_pybind_module() -> Any | None:
+    global _PYBIND_MODULE, _PYBIND_IMPORT_ATTEMPTED
+    if _PYBIND_IMPORT_ATTEMPTED:
+        return _PYBIND_MODULE
+    _PYBIND_IMPORT_ATTEMPTED = True
+    try:
+        _PYBIND_MODULE = _import_pybind_module()
+    except ImportError:
+        _PYBIND_MODULE = None
+    return _PYBIND_MODULE