rushdb 2.6.0__tar.gz → 2.9.0__tar.gz

{rushdb-2.6.0 → rushdb-2.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rushdb
-Version: 2.6.0
+Version: 2.9.0
 Summary: RushDB Python SDK — memory layer for AI agents and modern apps
 Project-URL: Homepage, https://github.com/rush-db/rushdb-python
 Project-URL: Repository, https://github.com/rush-db/rushdb-python
@@ -135,6 +135,28 @@ authored_posts = db.records.find({
     'limit': 10,
 })
 
+# Multi-hop: add hops to $relation — everyone in Alice's reporting chain, up to 4 levels
+chain = db.records.find({
+    'labels': ['EMPLOYEE'],
+    'where': {
+        'EMPLOYEE': {
+            '$relation': {'type': 'REPORTS_TO', 'direction': 'out', 'hops': {'min': 1, 'max': 4}},
+            'name': {'$contains': 'Alice'},
+        }
+    },
+})
+
+# Cycle detection: accounts on a circular transfer ring (fraud rings, circular ownership)
+ring_members = db.records.find({
+    'labels': ['ACCOUNT'],
+    'where': {
+        'RING': {  # key is a display name — the $cycle block holds only $relation
+            '$cycle': True,
+            '$relation': {'type': 'TRANSFERRED_TO', 'direction': 'out', 'hops': {'min': 2, 'max': 6}},
+        }
+    },
+})
+
 # Manage relationships explicitly
 user = db.records.find_uniq({'labels': ['USER'], 'where': {'name': 'Alice'}})
 company = db.records.find_uniq({'labels': ['COMPANY'], 'where': {'name': 'Acme Corp'}})

{rushdb-2.6.0 → rushdb-2.9.0}/README.md

@@ -110,6 +110,28 @@ authored_posts = db.records.find({
     'limit': 10,
 })
 
+# Multi-hop: add hops to $relation — everyone in Alice's reporting chain, up to 4 levels
+chain = db.records.find({
+    'labels': ['EMPLOYEE'],
+    'where': {
+        'EMPLOYEE': {
+            '$relation': {'type': 'REPORTS_TO', 'direction': 'out', 'hops': {'min': 1, 'max': 4}},
+            'name': {'$contains': 'Alice'},
+        }
+    },
+})
+
+# Cycle detection: accounts on a circular transfer ring (fraud rings, circular ownership)
+ring_members = db.records.find({
+    'labels': ['ACCOUNT'],
+    'where': {
+        'RING': {  # key is a display name — the $cycle block holds only $relation
+            '$cycle': True,
+            '$relation': {'type': 'TRANSFERRED_TO', 'direction': 'out', 'hops': {'min': 2, 'max': 6}},
+        }
+    },
+})
+
 # Manage relationships explicitly
 user = db.records.find_uniq({'labels': ['USER'], 'where': {'name': 'Alice'}})
 company = db.records.find_uniq({'labels': ['COMPANY'], 'where': {'name': 'Acme Corp'}})

{rushdb-2.6.0 → rushdb-2.9.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "rushdb"
-version = "2.6.0"
+version = "2.9.0"
 description = "RushDB Python SDK — memory layer for AI agents and modern apps"
 authors = [
     {name = "RushDB Team", email = "hi@rushdb.com"}

{rushdb-2.6.0 → rushdb-2.9.0}/src/rushdb/__init__.py

@@ -24,6 +24,9 @@ from .models.search_query import (
     RelationshipEndpointQuery,
     RelationshipSearchQuery,
     SearchQuery,
+    TraversalHops,
+    TraversalHopsRange,
+    TraversalRelationOptions,
 )
 from .models.transaction import Transaction
 
@@ -46,6 +49,9 @@ __all__ = [
     "SearchQuery",
     "RelationshipEndpointQuery",
     "RelationshipSearchQuery",
+    "TraversalHops",
+    "TraversalHopsRange",
+    "TraversalRelationOptions",
     "QueryAPI",
     "RelationsAPI",
     "RelationshipPatternsAPI",

{rushdb-2.6.0 → rushdb-2.9.0}/src/rushdb/api/transactions.py

@@ -51,7 +51,8 @@ class TransactionsAPI(BaseAPI):
         Args:
             ttl (Optional[int], optional): Time-to-live in milliseconds for the transaction.
                 After this time, the transaction will automatically expire and be rolled back.
-                If None, defaults to 5000ms (5 seconds). Defaults to None.
+                If None, defaults to 5000ms (5 seconds). Values above the server-side cap
+                (60 seconds) are clamped to it. Defaults to None.
 
         Returns:
             Transaction: A Transaction object that can be used with other API operations.

{rushdb-2.6.0 → rushdb-2.9.0}/src/rushdb/client.py

@@ -144,7 +144,9 @@ class RushDB:
             else None
         )
         effective_url = url or base_url or self.DEFAULT_BASE_URL
-        if url and "/api/" not in effective_url:
+        # Both `url` and its deprecated alias `base_url` are documented to get
+        # `/api/v1` appended when the path is absent.
+        if (url or base_url) and "/api/" not in effective_url:
             effective_url = effective_url.rstrip("/") + "/api/v1"
         self.base_url = effective_url.rstrip("/")
         self.api_key = api_key
@@ -245,8 +247,17 @@ class RushDB:
             with urllib.request.urlopen(request) as response:
                 return json.loads(response.read().decode("utf-8"))
         except urllib.error.HTTPError as e:
-            error_body = json.loads(e.read().decode("utf-8"))
-            raise RushDBError(error_body.get("message", str(e)), error_body)
+            # Error bodies are usually JSON, but a gateway/proxy may return HTML or an
+            # empty body — without this guard the JSONDecodeError raised here would
+            # escape as-is instead of surfacing as a RushDBError.
+            try:
+                error_body = json.loads(e.read().decode("utf-8"))
+            except (json.JSONDecodeError, UnicodeDecodeError):
+                error_body = {}
+            message = error_body.get("message")
+            if isinstance(message, list):
+                message = "; ".join(str(item) for item in message)
+            raise RushDBError(message or f"HTTP {e.code}", error_body, status=e.code)
         except urllib.error.URLError as e:
             raise RushDBError(f"Connection error: {str(e)}")
         except json.JSONDecodeError as e:

rushdb-2.9.0/src/rushdb/common.py

@@ -0,0 +1,33 @@
+from typing import Dict, Optional
+
+
+class RushDBError(Exception):
+    """Custom exception for RushDB client errors.
+
+    Attributes:
+        details: Parsed error response body from the server, if any.
+        status: HTTP status code for server errors (e.g. 403 for a write attempt
+            with a read-only API key, 408 for a server-side transaction timeout).
+            ``None`` for client-side errors such as connection failures.
+    """
+
+    def __init__(
+        self,
+        message: str,
+        details: Optional[Dict] = None,
+        status: Optional[int] = None,
+    ):
+        super().__init__(message)
+        self.details = details or {}
+        self.status = status
+
+
+class NonUniqueResultError(Exception):
+    """Raised by ``records.find_uniq()`` when more than one record matches the query."""
+
+    def __init__(self, count: int):
+        super().__init__(
+            f"Expected at most one record but found {count}. "
+            "Refine your search query or use records.find() instead."
+        )
+        self.count = count

{rushdb-2.6.0 → rushdb-2.9.0}/src/rushdb/models/__init__.py

@@ -11,6 +11,9 @@ from .search_query import (
     RelationshipEndpointQuery,
     RelationshipSearchQuery,
     SearchQuery,
+    TraversalHops,
+    TraversalHopsRange,
+    TraversalRelationOptions,
 )
 from .transaction import Transaction
 
@@ -26,5 +29,8 @@ __all__ = [
     "SearchQuery",
     "RelationshipEndpointQuery",
     "RelationshipSearchQuery",
+    "TraversalHops",
+    "TraversalHopsRange",
+    "TraversalRelationOptions",
     "Transaction",
 ]

{rushdb-2.6.0 → rushdb-2.9.0}/src/rushdb/models/result.py

@@ -246,5 +246,10 @@ class SearchResult(Generic[T]):
         return pd.DataFrame(rows)
 
 
-# Type alias for record search results
-RecordSearchResult = SearchResult[Record]
+class RecordSearchResult(SearchResult[Record]):
+    """Search result specialized for ``Record`` items.
+
+    A real subclass (not a ``SearchResult[Record]`` alias) so that
+    ``isinstance(result, RecordSearchResult)`` works — isinstance checks against
+    subscripted generics raise ``TypeError`` at runtime.
+    """

rushdb-2.9.0/src/rushdb/models/search_query.py

@@ -0,0 +1,93 @@
+from enum import Enum
+from typing import Any, Dict, List, Optional, TypedDict, Union
+
+
+class OrderDirection(str, Enum):
+    ASC = "asc"
+    DESC = "desc"
+
+
+class TraversalHopsRange(TypedDict, total=False):
+    """Hop range for variable-length traversal.
+
+    ``min`` defaults to 1 (2 inside a ``$cycle`` block). Omitting ``max``
+    requests unbounded traversal, which is only allowed on self-hosted
+    deployments and projects with a custom Neo4j instance; the shared cloud
+    connection caps ``max`` per deployment (default 25).
+    """
+
+    min: int
+    max: int
+
+
+TraversalHops = Union[int, TraversalHopsRange]
+"""Depth of a variable-length traversal: an exact hop count or a range."""
+
+
+class TraversalRelationOptions(TypedDict, total=False):
+    """Value for the ``$relation`` key of a traversal block in ``where``.
+
+    ``type`` and ``direction`` constrain every hop; ``hops`` makes the
+    traversal variable-length. The nested label constrains only the final
+    record — intermediate records are anonymous.
+
+    Example::
+
+        {
+            "labels": ["EMPLOYEE"],
+            "where": {
+                "EMPLOYEE": {
+                    "$alias": "$manager",
+                    "$relation": {
+                        "type": "REPORTS_TO",
+                        "direction": "out",
+                        "hops": {"min": 1, "max": 4},
+                    },
+                    "name": {"$contains": "Alice"},
+                }
+            },
+        }
+    """
+
+    type: str
+    direction: str  # 'in' | 'out'
+    hops: TraversalHops
+
+
+class SearchQuery(TypedDict, total=False):
+    """TypedDict representing the query structure for finding records.
+
+    Inside ``where``, a nested key that is a label name traverses to related
+    records. Traversal blocks accept ``$alias`` (name the endpoint for
+    ``select``/``groupBy``), ``$relation`` (a type string or
+    :class:`TraversalRelationOptions`, including variable-length ``hops``),
+    and ``$cycle`` (``True`` binds the traversal back to the parent record to
+    detect rings; requires ``$relation`` with ``hops`` where ``min`` >= 2 and
+    accepts no other keys).
+    """
+
+    where: Optional[Dict[str, Any]]
+    labels: Optional[List[str]]
+    skip: Optional[int]
+    limit: Optional[int]
+    orderBy: Optional[Union[Dict[str, OrderDirection], OrderDirection]]
+    select: Optional[Dict[str, Any]]
+    groupBy: Optional[List[str]]
+
+
+class RelationshipEndpointQuery(TypedDict, total=False):
+    """Endpoint record predicates for relationship search."""
+
+    where: Optional[Dict[str, Any]]
+    labels: Optional[List[str]]
+
+
+class RelationshipSearchQuery(TypedDict, total=False):
+    """TypedDict representing the query structure for finding relationship edges."""
+
+    where: Optional[Dict[str, Any]]
+    source: Optional[RelationshipEndpointQuery]
+    target: Optional[RelationshipEndpointQuery]
+    skip: Optional[int]
+    limit: Optional[int]
+    orderBy: Optional[Union[Dict[str, OrderDirection], OrderDirection]]

{rushdb-2.6.0 → rushdb-2.9.0}/tests/test_create_import.py

@@ -40,7 +40,7 @@ class TestCreateImport(TestBase):
         # Test new functionality
         self.assertEqual(record.get("name"), "Google LLC")
         self.assertEqual(record.get("nonexistent", "default"), "default")
-        self.assertTrue(record.exists())
+        self.assertTrue(record.exists)
 
     def test_record_methods(self):
         """Test Record class methods"""
@@ -201,7 +201,11 @@ class TestCreateImport(TestBase):
         )
 
     def test_create_with_nested_data(self):
-        """Test creating records with nested data structure"""
+        """Test importing records with nested data structure.
+
+        ``create_many`` accepts flat rows only (it raises ``ValueError`` for
+        nested payloads); nested JSON goes through ``import_json``.
+        """
         data = {
             "name": "Meta Platforms Inc",
             "rating": 4.6,
@@ -220,7 +224,9 @@ class TestCreateImport(TestBase):
                 }
             ],
         }
