memwal 0.1.8.dev2__tar.gz → 0.1.9.dev0__tar.gz

{memwal-0.1.8.dev2 → memwal-0.1.9.dev0}/CHANGELOG.md

@@ -1,5 +1,11 @@
 # memwal
 
+## 0.1.9
+
+### Fixed
+
+- `remember_bulk_async` rejects an empty `items` list before the request and raises when the relayer returns a `job_ids` length that does not match the batch.
+
 ## 0.1.8
 
 ### Added

{memwal-0.1.8.dev2 → memwal-0.1.9.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: memwal
-Version: 0.1.8.dev2
+Version: 0.1.9.dev0
 Summary: Python SDK for Walrus Memory — Privacy-first AI memory with Ed25519 signing
 Project-URL: Homepage, https://memory.walrus.xyz
 Project-URL: Documentation, https://memory.walrus.xyz

{memwal-0.1.8.dev2 → memwal-0.1.9.dev0}/memwal/__init__.py

@@ -122,4 +122,4 @@ __all__ = [
     "RecallManualResult",
 ]
 
-__version__ = "0.1.8.dev2"
+__version__ = "0.1.9.dev0"

{memwal-0.1.8.dev2 → memwal-0.1.9.dev0}/memwal/client.py

@@ -463,6 +463,9 @@ class MemWal:
         capped at ``MAX_REMEMBER_TEXT_BYTES`` (= 64 KiB).
         """
 
+        if not items:
+            raise ValueError("remember_bulk_async: items must be a non-empty array")
+
         payload_items: List[Dict[str, Any]] = [
             {
                 "text": item.text,
@@ -476,8 +479,15 @@ class MemWal:
             {"items": payload_items},
             accepted_statuses=(200, 202),
         )
+        job_ids = data.get("job_ids")
+        returned = len(job_ids) if job_ids else 0
+        if not job_ids or returned != len(payload_items):
+            raise ValueError(
+                f"remember_bulk_async: server returned {returned} job_ids for "
+                f"{len(payload_items)} items"
+            )
         return RememberBulkAcceptedResult(
-            job_ids=list(data.get("job_ids", [])),
+            job_ids=list(job_ids),
             total=int(data.get("total", len(payload_items))),
             status=data.get("status", "pending"),
         )

{memwal-0.1.8.dev2 → memwal-0.1.9.dev0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "memwal"
-version = "0.1.8.dev2"
+version = "0.1.9.dev0"
 description = "Python SDK for Walrus Memory — Privacy-first AI memory with Ed25519 signing"
 readme = "README.md"
 license = "MIT"

{memwal-0.1.8.dev2 → memwal-0.1.9.dev0}/tests/test_client.py

@@ -27,6 +27,8 @@ from memwal.client import (
 from memwal.types import (
     RecallManualOptions,
     RecallParams,
+    RememberBulkAcceptedResult,
+    RememberBulkItem,
     RememberManualOptions,
     ScoringWeights,
 )
@@ -343,6 +345,90 @@ class TestRemember:
         assert result.status == "pending"
 
 
+# ============================================================
+# remember_bulk_async() tests
+# ============================================================
+
+
+class TestRememberBulkAsync:
+    @respx.mock
+    async def test_empty_items_raises_without_http(self, memwal_client: MemWal) -> None:
+        mock_seal_session_prereqs()
+        route = respx.post(f"{_TEST_SERVER}/api/remember/bulk").mock(
+            return_value=httpx.Response(
+                202,
+                json={"job_ids": [], "total": 0, "status": "pending"},
+            )
+        )
+
+        with pytest.raises(
+            ValueError,
+            match="remember_bulk_async: items must be a non-empty array",
+        ):
+            await memwal_client.remember_bulk_async([])
+
+        assert not route.called
+
+    @respx.mock
+    async def test_mismatched_job_ids_raises(self, memwal_client: MemWal) -> None:
+        mock_seal_session_prereqs()
+        respx.post(f"{_TEST_SERVER}/api/remember/bulk").mock(
+            return_value=httpx.Response(
+                202,
+                json={
+                    "job_ids": ["job-1"],
+                    "total": 1,
+                    "status": "pending",
+                },
+            )
+        )
+
+        with pytest.raises(
+            ValueError,
+            match="remember_bulk_async: server returned 1 job_ids for 2 items",
+        ):
+            await memwal_client.remember_bulk_async(
+                [
+                    RememberBulkItem(text="I love coffee"),
+                    RememberBulkItem(text="I live in Tokyo"),
+                ]
+            )
+
+    @respx.mock
+    async def test_matching_job_ids_returns_accepted(
+        self, memwal_client: MemWal
+    ) -> None:
+        mock_seal_session_prereqs()
+        route = respx.post(f"{_TEST_SERVER}/api/remember/bulk").mock(
+            return_value=httpx.Response(
+                202,
+                json={
+                    "job_ids": ["job-1", "job-2"],
+                    "total": 2,
+                    "status": "pending",
+                },
+            )
+        )
+
+        result = await memwal_client.remember_bulk_async(
+            [
+                RememberBulkItem(text="I love coffee"),
+                RememberBulkItem(text="I live in Tokyo", namespace="profile"),
+            ]
+        )
+
+        assert route.called
+        body = json.loads(route.calls[0].request.content)
+        assert body["items"] == [
+            {"text": "I love coffee", "namespace": "default"},
+            {"text": "I live in Tokyo", "namespace": "profile"},
+        ]
+        assert isinstance(result, RememberBulkAcceptedResult)
+        assert result.job_ids == ["job-1", "job-2"]
+        assert result.total == 2
+        assert result.status == "pending"
+
+
 # ============================================================
 # recall() tests
 # ============================================================