memwal 0.1.11.dev0__tar.gz → 0.1.11.dev1__tar.gz

{memwal-0.1.11.dev0 → memwal-0.1.11.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: memwal
-Version: 0.1.11.dev0
+Version: 0.1.11.dev1
 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.11.dev0 → memwal-0.1.11.dev1}/memwal/__init__.py

@@ -126,4 +126,4 @@ __all__ = [
     "RecallManualResult",
 ]
 
-__version__ = "0.1.11.dev0"
+__version__ = "0.1.11.dev1"

{memwal-0.1.11.dev0 → memwal-0.1.11.dev1}/memwal/client.py

@@ -34,7 +34,7 @@ import re
 import time
 import uuid
 from datetime import datetime, timezone
-from typing import Any, Dict, List, Optional, Sequence, Tuple, TypeVar, Union
+from typing import Any, Dict, List, Optional, Sequence, Set, Tuple, TypeVar, Union
 from urllib.parse import ParseResult, urlencode, urlparse
 
 import httpx
@@ -624,7 +624,7 @@ class MemWal:
                     continue
                 raise
 
-            still_pending: List[str] = []
+            terminal_ids: Set[str] = set()
             for item in batch.results:
                 if item.status == "done":
                     results[item.job_id] = RememberBulkItemResult(
@@ -633,6 +633,7 @@ class MemWal:
                         status="done",
                         error=None,
                     )
+                    terminal_ids.add(item.job_id)
                 elif item.status in ("failed", "not_found"):
                     results[item.job_id] = RememberBulkItemResult(
                         id=item.job_id,
@@ -640,9 +641,8 @@ class MemWal:
                         status="failed",
                         error=item.error,
                     )
-                else:
-                    still_pending.append(item.job_id)
-            pending = still_pending
+                    terminal_ids.add(item.job_id)
+            pending = [jid for jid in pending if jid not in terminal_ids]
 
         ordered = [results[job_id] for job_id in job_ids]
         succeeded = sum(1 for r in ordered if r.status == "done")

{memwal-0.1.11.dev0 → memwal-0.1.11.dev1}/pyproject.toml

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

{memwal-0.1.11.dev0 → memwal-0.1.11.dev1}/tests/test_client.py

@@ -10,7 +10,7 @@ from __future__ import annotations
 import base64
 import json
 from datetime import datetime, timedelta, timezone
-from typing import Any
+from typing import Any, Sequence
 
 import httpx
 import nacl.signing
@@ -29,6 +29,9 @@ from memwal.types import (
     RecallParams,
     RememberBulkAcceptedResult,
     RememberBulkItem,
+    RememberBulkOptions,
+    RememberBulkStatusItem,
+    RememberBulkStatusResult,
     RememberManualOptions,
     ScoringWeights,
 )
@@ -428,6 +431,36 @@ class TestRememberBulkAsync:
         assert result.total == 2
         assert result.status == "pending"
 
+    async def test_job_omitted_from_a_poll_stays_pending(
+        self, memwal_client: MemWal
+    ) -> None:
+        polls = 0
+
+        async def partial_status(
+            job_ids: Sequence[str],
+        ) -> RememberBulkStatusResult:
+            nonlocal polls
+            polls += 1
+            job_id = "job-1" if polls == 1 else "job-2"
+            return RememberBulkStatusResult(
+                results=[
+                    RememberBulkStatusItem(
+                        job_id=job_id,
+                        status="done",
+                        blob_id=f"blob-for-{job_id}",
+                    )
+                ]
+            )
+
+        memwal_client.get_remember_bulk_status = partial_status  # type: ignore[method-assign]
+
+        result = await memwal_client.wait_for_remember_jobs(
+            ["job-1", "job-2"],
+            RememberBulkOptions(poll_interval_ms=1, timeout_ms=5_000),
+        )
+
+        assert [item.status for item in result.results] == ["done", "done"]
+
 
 # ============================================================
 # recall() tests