code-puppy 0.0.164__tar.gz → 0.0.166__tar.gz

{code_puppy-0.0.164 → code_puppy-0.0.166}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.164
+Version: 0.0.166
 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.164 → code_puppy-0.0.166}/code_puppy/message_history_processor.py

@@ -148,6 +148,81 @@ def split_messages_for_protected_summarization(
     return messages_to_summarize, protected_messages
 
 
+def deduplicate_tool_returns(messages: List[ModelMessage]) -> List[ModelMessage]:
+    """
+    Remove duplicate tool returns while preserving the first occurrence for each tool_call_id.
+
+    This function identifies tool-return parts that share the same tool_call_id and
+    removes duplicates, keeping only the first return for each id. This prevents
+    conversation corruption from duplicate tool_result blocks.
+    """
+    if not messages:
+        return messages
+
+    seen_tool_returns: Set[str] = set()
+    deduplicated: List[ModelMessage] = []
+    removed_count = 0
+
+    for msg in messages:
+        # Check if this message has any parts we need to filter
+        if not hasattr(msg, "parts") or not msg.parts:
+            deduplicated.append(msg)
+            continue
+
+        # Filter parts within this message
+        filtered_parts = []
+        msg_had_duplicates = False
+
+        for part in msg.parts:
+            tool_call_id = getattr(part, "tool_call_id", None)
+            part_kind = getattr(part, "part_kind", None)
+
+            # Check if this is a tool-return part
+            if tool_call_id and part_kind in {
+                "tool-return",
+                "tool-result",
+                "tool_result",
+            }:
+                if tool_call_id in seen_tool_returns:
+                    # This is a duplicate return, skip it
+                    msg_had_duplicates = True
+                    removed_count += 1
+                    continue
+                else:
+                    # First occurrence of this return, keep it
+                    seen_tool_returns.add(tool_call_id)
+                    filtered_parts.append(part)
+            else:
+                # Not a tool return, always keep
+                filtered_parts.append(part)
+
+        # If we filtered out parts, create a new message with filtered parts
+        if msg_had_duplicates and filtered_parts:
+            # Create a new message with the same attributes but filtered parts
+            new_msg = type(msg)(parts=filtered_parts)
+            # Copy over other attributes if they exist
+            for attr_name in dir(msg):
+                if (
+                    not attr_name.startswith("_")
+                    and attr_name != "parts"
+                    and hasattr(msg, attr_name)
+                ):
+                    try:
+                        setattr(new_msg, attr_name, getattr(msg, attr_name))
+                    except (AttributeError, TypeError):
+                        # Skip attributes that can't be set
+                        pass
+            deduplicated.append(new_msg)
+        elif filtered_parts:  # No duplicates but has parts
+            deduplicated.append(msg)
+        # If no parts remain after filtering, drop the entire message
+
+    if removed_count > 0:
+        emit_warning(f"Removed {removed_count} duplicate tool-return part(s)")
+
+    return deduplicated
+
+
 def summarize_messages(
     messages: List[ModelMessage], with_protection=True
 ) -> Tuple[List[ModelMessage], List[ModelMessage]]:

{code_puppy-0.0.164 → code_puppy-0.0.166}/code_puppy/tools/file_operations.py

@@ -208,17 +208,22 @@ def _list_files(
         files = result.stdout.strip().split("\n") if result.stdout.strip() else []
 
         # Create ListedFile objects with metadata
-        for file_path in files:
-            if not file_path:  # Skip empty lines
+        for full_path in files:
+            if not full_path:  # Skip empty lines
                 continue
 
-            full_path = os.path.join(directory, file_path)
-
             # Skip if file doesn't exist (though it should)
             if not os.path.exists(full_path):
                 continue
 
+            # Extract relative path from the full path
+            if full_path.startswith(directory):
+                file_path = full_path[len(directory):].lstrip(os.sep)
+            else:
+                file_path = full_path
+
             # For non-recursive mode, skip files in subdirectories
+            # Only check the relative path, not the full path
             if not recursive and os.sep in file_path:
                 continue
 
@@ -242,7 +247,7 @@ def _list_files(
                 if entry_type == "file":
                     size = actual_size
 
-                # Calculate depth
+                # Calculate depth based on the relative path
                 depth = file_path.count(os.sep)
 
                 # Add directory entries if needed for files
@@ -281,6 +286,33 @@ def _list_files(
             except (FileNotFoundError, PermissionError, OSError):
                 # Skip files we can't access
                 continue
+
+        # In non-recursive mode, we also need to explicitly list directories in the target directory
+        # ripgrep's --files option only returns files, not directories
+        if not recursive:
+            try:
+                entries = os.listdir(directory)
+                for entry in entries:
+                    full_entry_path = os.path.join(directory, entry)
+                    # Skip if it doesn't exist or if it's a file (since files are already listed by ripgrep)
+                    if not os.path.exists(full_entry_path) or os.path.isfile(full_entry_path):
+                        continue
+                    
+                    # For non-recursive mode, only include directories that are directly in the target directory
+                    if os.path.isdir(full_entry_path):
+                        # Create a ListedFile for the directory
+                        results.append(
+                            ListedFile(
+                                path=entry,
+                                type="directory",
+                                size=0,
+                                full_path=full_entry_path,
+                                depth=0,
+                            )
+                        )
+            except (FileNotFoundError, PermissionError, OSError):
+                # Skip directories we can't access
+                pass
     except subprocess.TimeoutExpired:
         error_msg = (
             "[red bold]Error:[/red bold] List files command timed out after 30 seconds"
@@ -337,9 +369,12 @@ def _list_files(
         else:
             return "\U0001f4c4"
 
+    # Count items in results
     dir_count = sum(1 for item in results if item.type == "directory")
     file_count = sum(1 for item in results if item.type == "file")
     total_size = sum(item.size for item in results if item.type == "file")
+    
+
 
     # Build the directory header section
     dir_name = os.path.basename(directory) or directory
@@ -393,8 +428,8 @@ def _list_files(
     final_divider = "[dim]" + "─" * 100 + "\n" + "[/dim]"
     output_lines.append(final_divider)
 
-    # Return both the content string and the list of ListedFile objects
-    return ListFileOutput(content="\n".join(output_lines), files=results)
+    # Return the content string
+    return ListFileOutput(content="\n".join(output_lines))
 
 
 def _read_file(

{code_puppy-0.0.164 → code_puppy-0.0.166}/code_puppy/tui/components/custom_widgets.py

@@ -21,8 +21,13 @@ class CustomTextArea(TextArea):
 
     def on_key(self, event):
         """Handle key events before they reach the internal _on_key handler."""
-        # Explicitly handle escape+enter/alt+enter sequences
-        if event.key == "escape+enter" or event.key == "alt+enter":
+        # Let the binding system handle alt+enter
+        if event.key == "alt+enter":
+            # Don't prevent default - let the binding system handle it
+            return
+        
+        # Handle escape+enter manually
+        if event.key == "escape+enter":
             self.action_insert_newline()
             event.prevent_default()
             event.stop()

{code_puppy-0.0.164 → code_puppy-0.0.166}/pyproject.toml

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