rushdb 3.1.0__tar.gz → 3.2.0__tar.gz

{rushdb-3.1.0 → rushdb-3.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: rushdb
-Version: 3.1.0
+Version: 3.2.0
 Summary: RushDB SDK for Python: the client for RushDB, the database for AI (RushDB Cloud or self-hosted)
 Project-URL: Homepage, https://github.com/rush-db/rushdb-python
 Project-URL: Repository, https://github.com/rush-db/rushdb-python
@@ -93,7 +93,7 @@ What each offers:
 | Feature | Cloud | Self-hosted engine |
 | --- | :---: | :---: |
 | Records: create, read, update, set, upsert, delete, `find` with `where`, `orderBy`, `select`, `groupBy` | ✓ | ✓ |
-| Relationships: `attach`/`detach`, `relationships.create_many`/`delete_many`/`find` | ✓ | ✓ |
+| Relationships: `attach`/`detach`, `relationships.create_many`/`delete_many`/`find`/`update` | ✓ | ✓ |
 | Labels, properties, property values | ✓ | ✓ |
 | Import JSON and CSV, export CSV | ✓ | ✓ |
 | External vector indexes; vector search with a `queryVector` you supply | ✓ | ✓ |
@@ -227,6 +227,13 @@ relationships = db.relationships.find({
     'target': {'labels': ['COMPANY']},
     'where': {'type': 'WORKS_AT', 'source': 'profile'},
 })
+
+# Change properties on existing relationships (RushDB 3.0.4+): set writes, unset removes
+db.relationships.update(user, company, type='WORKS_AT', set={'source': 'hr'}, unset=['legacy'])
+# types marks values JSON can't type: 'datetime' stores an ISO string as a datetime
+db.relationships.update(
+    user, company, type='WORKS_AT', set={'since': '2024-03-01'}, types={'since': 'datetime'}
+)
 ```
 
 ---

{rushdb-3.1.0 → rushdb-3.2.0}/README.md

@@ -79,7 +79,7 @@ What each offers:
 | Feature | Cloud | Self-hosted engine |
 | --- | :---: | :---: |
 | Records: create, read, update, set, upsert, delete, `find` with `where`, `orderBy`, `select`, `groupBy` | ✓ | ✓ |
-| Relationships: `attach`/`detach`, `relationships.create_many`/`delete_many`/`find` | ✓ | ✓ |
+| Relationships: `attach`/`detach`, `relationships.create_many`/`delete_many`/`find`/`update` | ✓ | ✓ |
 | Labels, properties, property values | ✓ | ✓ |
 | Import JSON and CSV, export CSV | ✓ | ✓ |
 | External vector indexes; vector search with a `queryVector` you supply | ✓ | ✓ |
@@ -213,6 +213,13 @@ relationships = db.relationships.find({
     'target': {'labels': ['COMPANY']},
     'where': {'type': 'WORKS_AT', 'source': 'profile'},
 })
+
+# Change properties on existing relationships (RushDB 3.0.4+): set writes, unset removes
+db.relationships.update(user, company, type='WORKS_AT', set={'source': 'hr'}, unset=['legacy'])
+# types marks values JSON can't type: 'datetime' stores an ISO string as a datetime
+db.relationships.update(
+    user, company, type='WORKS_AT', set={'since': '2024-03-01'}, types={'since': 'datetime'}
+)
 ```
 
 ---

{rushdb-3.1.0 → rushdb-3.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "rushdb"
-version = "3.1.0"
+version = "3.2.0"
 description = "RushDB SDK for Python: the client for RushDB, the database for AI (RushDB Cloud or self-hosted)"
 authors = [
     {name = "RushDB Team", email = "hi@rushdb.com"}

{rushdb-3.1.0 → rushdb-3.2.0}/src/rushdb/api/relationships.py

@@ -10,6 +10,7 @@ from .relationship_patterns import RelationshipPatternsAPI
 
 if TYPE_CHECKING:
     from ..client import RushDB
+    from ..models.record import RecordTarget, RelationTarget
 
 
 class PaginationParams(TypedDict, total=False):
@@ -181,6 +182,61 @@ class RelationsAPI(BaseAPI):
             "POST", "/relationships/create-many", payload, headers
         )
 
+    def update(
+        self,
+        source: "RecordTarget",
+        target: "RelationTarget",
+        *,
+        type: str,
+        direction: Optional[str] = None,
+        set: Optional[dict] = None,
+        unset: Optional[typing.List[str]] = None,
+        types: Optional[typing.Dict[str, str]] = None,
+    ) -> dict:
+        """Update the properties of existing relationships of one type.
+
+        Changes the ``type`` relationships from ``source`` to each ``target`` (or the
+        reverse with ``direction='in'``). Pairs without such a relationship are skipped.
+        Not available inside a transaction. Needs RushDB 3.0.4 or newer.
+
+        Args:
+            source: The source record (id, ``Record`` or dict with ``__id``).
+            target: The target record(s).
+            type (str): The relationship type to update.
+            direction (str, optional): ``'out'`` (default) or ``'in'``.
+            set (dict, optional): Properties to write; ``None`` values store an explicit null.
+            unset (list[str], optional): Names of properties to remove.
+            types (dict, optional): Type of ``set`` properties JSON can't express, one of
+                ``'string'``, ``'number'``, ``'boolean'``, ``'datetime'``;
+                ``{"since": "datetime"}`` stores an ISO string as a datetime.
+
+        Returns:
+            dict: API response; ``data["updated"]`` counts the relationships changed.
+
+        Example:
+            >>> db.relationships.update(
+            ...     customer, order, type="PLACES", set={"channel": "store"}, unset=["legacyCode"]
+            ... )
+        """
+        from .records import RecordsAPI
+
+        source_id = RecordsAPI._extract_target_ids(source)[0]
+        payload: dict = {
+            "targetIds": RecordsAPI._extract_target_ids(target),
+            "type": type,
+        }
+        if direction:
+            payload["direction"] = direction
+        if set is not None:
+            payload["set"] = set
+        if unset is not None:
+            payload["unset"] = unset
+        if types:
+            payload["types"] = types
+        return self.client._make_request(
+            "PATCH", f"/relationships/{source_id}", payload
+        )
+
     def delete_many(
         self,
         *,

{rushdb-3.1.0 → rushdb-3.2.0}/src/rushdb/direct/router.py

@@ -753,6 +753,39 @@ def _detach(request: _Request) -> Body:
     )
 
 
+def _update_relationships(request: _Request) -> Body:
+    body, (source_id,) = request.body, request.params
+    _not_in_transaction(request.tx_id, "relationships.update")
+    result = request.engine.execute(
+        {
+            "kind": "mutate",
+            "mutation": {
+                "kind": "updateRelationshipsById",
+                "command": {
+                    "relationType": body.get("type"),
+                    "direction": (
+                        "in" if _direction(body.get("direction")) == "in" else "out"
+                    ),
+                    "sourceIds": [source_id],
+                    "targetIds": _as_list(body.get("targetIds")),
+                    "set": body.get("set") or {},
+                    "unset": body.get("unset") or [],
+                    # Sent only when used, so untyped updates work on engines without it.
+                    **({"types": body["types"]} if body.get("types") else {}),
+                },
+            },
+        }
+    )
+    updated = int((result or {}).get("affected") or 0)
+    plural = "" if updated == 1 else "s"
+    return _ok(
+        {
+            "message": f"{updated} relationship{plural} of Record {source_id} updated",
+            "updated": updated,
+        }
+    )
+
+
 def _link_many(kind: str, message: str, what: str) -> Callable[[_Request], Body]:
     def handle(request: _Request) -> Body:
         _not_in_transaction(request.tx_id, what)
@@ -1283,6 +1316,7 @@ _ROUTES: List[Tuple[str, "re.Pattern[str]", Callable[[_Request], Body]]] = [
     ("POST", re.compile(r"^/records/export/csv$"), _export_csv),
     ("POST", re.compile(rf"^/relationships/{_ID}$"), _attach),
     ("PUT", re.compile(rf"^/relationships/{_ID}$"), _detach),
+    ("PATCH", re.compile(rf"^/relationships/{_ID}$"), _update_relationships),
     (
         "POST",
         re.compile(r"^/relationships/create-many$"),

{rushdb-3.1.0 → rushdb-3.2.0}/tests/test_direct_mode.py

@@ -191,6 +191,9 @@ EVERY_METHOD = {
     "records.detach": lambda db: db.records.detach(
         RECORD["id"], RECORD["id"], options={"typeOrTypes": "LIKES"}
     ),
+    "relationships.update": lambda db: db.relationships.update(
+        RECORD["id"], RECORD["id"], type="LIKES", set={"weight": 1}
+    ),
     "records.create": lambda db: db.records.create("Book", {"title": "Dune"}),
     "records.create_many": lambda db: db.records.create_many(
         "Book", [{"title": "Dune"}]
@@ -531,6 +534,39 @@ def test_detach_without_type_or_direction_removes_every_link(engine_calls):
     ]
 
 
+def test_relationship_properties_are_updated_by_record_ids(engine_calls):
+    result = direct_db().relationships.update(
+        RECORD["id"],
+        ["t1", "t2"],
+        type="PLACES",
+        direction="in",
+        set={"channel": "web", "note": None, "since": "2026-11-06T10:00:00.000Z"},
+        unset=["old"],
+        types={"since": "datetime"},
+    )
+    assert result["data"]["updated"] == 1
+    commands = [
+        op["mutation"]["command"]
+        for op in operations(engine_calls)
+        if op.get("mutation", {}).get("kind") == "updateRelationshipsById"
+    ]
+    assert commands == [
+        {
+            "relationType": "PLACES",
+            "direction": "in",
+            "sourceIds": [RECORD["id"]],
+            "targetIds": ["t1", "t2"],
+            "set": {
+                "channel": "web",
+                "note": None,
+                "since": "2026-11-06T10:00:00.000Z",
+            },
+            "unset": ["old"],
+            "types": {"since": "datetime"},
+        }
+    ]
+
+
 def test_keyword_search_uses_the_labels_text_index(engine_calls):
     result = direct_db().records.text_search({"query": "dune", "labels": ["Book"]})
     hybrid = [op for op in operations(engine_calls) if op.get("kind") == "query"][0]

{rushdb-3.1.0 → rushdb-3.2.0}/tests/test_direct_mode_live.py

@@ -85,6 +85,27 @@ def test_relationships(db):
     )
     assert found.total == 1
 
+    edge = found.data[0]
+    result = db.relationships.update(
+        edge["sourceId"],
+        edge["targetId"],
+        type="AUTHORED",
+        set={"year": 1965, "role": "author", "since": "1965-08-01T10:30:00.000Z"},
+        types={"since": "datetime"},
+    )
+    assert result["data"]["updated"] == 1
+    db.relationships.update(
+        edge["sourceId"], edge["targetId"], type="AUTHORED", unset=["role"]
+    )
+    updated = db.relationships.find(
+        {"where": {"type": "AUTHORED"}, "source": {"labels": [AUTHOR]}}
+    )
+    assert updated.data[0].get("properties") == {
+        "year": 1965,
+        "since": "1965-08-01T10:30:00.000Z",
+    }
+    assert updated.data[0].get("propertyTypes", {}).get("since") == "datetime"
+
 
 def test_labels_properties_schema(db):
     assert db.labels.find({})["data"][BOOK] >= 3

{rushdb-3.1.0 → rushdb-3.2.0}/uv.lock

@@ -710,7 +710,7 @@ wheels = [
 
 [[package]]
 name = "rushdb"
-version = "3.1.0"
+version = "3.2.0"
 source = { editable = "." }
 
 [package.dev-dependencies]