code-puppy-core-plugins 0.0.31__tar.gz → 0.0.32__tar.gz

{code_puppy_core_plugins-0.0.31 → code_puppy_core_plugins-0.0.32}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: code-puppy-core-plugins
-Version: 0.0.31
+Version: 0.0.32
 Summary: Official plugin bundle for Code Puppy
 Project-URL: Repository, https://github.com/mpfaffenberger/code_puppy_core_plugins
 Project-URL: Homepage, https://github.com/mpfaffenberger/code_puppy_core_plugins

code_puppy_core_plugins-0.0.32/code_puppy_core_plugins/auto_continue/classifier.py

@@ -0,0 +1,91 @@
+"""Private, one-shot model classifier for automatic continuation."""
+
+from __future__ import annotations
+
+import asyncio
+from typing import Literal
+
+from pydantic import BaseModel
+
+Approval = Literal["yes, go.", "continue", "okay"]
+
+_INSTRUCTIONS = """Decide whether the assistant is merely waiting for permission to
+continue routine work the user already requested.
+
+Approve only a clear request to continue, proceed, or perform an explicitly
+described next step within the existing task. Do not approve destructive or
+risky actions, credential or payment requests, ambiguous choices, requests for
+new requirements, completed work, or anything needing a real human decision.
+Return no approval when uncertain."""
+
+_NON_THINKING_OVERRIDES = {
+    "reasoning_effort": "none",
+    "extended_thinking": "off",
+    "thinking_type": "disabled",
+    "thinking_enabled": False,
+    "verbosity": "low",
+}
+
+
+class ContinuationDecision(BaseModel):
+    """Strict classifier result; ``None`` means wait for the user."""
+
+    approval: Approval | None = None
+
+
+def _model_name() -> str | None:
+    """Use the dedicated override when available, otherwise the global model."""
+    try:
+        from code_puppy.config import get_auto_continue_model_name
+
+        return get_auto_continue_model_name()
+    except ImportError:  # Compatibility with core versions before the setting.
+        from code_puppy.config import get_global_model_name, get_value
+
+        return get_value("auto_continue_model") or get_global_model_name()
+
+
+async def classify(response: str) -> Approval | None:
+    """Classify one response without entering Code Puppy's sub-agent runtime."""
+    from pydantic_ai import Agent, UsageLimits
+
+    from code_puppy.model_factory import ModelFactory, make_model_settings
+    from code_puppy.model_utils import prepare_prompt_for_model
+
+    model_name = _model_name()
+    if not model_name:
+        return None
+    models_config = ModelFactory.load_config()
+    if model_name not in models_config:
+        return None
+
+    model = ModelFactory.get_model(model_name, models_config)
+    prepared = prepare_prompt_for_model(
+        model_name,
+        _INSTRUCTIONS,
+        f"Classify this assistant response:\n\n{response}",
+        prepend_system_to_user=True,
+    )
+    agent = Agent(
+        model=model,
+        instructions=prepared.instructions,
+        output_type=ContinuationDecision,
+        retries=0,
+        model_settings=make_model_settings(
+            model_name,
+            max_tokens=64,
+            overrides=_NON_THINKING_OVERRIDES,
+        ),
+    )
+    try:
+        async with asyncio.timeout(30):
+            result = await agent.run(
+                prepared.user_prompt,
+                usage_limits=UsageLimits(request_limit=1),
+            )
+    except Exception:
+        return None
+    return result.output.approval
+
+
+__all__ = ["Approval", "ContinuationDecision", "classify"]

{code_puppy_core_plugins-0.0.31 → code_puppy_core_plugins-0.0.32}/code_puppy_core_plugins/auto_continue/register_callbacks.py

@@ -6,14 +6,7 @@ from typing import Any
 
 from code_puppy.callbacks import register_callback
 
-from .agent import AutoContinueAgent
-
-_AGENT_NAME = "auto-continue"
-_APPROVALS = {"yes, go.", "continue", "okay"}
-
-
-def _register_agents() -> list[dict[str, object]]:
-    return [{"name": _AGENT_NAME, "class": AutoContinueAgent}]
+from .classifier import classify
 
 
 def _response_text(result: Any) -> str | None:
@@ -36,22 +29,13 @@ async def _on_interactive_turn_end(
     if not success or error is not None or not response:
         return None
 
-    from code_puppy.tools.subagent_invocation import _invoke_agent_impl
-
-    verdict = await _invoke_agent_impl(
-        context=None,
-        agent_name=_AGENT_NAME,
-        prompt=f"Classify this assistant response:\n\n{response}",
-        emit_response_message=False,
-    )
-    approval = (verdict.response or "").strip().lower()
-    if approval not in _APPROVALS:
+    approval = await classify(response)
+    if approval is None:
         return None
     return {"prompt": approval}
 
 
-register_callback("register_agents", _register_agents)
 register_callback("interactive_turn_end", _on_interactive_turn_end)
 
 
-__all__ = ["_on_interactive_turn_end", "_register_agents"]
+__all__ = ["_on_interactive_turn_end"]

{code_puppy_core_plugins-0.0.31 → code_puppy_core_plugins-0.0.32}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "code-puppy-core-plugins"
-version = "0.0.31"
+version = "0.0.32"
 description = "Official plugin bundle for Code Puppy"
 readme = "README.md"
 requires-python = ">=3.11,<3.15"

code_puppy_core_plugins-0.0.31/code_puppy_core_plugins/auto_continue/agent.py

@@ -1,38 +0,0 @@
-"""The deliberately tiny continuation classifier agent."""
-
-from code_puppy.agents.base_agent import BaseAgent
-
-
-class AutoContinueAgent(BaseAgent):
-    """Decide whether a completed response merely needs permission to continue."""
-
-    @property
-    def name(self) -> str:
-        return "auto-continue"
-
-    @property
-    def display_name(self) -> str:
-        return "Auto Continue "
-
-    @property
-    def description(self) -> str:
-        return "Approves routine requests to continue an already requested task"
-
-    def get_available_tools(self) -> list[str]:
-        return []
-
-    def get_system_prompt(self) -> str:
-        return """You classify whether an assistant is merely waiting for permission to
-continue routine work the user already requested.
-
-Reply with exactly one of these strings and nothing else:
-- yes, go.
-- continue
-- okay
-- NO
-
-Approve only a clear request to continue, proceed, or perform an explicitly
-described next step within the existing task. Use NO for destructive or risky
-actions, credential or payment requests, ambiguous choices, requests for new
-requirements, completed work, or anything needing a real human decision.
-Never use tools and never explain your answer."""