first commit

test_memory.py

@@ -0,0 +1,113 @@
+"""Quick smoke test for the memory system."""
+
+import asyncio
+import os
+import shutil
+
+from memory import MemoryManager
+from memory.types import MemoryConfig
+from memory.internal import chunk_markdown, hash_text, list_memory_files
+
+
+def test_internals():
+    print("── Internal utilities ──")
+
+    # Hashing
+    h = hash_text("hello world")
+    assert len(h) == 64
+    print(f"✅ hash_text: {h[:16]}...")
+
+    # Chunking
+    text = "# Title\n\nLine1\nLine2\nLine3\n\n## Section\n\nMore text here"
+    chunks = chunk_markdown(text, chunk_tokens=50, chunk_overlap=10)
+    assert len(chunks) >= 1
+    print(f"✅ chunk_markdown: {len(chunks)} chunks")
+    for c in chunks:
+        print(f"   lines {c.start_line}-{c.end_line}: {c.text[:50]!r}")
+
+    print()
+
+
+async def test_manager():
+    print("── MemoryManager ──")
+
+    # Clean slate
+    test_dir = "/tmp/aetheel_test_workspace"
+    test_db = "/tmp/aetheel_test_memory.db"
+    for p in [test_dir, test_db]:
+        if os.path.exists(p):
+            if os.path.isdir(p):
+                shutil.rmtree(p)
+            else:
+                os.remove(p)
+
+    config = MemoryConfig(
+        workspace_dir=test_dir,
+        db_path=test_db,
+    )
+
+    mgr = MemoryManager(config)
+    print(f"✅ Created: workspace={mgr._workspace_dir}")
+
+    # Identity files
+    soul = mgr.read_soul()
+    assert soul and len(soul) > 0
+    print(f"✅ SOUL.md: {len(soul)} chars")
+
+    user = mgr.read_user()
+    assert user and len(user) > 0
+    print(f"✅ USER.md: {len(user)} chars")
+
+    memory = mgr.read_long_term_memory()
+    assert memory and len(memory) > 0
+    print(f"✅ MEMORY.md: {len(memory)} chars")
+
+    # Append to memory
+    mgr.append_to_memory("Test entry: Python 3.14 works great!")
+    memory2 = mgr.read_long_term_memory()
+    assert len(memory2) > len(memory)
+    print(f"✅ Appended to MEMORY.md: {len(memory2)} chars")
+
+    # Log a session
+    log_path = mgr.log_session(
+        "User: Hello!\nAssistant: Hi, how can I help?",
+        channel="terminal",
+    )
+    assert os.path.exists(log_path)
+    print(f"✅ Session logged: {log_path}")
+
+    # Sync
+    print("\n⏳ Syncing (loading embedding model on first run)...")
+    stats = await mgr.sync()
+    print(f"✅ Sync complete:")
+    for k, v in stats.items():
+        print(f"   {k}: {v}")
+
+    # Search
+    print("\n🔍 Searching for 'personality values'...")
+    results = await mgr.search("personality values")
+    print(f"✅ Found {len(results)} results:")
+    for i, r in enumerate(results[:3]):
+        print(f"   [{i+1}] score={r.score:.3f}  path={r.path}  lines={r.start_line}-{r.end_line}")
+        print(f"       {r.snippet[:80]}...")
+
+    print("\n🔍 Searching for 'preferences'...")
+    results2 = await mgr.search("preferences")
+    print(f"✅ Found {len(results2)} results:")
+    for i, r in enumerate(results2[:3]):
+        print(f"   [{i+1}] score={r.score:.3f}  path={r.path}  lines={r.start_line}-{r.end_line}")
+        print(f"       {r.snippet[:80]}...")
+
+    # Status
+    print("\n📊 Status:")
+    status = mgr.status()
+    for k, v in status.items():
+        print(f"   {k}: {v}")
+
+    mgr.close()
+    print("\n✅ All memory system tests passed!")
+
+
+if __name__ == "__main__":
+    test_internals()
+    asyncio.run(test_manager())