codespy-ai 1.2.4__tar.gz → 2.0.1__tar.gz

{codespy_ai-1.2.4 → codespy_ai-2.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codespy-ai
-Version: 1.2.4
+Version: 2.0.1
 Summary: Code review agent powered by DSPy
 License: MIT
 License-File: LICENSE
@@ -23,11 +23,12 @@ Requires-Dist: cloudpickle (>=3.1.2,<4.0.0)
 Requires-Dist: ddgs (>=8.0.0)
 Requires-Dist: dspy[mcp] (>=3.3.0,<4.0.0)
 Requires-Dist: gitpython (>=3.1.42)
+Requires-Dist: hindsight-api-slim (>=0.10.1)
 Requires-Dist: httpx (>=0.28.0)
 Requires-Dist: json-repair (>=0.60.1)
 Requires-Dist: litellm (>=1.84.0,<2.0.0)
 Requires-Dist: markdownify (>=0.15.0)
-Requires-Dist: mcp (>=1.29.0,<2.0.0)
+Requires-Dist: mcp (>=2.2,<3)
 Requires-Dist: pg0-embedded (>=0.15)
 Requires-Dist: psycopg[binary,pool] (>=3.1)
 Requires-Dist: pydantic (>=2.10.0)
@@ -204,7 +205,7 @@ codespy auto-discovers credentials from standard locations (`~/.aws/credentials`
 | **[Usage](docs/usage.md)** | CLI commands, Docker, GitHub Action, MCP server, output formats |
 | **[Configuration](docs/configuration.md)** | Environment variables, YAML config, model strategy, per-signature settings |
 | **[Architecture](docs/architecture.md)** | Pipeline design, DSPy signatures, supported languages |
-| **[Memory System](docs/memory.md)** | Hippocampus episodic memory for cross-review knowledge |
+| **[Memory System](docs/memory.md)** | Hippocampus (episodic), Cerebral (semantic store), Prefrontal (recall) |
 | **[Development](docs/development.md)** | Setup, build, test, lint |
 
 ---

{codespy_ai-1.2.4 → codespy_ai-2.0.1}/README.md

@@ -151,7 +151,7 @@ codespy auto-discovers credentials from standard locations (`~/.aws/credentials`
 | **[Usage](docs/usage.md)** | CLI commands, Docker, GitHub Action, MCP server, output formats |
 | **[Configuration](docs/configuration.md)** | Environment variables, YAML config, model strategy, per-signature settings |
 | **[Architecture](docs/architecture.md)** | Pipeline design, DSPy signatures, supported languages |
-| **[Memory System](docs/memory.md)** | Hippocampus episodic memory for cross-review knowledge |
+| **[Memory System](docs/memory.md)** | Hippocampus (episodic), Cerebral (semantic store), Prefrontal (recall) |
 | **[Development](docs/development.md)** | Setup, build, test, lint |
 
 ---

{codespy_ai-1.2.4 → codespy_ai-2.0.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "codespy-ai"
-version = "1.2.4"
+version = "2.0.1"
 description = "Code review agent powered by DSPy"
 readme = "README.md"
 license = "MIT"
@@ -47,12 +47,13 @@ tree-sitter-kotlin = ">=1.0"
 tree-sitter-objc = ">=3.0"
 tree-sitter-rust = ">=0.23"
 tree-sitter-hcl = ">=1.2.0"
-mcp = ">=1.29.0,<2.0.0"
+mcp = ">=2.2,<3"
 psycopg = {version = ">=3.1", extras = ["binary", "pool"]}
 beautifulsoup4 = ">=4.12.0"
 markdownify = ">=0.15.0"
 ddgs = ">=8.0.0"
 pg0-embedded = ">=0.15"
+hindsight-api-slim = ">=0.10.1"
 
 [tool.poetry.group.dev.dependencies]
 pytest = ">=8.0.0"

{codespy_ai-1.2.4 → codespy_ai-2.0.1}/src/codespy/__init__.py

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

{codespy_ai-1.2.4 → codespy_ai-2.0.1}/src/codespy/agents/context_safe.py

@@ -4,6 +4,7 @@ Provides transparent fallback to RLM (Recursive Language Model) when inputs
 exceed the model's context window. Used to wrap all DSPy signature modules.
 """
 
+import enum
 import logging
 import re
 
@@ -19,6 +20,47 @@ _RE_CONTEXT_LENGTH = re.compile(r"maximum context length is \d+ tokens", re.IGNO
 # Below this, context rot is not a concern regardless of ratio.
 _MIN_RLM_THRESHOLD_TOKENS = 8192
 
+# Replaces dspy.RLM's stock `reasoning` desc ("Think step-by-step: what do you
+# know? What remains? Plan your next action."), which on its own triggers Bedrock
+# content_filter on Claude Opus 5.5.
+RLM_REASONING_DESC = "Short plan for the next step."
+
+
+def _plain(value):  # type: ignore[no-untyped-def]
+    """Replace Enum members with their values, recursively.
+
+    dspy's PythonInterpreter injects inputs as `name = repr(value)`; an Enum repr
+    (`<IssueCategory.BUG: 'bug'>`) is not valid Python, so every REPL execution
+    fails with an empty SyntaxError.
+    """
+    if isinstance(value, enum.Enum):
+        return value.value
+    if isinstance(value, (list, tuple, set)):
+        return [_plain(v) for v in value]
+    if isinstance(value, dict):
+        return {_plain(k): _plain(v) for k, v in value.items()}
+    return value
+
+
+class CodespyRLM(dspy.RLM):
+    """dspy.RLM adjusted for codespy.
+
+    - Neutral action-signature `reasoning` description (Bedrock content_filter).
+    - Enum inputs passed as plain values (REPL variable injection).
+    """
+
+    def forward(self, **kwargs):  # type: ignore[no-untyped-def]
+        return super().forward(**{k: _plain(v) for k, v in kwargs.items()})
+
+    async def aforward(self, **kwargs):  # type: ignore[no-untyped-def]
+        return await super().aforward(**{k: _plain(v) for k, v in kwargs.items()})
+
+    def _build_signatures(self):  # type: ignore[no-untyped-def]
+        action_sig, extract_sig = super()._build_signatures()
+        if "reasoning" in action_sig.output_fields:
+            action_sig = action_sig.with_updated_fields("reasoning", desc=RLM_REASONING_DESC)
+        return action_sig, extract_sig
+
 
 def estimate_context_overflow(model: str, max_tokens: int, input_text: str) -> bool:
     """Estimate whether input + max_tokens would exceed the model's context window.
@@ -60,7 +102,7 @@ class ContextSafe(dspy.Module):
     - Try/catch: if the inner module raises a context overflow error (unknown
       models where pre-flight can't estimate), catches it and retries with RLM.
 
-    Transparent to inject_context_memory — delegates .signature to inner module.
+    Transparent to signature injection — delegates .signature to inner module.
     """
 
     def __init__(
@@ -186,7 +228,7 @@ class ContextSafe(dspy.Module):
             return False, ""
 
     def _get_current_signature(self):
-        """Get current signature (may include context_memory if inject_context_memory was called)."""
+        """Get current signature."""
         sig = getattr(self.module, "signature", None)
         if sig is not None:
             return sig
@@ -197,9 +239,10 @@ class ContextSafe(dspy.Module):
 
     def _create_rlm_fallback(self) -> dspy.RLM:
         """Create RLM with current signature and tools."""
-        return dspy.RLM(
+        return CodespyRLM(
             self._get_current_signature(),
             tools=self._tools,
             max_iters=self._max_iters,
             max_llm_calls=self._max_llm_calls,
+            verbose=logger.isEnabledFor(logging.DEBUG),
         )