-        self.client.records.create_many("COMPANY", data)
+        with self.assertRaises(ValueError):
+            self.client.records.create_many("COMPANY", data)
+        self.client.records.import_json(data=data, label="COMPANY")
 
     def test_transaction_rollback(self):
         """Test transaction rollback"""

{rushdb-2.6.0 → rushdb-2.9.0}/tests/test_search_query.py

@@ -1,6 +1,10 @@
 """Test cases for RushDB search query functionality."""
 
+import time
 import unittest
+import uuid
+
+from src.rushdb import RushDB, RushDBError
 
 from .test_base_setup import TestBase
 
@@ -389,5 +393,225 @@ class TestSearchQuery(TestBase):
         self.client.records.find(query)
 
 
+class TestMultihopAndCycles(TestBase):
+    """Variable-length traversal ($relation.hops) and cycle detection ($cycle).
+
+    Seeds its own graph, isolated by a unique tenantId:
+
+        Reporting chain (MHEmployee, REPORTS_TO, directed "up"):
+            E1 -> E2 -> E3 -> E4
+
+        Transfer ring + linear chain (MHAccount, TRANSFERRED_TO):
+            A -> B -> C -> A        (3-hop directed ring)
+            X -> Y -> Z             (no cycle)
+    """
+
+    tenant: str
+
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        cls.tenant = f"multihop-{uuid.uuid4().hex[:8]}"
+        # TestBase only creates a client per-test (setUp); seeding needs one here.
+        cls.client = RushDB(cls.token, base_url=cls.base_url)
+
+        # TestBase skips per-test in setUp when the server is unreachable; this
+        # class seeds data in setUpClass, so it must skip here for the same
+        # reason (e.g. CI without a running RushDB) instead of erroring.
+        try:
+            if not cls.client.ping():
+                raise unittest.SkipTest(
+                    f"Could not connect to RushDB at {cls.base_url}"
+                )
+        except RushDBError as e:
+            raise unittest.SkipTest(f"RushDB connection error: {str(e)}")
+
+        cls.client.records.create_many(
+            "MHEmployee",
+            [
+                {"name": "E1", "managerName": "E2", "tenantId": cls.tenant},
+                {"name": "E2", "managerName": "E3", "tenantId": cls.tenant},
+                {"name": "E3", "managerName": "E4", "tenantId": cls.tenant},
+                {"name": "E4", "tenantId": cls.tenant},
+            ],
+        )
+        cls.client.relationships.create_many(
+            source={
+                "label": "MHEmployee",
+                "key": "managerName",
+                "where": {"tenantId": cls.tenant},
+            },
+            target={
+                "label": "MHEmployee",
+                "key": "name",
+                "where": {"tenantId": cls.tenant},
+            },
+            type="REPORTS_TO",
+            direction="out",
+        )
+
+        cls.client.records.create_many(
+            "MHAccount",
+            [
+                {"name": "A", "sendsTo": "B", "tenantId": cls.tenant},
+                {"name": "B", "sendsTo": "C", "tenantId": cls.tenant},
+                {"name": "C", "sendsTo": "A", "tenantId": cls.tenant},
+                {"name": "X", "sendsTo": "Y", "tenantId": cls.tenant},
+                {"name": "Y", "sendsTo": "Z", "tenantId": cls.tenant},
+                {"name": "Z", "tenantId": cls.tenant},
+            ],
+        )
+        cls.client.relationships.create_many(
+            source={
+                "label": "MHAccount",
+                "key": "sendsTo",
+                "where": {"tenantId": cls.tenant},
+            },
+            target={
+                "label": "MHAccount",
+                "key": "name",
+                "where": {"tenantId": cls.tenant},
+            },
+            type="TRANSFERRED_TO",
+            direction="out",  # sender -> receiver
+        )
+
+        # relationships.create_many is applied via apoc.periodic.iterate — poll
+        # until both relationship sets are visible.
+        for _ in range(10):
+            employee_rels = cls.client.relationships.find(
+                {
+                    "source": {
+                        "labels": ["MHEmployee"],
+                        "where": {"tenantId": cls.tenant},
+                    },
+                    "limit": 100,
+                }
+            )
+            account_rels = cls.client.relationships.find(
+                {
+                    "source": {
+                        "labels": ["MHAccount"],
+                        "where": {"tenantId": cls.tenant},
+                    },
+                    "limit": 100,
+                }
+            )
+            if len(employee_rels.data) >= 3 and len(account_rels.data) >= 5:
+                break
+            time.sleep(1)
+
+    @classmethod
+    def tearDownClass(cls):
+        cls.client.records.delete({"where": {"tenantId": cls.tenant}})
+        super().tearDownClass()
+
+    def _names(self, result):
+        return sorted(record.get("name") for record in result)
+
+    def test_hops_range_reaches_chain_top(self):
+        """hops {max: 3} reaches E4 from every subordinate"""
+        result = self.client.records.find(
+            {
+                "labels": ["MHEmployee"],
+                "where": {
+                    "tenantId": self.tenant,
+                    "MHEmployee": {
+                        "$relation": {
+                            "type": "REPORTS_TO",
+                            "direction": "out",
+                            "hops": {"max": 3},
+                        },
+                        "name": "E4",
+                    },
+                },
+            }
+        )
+        self.assertEqual(self._names(result), ["E1", "E2", "E3"])
+
+    def test_hops_range_is_bounded(self):
+        """hops {max: 2} does not reach 3 hops away"""
+        result = self.client.records.find(
+            {
+                "labels": ["MHEmployee"],
+                "where": {
+                    "tenantId": self.tenant,
+                    "MHEmployee": {
+                        "$relation": {
+                            "type": "REPORTS_TO",
+                            "direction": "out",
+                            "hops": {"max": 2},
+                        },
+                        "name": "E4",
+                    },
+                },
+            }
+        )
+        self.assertEqual(self._names(result), ["E2", "E3"])
+
+    def test_hops_exact_count(self):
+        """hops: 3 matches exactly 3 hops"""
+        result = self.client.records.find(
+            {
+                "labels": ["MHEmployee"],
+                "where": {
+                    "tenantId": self.tenant,
+                    "MHEmployee": {
+                        "$relation": {
+                            "type": "REPORTS_TO",
+                            "direction": "out",
+                            "hops": 3,
+                        },
+                        "name": "E4",
+                    },
+                },
+            }
+        )
+        self.assertEqual(self._names(result), ["E1"])
+
+    def test_cycle_flags_ring_members(self):
+        """$cycle returns exactly the ring participants with a deduplicated total"""
+        result = self.client.records.find(
+            {
+                "labels": ["MHAccount"],
+                "where": {
+                    "tenantId": self.tenant,
+                    "RING": {
+                        "$cycle": True,
+                        "$relation": {
+                            "type": "TRANSFERRED_TO",
+                            "direction": "out",
+                            "hops": {"min": 2, "max": 6},
+                        },
+                    },
+                },
+            }
+        )
+        self.assertEqual(self._names(result), ["A", "B", "C"])
+        self.assertEqual(result.total, 3)
+
+    def test_not_cycle_excludes_ring_members(self):
+        """$not around a $cycle block finds acyclic accounts"""
+        result = self.client.records.find(
+            {
+                "labels": ["MHAccount"],
+                "where": {
+                    "tenantId": self.tenant,
+                    "$not": {
+                        "RING": {
+                            "$cycle": True,
+                            "$relation": {
+                                "type": "TRANSFERRED_TO",
+                                "direction": "out",
+                                "hops": {"min": 2, "max": 6},
+                            },
+                        }
+                    },
+                },
+            }
+        )
+        self.assertEqual(self._names(result), ["X", "Y", "Z"])
+
+
 if __name__ == "__main__":
     unittest.main()

