code-puppy 0.0.376__tar.gz → 0.0.377__tar.gz

{code_puppy-0.0.376 → code_puppy-0.0.377}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.376
+Version: 0.0.377
 Summary: Code generation agent
 Project-URL: repository, https://github.com/mpfaffenberger/code_puppy
 Project-URL: HomePage, https://github.com/mpfaffenberger/code_puppy

{code_puppy-0.0.376 → code_puppy-0.0.377}/code_puppy/command_line/model_settings_menu.py

@@ -106,6 +106,19 @@ SETTING_DEFINITIONS: Dict[str, Dict] = {
         "type": "boolean",
         "default": False,
     },
+    "thinking_enabled": {
+        "name": "Thinking Enabled",
+        "description": "Enable thinking mode for Gemini 3 Pro models. When enabled, the model will show its reasoning process.",
+        "type": "boolean",
+        "default": True,
+    },
+    "thinking_level": {
+        "name": "Thinking Level",
+        "description": "Controls the depth of thinking for Gemini 3 Pro models. Low = faster responses, High = more thorough reasoning.",
+        "type": "choice",
+        "choices": ["low", "high"],
+        "default": "low",
+    },
 }
 
 

{code_puppy-0.0.376 → code_puppy-0.0.377}/code_puppy/gemini_model.py

@@ -40,6 +40,10 @@ from pydantic_ai.usage import RequestUsage
 
 logger = logging.getLogger(__name__)
 
+# Bypass thought signature for Gemini when no pending signature is available
+# This allows function calls to work with thinking models
+BYPASS_THOUGHT_SIGNATURE = "context_engineering_is_the_way_to_go"
+
 
 def generate_tool_call_id() -> str:
     """Generate a unique tool call ID."""
@@ -410,27 +414,48 @@ class GeminiModel(Model):
         return system_instruction, contents
 
     def _map_model_response(self, m: ModelResponse) -> dict[str, Any] | None:
-        """Map a ModelResponse to Gemini content format."""
+        """Map a ModelResponse to Gemini content format.
+
+        For Gemini thinking models, we need to track thought signatures from
+        ThinkingParts and apply them to subsequent function_call parts.
+        """
         parts: list[dict[str, Any]] = []
+        pending_signature: str | None = None
 
         for item in m.parts:
             if isinstance(item, ToolCallPart):
