deepwrap 0.2.2__tar.gz → 0.2.3__tar.gz

{deepwrap-0.2.2 → deepwrap-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: deepwrap
-Version: 0.2.2
+Version: 0.2.3
 Summary: Python SDK, CLI, and local FastAPI wrapper for DeepSeek Chat.
 Author-email: Nika Kudukhashvili <nikakuduxashvili0@gmail.com>
 Maintainer-email: Nika Kudukhashvili <nikakuduxashvili0@gmail.com>
@@ -749,7 +749,7 @@ Example response:
 {
   "ok": true,
   "app": "deepwrap",
-  "version": "0.2.2",
+  "version": "0.2.3",
   "token_configured": true,
   "cached_clients": 1,
   "active_sessions": 0

{deepwrap-0.2.2 → deepwrap-0.2.3}/README.md

@@ -694,7 +694,7 @@ Example response:
 {
   "ok": true,
   "app": "deepwrap",
-  "version": "0.2.2",
+  "version": "0.2.3",
   "token_configured": true,
   "cached_clients": 1,
   "active_sessions": 0

{deepwrap-0.2.2 → deepwrap-0.2.3}/deepwrap/client.py

@@ -96,6 +96,7 @@ class Client:
 
         self._conversations: dict[str, ChatSession] = {}
         self._responses: dict[str, str]             = {}
+        self._loaded_sessions: set[str]             = set()
 
     def register_tool(
         self,
@@ -145,6 +146,14 @@ class Client:
             return ""
         sections: list[str] = []
         if session_id:
+            if session_id not in self._loaded_sessions:
+                self._loaded_sessions.add(session_id)
+                priority_mems = self.memory.get_priority_memories(limit=20, min_importance=0.9)
+                if priority_mems:
+                    sections.append(
+                        "[PRIORITY MEMORIES]\n"
+                        + "\n".join(f"- ({item['id']}) {item['content']}" for item in priority_mems)
+                    )
             context = self.memory.session_context(session_id, limit=20)
             if context:
                 sections.append("[RESUMED SESSION]\n" + context)

{deepwrap-0.2.2 → deepwrap-0.2.3}/deepwrap/config.py

@@ -3,7 +3,7 @@ import os
 from dataclasses import dataclass, field
 from typing import Dict, Optional
 
-PROJECT_VERSION = "0.2.2"
+PROJECT_VERSION = "0.2.3"
 
 @dataclass(frozen=True)
 class Settings:

{deepwrap-0.2.2 → deepwrap-0.2.3}/deepwrap/memory.py

@@ -146,6 +146,22 @@ class MemoryStore:
             ).fetchall()
         return [self._row(row) for row in rows]
 
+    def get_priority_memories(self, limit: int = 20, min_importance: float = 0.9) -> list[dict[str, Any]]:
+        with self._lock:
+            rows = self._connection.execute(
+                "SELECT * FROM memories WHERE namespace = ? AND importance >= ? ORDER BY updated_at DESC LIMIT ?",
+                (self.namespace, min_importance, limit),
+            ).fetchall()
+        return [self._row(row) for row in rows]
+
+    def has_turns(self, session_id: str) -> bool:
+        with self._lock:
+            row = self._connection.execute(
+                "SELECT COUNT(*) FROM turns WHERE session_id = ?",
+                (session_id,),
+            ).fetchone()
+        return row[0] > 0 if row else False
+
     def forget(self, memory_id: str) -> dict[str, Any]:
         memory = self.get(memory_id)
         with self._lock, self._connection:

{deepwrap-0.2.2 → deepwrap-0.2.3}/deepwrap.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: deepwrap
-Version: 0.2.2
+Version: 0.2.3
 Summary: Python SDK, CLI, and local FastAPI wrapper for DeepSeek Chat.
 Author-email: Nika Kudukhashvili <nikakuduxashvili0@gmail.com>
 Maintainer-email: Nika Kudukhashvili <nikakuduxashvili0@gmail.com>
@@ -749,7 +749,7 @@ Example response:
 {
   "ok": true,
   "app": "deepwrap",
-  "version": "0.2.2",
+  "version": "0.2.3",
   "token_configured": true,
   "cached_clients": 1,
   "active_sessions": 0

{deepwrap-0.2.2 → deepwrap-0.2.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "deepwrap"
-version = "0.2.2"
+version = "0.2.3"
 description = "Python SDK, CLI, and local FastAPI wrapper for DeepSeek Chat."
 readme = "README.md"
 requires-python = ">=3.10"

{deepwrap-0.2.2 → deepwrap-0.2.3}/tests/test_memory.py

@@ -55,10 +55,47 @@ class MemoryStoreTests(unittest.TestCase):
             {tool.name for tool in self.memory.definitions},
             {"remember", "recall", "list_memories", "forget", "list_sessions"},
         )
-        self.assertEqual(
-            {tool.name for tool in self.memory.definitions},
-            set(self.memory.functions),
-        )
+    def test_priority_memories_and_turns(self):
+        # 1. Test has_turns
+        session_id = self.memory.create_session("expert")
+        self.assertFalse(self.memory.has_turns(session_id))
+        self.memory.add_turn(session_id, "user", "hi")
+        self.assertTrue(self.memory.has_turns(session_id))
+
+        # 2. Test get_priority_memories
+        self.memory.remember("Low priority memory", importance=0.5)
+        high1 = self.memory.remember("High priority memory 1", importance=0.9)
+        high2 = self.memory.remember("High priority memory 2", importance=0.95)
+        
+        priority = self.memory.get_priority_memories(limit=10, min_importance=0.9)
+        self.assertEqual(len(priority), 2)
+        # Verify ordering is updated_at DESC (last inserted first)
+        self.assertEqual(priority[0]["id"], high2["id"])
+        self.assertEqual(priority[1]["id"], high1["id"])
+
+    def test_client_memory_context_priority_load(self):
+        from deepwrap.client import Client
+        
+        # Write some memories
+        self.memory.remember("Low priority", importance=0.5)
+        self.memory.remember("High priority 1", importance=0.9)
+        self.memory.remember("High priority 2", importance=0.95)
+        
+        client = Client(api_key="test-key", memory_path=self.path, memory_namespace="project-a")
+        session_id = client.memory.create_session("expert")
+        
+        # 1. First load of a session
+        context = client.memory_context("query", session_id)
+        self.assertIn("[PRIORITY MEMORIES]", context)
+        self.assertIn("High priority 1", context)
+        self.assertIn("High priority 2", context)
+        self.assertNotIn("Low priority", context)
+        
+        # 2. Second load of the same session -> should not load priority memories again
+        context2 = client.memory_context("query", session_id)
+        self.assertNotIn("[PRIORITY MEMORIES]", context2)
+
+
 
 
 if __name__ == "__main__":