code-puppy-core-plugins 0.0.62__tar.gz → 0.0.64__tar.gz

{code_puppy_core_plugins-0.0.62 → code_puppy_core_plugins-0.0.64}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: code-puppy-core-plugins
-Version: 0.0.62
+Version: 0.0.64
 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.62 → code_puppy_core_plugins-0.0.64}/code_puppy_core_plugins/chatgpt_oauth/image_tool.py

@@ -19,7 +19,7 @@ TOOL_NAME = "codex_imagegen"
 def register_codex_imagegen(agent: Any) -> None:
     """Register image generation on a pydantic-ai agent."""
 
-    @agent.tool
+    @agent.tool(metadata={"speculatable": True})
     async def codex_imagegen(
         context: RunContext,
         prompt: str,

{code_puppy_core_plugins-0.0.62 → code_puppy_core_plugins-0.0.64}/code_puppy_core_plugins/chatgpt_oauth/test_plugin.py

@@ -355,11 +355,23 @@ def test_reasoning_gates_by_gpt_generation(name, xhigh, max_effort, responses_co
     assert utils._supports_responses_reasoning_controls(name) is responses_controls
 
 
+def test_codex_imagegen_is_speculatable():
+    from pydantic_ai import Agent
+
+    agent = Agent("test")
+    image_tool.register_codex_imagegen(agent)
+    assert (
+        agent._function_toolset.tools["codex_imagegen"].metadata["speculatable"] is True
+    )
+
+
 def test_codex_imagegen_agent_tool(tmp_path):
     registered = {}
 
     class FakeAgent:
-        def tool(self, function):
+        def tool(self, function=None, **_kwargs):
+            if function is None:
+                return lambda fn: self.tool(fn)
             registered[function.__name__] = function
             return function
 
@@ -385,7 +397,9 @@ def test_codex_imagegen_agent_tool_forwards_reference_images(tmp_path):
     registered = {}
 
     class FakeAgent:
-        def tool(self, function):
+        def tool(self, function=None, **_kwargs):
+            if function is None:
+                return lambda fn: self.tool(fn)
             registered[function.__name__] = function
             return function
 

{code_puppy_core_plugins-0.0.62 → code_puppy_core_plugins-0.0.64}/code_puppy_core_plugins/claude_code_oauth/config.py

@@ -30,6 +30,7 @@ CLAUDE_CODE_OAUTH_CONFIG: Dict[str, Any] = {
         "claude-opus-4-7",
         "claude-opus-4-8",
         "claude-opus-5",
+        "claude-opus-5-5",
         "claude-sonnet-5",
         "claude-fable-5",
         "claude-fable-5-1",

{code_puppy_core_plugins-0.0.62 → code_puppy_core_plugins-0.0.64}/code_puppy_core_plugins/claude_code_oauth/model_filter.py

@@ -7,6 +7,25 @@ from typing import Dict, List, Tuple, Union
 
 logger = logging.getLogger(__name__)
 
+# claude-{family}-{major}[-|.{minor}][-{yyyymmdd}], e.g. claude-opus-5,
+# claude-opus-5-5, claude-sonnet-4.5-20250929, claude-opus-4-20250514.
+# The lookaheads stop a date from being misread as a minor version
+# (claude-opus-4-20250514 is 4.0, not 4.20). Undated names rank date 0.
+_MODEL_RE = re.compile(
+    r"claude-(haiku|sonnet|opus|fable)-(\d+)"
+    r"(?:[-.](\d{1,2})(?!\d))?"
+    r"(?:-(\d{8}))?(?![\d.])"
+)
+
+
+def _parse_model(model_name: str) -> Tuple[str, int, int, int] | None:
+    """Parse a model name into ``(family, major, minor, date)``."""
+    match = _MODEL_RE.match(model_name)
+    if not match:
+        return None
+    family, major, minor, date = match.groups()
+    return family, int(major), int(minor or 0), int(date or 0)
+
 
 def filter_latest_claude_models(
     models: List[str], max_per_family: Union[int, Dict[str, int]] = 2
@@ -34,50 +53,10 @@ def filter_latest_claude_models(
     family_models: Dict[str, List[Tuple[str, int, int, int]]] = {}
 
     for model_name in models:
-        if model_name == "claude-opus-5-5":
-            family_models.setdefault("opus", []).append((model_name, 5, 5, 0))
-            continue
-        if model_name == "claude-opus-5":
-            family_models.setdefault("opus", []).append((model_name, 5, 0, 0))
-            continue
-        if model_name == "claude-opus-4-8":
-            family_models.setdefault("opus", []).append((model_name, 4, 8, 20250301))
-            continue
-        if model_name == "claude-opus-4-7":
-            family_models.setdefault("opus", []).append((model_name, 4, 7, 20250219))
-            continue
-        if model_name == "claude-opus-4-6":
-            family_models.setdefault("opus", []).append((model_name, 4, 6, 20260205))
-            continue
-        if model_name == "claude-sonnet-4-6":
-            family_models.setdefault("sonnet", []).append((model_name, 4, 6, 20250610))
+        parsed = _parse_model(model_name)
+        if parsed is None:
             continue
-        if model_name == "claude-sonnet-5":
-            family_models.setdefault("sonnet", []).append((model_name, 5, 0, 0))
-            continue
-        if model_name == "claude-fable-5":
-            family_models.setdefault("fable", []).append((model_name, 5, 0, 0))
-            continue
-        if model_name == "claude-fable-5-1":
-            family_models.setdefault("fable", []).append((model_name, 5, 1, 0))
-            continue
-        # Match pattern: claude-{family}-{major}-{minor}-{date}
-        # Examples: claude-haiku-3-5-20241022, claude-sonnet-4-5-20250929
-        match = re.match(r"claude-(haiku|sonnet|opus)-(\d+)-(\d+)-(\d+)", model_name)
-        if not match:
-            # Also try pattern with dots: claude-{family}-{major}.{minor}-{date}
-            match = re.match(
-                r"claude-(haiku|sonnet|opus)-(\d+)\.(\d+)-(\d+)", model_name
-            )
-
-        if not match:
-            continue
-
-        family = match.group(1)
-        major = int(match.group(2))
-        minor = int(match.group(3))
-        date = int(match.group(4))
-
+        family, major, minor, date = parsed
         family_models.setdefault(family, []).append((model_name, major, minor, date))
 
     # Sort each family descending and keep the top N

{code_puppy_core_plugins-0.0.62 → code_puppy_core_plugins-0.0.64}/code_puppy_core_plugins/logfire_oauth/query_tool.py

@@ -47,7 +47,7 @@ async def _query_mcp(
 def register_logfire_query(agent: Any) -> None:
     """Register the ``logfire_query`` tool on an agent."""
 
-    @agent.tool
+    @agent.tool(metadata={"speculatable": True})
     async def logfire_query(
         context: RunContext,
         query: str = Field(description="Read-only SQL query over Logfire records."),

{code_puppy_core_plugins-0.0.62 → code_puppy_core_plugins-0.0.64}/code_puppy_core_plugins/namespace_skill_search/search_tool.py

@@ -58,7 +58,7 @@ def _query_terms(query: Optional[str]) -> List[str]:
 def register_browse_skill_namespace(agent):
     """Register the browse_skill_namespace tool on an agent."""
 
-    @agent.tool
+    @agent.tool(metadata={"speculatable": True})
     async def browse_skill_namespace(
         context: RunContext,
         namespace: Optional[str] = None,

{code_puppy_core_plugins-0.0.62 → code_puppy_core_plugins-0.0.64}/code_puppy_core_plugins/puppy_kennel/tools.py

@@ -157,7 +157,7 @@ def _resolve_scope(wing: str, scope: str, agent_name: str, cwd: Any) -> list[str
 def register_kennel_recall(agent: Any) -> None:
     """Register the ``kennel_recall`` tool on the given agent."""
 
-    @agent.tool
+    @agent.tool(metadata={"speculatable": True})
     async def kennel_recall(
         context: RunContext,
         query: str,
@@ -314,7 +314,7 @@ def register_kennel_remember(agent: Any) -> None:
 def register_kennel_recent(agent: Any) -> None:
     """Register the ``kennel_recent`` tool — time-ordered drawer browsing."""
 
-    @agent.tool
+    @agent.tool(metadata={"speculatable": True})
     async def kennel_recent(
         context: RunContext,
         wing: str = "",
@@ -366,7 +366,7 @@ def register_kennel_recent(agent: Any) -> None:
 def register_kennel_list_wings(agent: Any) -> None:
     """Register the ``kennel_list_wings`` tool — wing discovery."""
 
-    @agent.tool
+    @agent.tool(metadata={"speculatable": True})
     async def kennel_list_wings(context: RunContext) -> KennelWingsOutput:
         """List every wing in the kennel with its drawer count.
 
@@ -392,7 +392,7 @@ def register_kennel_list_wings(agent: Any) -> None:
 def register_kennel_stats(agent: Any) -> None:
     """Register the ``kennel_stats`` tool — kennel-wide totals + size."""
 
-    @agent.tool
+    @agent.tool(metadata={"speculatable": True})
     async def kennel_stats(context: RunContext) -> KennelStatsOutput:
         """Return kennel-wide totals: drawers, wings, and on-disk size.
 

{code_puppy_core_plugins-0.0.62 → code_puppy_core_plugins-0.0.64}/pyproject.toml

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