codespy-ai 2.1.0__tar.gz → 2.1.1__tar.gz

{codespy_ai-2.1.0 → codespy_ai-2.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codespy-ai
-Version: 2.1.0
+Version: 2.1.1
 Summary: Code review agent powered by DSPy
 License: MIT
 License-File: LICENSE

{codespy_ai-2.1.0 → codespy_ai-2.1.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "codespy-ai"
-version = "2.1.0"
+version = "2.1.1"
 description = "Code review agent powered by DSPy"
 readme = "README.md"
 license = "MIT"

{codespy_ai-2.1.0 → codespy_ai-2.1.1}/src/codespy/__init__.py

@@ -1,3 +1,3 @@
 """codespy - Code review agent powered by DSPy."""
 
-__version__ = "2.1.0"
+__version__ = "2.1.1"

{codespy_ai-2.1.0 → codespy_ai-2.1.1}/src/codespy/agents/memory/postgres.py

@@ -33,6 +33,7 @@ class EpisodeStore:
         schema: str | None = None,
         min_size: int = 1,
         max_size: int = 4,
+        persist_recalls: bool = False,
     ):
         """Create store with a psycopg ConnectionPool, scoped to a bank.
 
@@ -45,8 +46,12 @@ class EpisodeStore:
                     (e.g. "episodic", "semantic"). None = server default (public).
             min_size: Minimum connections in pool
             max_size: Maximum connections in pool
+            persist_recalls: Write Prefrontal recall records to the ``recalls``
+                    table. When False (default), ``save_episode`` skips them
+                    and ``save_recalls`` is a no-op. The table is still created.
         """
         self.bank_id = bank_id
+        self.persist_recalls = persist_recalls
         self._schema = schema
         # Inject search_path into the connection string so every pooled
         # connection targets the right schema automatically.
@@ -567,19 +572,20 @@ class EpisodeStore:
                         (self.bank_id, str(episode.id), name, content),
                     )
 
-                # 8. Insert recalls (Prefrontal monitoring) - now with episode_id set
-                self._insert_recalls(
-                    cur, episode.recalls or [], run_id=episode.run_id, episode_id=episode.id, task=episode.task
-                )
+                # 8. Insert recalls (Prefrontal monitoring, opt-in)
+                if self.persist_recalls:
+                    self._insert_recalls(
+                        cur, episode.recalls or [], run_id=episode.run_id, episode_id=episode.id, task=episode.task
+                    )
 
                 conn.commit()
 
         logger.info(
-            "save_episode: persisted episode %s (bank=%s, task=%s, topics=%d, observations=%d, recalls=%d)",
+            "save_episode: persisted episode %s (bank=%s, task=%s, topics=%d, observations=%d, recalls=%s)",
             episode.id, self.bank_id, episode.task,
             len(episode.context_memory.topics),
             len(episode.context_memory.all_observations()),
-            len(episode.recalls or []),
+            len(episode.recalls or []) if self.persist_recalls else "off",
         )
 
     def load_context(
@@ -836,8 +842,10 @@ class EpisodeStore:
             run_id: Run identifier
             episode_id: Optional episode UUID (None for run-level recalls)
             task: Task name (consumer task or "review" for run-level load)
+
+        No-op unless the store was built with ``persist_recalls=True``.
         """
-        if not records:
+        if not self.persist_recalls or not records:
             return
 
         with self._pool.connection() as conn:

{codespy_ai-2.1.0 → codespy_ai-2.1.1}/src/codespy/config_memory.py

@@ -81,6 +81,11 @@ class HippocampusConfig(BaseModel):
     # Whether to apply head+tail trajectory bounding before distillation.
     compact_trajectory: bool = True
 
+    # Store Prefrontal recalls in the episodic ``recalls`` table (monitoring
+    # only). Off by default: every row holds the full recalled text, so the
+    # table grows fast.
+    persist_recalls: bool = False
+
     # Token budgets
     max_hippocampus_tokens: int = Field(default=16384)
     max_hippocampus_item_tokens: int = Field(default=512)
@@ -343,13 +348,14 @@ def get_episode_store(settings: Settings) -> EpisodeStore | None:
     mem = settings.memory
     bank_id = mem.bank_id or DEFAULT_BANK_ID
     schema = EPISODIC_SCHEMA
+    persist_recalls = mem.hippocampus.persist_recalls
 
     # Try external PostgreSQL first
     uri = mem.postgres.build_uri()
     if uri:
         from codespy.agents.memory.postgres import EpisodeStore
 
-        _store = EpisodeStore(uri, bank_id, schema=schema)
+        _store = EpisodeStore(uri, bank_id, schema=schema, persist_recalls=persist_recalls)
         logger.info(f"EpisodeStore connected to external PostgreSQL (bank={bank_id}, schema={schema})")
     else:
         # Try pg0-embedded for local dev
@@ -359,7 +365,7 @@ def get_episode_store(settings: Settings) -> EpisodeStore | None:
             uri = get_pg0_uri(name=mem.pg0.name, port=mem.pg0.port, data_dir=mem.pg0.data_dir)
             from codespy.agents.memory.postgres import EpisodeStore
 
-            _store = EpisodeStore(uri, bank_id, schema=schema)
+            _store = EpisodeStore(uri, bank_id, schema=schema, persist_recalls=persist_recalls)
             logger.info(f"EpisodeStore connected to pg0-embedded PostgreSQL (bank={bank_id}, schema={schema})")
         except ImportError:
             logger.warning(

{codespy_ai-2.1.0 → codespy_ai-2.1.1}/src/codespy/workflows/review/pipeline.py

@@ -299,10 +299,10 @@ class ReviewPipeline(dspy.Module):
                 packages=all_packages[:MAX_ENTITY_TERMS],
             )
             pf_text = run_pf.load(facets)
-            # Persist run-level recalls
+            # Persist run-level recalls (opt-in: memory.hippocampus.persist_recalls)
             try:
                 store = get_episode_store(self.settings)
-                if store and run_pf.recalls:
+                if store and store.persist_recalls and run_pf.recalls:
                     store.save_recalls(
                         run_pf.recalls,
                         run_id=run_id,