memwal 0.1.0__tar.gz

memwal-0.1.0/.gitignore

@@ -0,0 +1,65 @@
+# Python
+__pycache__/
+*.py[cod]
+*.pyo
+*.pyd
+*.so
+*.egg
+*.egg-info/
+dist/
+build/
+wheels/
+*.whl
+.eggs/
+
+# Virtual environments
+.venv/
+venv/
+env/
+ENV/
+
+# Test / coverage
+.pytest_cache/
+.coverage
+.coverage.*
+coverage.xml
+htmlcov/
+.tox/
+
+# Type checking
+.mypy_cache/
+.dmypy.json
+dmypy.json
+.pyright/
+
+# Linting
+.ruff_cache/
+
+# Distribution / packaging
+MANIFEST
+pip-wheel-metadata/
+share/python-wheels/
+
+# IDE
+.vscode/
+.idea/
+*.swp
+*.swo
+*~
+
+# macOS
+.DS_Store
+.AppleDouble
+.LSOverride
+
+# Env files
+.env
+.env.*
+!.env.example
+
+# Jupyter
+.ipynb_checkpoints/
+*.ipynb
+
+# Docs build
+site/

memwal-0.1.0/PKG-INFO

@@ -0,0 +1,237 @@
+Metadata-Version: 2.4
+Name: memwal
+Version: 0.1.0
+Summary: Python SDK for MemWal — Privacy-first AI memory with Ed25519 signing
+Project-URL: Homepage, https://memwal.ai
+Project-URL: Documentation, https://docs.memwal.ai
+Project-URL: Repository, https://github.com/MystenLabs/MemWal
+Author: MemWal Team
+License-Expression: MIT
+Keywords: ai,ed25519,memory,memwal,privacy,sui,walrus
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Typing :: Typed
+Requires-Python: >=3.9
+Requires-Dist: httpx>=0.27.0
+Requires-Dist: pynacl>=1.5.0
+Provides-Extra: all
+Requires-Dist: langchain-core>=0.2.0; extra == 'all'
+Requires-Dist: openai>=1.0.0; extra == 'all'
+Provides-Extra: dev
+Requires-Dist: langchain-core>=0.2.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.23.0; extra == 'dev'
+Requires-Dist: pytest>=7.0; extra == 'dev'
+Requires-Dist: respx>=0.21.0; extra == 'dev'
+Requires-Dist: ruff>=0.4.0; extra == 'dev'
+Provides-Extra: langchain
+Requires-Dist: langchain-core>=0.2.0; extra == 'langchain'
+Provides-Extra: openai
+Requires-Dist: openai>=1.0.0; extra == 'openai'
+Description-Content-Type: text/markdown
+
+# memwal
+
+Python SDK for [MemWal](https://memwal.ai) — Privacy-first AI memory with Ed25519 signing.
+
+All data processing (encryption, embedding, Walrus storage) happens server-side in a TEE. The SDK signs requests with your Ed25519 delegate key and sends text over HTTPS.
+
+## Installation
+
+```bash
+pip install memwal
+```
+
+With optional integrations:
+
+```bash
+pip install memwal[langchain]   # LangChain support
+pip install memwal[openai]      # OpenAI SDK support
+pip install memwal[all]         # Everything
+```
+
+## Quick Start
+
+Set your environment variables first:
+
+```bash
+export MEMWAL_KEY="your-ed25519-delegate-key-hex"
+export MEMWAL_ACCOUNT_ID="0x-your-memwal-account-id"
+export MEMWAL_SERVER_URL="https://relayer.memwal.ai"
+```
+
+### Async (recommended)
+
+```python
+import asyncio
+import os
+from memwal import MemWal
+
+async def main():
+    memwal = MemWal.create(
+        key=os.environ["MEMWAL_KEY"],
+        account_id=os.environ["MEMWAL_ACCOUNT_ID"],
+        server_url=os.environ.get("MEMWAL_SERVER_URL", "https://relayer.memwal.ai"),
+    )
+
+    # Store a memory
+    result = await memwal.remember("I'm allergic to peanuts")
+    print(result.blob_id)
+
+    # Recall memories
+    matches = await memwal.recall("food allergies")
+    for memory in matches.results:
+        print(f"{memory.text} (relevance: {1 - memory.distance:.2f})")
+
+    # Analyze conversation for facts
+    analysis = await memwal.analyze("I love coffee and live in Tokyo")
+    for fact in analysis.facts:
+        print(fact.text)
+
+    await memwal.close()
+
+asyncio.run(main())
+```
+
+### Sync
+
+```python
+import os
+from memwal import MemWalSync
+
+client = MemWalSync.create(
+    key=os.environ["MEMWAL_KEY"],
+    account_id=os.environ["MEMWAL_ACCOUNT_ID"],
+    server_url=os.environ.get("MEMWAL_SERVER_URL", "https://relayer.memwal.ai"),
+)
+
+result = client.remember("I'm allergic to peanuts")
+matches = client.recall("food allergies")
+client.close()
+```
+
+### Context Manager
+
+```python
+import os
+from memwal import MemWal
+
+async with MemWal.create(
+    key=os.environ["MEMWAL_KEY"],
+    account_id=os.environ["MEMWAL_ACCOUNT_ID"],
+) as memwal:
+    await memwal.remember("I prefer dark mode")
+```
+
+## Environment Presets
+
+Instead of hardcoding a relayer URL, pass `env` to target a hosted relayer.
+Same shorthand as the TypeScript SDK and MCP package.
+
+```python
+from memwal import MemWal
+
+memwal = MemWal.create(
+    key=os.environ["MEMWAL_KEY"],
+    account_id=os.environ["MEMWAL_ACCOUNT_ID"],
+    env="prod",   # prod | dev | staging | local
+)
+```
+
+| `env` | Relayer URL |
+|-------|-------------|
+| `prod` | `https://relayer.memwal.ai` |
+| `dev` | `https://relayer.dev.memwal.ai` |
+| `staging` | `https://relayer.staging.memwal.ai` |
+| `local` | `http://127.0.0.1:8000` |
+
+Precedence: an explicit non-default **`server_url` wins over `env`**, which wins
+over the default. An unknown preset raises `ValueError`. `env` is also accepted
+by `MemWalSync.create`, `with_memwal_langchain`, and `with_memwal_openai`.
+
+## AI Middleware
+
+### LangChain
+
+```python
+import os
+from langchain_openai import ChatOpenAI
+from langchain_core.messages import HumanMessage
+from memwal import with_memwal_langchain
+
+llm = ChatOpenAI(model="gpt-4o")
+smart_llm = with_memwal_langchain(
+    llm,
+    key=os.environ["MEMWAL_KEY"],
+    account_id=os.environ["MEMWAL_ACCOUNT_ID"],
+    server_url=os.environ.get("MEMWAL_SERVER_URL", "https://relayer.memwal.ai"),
+    max_memories=5,
+    min_relevance=0.3,
+)
+
+# Memories are automatically recalled and injected
+response = await smart_llm.ainvoke([HumanMessage("What are my food allergies?")])
+```
+
+### OpenAI SDK
+
+```python
+import os
+from openai import AsyncOpenAI
+from memwal import with_memwal_openai
+
+client = AsyncOpenAI()
+smart_client = with_memwal_openai(
+    client,
+    key=os.environ["MEMWAL_KEY"],
+    account_id=os.environ["MEMWAL_ACCOUNT_ID"],
+    server_url=os.environ.get("MEMWAL_SERVER_URL", "https://relayer.memwal.ai"),
+)
+
+# Memories are automatically recalled and injected
+response = await smart_client.chat.completions.create(
+    model="gpt-4o",
+    messages=[{"role": "user", "content": "What are my food allergies?"}],
+)
+```
+
+## API Reference
+
+### `MemWal.create(key, account_id, server_url?, namespace?)`
+
+Create a new async client.
+
+### Methods
+
+| Method | Description |
+|--------|-------------|
+| `await remember(text, namespace?)` | Store a memory |
+| `await recall(query, limit?, namespace?)` | Search memories |
+| `await analyze(text, namespace?)` | Extract and store facts |
+| `await ask(question, limit?, namespace?)` | Ask a question answered using memories |
+| `await restore(namespace, limit?)` | Restore a namespace |
+| `await health()` | Check server health |
+| `await remember_manual(opts)` | Store with pre-computed vector |
+| `await recall_manual(opts)` | Search with pre-computed vector |
+| `await get_public_key_hex()` | Get Ed25519 public key |
+
+## Authentication
+
+Every request is signed with Ed25519:
+
+```
+message = f"{timestamp}.{method}.{path}.{sha256(body)}"
+```
+
+Headers sent: `x-public-key`, `x-signature`, `x-timestamp`, `x-delegate-key`, `x-account-id`.
+
+## License
+
+MIT

memwal-0.1.0/README.md

@@ -0,0 +1,198 @@
+# memwal
+
+Python SDK for [MemWal](https://memwal.ai) — Privacy-first AI memory with Ed25519 signing.
+
+All data processing (encryption, embedding, Walrus storage) happens server-side in a TEE. The SDK signs requests with your Ed25519 delegate key and sends text over HTTPS.
+
+## Installation
+
+```bash
+pip install memwal
+```
+
+With optional integrations:
+
+```bash
+pip install memwal[langchain]   # LangChain support
+pip install memwal[openai]      # OpenAI SDK support
+pip install memwal[all]         # Everything
+```
+
+## Quick Start
+
+Set your environment variables first:
+
+```bash
+export MEMWAL_KEY="your-ed25519-delegate-key-hex"
+export MEMWAL_ACCOUNT_ID="0x-your-memwal-account-id"
+export MEMWAL_SERVER_URL="https://relayer.memwal.ai"
+```
+
+### Async (recommended)
+
+```python
+import asyncio
+import os
+from memwal import MemWal
+
+async def main():
+    memwal = MemWal.create(
+        key=os.environ["MEMWAL_KEY"],
+        account_id=os.environ["MEMWAL_ACCOUNT_ID"],
+        server_url=os.environ.get("MEMWAL_SERVER_URL", "https://relayer.memwal.ai"),
+    )
+
+    # Store a memory
+    result = await memwal.remember("I'm allergic to peanuts")
+    print(result.blob_id)
+
+    # Recall memories
+    matches = await memwal.recall("food allergies")
+    for memory in matches.results:
+        print(f"{memory.text} (relevance: {1 - memory.distance:.2f})")
+
+    # Analyze conversation for facts
+    analysis = await memwal.analyze("I love coffee and live in Tokyo")
+    for fact in analysis.facts:
+        print(fact.text)
+
+    await memwal.close()
+
+asyncio.run(main())
+```
+
+### Sync
+
+```python
+import os
+from memwal import MemWalSync
+
+client = MemWalSync.create(
+    key=os.environ["MEMWAL_KEY"],
+    account_id=os.environ["MEMWAL_ACCOUNT_ID"],
+    server_url=os.environ.get("MEMWAL_SERVER_URL", "https://relayer.memwal.ai"),
+)
+
+result = client.remember("I'm allergic to peanuts")
+matches = client.recall("food allergies")
+client.close()
+```
+
+### Context Manager
+
+```python
+import os
+from memwal import MemWal
+
+async with MemWal.create(
+    key=os.environ["MEMWAL_KEY"],
+    account_id=os.environ["MEMWAL_ACCOUNT_ID"],
+) as memwal:
+    await memwal.remember("I prefer dark mode")
+```
+
+## Environment Presets
+
+Instead of hardcoding a relayer URL, pass `env` to target a hosted relayer.
+Same shorthand as the TypeScript SDK and MCP package.
+
+```python
+from memwal import MemWal
+
+memwal = MemWal.create(
+    key=os.environ["MEMWAL_KEY"],
+    account_id=os.environ["MEMWAL_ACCOUNT_ID"],
+    env="prod",   # prod | dev | staging | local
+)
+```
+
+| `env` | Relayer URL |
+|-------|-------------|
+| `prod` | `https://relayer.memwal.ai` |
+| `dev` | `https://relayer.dev.memwal.ai` |
+| `staging` | `https://relayer.staging.memwal.ai` |
+| `local` | `http://127.0.0.1:8000` |
+
+Precedence: an explicit non-default **`server_url` wins over `env`**, which wins
+over the default. An unknown preset raises `ValueError`. `env` is also accepted
+by `MemWalSync.create`, `with_memwal_langchain`, and `with_memwal_openai`.
+
+## AI Middleware
+
+### LangChain
+
+```python
+import os
+from langchain_openai import ChatOpenAI
+from langchain_core.messages import HumanMessage
+from memwal import with_memwal_langchain
+
+llm = ChatOpenAI(model="gpt-4o")
+smart_llm = with_memwal_langchain(
+    llm,
+    key=os.environ["MEMWAL_KEY"],
+    account_id=os.environ["MEMWAL_ACCOUNT_ID"],
+    server_url=os.environ.get("MEMWAL_SERVER_URL", "https://relayer.memwal.ai"),
+    max_memories=5,
+    min_relevance=0.3,
+)
+
+# Memories are automatically recalled and injected
+response = await smart_llm.ainvoke([HumanMessage("What are my food allergies?")])
+```
+
+### OpenAI SDK
+
+```python
+import os
+from openai import AsyncOpenAI
+from memwal import with_memwal_openai
+
+client = AsyncOpenAI()
+smart_client = with_memwal_openai(
+    client,
+    key=os.environ["MEMWAL_KEY"],
+    account_id=os.environ["MEMWAL_ACCOUNT_ID"],
+    server_url=os.environ.get("MEMWAL_SERVER_URL", "https://relayer.memwal.ai"),
+)
+
+# Memories are automatically recalled and injected
+response = await smart_client.chat.completions.create(
+    model="gpt-4o",
+    messages=[{"role": "user", "content": "What are my food allergies?"}],
+)
+```
+
+## API Reference
+
+### `MemWal.create(key, account_id, server_url?, namespace?)`
+
+Create a new async client.
+
+### Methods
+
+| Method | Description |
+|--------|-------------|
+| `await remember(text, namespace?)` | Store a memory |
+| `await recall(query, limit?, namespace?)` | Search memories |
+| `await analyze(text, namespace?)` | Extract and store facts |
+| `await ask(question, limit?, namespace?)` | Ask a question answered using memories |
+| `await restore(namespace, limit?)` | Restore a namespace |
+| `await health()` | Check server health |
+| `await remember_manual(opts)` | Store with pre-computed vector |
+| `await recall_manual(opts)` | Search with pre-computed vector |
+| `await get_public_key_hex()` | Get Ed25519 public key |
+
+## Authentication
+
+Every request is signed with Ed25519:
+
+```
+message = f"{timestamp}.{method}.{path}.{sha256(body)}"
+```
+
+Headers sent: `x-public-key`, `x-signature`, `x-timestamp`, `x-delegate-key`, `x-account-id`.
+
+## License
+
+MIT

memwal-0.1.0/examples/.env.example

@@ -0,0 +1,8 @@
+# Local server (default) or remote relayer
+MEMWAL_SERVER_URL=http://localhost:3001
+# Ed25519 delegate private key (64-hex). Get from MemWal dashboard.
+MEMWAL_KEY=21b423e72282dcc47805de48ef9130331b642667b7b2a5cd621767928205e360
+# MemWalAccount object ID on Sui (the wallet's account)
+MEMWAL_ACCOUNT_ID=0x8a1121b8f95d79e68bd07efaf71689ce6fd832b369cdb1b2a943ec7beb822392
+# Namespace for these test memories
+MEMWAL_NAMESPACE=python-sdk-example

memwal-0.1.0/examples/.gitignore

@@ -0,0 +1,2 @@
+# Local credentials — never commit. Use .env.example as template.
+.env

memwal-0.1.0/examples/async_remember_demo.py

@@ -0,0 +1,179 @@
+"""End-to-end smoke test for the async remember family (PR #121 / ENG-1406+1408).
+
+Loads credentials from ./.env (see ``.env.example``), then walks every new
+method that was added to mirror the TypeScript SDK:
+
+    1. health()                   sanity check
+    2. remember()                 returns RememberAcceptedResult (~500ms)
+    3. wait_for_remember_job()    polls until ``done`` (~upload time)
+    4. remember_and_wait()        single call doing both
+    5. recall()                   verify the new memory came back
+    6. remember_bulk_and_wait()   3 facts in one call
+    7. analyze_and_wait()         LLM extracts facts + waits for them all
+    8. embed()                    raw embedding vector
+
+Run::
+
+    cd packages/python-sdk-memwal
+    cp examples/.env.example examples/.env  # then fill in real values
+    python3 examples/async_remember_demo.py
+"""
+
+from __future__ import annotations
+
+import os
+import sys
+import time
+from pathlib import Path
+from typing import Optional
+
+
+def _load_env() -> None:
+    """Tiny .env loader so the demo doesn't need python-dotenv."""
+
+    env_path = Path(__file__).parent / ".env"
+    if not env_path.exists():
+        return
+    for line in env_path.read_text().splitlines():
+        line = line.strip()
+        if not line or line.startswith("#") or "=" not in line:
+            continue
+        key, _, value = line.partition("=")
+        os.environ.setdefault(key.strip(), value.strip())
+
+
+_load_env()
+
+# Make ``import memwal`` work when running directly from the repo without
+# `pip install -e .`.
+sys.path.insert(0, str(Path(__file__).parent.parent))
+
+from memwal import (  # noqa: E402
+    MemWal,
+    MemWalRememberJobFailed,
+    RememberBulkItem,
+    RememberBulkOptions,
+)
+
+
+def _section(title: str) -> None:
+    print(f"\n──────── {title} ────────")
+
+
+def _ms(start: float) -> int:
+    return int((time.monotonic() - start) * 1000)
+
+
+async def main() -> None:
+    server_url = os.environ.get("MEMWAL_SERVER_URL", "http://localhost:3001")
+    key = os.environ.get("MEMWAL_KEY")
+    account_id = os.environ.get("MEMWAL_ACCOUNT_ID")
+    namespace = os.environ.get("MEMWAL_NAMESPACE", "python-sdk-example")
+
+    if not key or not account_id:
+        print("ERROR: set MEMWAL_KEY + MEMWAL_ACCOUNT_ID in examples/.env")
+        sys.exit(2)
+
+    print(f"server      : {server_url}")
+    print(f"account_id  : {account_id[:14]}...")
+    print(f"namespace   : {namespace}")
+
+    async with MemWal.create(
+        key=key,
+        account_id=account_id,
+        server_url=server_url,
+        namespace=namespace,
+    ) as memwal:
+        # 1. health
+        _section("1. health()")
+        h = await memwal.health()
+        print(f"  status={h.status}  version={h.version}")
+
+        # 2. remember() → 202 + job_id (PR #121: should return in ~500ms)
+        _section("2. remember()  — returns 202 + job_id")
+        t0 = time.monotonic()
+        accepted = await memwal.remember(
+            "Python SDK demo: I prefer FastAPI over Flask for async HTTP services."
+        )
+        accept_ms = _ms(t0)
+        print(f"  accept_ms={accept_ms}  job_id={accepted.job_id}  status={accepted.status}")
+        assert accepted.job_id, "expected job_id"
+        assert accept_ms < 5000, f"expected <5s response, got {accept_ms}ms"
+
+        # 3. wait_for_remember_job → polls until "done"
+        _section("3. wait_for_remember_job(job_id)")
+        t1 = time.monotonic()
+        result = await memwal.wait_for_remember_job(accepted.job_id, timeout_ms=60_000)
+        wait_ms = _ms(t1)
+        print(f"  wait_ms={wait_ms}  blob_id={result.blob_id}  ns={result.namespace}")
+
+        # 4. remember_and_wait — convenience
+        _section("4. remember_and_wait()")
+        t2 = time.monotonic()
+        full = await memwal.remember_and_wait(
+            "Python SDK demo: I drink black coffee in the morning.",
+            timeout_ms=60_000,
+        )
+        total_ms = _ms(t2)
+        print(f"  total_ms={total_ms}  blob_id={full.blob_id}")
+
+        # 5. recall — confirm the memories are searchable
+        _section("5. recall('coffee')")
+        rc = await memwal.recall("coffee", limit=3)
+        print(f"  found {len(rc.results)} / total={rc.total}")
+        for m in rc.results[:3]:
+            print(f"    [{m.distance:.3f}] {m.text[:80]}")
+
+        # 6. remember_bulk_and_wait — 3 items in one call
+        _section("6. remember_bulk_and_wait()  — 3 items")
+        t3 = time.monotonic()
+        bulk = await memwal.remember_bulk_and_wait(
+            [
+                RememberBulkItem(text="Bulk demo 1: Trees clean the air."),
+                RememberBulkItem(text="Bulk demo 2: Coffee comes from beans."),
+                RememberBulkItem(text="Bulk demo 3: Mountains are usually cold."),
+            ],
+            opts=RememberBulkOptions(poll_interval_ms=2000, timeout_ms=120_000),
+        )
+        bulk_ms = _ms(t3)
+        print(
+            f"  bulk_ms={bulk_ms}  total={bulk.total}  ok={bulk.succeeded}  "
+            f"failed={bulk.failed}  timed_out={bulk.timed_out}"
+        )
+        for r in bulk.results:
+            print(f"    [{r.status}] id={r.id[:8]}... blob={r.blob_id[:20]}...")
+
+        # 7. analyze_and_wait — LLM splits text into facts + persists each
+        _section("7. analyze_and_wait()")
+        t4 = time.monotonic()
+        try:
+            an = await memwal.analyze_and_wait(
+                "Today I learned that the Pacific is the largest ocean and "
+                "that octopuses have three hearts.",
+                opts=RememberBulkOptions(timeout_ms=120_000),
+            )
+            ana_ms = _ms(t4)
+            print(
+                f"  analyze_ms={ana_ms}  facts={len(an.facts)}  "
+                f"ok={an.succeeded}  failed={an.failed}  timed_out={an.timed_out}"
+            )
+            for f, r in zip(an.facts, an.results):
+                print(f"    fact='{f.text[:60]}' status={r.status} blob={r.blob_id[:20]}...")
+        except Exception as e:
+            print(f"  skipped (analyze requires server LLM key): {e}")
+
+        # 8. embed — raw embedding vector
+        _section("8. embed('hello world')")
+        try:
+            emb = await memwal.embed("hello world")
+            print(f"  vector dims={len(emb.vector)}  first 5={emb.vector[:5]}")
+        except Exception as e:
+            print(f"  skipped (server may not expose /api/embed yet): {e}")
+
+    print("\n✅ all sections completed")
+
+
+if __name__ == "__main__":
+    import asyncio
+
+    asyncio.run(main())