audrey-memory 1.2.2__tar.gz → 2.1.0__tar.gz

{audrey_memory-1.2.2 → audrey_memory-2.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: audrey-memory
-Version: 1.2.2
+Version: 2.1.0
 Summary: Typed Python client for the Audrey LLM memory server
 Author: evilander
 License: MIT
@@ -45,7 +45,7 @@ python -m pip install -e .
 Start Audrey's REST API:
 
 ```bash
-npm install -g audrey --allow-scripts=better-sqlite3,onnxruntime-node,sharp,protobufjs
+npm install -g audrey
 audrey serve
 ```
 
@@ -67,11 +67,22 @@ memory_id = brain.encode(
 )
 
 results = brain.recall("stripe rate limits", limit=5)
-snapshot = brain.snapshot()
 brain.close()
 ```
 
-Restore snapshots only into an empty Audrey store, such as a sidecar started with a fresh `AUDREY_DATA_DIR`:
+Context keys and values must be strings, for example `context={"project": "support"}`.
+`brain.status()` returns the server's `episodes`, `semantics`, `procedures`, and
+`healthy` fields along with vector-index health details.
+
+Snapshot export and restore require the sidecar to be started with
+`AUDREY_ENABLE_ADMIN_TOOLS=1`. With that explicitly enabled:
+
+```python
+with Audrey(base_url="http://127.0.0.1:7437", api_key="secret") as brain:
+    snapshot = brain.snapshot()
+```
+
+Snapshots cover the whole store. Restore only into an empty Audrey store, such as a sidecar started with a fresh `AUDREY_DATA_DIR` and admin tools enabled:
 
 ```python
 restore_target = Audrey(base_url="http://127.0.0.1:7437", api_key="secret")

{audrey_memory-1.2.2 → audrey_memory-2.1.0}/README.md

@@ -20,7 +20,7 @@ python -m pip install -e .
 Start Audrey's REST API:
 
 ```bash
-npm install -g audrey --allow-scripts=better-sqlite3,onnxruntime-node,sharp,protobufjs
+npm install -g audrey
 audrey serve
 ```
 
@@ -42,11 +42,22 @@ memory_id = brain.encode(
 )
 
 results = brain.recall("stripe rate limits", limit=5)
-snapshot = brain.snapshot()
 brain.close()
 ```
 
-Restore snapshots only into an empty Audrey store, such as a sidecar started with a fresh `AUDREY_DATA_DIR`:
+Context keys and values must be strings, for example `context={"project": "support"}`.
+`brain.status()` returns the server's `episodes`, `semantics`, `procedures`, and
+`healthy` fields along with vector-index health details.
+
+Snapshot export and restore require the sidecar to be started with
+`AUDREY_ENABLE_ADMIN_TOOLS=1`. With that explicitly enabled:
+
+```python
+with Audrey(base_url="http://127.0.0.1:7437", api_key="secret") as brain:
+    snapshot = brain.snapshot()
+```
+
+Snapshots cover the whole store. Restore only into an empty Audrey store, such as a sidecar started with a fresh `AUDREY_DATA_DIR` and admin tools enabled:
 
 ```python
 restore_target = Audrey(base_url="http://127.0.0.1:7437", api_key="secret")

audrey_memory-2.1.0/audrey_memory/_version.py

@@ -0,0 +1 @@
+__version__ = "2.1.0"

{audrey_memory-1.2.2 → audrey_memory-2.1.0}/audrey_memory/types.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 from typing import Any
 
-from pydantic import BaseModel, ConfigDict, Field
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
 
 
 class AudreyModel(BaseModel):
@@ -31,14 +31,39 @@ class ContradictionStatus(AudreyModel):
 
 
 class StatusResponse(AudreyModel):
-    episodic: int | None = None
-    semantic: int | None = None
-    procedural: int | None = None
-    causalLinks: int | None = None
-    contradictions: ContradictionStatus | None = None
-    dormant: int | None = None
-    lastConsolidation: str | None = None
-    totalConsolidationRuns: int | None = None
+    episodes: int
+    semantics: int
+    procedures: int
+    healthy: bool
+    vec_episodes: int = 0
+    vec_semantics: int = 0
+    vec_procedures: int = 0
+    searchable_episodes: int = 0
+    searchable_semantics: int = 0
+    searchable_procedures: int = 0
+    dimensions: int | None = None
+    schema_version: int | None = None
+    device: str | None = None
+    reembed_recommended: bool = False
+    pending_consolidation_count: int = 0
+    embedding_warm: bool = False
+    warmup_duration_ms: float | None = None
+    default_retrieval_mode: str | None = None
+    recall_degraded: bool = False
+    last_recall_check_at: str | None = None
+    last_recall_errors: list[dict[str, Any]] = Field(default_factory=list)
+
+    @property
+    def episodic(self) -> int:
+        return self.episodes
+
+    @property
+    def semantic(self) -> int:
+        return self.semantics
+
+    @property
+    def procedural(self) -> int:
+        return self.procedures
 
 
 class AnalyticsRow(AudreyModel):
@@ -60,7 +85,7 @@ class EncodeRequest(AudreyModel):
     source: str
     salience: float | None = Field(default=None, ge=0, le=1)
     tags: list[str] | None = None
-    context: dict[str, Any] | None = None
+    context: dict[StrictStr, StrictStr] | None = None
     affect: Affect | None = None
     causal: dict[str, Any] | None = None
     supersedes: str | None = None
@@ -75,7 +100,7 @@ class EncodeResponse(AudreyModel):
 class RecallRequest(AudreyModel):
     query: str
     limit: int | None = Field(default=None, ge=1, le=50)
-    context: dict[str, Any] | None = None
+    context: dict[StrictStr, StrictStr] | None = None
     mood: dict[str, Any] | None = None
     types: list[str] | None = None
     scope: str | None = None

{audrey_memory-1.2.2 → audrey_memory-2.1.0}/audrey_memory.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: audrey-memory
-Version: 1.2.2
+Version: 2.1.0
 Summary: Typed Python client for the Audrey LLM memory server
 Author: evilander
 License: MIT
@@ -45,7 +45,7 @@ python -m pip install -e .
 Start Audrey's REST API:
 
 ```bash
-npm install -g audrey --allow-scripts=better-sqlite3,onnxruntime-node,sharp,protobufjs
+npm install -g audrey
 audrey serve
 ```
 
@@ -67,11 +67,22 @@ memory_id = brain.encode(
 )
 
 results = brain.recall("stripe rate limits", limit=5)
-snapshot = brain.snapshot()
 brain.close()
 ```
 
-Restore snapshots only into an empty Audrey store, such as a sidecar started with a fresh `AUDREY_DATA_DIR`:
+Context keys and values must be strings, for example `context={"project": "support"}`.
+`brain.status()` returns the server's `episodes`, `semantics`, `procedures`, and
+`healthy` fields along with vector-index health details.
+
+Snapshot export and restore require the sidecar to be started with
+`AUDREY_ENABLE_ADMIN_TOOLS=1`. With that explicitly enabled:
+
+```python
+with Audrey(base_url="http://127.0.0.1:7437", api_key="secret") as brain:
+    snapshot = brain.snapshot()
+```
+
+Snapshots cover the whole store. Restore only into an empty Audrey store, such as a sidecar started with a fresh `AUDREY_DATA_DIR` and admin tools enabled:
 
 ```python
 restore_target = Audrey(base_url="http://127.0.0.1:7437", api_key="secret")

{audrey_memory-1.2.2 → audrey_memory-2.1.0}/tests/test_client.py

@@ -12,6 +12,7 @@ import unittest
 from pathlib import Path
 
 import httpx
+from pydantic import ValidationError
 
 PYTHON_ROOT = Path(__file__).resolve().parents[1]
 REPO_ROOT = PYTHON_ROOT.parent
@@ -29,6 +30,36 @@ def _free_port() -> int:
 
 
 class AudreyClientUnitTests(unittest.TestCase):
+    def test_status_models_counts_and_health_from_status_endpoint(self) -> None:
+        def handler(request: httpx.Request) -> httpx.Response:
+            self.assertEqual(request.url.path, "/v1/status")
+            return httpx.Response(200, json={
+                "episodes": 1, "semantics": 2, "procedures": 3, "healthy": True,
+                "vec_episodes": 1, "dimensions": 8, "recall_degraded": False,
+            })
+
+        with Audrey(base_url="http://audrey.test", transport=httpx.MockTransport(handler)) as client:
+            status = client.status()
+        self.assertEqual((status.episodes, status.semantics, status.procedures), (1, 2, 3))
+        self.assertEqual(status.episodic, status.episodes)
+        self.assertTrue(status.healthy)
+        self.assertEqual(status.dimensions, 8)
+
+    def test_context_rejects_non_strings_before_sending(self) -> None:
+        requests: list[httpx.Request] = []
+
+        def handler(request: httpx.Request) -> httpx.Response:
+            requests.append(request)
+            return httpx.Response(200, json={"id": "memory"})
+
+        with Audrey(base_url="http://audrey.test", transport=httpx.MockTransport(handler)) as client:
+            for value in (0, False, 1, None):
+                with self.subTest(value=value), self.assertRaises(ValidationError):
+                    client.encode("Context fixture", source="told-by-user", context={"retry": value})
+            client.encode("String context", source="told-by-user", context={"project": "billing"})
+        self.assertEqual(len(requests), 1)
+        self.assertEqual(json.loads(requests[0].content)["context"], {"project": "billing"})
+
     def test_sync_client_sends_auth_and_agent_headers(self) -> None:
         seen: dict[str, object] = {}
 
@@ -99,6 +130,19 @@ class AudreyClientUnitTests(unittest.TestCase):
 
 
 class AudreyAsyncClientUnitTests(unittest.IsolatedAsyncioTestCase):
+    async def test_status_models_counts_and_health_from_status_endpoint(self) -> None:
+        def handler(request: httpx.Request) -> httpx.Response:
+            self.assertEqual(request.url.path, "/v1/status")
+            return httpx.Response(200, json={
+                "episodes": 1, "semantics": 0, "procedures": 0, "healthy": True,
+            })
+
+        async with AsyncAudrey(base_url="http://audrey.test", transport=httpx.MockTransport(handler)) as client:
+            status = await client.status()
+        self.assertEqual(status.episodes, 1)
+        self.assertEqual(status.episodic, 1)
+        self.assertTrue(status.healthy)
+
     async def test_async_restore_accepts_the_server_imported_response(self) -> None:
         def handler(request: httpx.Request) -> httpx.Response:
             self.assertEqual(request.url.path, "/v1/import")
@@ -242,6 +286,11 @@ class AudreyClientIntegrationTests(unittest.TestCase):
 
             client.mark_used(memory_id)
 
+            status = client.status()
+            self.assertGreaterEqual(status.episodes, 1)
+            self.assertEqual(status.episodic, status.episodes)
+            self.assertTrue(status.healthy)
+
             results = client.recall("stripe rate limits", limit=5, scope="agent")
             self.assertGreaterEqual(len(results), 1)
             self.assertIn("Stripe", results[0].content)

audrey_memory-1.2.2/audrey_memory/_version.py

@@ -1 +0,0 @@
-__version__ = "1.2.2"