-                parts.append(
-                    {
-                        "function_call": {
-                            "name": item.tool_name,
-                            "args": item.args_as_dict(),
-                            "id": item.tool_call_id,
-                        }
+                part_dict: dict[str, Any] = {
+                    "function_call": {
+                        "name": item.tool_name,
+                        "args": item.args_as_dict(),
+                        "id": item.tool_call_id,
                     }
+                }
+                # Gemini thinking models REQUIRE thoughtSignature on function calls
+                # Use pending signature from thinking or bypass signature
+                part_dict["thoughtSignature"] = (
+                    pending_signature
+                    if pending_signature is not None
+                    else BYPASS_THOUGHT_SIGNATURE
                 )
+                parts.append(part_dict)
             elif isinstance(item, TextPart):
-                parts.append({"text": item.content})
+                part_dict = {"text": item.content}
+                # Apply pending signature to text parts too if present
+                if pending_signature is not None:
+                    part_dict["thoughtSignature"] = pending_signature
+                    pending_signature = None
+                parts.append(part_dict)
             elif isinstance(item, ThinkingPart):
                 if item.content:
-                    part_dict: dict[str, Any] = {"text": item.content, "thought": True}
+                    part_dict = {"text": item.content, "thought": True}
                     if item.signature:
                         part_dict["thoughtSignature"] = item.signature
+                        # Store signature for subsequent parts
+                        pending_signature = item.signature
+                    else:
+                        # No signature on thinking part, use bypass
+                        pending_signature = BYPASS_THOUGHT_SIGNATURE
                     parts.append(part_dict)
 
         if not parts:
@@ -462,18 +487,34 @@ class GeminiModel(Model):
         config: dict[str, Any] = {}
 
         if model_settings:
-            if (
-                hasattr(model_settings, "temperature")
-                and model_settings.temperature is not None
-            ):
-                config["temperature"] = model_settings.temperature
-            if hasattr(model_settings, "top_p") and model_settings.top_p is not None:
-                config["topP"] = model_settings.top_p
-            if (
-                hasattr(model_settings, "max_tokens")
-                and model_settings.max_tokens is not None
-            ):
-                config["maxOutputTokens"] = model_settings.max_tokens
+            # ModelSettings is a TypedDict, so use .get() for all access
+            temperature = model_settings.get("temperature")
+            if temperature is not None:
+                config["temperature"] = temperature
+
+            top_p = model_settings.get("top_p")
+            if top_p is not None:
+                config["topP"] = top_p
+
+            max_tokens = model_settings.get("max_tokens")
+            if max_tokens is not None:
+                config["maxOutputTokens"] = max_tokens
+
+            # Handle Gemini 3 Pro thinking settings
+            thinking_enabled = model_settings.get("thinking_enabled")
+            thinking_level = model_settings.get("thinking_level")
+
+            # Build thinkingConfig if thinking settings are present
+            if thinking_enabled is False:
+                # Disable thinking by not including thinkingConfig
+                pass
+            elif thinking_level is not None:
+                # Gemini 3 Pro uses thinkingLevel with values "low" or "high"
+                # includeThoughts=True is required to surface the thinking in the response
+                config["thinkingConfig"] = {
+                    "thinkingLevel": thinking_level,
+                    "includeThoughts": True,
+                }
 
         return config
 

{code_puppy-0.0.376 → code_puppy-0.0.377}/code_puppy/model_factory.py

@@ -74,6 +74,7 @@ def make_model_settings(
         get_effective_model_settings,
         get_openai_reasoning_effort,
         get_openai_verbosity,
+        model_supports_setting,
     )
 
     model_settings_dict: dict = {}
@@ -131,6 +132,18 @@ def make_model_settings(
             }
         model_settings = AnthropicModelSettings(**model_settings_dict)
 
+    # Handle Gemini thinking models (Gemini-3)
+    # Check if model supports thinking settings and apply defaults
+    if model_supports_setting(model_name, "thinking_level"):
+        # Apply defaults if not explicitly set by user
+        # Default: thinking_enabled=True, thinking_level="low"
+        if "thinking_enabled" not in model_settings_dict:
+            model_settings_dict["thinking_enabled"] = True
+        if "thinking_level" not in model_settings_dict:
+            model_settings_dict["thinking_level"] = "low"
+        # Recreate settings with Gemini thinking config
+        model_settings = ModelSettings(**model_settings_dict)
+
     return model_settings
 
 

{code_puppy-0.0.376 → code_puppy-0.0.377}/code_puppy/models.json

@@ -33,13 +33,13 @@
     "type": "gemini",
     "name": "gemini-3-pro-preview",
     "context_length": 200000,
-    "supported_settings": ["temperature", "top_p"]
+    "supported_settings": ["temperature", "top_p", "thinking_enabled", "thinking_level"]
   },
   "Gemini-3-Long-Context": {
     "type": "gemini",
     "name": "gemini-3-pro-preview",
     "context_length": 1000000,
-    "supported_settings": ["temperature", "top_p"]
+    "supported_settings": ["temperature", "top_p", "thinking_enabled", "thinking_level"]
   },
   "gpt-5.1": {
     "type": "openai",

{code_puppy-0.0.376 → code_puppy-0.0.377}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "code-puppy"
-version = "0.0.376"
+version = "0.0.377"
 description = "Code generation agent"
 readme = "README.md"
 requires-python = ">=3.11,<3.14"