{rushdb-2.6.0 → rushdb-2.9.0}/uv.lock

@@ -1,5 +1,5 @@
 version = 1
-revision = 3
+revision = 2
 requires-python = ">=3.8"
 resolution-markers = [
     "python_full_version >= '3.10'",
@@ -15,13 +15,13 @@ resolution-markers = [
     "python_full_version < '3.9'",
 ]
 dependencies = [
-    { name = "click", version = "8.1.8", source = { registry = "https://pypi.org/simple" } },
-    { name = "mypy-extensions" },
-    { name = "packaging" },
-    { name = "pathspec" },
-    { name = "platformdirs", version = "4.3.6", source = { registry = "https://pypi.org/simple" } },
-    { name = "tomli" },
-    { name = "typing-extensions", version = "4.13.2", source = { registry = "https://pypi.org/simple" } },
+    { name = "click", version = "8.1.8", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.9'" },
+    { name = "mypy-extensions", marker = "python_full_version < '3.9'" },
+    { name = "packaging", marker = "python_full_version < '3.9'" },
+    { name = "pathspec", marker = "python_full_version < '3.9'" },
+    { name = "platformdirs", version = "4.3.6", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.9'" },
+    { name = "tomli", marker = "python_full_version < '3.9'" },
+    { name = "typing-extensions", version = "4.13.2", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.9'" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/04/b0/46fb0d4e00372f4a86a6f8efa3cb193c9f64863615e39010b1477e010578/black-24.8.0.tar.gz", hash = "sha256:2500945420b6784c38b9ee885af039f5e7471ef284ab03fa35ecdde4688cd83f", size = 644810, upload-time = "2024-08-02T17:43:18.405Z" }
 wheels = [
@@ -57,14 +57,14 @@ resolution-markers = [
     "python_full_version == '3.9.*'",
 ]
 dependencies = [
-    { name = "click", version = "8.1.8", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.10'" },
+    { name = "click", version = "8.1.8", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version == '3.9.*'" },
     { name = "click", version = "8.2.1", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.10'" },
-    { name = "mypy-extensions" },
-    { name = "packaging" },
-    { name = "pathspec" },
-    { name = "platformdirs", version = "4.3.8", source = { registry = "https://pypi.org/simple" } },
-    { name = "tomli", marker = "python_full_version < '3.11'" },
-    { name = "typing-extensions", version = "4.14.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.11'" },
+    { name = "mypy-extensions", marker = "python_full_version >= '3.9'" },
+    { name = "packaging", marker = "python_full_version >= '3.9'" },
+    { name = "pathspec", marker = "python_full_version >= '3.9'" },
+    { name = "platformdirs", version = "4.3.8", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.9'" },
+    { name = "tomli", marker = "python_full_version >= '3.9' and python_full_version < '3.11'" },
+    { name = "typing-extensions", version = "4.14.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.9' and python_full_version < '3.11'" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/94/49/26a7b0f3f35da4b5a65f081943b7bcd22d7002f5f0fb8098ec1ff21cb6ef/black-25.1.0.tar.gz", hash = "sha256:33496d5cd1222ad73391352b4ae8da15253c5de89b93a80b3e2c8d9a19ec2666", size = 649449, upload-time = "2025-01-29T04:15:40.373Z" }
 wheels = [
@@ -196,7 +196,7 @@ resolution-markers = [
     "python_full_version < '3.9'",
 ]
 dependencies = [
-    { name = "colorama", marker = "sys_platform == 'win32'" },
+    { name = "colorama", marker = "python_full_version < '3.10' and sys_platform == 'win32'" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/b9/2e/0090cbf739cee7d23781ad4b89a9894a41538e4fcf4c31dcdd705b78eb8b/click-8.1.8.tar.gz", hash = "sha256:ed53c9d8990d83c2a27deae68e4ee337473f6330c040a31d4225c9574d16096a", size = 226593, upload-time = "2024-12-21T18:38:44.339Z" }
 wheels = [
@@ -211,7 +211,7 @@ resolution-markers = [
     "python_full_version >= '3.10'",
 ]
 dependencies = [
-    { name = "colorama", marker = "sys_platform == 'win32'" },
+    { name = "colorama", marker = "python_full_version >= '3.10' and sys_platform == 'win32'" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/60/6c/8ca2efa64cf75a977a0d7fac081354553ebe483345c734fb6b6515d96bbc/click-8.2.1.tar.gz", hash = "sha256:27c491cc05d968d271d5a1db13e3b5a184636d9d930f148c50b038f0d0646202", size = 286342, upload-time = "2025-05-20T23:19:49.832Z" }
 wheels = [
@@ -311,7 +311,7 @@ wheels = [
 
 [package.optional-dependencies]
 toml = [
-    { name = "tomli" },
+    { name = "tomli", marker = "python_full_version < '3.9'" },
 ]
 
 [[package]]
@@ -394,7 +394,7 @@ wheels = [
 
 [package.optional-dependencies]
 toml = [
-    { name = "tomli", marker = "python_full_version <= '3.11'" },
+    { name = "tomli", marker = "python_full_version >= '3.9' and python_full_version <= '3.11'" },
 ]
 
 [[package]]
@@ -402,8 +402,8 @@ name = "exceptiongroup"
 version = "1.3.0"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
-    { name = "typing-extensions", version = "4.13.2", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.9' or python_full_version >= '3.11'" },
-    { name = "typing-extensions", version = "4.14.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.9' and python_full_version < '3.11'" },
+    { name = "typing-extensions", version = "4.13.2", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.9'" },
+    { name = "typing-extensions", version = "4.14.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.9' and python_full_version < '3.13'" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/0b/9f/a65090624ecf468cdca03533906e7c69ed7588582240cfe7cc9e770b50eb/exceptiongroup-1.3.0.tar.gz", hash = "sha256:b241f5885f560bc56a59ee63ca4c6a8bfa46ae4ad651af316d4e81817bb9fd88", size = 29749, upload-time = "2025-05-10T17:42:51.123Z" }
 wheels = [
@@ -461,9 +461,9 @@ resolution-markers = [
     "python_full_version < '3.9'",
 ]
 dependencies = [
-    { name = "mypy-extensions" },
-    { name = "tomli" },
-    { name = "typing-extensions", version = "4.13.2", source = { registry = "https://pypi.org/simple" } },
+    { name = "mypy-extensions", marker = "python_full_version < '3.9'" },
+    { name = "tomli", marker = "python_full_version < '3.9'" },
+    { name = "typing-extensions", version = "4.13.2", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.9'" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/b9/eb/2c92d8ea1e684440f54fa49ac5d9a5f19967b7b472a281f419e69a8d228e/mypy-1.14.1.tar.gz", hash = "sha256:7ec88144fe9b510e8475ec2f5f251992690fcf89ccb4500b214b4226abcd32d6", size = 3216051, upload-time = "2024-12-30T16:39:07.335Z" }
 wheels = [
@@ -515,10 +515,10 @@ resolution-markers = [
     "python_full_version == '3.9.*'",
 ]
 dependencies = [
-    { name = "mypy-extensions" },
-    { name = "pathspec" },
-    { name = "tomli", marker = "python_full_version < '3.11'" },
-    { name = "typing-extensions", version = "4.14.0", source = { registry = "https://pypi.org/simple" } },
+    { name = "mypy-extensions", marker = "python_full_version >= '3.9'" },
+    { name = "pathspec", marker = "python_full_version >= '3.9'" },
+    { name = "tomli", marker = "python_full_version >= '3.9' and python_full_version < '3.11'" },
+    { name = "typing-extensions", version = "4.14.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.9'" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/81/69/92c7fa98112e4d9eb075a239caa4ef4649ad7d441545ccffbd5e34607cbb/mypy-1.16.1.tar.gz", hash = "sha256:6bd00a0a2094841c5e47e7374bb42b83d64c527a502e3334e1173a0c24437bab", size = 3324747, upload-time = "2025-06-16T16:51:35.145Z" }
 wheels = [
@@ -649,12 +649,12 @@ resolution-markers = [
     "python_full_version < '3.9'",
 ]
 dependencies = [
-    { name = "colorama", marker = "sys_platform == 'win32'" },
-    { name = "exceptiongroup" },
-    { name = "iniconfig" },
-    { name = "packaging" },
-    { name = "pluggy", version = "1.5.0", source = { registry = "https://pypi.org/simple" } },
-    { name = "tomli" },
+    { name = "colorama", marker = "python_full_version < '3.9' and sys_platform == 'win32'" },
+    { name = "exceptiongroup", marker = "python_full_version < '3.9'" },
+    { name = "iniconfig", marker = "python_full_version < '3.9'" },
+    { name = "packaging", marker = "python_full_version < '3.9'" },
+    { name = "pluggy", version = "1.5.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.9'" },
+    { name = "tomli", marker = "python_full_version < '3.9'" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/ae/3c/c9d525a414d506893f0cd8a8d0de7706446213181570cdbd766691164e40/pytest-8.3.5.tar.gz", hash = "sha256:f4efe70cc14e511565ac476b57c279e12a855b11f48f212af1080ef2263d3845", size = 1450891, upload-time = "2025-03-02T12:54:54.503Z" }
 wheels = [
@@ -670,13 +670,13 @@ resolution-markers = [
     "python_full_version == '3.9.*'",
 ]
 dependencies = [
-    { name = "colorama", marker = "sys_platform == 'win32'" },
-    { name = "exceptiongroup", marker = "python_full_version < '3.11'" },
-    { name = "iniconfig" },
-    { name = "packaging" },
-    { name = "pluggy", version = "1.6.0", source = { registry = "https://pypi.org/simple" } },
-    { name = "pygments" },
-    { name = "tomli", marker = "python_full_version < '3.11'" },
+    { name = "colorama", marker = "python_full_version >= '3.9' and sys_platform == 'win32'" },
+    { name = "exceptiongroup", marker = "python_full_version >= '3.9' and python_full_version < '3.11'" },
+    { name = "iniconfig", marker = "python_full_version >= '3.9'" },
+    { name = "packaging", marker = "python_full_version >= '3.9'" },
+    { name = "pluggy", version = "1.6.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.9'" },
+    { name = "pygments", marker = "python_full_version >= '3.9'" },
+    { name = "tomli", marker = "python_full_version >= '3.9' and python_full_version < '3.11'" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/08/ba/45911d754e8eba3d5a841a5ce61a65a685ff1798421ac054f85aa8747dfb/pytest-8.4.1.tar.gz", hash = "sha256:7c67fd69174877359ed9371ec3af8a3d2b04741818c51e5e99cc1742251fa93c", size = 1517714, upload-time = "2025-06-18T05:48:06.109Z" }
 wheels = [
@@ -691,8 +691,8 @@ resolution-markers = [
     "python_full_version < '3.9'",
 ]
 dependencies = [
-    { name = "coverage", version = "7.6.1", source = { registry = "https://pypi.org/simple" }, extra = ["toml"] },
-    { name = "pytest", version = "8.3.5", source = { registry = "https://pypi.org/simple" } },
+    { name = "coverage", version = "7.6.1", source = { registry = "https://pypi.org/simple" }, extra = ["toml"], marker = "python_full_version < '3.9'" },
+    { name = "pytest", version = "8.3.5", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.9'" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/74/67/00efc8d11b630c56f15f4ad9c7f9223f1e5ec275aaae3fa9118c6a223ad2/pytest-cov-5.0.0.tar.gz", hash = "sha256:5837b58e9f6ebd335b0f8060eecce69b662415b16dc503883a02f45dfeb14857", size = 63042, upload-time = "2024-03-24T20:16:34.856Z" }
 wheels = [
@@ -708,9 +708,9 @@ resolution-markers = [
     "python_full_version == '3.9.*'",
 ]
 dependencies = [
-    { name = "coverage", version = "7.9.1", source = { registry = "https://pypi.org/simple" }, extra = ["toml"] },
-    { name = "pluggy", version = "1.6.0", source = { registry = "https://pypi.org/simple" } },
-    { name = "pytest", version = "8.4.1", source = { registry = "https://pypi.org/simple" } },
+    { name = "coverage", version = "7.9.1", source = { registry = "https://pypi.org/simple" }, extra = ["toml"], marker = "python_full_version >= '3.9'" },
+    { name = "pluggy", version = "1.6.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.9'" },
+    { name = "pytest", version = "8.4.1", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.9'" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/18/99/668cade231f434aaa59bbfbf49469068d2ddd945000621d3d165d2e7dd7b/pytest_cov-6.2.1.tar.gz", hash = "sha256:25cc6cc0a5358204b8108ecedc51a9b57b34cc6b8c967cc2c01a4e00d8a67da2", size = 69432, upload-time = "2025-06-12T10:47:47.684Z" }
 wheels = [
@@ -785,7 +785,7 @@ wheels = [
 
 [[package]]
 name = "rushdb"
-version = "2.6.0"
+version = "2.9.0"
 source = { editable = "." }
 dependencies = [
     { name = "python-dotenv", version = "1.0.1", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.9'" },
@@ -930,7 +930,7 @@ resolution-markers = [
     "python_full_version < '3.9'",
 ]
 dependencies = [
-    { name = "urllib3", version = "2.2.3", source = { registry = "https://pypi.org/simple" } },
+    { name = "urllib3", version = "2.2.3", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.9'" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/fa/3c/4f2a430c01a22abd49a583b6b944173e39e7d01b688190a5618bd59a2e22/types-requests-2.32.0.20241016.tar.gz", hash = "sha256:0d9cad2f27515d0e3e3da7134a1b6f28fb97129d86b867f24d9c726452634d95", size = 18065, upload-time = "2024-10-16T02:46:10.818Z" }
 wheels = [
@@ -946,7 +946,7 @@ resolution-markers = [
     "python_full_version == '3.9.*'",
 ]
 dependencies = [
-    { name = "urllib3", version = "2.5.0", source = { registry = "https://pypi.org/simple" } },
+    { name = "urllib3", version = "2.5.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.9'" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/6d/7f/73b3a04a53b0fd2a911d4ec517940ecd6600630b559e4505cc7b68beb5a0/types_requests-2.32.4.20250611.tar.gz", hash = "sha256:741c8777ed6425830bf51e54d6abe245f79b4dcb9019f1622b773463946bf826", size = 23118, upload-time = "2025-06-11T03:11:41.272Z" }
 wheels = [

rushdb-2.6.0/src/rushdb/common.py

@@ -1,20 +0,0 @@
-from typing import Dict, Optional
-
-
-class RushDBError(Exception):
-    """Custom exception for RushDB client errors."""
-
-    def __init__(self, message: str, details: Optional[Dict] = None):
-        super().__init__(message)
-        self.details = details or {}
-
-
-class NonUniqueResultError(Exception):
-    """Raised by ``records.find_uniq()`` when more than one record matches the query."""
-
-    def __init__(self, count: int):
-        super().__init__(
-            f"Expected at most one record but found {count}. "
-            "Refine your search query or use records.find() instead."
-        )
-        self.count = count

rushdb-2.6.0/src/rushdb/models/search_query.py

@@ -1,37 +0,0 @@
-from enum import Enum
-from typing import Any, Dict, List, Optional, TypedDict, Union
-
-
-class OrderDirection(str, Enum):
-    ASC = "asc"
-    DESC = "desc"
-
-
-class SearchQuery(TypedDict, total=False):
-    """TypedDict representing the query structure for finding records."""
-
-    where: Optional[Dict[str, Any]]
-    labels: Optional[List[str]]
-    skip: Optional[int]
-    limit: Optional[int]
-    orderBy: Optional[Union[Dict[str, OrderDirection], OrderDirection]]
-    select: Optional[Dict[str, Any]]
-    groupBy: Optional[List[str]]
-
-
-class RelationshipEndpointQuery(TypedDict, total=False):
-    """Endpoint record predicates for relationship search."""
-
-    where: Optional[Dict[str, Any]]
-    labels: Optional[List[str]]
-
-
-class RelationshipSearchQuery(TypedDict, total=False):
-    """TypedDict representing the query structure for finding relationship edges."""
-
-    where: Optional[Dict[str, Any]]
-    source: Optional[RelationshipEndpointQuery]
-    target: Optional[RelationshipEndpointQuery]
-    skip: Optional[int]
-    limit: Optional[int]
-    orderBy: Optional[Union[Dict[str, OrderDirection], OrderDirection]]