selectools 0.16.0__tar.gz → 0.16.2__tar.gz

{selectools-0.16.0/src/selectools.egg-info → selectools-0.16.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: selectools
-Version: 0.16.0
+Version: 0.16.2
 Summary: Production-ready AI agents with tool calling, structured output, execution traces, and RAG. Provider-agnostic (OpenAI, Anthropic, Gemini, Ollama) with fallback chains, batch processing, tool policies, streaming, caching, and cost tracking.
 Author-email: John <johnnichev@gmail.com>
 License-Expression: LGPL-3.0-or-later
@@ -65,7 +65,7 @@ Dynamic: license-file
 - **Entity Memory** — Auto-extract named entities (person, org, project) across turns with LRU-pruned registry and system prompt injection.
 - **Knowledge Graph** — Relationship triple extraction with in-memory and SQLite storage. Query-relevant triples auto-injected into prompts.
 - **Cross-Session Knowledge** — Daily logs + persistent facts with auto-registered `remember` tool. Give your agent durable memory across conversations.
-- **182 new tests** (total: 1365)
+- **182 new tests** (total: 1421 with v0.16.1 consolidation)
 
 > Full changelog: [CHANGELOG.md](https://github.com/johnnichev/selectools/blob/main/CHANGELOG.md)
 
@@ -135,7 +135,7 @@ Dynamic: license-file
 - **Cross-Session Knowledge**: Daily logs + persistent memory with `remember` tool
 - **37 Examples**: RAG, hybrid search, streaming, structured output, traces, batch, policy, observer, guardrails, audit, sessions, entity memory, knowledge graph, and more
 - **AgentObserver Protocol**: 19 lifecycle events with `run_id` correlation, `LoggingObserver`, OTel export
-- **1365 Tests**: Unit, integration, regression, and E2E with real API calls
+- **1421 Tests**: Unit, integration, regression, and E2E with real API calls
 
 ## Install
 

{selectools-0.16.0 → selectools-0.16.2}/README.md

@@ -16,7 +16,7 @@
 - **Entity Memory** — Auto-extract named entities (person, org, project) across turns with LRU-pruned registry and system prompt injection.
 - **Knowledge Graph** — Relationship triple extraction with in-memory and SQLite storage. Query-relevant triples auto-injected into prompts.
 - **Cross-Session Knowledge** — Daily logs + persistent facts with auto-registered `remember` tool. Give your agent durable memory across conversations.
-- **182 new tests** (total: 1365)
+- **182 new tests** (total: 1421 with v0.16.1 consolidation)
 
 > Full changelog: [CHANGELOG.md](https://github.com/johnnichev/selectools/blob/main/CHANGELOG.md)
 
@@ -86,7 +86,7 @@
 - **Cross-Session Knowledge**: Daily logs + persistent memory with `remember` tool
 - **37 Examples**: RAG, hybrid search, streaming, structured output, traces, batch, policy, observer, guardrails, audit, sessions, entity memory, knowledge graph, and more
 - **AgentObserver Protocol**: 19 lifecycle events with `run_id` correlation, `LoggingObserver`, OTel export
-- **1365 Tests**: Unit, integration, regression, and E2E with real API calls
+- **1421 Tests**: Unit, integration, regression, and E2E with real API calls
 
 ## Install
 

{selectools-0.16.0 → selectools-0.16.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "selectools"
-version = "0.16.0"
+version = "0.16.2"
 description = "Production-ready AI agents with tool calling, structured output, execution traces, and RAG. Provider-agnostic (OpenAI, Anthropic, Gemini, Ollama) with fallback chains, batch processing, tool policies, streaming, caching, and cost tracking."
 readme = "README.md"
 requires-python = ">=3.9"

{selectools-0.16.0 → selectools-0.16.2}/src/selectools/__init__.py

@@ -1,6 +1,6 @@
 """Public exports for the selectools package."""
 
-__version__ = "0.16.0"
+__version__ = "0.16.2"
 
 # Import submodules (lazy loading for optional dependencies)
 from . import embeddings, guardrails, models, rag, toolbox

{selectools-0.16.0 → selectools-0.16.2}/src/selectools/agent/core.py

@@ -1976,6 +1976,7 @@ class Agent:
         _history_checkpoint = len(self._history)
         iteration = 0
 
+        original_system_prompt = self._system_prompt
         self._wire_fallback_observer(run_id)
         self._notify_observers("on_run_start", run_id, messages, self._system_prompt)
 
@@ -2112,8 +2113,8 @@ class Agent:
                         usage=copy.copy(self.usage),
                     )
                     self._notify_observers("on_run_end", run_id, _result)
-                    yield _result
                     self._call_hook("on_agent_end", final_response, self.usage)
+                    yield _result
                     return
 
                 if self.config.routing_only:
@@ -2260,9 +2261,9 @@ class Agent:
                 usage=copy.copy(self.usage),
             )
             self._notify_observers("on_run_end", run_id, _result)
-            yield _result
             self._memory_add(final_msg, run_id)
             self._call_hook("on_agent_end", final_msg, self.usage)
+            yield _result
             return
         except Exception as exc:
             if not self.memory:
@@ -2274,6 +2275,7 @@ class Agent:
             raise
         finally:
             self._unwire_fallback_observer()
+            self._system_prompt = original_system_prompt
 
     async def arun(
         self,
@@ -2637,6 +2639,14 @@ class Agent:
                                     cost=0.0,
                                     chunk_count=chunk_counter["count"],
                                 )
+                            if self.usage.iterations:
+                                self.usage.tool_usage[tool.name] = (
+                                    self.usage.tool_usage.get(tool.name, 0) + 1
+                                )
+                                self.usage.tool_tokens[tool.name] = (
+                                    self.usage.tool_tokens.get(tool.name, 0)
+                                    + self.usage.iterations[-1].total_tokens
+                                )
                         except Exception as exc:
                             duration = time.time() - start_time
                             self._call_hook("on_tool_error", tool.name, exc, parameters)

{selectools-0.16.0 → selectools-0.16.2}/src/selectools/entity_memory.py

@@ -8,6 +8,7 @@ extraction, deduplication, and LRU pruning.
 from __future__ import annotations
 
 import json
+import threading
 import time
 from dataclasses import dataclass, field
 from typing import Any, Dict, List, Optional
@@ -92,15 +93,17 @@ class EntityMemory:
         self._max_entities = max_entities
         self._relevance_window = relevance_window
         self._entities: Dict[str, Entity] = {}
+        self._lock = threading.Lock()
 
     @property
     def entities(self) -> List[Entity]:
         """All tracked entities, sorted by most recently mentioned."""
-        return sorted(
-            self._entities.values(),
-            key=lambda e: e.last_mentioned,
-            reverse=True,
-        )
+        with self._lock:
+            return sorted(
+                self._entities.values(),
+                key=lambda e: e.last_mentioned,
+                reverse=True,
+            )
 
     def extract_entities(
         self,
@@ -174,28 +177,31 @@ class EntityMemory:
 
         Deduplicates by name (case-insensitive), updates mention counts
         and attributes, and prunes if over ``max_entities``.
+
+        Thread-safe: protected by an internal lock.
         """
         now = time.time()
-        for entity in entities:
-            key = entity.name.lower()
-            if key in self._entities:
-                existing = self._entities[key]
-                existing.mention_count += 1
-                existing.last_mentioned = now
-                existing.attributes.update(entity.attributes)
-            else:
-                entity.last_mentioned = now
-                self._entities[key] = entity
-
-        # LRU prune: remove least recently mentioned
-        if len(self._entities) > self._max_entities:
-            sorted_keys = sorted(
-                self._entities.keys(),
-                key=lambda k: self._entities[k].last_mentioned,
-            )
-            excess = len(self._entities) - self._max_entities
-            for k in sorted_keys[:excess]:
-                del self._entities[k]
+        with self._lock:
+            for entity in entities:
+                key = entity.name.lower()
+                if key in self._entities:
+                    existing = self._entities[key]
+                    existing.mention_count += 1
+                    existing.last_mentioned = now
+                    existing.attributes.update(entity.attributes)
+                else:
+                    entity.last_mentioned = now
+                    self._entities[key] = entity
+
+            # LRU prune: remove least recently mentioned
+            if len(self._entities) > self._max_entities:
+                sorted_keys = sorted(
+                    self._entities.keys(),
+                    key=lambda k: self._entities[k].last_mentioned,
+                )
+                excess = len(self._entities) - self._max_entities
+                for k in sorted_keys[:excess]:
+                    del self._entities[k]
 
     def build_context(self) -> str:
         """Build a context string for prompt injection.

{selectools-0.16.0 → selectools-0.16.2}/src/selectools/knowledge.py

@@ -9,6 +9,7 @@ for long-term facts.
 from __future__ import annotations
 
 import os
+import threading
 import time
 from datetime import datetime, timedelta
 from typing import Any, Dict, List, Optional
@@ -39,6 +40,7 @@ class KnowledgeMemory:
         self._directory = directory
         self._recent_days = recent_days
         self._max_context_chars = max_context_chars
+        self._lock = threading.Lock()
         os.makedirs(directory, exist_ok=True)
 
     @property
@@ -66,17 +68,18 @@ class KnowledgeMemory:
         timestamp = now.strftime("%Y-%m-%d %H:%M:%S")
         entry = f"[{timestamp}] [{category}] {content}"
 
-        # Write to daily log
+        # Write to daily log (thread-safe)
         today = now.strftime("%Y-%m-%d")
         log_path = os.path.join(self._directory, f"{today}.log")
-        with open(log_path, "a", encoding="utf-8") as f:
-            f.write(entry + "\n")
-
-        # Optionally write to persistent memory
-        if persistent:
-            mem_path = os.path.join(self._directory, "MEMORY.md")
-            with open(mem_path, "a", encoding="utf-8") as f:
-                f.write(f"- [{category}] {content}\n")
+        with self._lock:
+            with open(log_path, "a", encoding="utf-8") as f:
+                f.write(entry + "\n")
+
+            # Optionally write to persistent memory
+            if persistent:
+                mem_path = os.path.join(self._directory, "MEMORY.md")
+                with open(mem_path, "a", encoding="utf-8") as f:
+                    f.write(f"- [{category}] {content}\n")
 
         return f"Remembered: {content}"
 

{selectools-0.16.0 → selectools-0.16.2}/src/selectools/knowledge_graph.py

@@ -133,7 +133,9 @@ class SQLiteTripleStore:
         self._ensure_table()
 
     def _connect(self) -> sqlite3.Connection:
-        return sqlite3.connect(self._db_path)
+        conn = sqlite3.connect(self._db_path)
+        conn.execute("PRAGMA journal_mode=WAL")
+        return conn
 
     def _ensure_table(self) -> None:
         conn = self._connect()
@@ -249,7 +251,7 @@ class SQLiteTripleStore:
     def count(self) -> int:
         conn = self._connect()
         try:
-            return conn.execute("SELECT COUNT(*) FROM triples").fetchone()[0]
+            return int(conn.execute("SELECT COUNT(*) FROM triples").fetchone()[0])
         finally:
             conn.close()
 
@@ -416,7 +418,7 @@ class KnowledgeGraphMemory:
         keywords = [w for w in query.lower().split() if len(w) > 2]
         if not keywords:
             return []
-        results = self._store.query(keywords)
+        results: List[Triple] = self._store.query(keywords)
         return results[: self._max_context_triples]
 
     def build_context(self, query: str = "") -> str:

{selectools-0.16.0 → selectools-0.16.2}/src/selectools/memory.py

@@ -156,6 +156,7 @@ class ConversationMemory:
         mem._messages = [Message.from_dict(m) for m in data.get("messages", [])]
         mem._summary = data.get("summary")
         mem._last_trimmed = []
+        mem._fix_tool_pair_boundary()
         return mem
 
     def _enforce_limits(self) -> None:

{selectools-0.16.0 → selectools-0.16.2}/src/selectools/sessions.py

@@ -92,7 +92,7 @@ class JsonFileSessionStore:
     def _is_expired(self, data: Dict[str, Any]) -> bool:
         if self._default_ttl is None:
             return False
-        updated_at = data.get("updated_at", data.get("created_at", 0))
+        updated_at: float = data.get("updated_at", data.get("created_at", 0))
         return (time.time() - updated_at) > self._default_ttl
 
     # -- public API --------------------------------------------------------
@@ -304,7 +304,7 @@ class SQLiteSessionStore:
         try:
             cursor = conn.execute("DELETE FROM sessions WHERE session_id = ?", (session_id,))
             conn.commit()
-            return cursor.rowcount > 0
+            return int(cursor.rowcount) > 0
         finally:
             conn.close()
 
@@ -433,7 +433,7 @@ class RedisSessionStore:
         key = self._key(session_id)
         meta_key = self._meta_key(session_id)
         removed = self._client.delete(key, meta_key)
-        return removed > 0
+        return int(removed) > 0
 
     def exists(self, session_id: str) -> bool:
         return bool(self._client.exists(self._key(session_id)))

{selectools-0.16.0 → selectools-0.16.2/src/selectools.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: selectools
-Version: 0.16.0
+Version: 0.16.2
 Summary: Production-ready AI agents with tool calling, structured output, execution traces, and RAG. Provider-agnostic (OpenAI, Anthropic, Gemini, Ollama) with fallback chains, batch processing, tool policies, streaming, caching, and cost tracking.
 Author-email: John <johnnichev@gmail.com>
 License-Expression: LGPL-3.0-or-later
@@ -65,7 +65,7 @@ Dynamic: license-file
 - **Entity Memory** — Auto-extract named entities (person, org, project) across turns with LRU-pruned registry and system prompt injection.
 - **Knowledge Graph** — Relationship triple extraction with in-memory and SQLite storage. Query-relevant triples auto-injected into prompts.
 - **Cross-Session Knowledge** — Daily logs + persistent facts with auto-registered `remember` tool. Give your agent durable memory across conversations.
-- **182 new tests** (total: 1365)
+- **182 new tests** (total: 1421 with v0.16.1 consolidation)
 
 > Full changelog: [CHANGELOG.md](https://github.com/johnnichev/selectools/blob/main/CHANGELOG.md)
 
@@ -135,7 +135,7 @@ Dynamic: license-file
 - **Cross-Session Knowledge**: Daily logs + persistent memory with `remember` tool
 - **37 Examples**: RAG, hybrid search, streaming, structured output, traces, batch, policy, observer, guardrails, audit, sessions, entity memory, knowledge graph, and more
 - **AgentObserver Protocol**: 19 lifecycle events with `run_id` correlation, `LoggingObserver`, OTel export
-- **1365 Tests**: Unit, integration, regression, and E2E with real API calls
+- **1421 Tests**: Unit, integration, regression, and E2E with real API calls
 
 ## Install
 

{selectools-0.16.0 → selectools-0.16.2}/src/selectools.egg-info/SOURCES.txt

@@ -87,18 +87,24 @@ tests/test_cache.py
 tests/test_cache_redis.py
 tests/test_cli.py
 tests/test_coherence.py
+tests/test_consolidation_regression.py
 tests/test_entity_memory.py
 tests/test_env.py
 tests/test_guardrails.py
 tests/test_knowledge.py
 tests/test_knowledge_graph.py
 tests/test_memory.py
+tests/test_memory_async.py
+tests/test_memory_boundary.py
+tests/test_memory_integration.py
 tests/test_parser.py
 tests/test_policy.py
 tests/test_prompt.py
 tests/test_routing_mode.py
 tests/test_security.py
 tests/test_sessions.py
+tests/test_sessions_edge_cases.py
+tests/test_sessions_redis.py
 tests/test_structured.py
 tests/test_summarize_on_trim.py
 tests/test_trace.py