code-puppy 0.0.751__tar.gz → 0.0.753__tar.gz

{code_puppy-0.0.751 → code_puppy-0.0.753}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: code-puppy
-Version: 0.0.751
+Version: 0.0.753
 Summary: Code generation agent
 Project-URL: repository, https://github.com/mpfaffenberger/code_puppy
 Project-URL: HomePage, https://github.com/mpfaffenberger/code_puppy
@@ -20,7 +20,7 @@ Requires-Dist: anthropic<1,>=0.108.0
 Requires-Dist: anyio>=4.7.0
 Requires-Dist: azure-identity>=1.15.0
 Requires-Dist: boto3>=1.43.9
-Requires-Dist: code-puppy-core-plugins>=0.0.8
+Requires-Dist: code-puppy-core-plugins>=0.0.12
 Requires-Dist: httpx[http2]>=0.24.1
 Requires-Dist: json-repair>=0.46.2
 Requires-Dist: keyring>=24.0.0

{code_puppy-0.0.751 → code_puppy-0.0.753}/code_puppy/agents/_builder.py

@@ -19,6 +19,7 @@ from pydantic_ai import Agent as PydanticAgent
 from pydantic_ai.capabilities import ProcessHistory
 
 from code_puppy.agents._compaction import make_history_processor
+from code_puppy.agents._model_message_transform import build_model_message_transform
 from code_puppy.agents._output_limits import (
     build_response_clamp,
     build_tool_output_limits,
@@ -657,13 +658,14 @@ def build_pydantic_agent(
             # `history_processors=` kwarg, removed in pydantic-ai v2).
             # ToolOutputLimits reduces oversized tool returns on a different
             # hook (after_tool_execute), so its position is inert; the
-            # response clamp runs before_model_request and sits LAST so it
-            # sees the final, steer-injected history.
+            # response clamp runs before_model_request after both history
+            # processors. The plugin transform wraps the final model request.
             capabilities=[
                 *build_tool_output_limits(),
                 ProcessHistory(history_processor),
                 ProcessHistory(steer_processor),
                 build_response_clamp(),
+                build_model_message_transform(logical_agent_name),
             ],
             model_settings=model_settings,
         )

code_puppy-0.0.753/code_puppy/agents/_model_message_transform.py

@@ -0,0 +1,28 @@
+"""Plugin transforms for final outbound model messages."""
+
+from copy import copy
+from typing import Any
+
+from pydantic_ai import RunContext
+from pydantic_ai.capabilities import Hooks, WrapModelRequestHandler
+from pydantic_ai.messages import ModelResponse
+from pydantic_ai.models import ModelRequestContext
+
+from code_puppy.callbacks import on_transform_model_messages
+
+
+def build_model_message_transform(agent_name: str | None) -> Hooks:
+    """Build the request-only plugin transform for an agent."""
+
+    async def transform(
+        _ctx: RunContext[Any],
+        *,
+        request_context: ModelRequestContext,
+        handler: WrapModelRequestHandler,
+    ) -> ModelResponse:
+        transformed_context = copy(request_context)
+        transformed_context.messages = list(request_context.messages)
+        await on_transform_model_messages(agent_name, transformed_context.messages)
+        return await handler(transformed_context)
+
+    return Hooks(model_request=transform)

{code_puppy-0.0.751 → code_puppy-0.0.753}/code_puppy/callbacks.py

@@ -3,6 +3,8 @@ import logging
 import traceback
 from typing import Any, Callable, Dict, List, Literal, Optional, Set, Tuple
 
+from pydantic_ai.messages import ModelMessage
+
 PhaseType = Literal[
     "startup",
     "shutdown",
@@ -73,6 +75,7 @@ PhaseType = Literal[
     "awaiting_user_input",
     "git_branch_provider",
     "feature_capability",
+    "transform_model_messages",
 ]
 CallbackFunc = Callable[..., Any]
 
@@ -160,6 +163,7 @@ _callbacks: Dict[PhaseType, List[CallbackFunc]] = {
     "awaiting_user_input": [],
     "git_branch_provider": [],
     "feature_capability": [],
+    "transform_model_messages": [],
 }
 
 logger = logging.getLogger(__name__)
@@ -886,6 +890,13 @@ async def on_stream_event(
     )
 
 
+async def on_transform_model_messages(
+    agent_name: str | None, messages: List[ModelMessage]
+) -> List[Any]:
+    """Let plugins mutate the final outbound model messages in place."""
+    return await _trigger_callbacks("transform_model_messages", agent_name, messages)
+
+
 def on_register_tools() -> List[Dict[str, Any]]:
     """Collect custom tool registrations from plugins.
 

{code_puppy-0.0.751 → code_puppy-0.0.753}/code_puppy/tools/subagent_invocation.py

@@ -401,6 +401,9 @@ async def _invoke_agent_impl(
                 mcp_servers = manager.get_servers_for_agent(agent_name=bound_agent_name)
 
             from code_puppy.agents._compaction import make_history_processor
+            from code_puppy.agents._model_message_transform import (
+                build_model_message_transform,
+            )
 
             # Build the pydantic-ai agent. MCP servers always included; plugins
             # (e.g. DBOS) may swap them via the agent_run_context hook.
@@ -417,7 +420,10 @@ async def _invoke_agent_impl(
                 toolsets=mcp_servers,
                 # ProcessHistory capability replaces the deprecated
                 # `history_processors=` kwarg (removed in pydantic-ai v2).
-                capabilities=[ProcessHistory(make_history_processor(agent_config))],
+                capabilities=[
+                    ProcessHistory(make_history_processor(agent_config)),
+                    build_model_message_transform(agent_name),
+                ],
                 model_settings=model_settings,
             )
 

{code_puppy-0.0.751 → code_puppy-0.0.753}/pyproject.toml

@@ -4,12 +4,12 @@ build-backend = "hatchling.build"
 
 [project]
 name = "code-puppy"
-version = "0.0.751"
+version = "0.0.753"
 description = "Code generation agent"
 readme = "README.md"
 requires-python = ">=3.11,<3.15"
 dependencies = [
-    "code-puppy-core-plugins>=0.0.8",
+    "code-puppy-core-plugins>=0.0.12",
     "logfire>=4.0",
     "pydantic-ai-slim[openai,anthropic,mcp]==2.31.0",
     "pydantic-ai-harness==0.23.0",