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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: code-puppy-core-plugins
-Version: 0.0.32
+Version: 0.0.34
 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-0.0.34}/code_puppy_core_plugins/auto_continue/classifier.py

@@ -47,37 +47,37 @@ def _model_name() -> str | None:
 
 async def classify(response: str) -> Approval | None:
     """Classify one response without entering Code Puppy's sub-agent runtime."""
-    from pydantic_ai import Agent, UsageLimits
+    try:
+        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
+        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_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 = ModelFactory.get_model(model_name, models_config)
+        prepared = prepare_prompt_for_model(
             model_name,
-            max_tokens=64,
-            overrides=_NON_THINKING_OVERRIDES,
-        ),
-    )
-    try:
+            _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,
+            ),
+        )
         async with asyncio.timeout(30):
             result = await agent.run(
                 prepared.user_prompt,

code_puppy_core_plugins-0.0.34/code_puppy_core_plugins/auto_continue/register_callbacks.py

@@ -0,0 +1,100 @@
+"""Register automatic continuation after routine permission requests."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from code_puppy.callbacks import register_callback
+
+from .classifier import classify
+
+_COMMAND = "auto-continue"
+_ENABLED_KEY = "auto_continue_enabled"
+
+
+def _is_enabled() -> bool:
+    from code_puppy.config import get_truthy_bool_value
+
+    return get_truthy_bool_value(_ENABLED_KEY, True)
+
+
+def _set_enabled(enabled: bool) -> None:
+    from code_puppy.config import set_value
+
+    set_value(_ENABLED_KEY, "true" if enabled else "false")
+
+
+def _emit_status() -> None:
+    from code_puppy.config import get_auto_continue_model_name
+    from code_puppy.messaging import emit_info
+
+    state = "enabled" if _is_enabled() else "disabled"
+    model = get_auto_continue_model_name() or "not configured"
+    emit_info(f"Auto-continue is {state}; classifier model: {model}.")
+
+
+def _handle_custom_command(command: str, name: str) -> bool | None:
+    if name != _COMMAND:
+        return None
+
+    from code_puppy.messaging import emit_error, emit_success
+
+    tokens = command.strip().split()
+    action = tokens[1].lower() if len(tokens) > 1 else "status"
+    if action == "enable":
+        _set_enabled(True)
+        emit_success("Auto-continue enabled.")
+    elif action == "disable":
+        _set_enabled(False)
+        emit_success("Auto-continue disabled.")
+    elif action == "status":
+        _emit_status()
+    else:
+        emit_error("Usage: /auto-continue [enable | disable | status]")
+    return True
+
+
+def _custom_help() -> list[tuple[str, str]]:
+    return [(_COMMAND, "Enable, disable, or inspect automatic continuation")]
+
+
+def _response_text(result: Any) -> str | None:
+    if result is None:
+        return None
+    output = getattr(result, "output", result)
+    return output if isinstance(output, str) else None
+
+
+async def _on_interactive_turn_end(
+    agent: Any,
+    prompt: str,
+    result: Any = None,
+    *,
+    success: bool = True,
+    error: BaseException | None = None,
+) -> dict[str, str] | None:
+    """Ask the classifier whether the completed response should be continued."""
+    response = _response_text(result)
+    if not _is_enabled() or not success or error is not None or not response:
+        return None
+
+    approval = await classify(response)
+    if approval is None:
+        return None
+    from code_puppy.messaging import emit_info
+
+    emit_info(f"Auto-continue: submitting '{approval}'.")
+    return {"prompt": approval}
+
+
+register_callback("interactive_turn_end", _on_interactive_turn_end)
+register_callback("custom_command", _handle_custom_command)
+register_callback("custom_command_help", _custom_help)
+
+
+__all__ = [
+    "_custom_help",
+    "_handle_custom_command",
+    "_is_enabled",
+    "_on_interactive_turn_end",
+]

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

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

code_puppy_core_plugins-0.0.32/code_puppy_core_plugins/auto_continue/register_callbacks.py

@@ -1,41 +0,0 @@
-"""Register automatic continuation after routine permission requests."""
-
-from __future__ import annotations
-
-from typing import Any
-
-from code_puppy.callbacks import register_callback
-
-from .classifier import classify
-
-
-def _response_text(result: Any) -> str | None:
-    if result is None:
-        return None
-    output = getattr(result, "output", result)
-    return output if isinstance(output, str) else None
-
-
-async def _on_interactive_turn_end(
-    agent: Any,
-    prompt: str,
-    result: Any = None,
-    *,
-    success: bool = True,
-    error: BaseException | None = None,
-) -> dict[str, str] | None:
-    """Ask the classifier whether the completed response should be continued."""
-    response = _response_text(result)
-    if not success or error is not None or not response:
-        return None
-
-    approval = await classify(response)
-    if approval is None:
-        return None
-    return {"prompt": approval}
-
-
-register_callback("interactive_turn_end", _on_interactive_turn_end)
-
-
-__all__ = ["_on_interactive_turn_end"]