memwal 0.1.2.dev2__tar.gz → 0.1.2.dev3__tar.gz

{memwal-0.1.2.dev2 → memwal-0.1.2.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: memwal
-Version: 0.1.2.dev2
+Version: 0.1.2.dev3
 Summary: Python SDK for Walrus Memory — Privacy-first AI memory with Ed25519 signing
 Project-URL: Homepage, https://memwal.ai
 Project-URL: Documentation, https://docs.memwal.ai

{memwal-0.1.2.dev2 → memwal-0.1.2.dev3}/memwal/__init__.py

@@ -32,7 +32,6 @@ from .client import (
     MemWalSync,
 )
 from .middleware import with_memwal_langchain, with_memwal_openai
-from .utils import delegate_key_to_sui_address, delegate_key_to_public_key
 from .types import (
     ENV_PRESETS,
     AnalyzedFact,
@@ -61,7 +60,9 @@ from .types import (
     RememberManualResult,
     RememberResult,
     RestoreResult,
+    ScoringWeights,
 )
+from .utils import delegate_key_to_public_key, delegate_key_to_sui_address
 
 # JS-style alias for developers coming from the TypeScript SDK
 withMemWal = with_memwal_langchain
@@ -105,6 +106,7 @@ __all__ = [
     "AnalyzedFact",
     "HealthResult",
     "RestoreResult",
+    "ScoringWeights",
     "RememberManualOptions",
     "RememberManualResult",
     "RecallManualOptions",
@@ -112,4 +114,4 @@ __all__ = [
     "RecallManualResult",
 ]
 
-__version__ = "0.1.2.dev2"
+__version__ = "0.1.2.dev3"

{memwal-0.1.2.dev2 → memwal-0.1.2.dev3}/memwal/client.py

@@ -58,7 +58,6 @@ from .types import (
     RememberBulkResult,
     RememberBulkStatusItem,
     RememberBulkStatusResult,
-    RememberJobStatus,
     RememberManualOptions,
     RememberManualResult,
     RememberResult,
@@ -728,14 +727,18 @@ class MemWal:
         Returns:
             :class:`RecallManualResult` with blob_id + distance pairs.
         """
+        body: Dict[str, Any] = {
+            "vector": opts.vector,
+            "limit": opts.limit,
+            "namespace": opts.namespace or self._namespace,
+        }
+        if opts.scoring_weights is not None:
+            body["scoring_weights"] = opts.scoring_weights.to_wire()
+
         data = await self._signed_request(
             "POST",
             "/api/recall/manual",
-            {
-                "vector": opts.vector,
-                "limit": opts.limit,
-                "namespace": opts.namespace or self._namespace,
-            },
+            body,
             include_seal_session=False,
         )
         hits = [
@@ -840,7 +843,8 @@ class MemWal:
                     version = obj.get("version")
         if str(version) != "1":
             raise MemWalError(
-                f"SEAL package {package_id} must be at version 1 to build x-seal-session, got {version!r}"
+                f"SEAL package {package_id} must be at version 1 to build "
+                f"x-seal-session, got {version!r}"
             )
 
     async def _build_seal_session_inner(self) -> str:

{memwal-0.1.2.dev2 → memwal-0.1.2.dev3}/memwal/types.py

@@ -36,7 +36,8 @@ class MemWalConfig:
     """Configuration for creating a Walrus Memory client.
 
     Attributes:
-        key: Ed25519 private key (hex string). This is the delegate key from the Walrus Memory dashboard.
+        key: Ed25519 private key (hex string). This is the delegate key from the
+            Walrus Memory dashboard.
         account_id: Walrus Memory account object ID on Sui.
         server_url: Server URL (default: http://localhost:8000). An explicit
             non-default value always wins over ``env``.
@@ -99,6 +100,37 @@ class RecallResult:
     total: int
 
 
+@dataclass
+class ScoringWeights:
+    """Optional composite-scoring weights for recall ranking.
+
+    Attributes:
+        semantic: Weight applied to semantic similarity (default: 1).
+        recency: Weight applied to recency decay (default: 0).
+        recency_half_life_days: Half-life for the recency term, in days (default: 30).
+        importance: Weight applied to memory importance (default: 0).
+    """
+
+    semantic: Optional[float] = None
+    recency: Optional[float] = None
+    recency_half_life_days: Optional[float] = None
+    importance: Optional[float] = None
+
+    def to_wire(self) -> Dict[str, float]:
+        """Return the snake_case payload expected by the relayer."""
+
+        payload: Dict[str, float] = {}
+        if self.semantic is not None:
+            payload["semantic"] = self.semantic
+        if self.recency is not None:
+            payload["recency"] = self.recency
+        if self.recency_half_life_days is not None:
+            payload["recency_half_life_days"] = self.recency_half_life_days
+        if self.importance is not None:
+            payload["importance"] = self.importance
+        return payload
+
+
 @dataclass
 class AnalyzedFact:
     """A single extracted fact."""
@@ -212,11 +244,13 @@ class RecallManualOptions:
         vector: Pre-computed query embedding vector.
         limit: Max number of results (default: 10).
         namespace: Namespace (default: config namespace or "default").
+        scoring_weights: Optional composite-scoring weights applied before returning hits.
     """
 
     vector: List[float]
     limit: int = 10
     namespace: Optional[str] = None
+    scoring_weights: Optional[ScoringWeights] = None
 
 
 @dataclass

{memwal-0.1.2.dev2 → memwal-0.1.2.dev3}/pyproject.toml

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

{memwal-0.1.2.dev2 → memwal-0.1.2.dev3}/tests/test_client.py

@@ -17,7 +17,7 @@ import pytest
 import respx
 
 from memwal.client import MemWal, MemWalCompatibilityError, MemWalError
-from memwal.types import RecallManualOptions, RememberManualOptions
+from memwal.types import RecallManualOptions, RememberManualOptions, ScoringWeights
 from memwal.utils import build_signature_message, bytes_to_hex, sha256_hex
 
 # ============================================================
@@ -327,7 +327,11 @@ class TestRecall:
             )
         )
 
-        result = await memwal_client.wait_for_remember_job("job-1", poll_interval_ms=0, timeout_ms=100)
+        result = await memwal_client.wait_for_remember_job(
+            "job-1",
+            poll_interval_ms=0,
+            timeout_ms=100,
+        )
 
         request = route.calls[0].request
         assert request.content == b""
@@ -583,6 +587,33 @@ class TestManualAPI:
         assert len(result.results) == 1
         assert result.results[0].blob_id == "b1"
 
+    @respx.mock
+    async def test_recall_manual_forwards_scoring_weights(self, memwal_client: MemWal) -> None:
+        _mock_version()
+        route = respx.post(f"{_TEST_SERVER}/api/recall/manual").mock(
+            return_value=httpx.Response(200, json={"results": [], "total": 0})
+        )
+
+        opts = RecallManualOptions(
+            vector=[0.1, 0.2, 0.3],
+            limit=5,
+            scoring_weights=ScoringWeights(
+                semantic=1.0,
+                recency=0.5,
+                recency_half_life_days=7.0,
+                importance=2.0,
+            ),
+        )
+        await memwal_client.recall_manual(opts)
+
+        body = json.loads(route.calls[0].request.content)
+        assert body["scoring_weights"] == {
+            "semantic": 1.0,
+            "recency": 0.5,
+            "recency_half_life_days": 7.0,
+            "importance": 2.0,
+        }
+
 
 class TestPublicKey:
     async def test_get_public_key_hex(self, memwal_client: MemWal) -> None: