code-puppy 0.0.204__tar.gz → 0.0.205__tar.gz

{code_puppy-0.0.204 → code_puppy-0.0.205}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.204
+Version: 0.0.205
 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.204 → code_puppy-0.0.205}/code_puppy/command_line/attachments.py

@@ -13,6 +13,10 @@ from pydantic_ai import BinaryContent, DocumentUrl, ImageUrl
 
 SUPPORTED_INLINE_SCHEMES = {"http", "https"}
 
+# Maximum path length to consider - conservative limit to avoid OS errors
+# Most OS have limits around 4096, but we set lower to catch garbage early
+MAX_PATH_LENGTH = 1024
+
 # Allow common extensions people drag in the terminal.
 DEFAULT_ACCEPTED_IMAGE_EXTENSIONS = {
     ".png",
@@ -61,6 +65,9 @@ def _is_probable_path(token: str) -> bool:
 
     if not token:
         return False
+    # Reject absurdly long tokens before any processing to avoid OS errors
+    if len(token) > MAX_PATH_LENGTH:
+        return False
     if token.startswith("#"):
         return False
     # Windows drive letters or Unix absolute/relative paths
@@ -69,7 +76,7 @@ def _is_probable_path(token: str) -> bool:
     if len(token) >= 2 and token[1] == ":":
         return True
     # Things like `path/to/file.png`
-    return os.sep in token or "\"" in token
+    return os.sep in token or '"' in token
 
 
 def _unescape_dragged_path(token: str) -> str:
@@ -107,9 +114,13 @@ def _load_binary(path: Path) -> bytes:
     except FileNotFoundError as exc:
         raise AttachmentParsingError(f"Attachment not found: {path}") from exc
     except PermissionError as exc:
-        raise AttachmentParsingError(f"Cannot read attachment (permission denied): {path}") from exc
+        raise AttachmentParsingError(
+            f"Cannot read attachment (permission denied): {path}"
+        ) from exc
     except OSError as exc:
-        raise AttachmentParsingError(f"Failed to read attachment {path}: {exc}") from exc
+        raise AttachmentParsingError(
+            f"Failed to read attachment {path}: {exc}"
+        ) from exc
 
 
 def _tokenise(prompt: str) -> Iterable[str]:
@@ -147,7 +158,10 @@ def _candidate_paths(
 
 def _is_supported_extension(path: Path) -> bool:
     suffix = path.suffix.lower()
-    return suffix in DEFAULT_ACCEPTED_IMAGE_EXTENSIONS | DEFAULT_ACCEPTED_DOCUMENT_EXTENSIONS
+    return (
+        suffix
+        in DEFAULT_ACCEPTED_IMAGE_EXTENSIONS | DEFAULT_ACCEPTED_DOCUMENT_EXTENSIONS
+    )
 
 
 def _parse_link(token: str) -> PromptLinkAttachment | None:
@@ -203,6 +217,11 @@ def _detect_path_tokens(prompt: str) -> tuple[list[_DetectedPath], list[str]]:
             index += 1
             continue
 
+        # Additional guard: skip if stripped token exceeds reasonable path length
+        if len(stripped_token) > MAX_PATH_LENGTH:
+            index += 1
+            continue
+
         start_index = index
         consumed_until = index + 1
         candidate_path_token = stripped_token
@@ -223,7 +242,16 @@ def _detect_path_tokens(prompt: str) -> tuple[list[_DetectedPath], list[str]]:
             index = consumed_until
             continue
 
-        if not path.exists() or not path.is_file():
+        # Guard filesystem operations against OS errors (ENAMETOOLONG, etc.)
+        try:
+            path_exists = path.exists()
+            path_is_file = path.is_file() if path_exists else False
+        except OSError:
+            # Skip this token if filesystem check fails (path too long, etc.)
+            index = consumed_until
+            continue
+
+        if not path_exists or not path_is_file:
             found_span = False
             last_path = path
             for joined, end_index in _candidate_paths(tokens, index):
@@ -328,7 +356,8 @@ def parse_prompt_attachments(prompt: str) -> ProcessedPrompt:
     spans = [
         (d.start_index, d.consumed_until)
         for d in detections
-        if (d.path is not None and not d.unsupported) or (d.link is not None and d.path is None)
+        if (d.path is not None and not d.unsupported)
+        or (d.link is not None and d.path is None)
     ]
     cleaned_parts: list[str] = []
     i = 0
@@ -360,4 +389,4 @@ __all__ = [
     "PromptLinkAttachment",
     "AttachmentParsingError",
     "parse_prompt_attachments",
-]
+]

{code_puppy-0.0.204 → code_puppy-0.0.205}/code_puppy/command_line/prompt_toolkit_completion.py

@@ -12,14 +12,20 @@ from typing import Optional
 
 from prompt_toolkit import PromptSession
 from prompt_toolkit.completion import Completer, Completion, merge_completers
+from prompt_toolkit.filters import is_searching
 from prompt_toolkit.formatted_text import FormattedText
 from prompt_toolkit.history import FileHistory
-from prompt_toolkit.filters import is_searching
 from prompt_toolkit.key_binding import KeyBindings
 from prompt_toolkit.keys import Keys
 from prompt_toolkit.layout.processors import Processor, Transformation
 from prompt_toolkit.styles import Style
 
+from code_puppy.command_line.attachments import (
+    DEFAULT_ACCEPTED_DOCUMENT_EXTENSIONS,
+    DEFAULT_ACCEPTED_IMAGE_EXTENSIONS,
+    _detect_path_tokens,
+    _tokenise,
+)
 from code_puppy.command_line.file_path_completion import FilePathCompleter
 from code_puppy.command_line.load_context_completion import LoadContextCompleter
 from code_puppy.command_line.model_picker_completion import (
@@ -34,11 +40,6 @@ from code_puppy.config import (
     get_puppy_name,
     get_value,
 )
-from code_puppy.command_line.attachments import (
-    DEFAULT_ACCEPTED_DOCUMENT_EXTENSIONS,
-    DEFAULT_ACCEPTED_IMAGE_EXTENSIONS,
-    _detect_path_tokens, _tokenise,
-)
 
 
 class SetCompleter(Completer):
@@ -108,6 +109,8 @@ class AttachmentPlaceholderProcessor(Processor):
     """Display friendly placeholders for recognised attachments."""
 
     _PLACEHOLDER_STYLE = "class:attachment-placeholder"
+    # Skip expensive path detection for very long input (likely pasted content)
+    _MAX_TEXT_LENGTH_FOR_REALTIME = 500
 
     def apply_transformation(self, transformation_input):
         document = transformation_input.document
@@ -115,6 +118,10 @@ class AttachmentPlaceholderProcessor(Processor):
         if not text:
             return Transformation(list(transformation_input.fragments))
 
+        # Skip real-time path detection for long text to avoid slowdown
+        if len(text) > self._MAX_TEXT_LENGTH_FOR_REALTIME:
+            return Transformation(list(transformation_input.fragments))
+
         detections, _warnings = _detect_path_tokens(text)
         replacements: list[tuple[int, int, str]] = []
         search_cursor = 0
@@ -138,7 +145,7 @@ class AttachmentPlaceholderProcessor(Processor):
                 continue
 
             # Use token-span for robust lookup (handles escaped spaces)
-            span_tokens = token_view[detection.start_index:detection.consumed_until]
+            span_tokens = token_view[detection.start_index : detection.consumed_until]
             raw_span = " ".join(span_tokens).replace(ESCAPE_MARKER, r"\ ")
             index = text.find(raw_span, search_cursor)
             span_len = len(raw_span)
@@ -188,7 +195,9 @@ class AttachmentPlaceholderProcessor(Processor):
                     display_index += 1
 
             for _ in text[source_index:end]:
-                source_to_display_map.append(placeholder_start if placeholder else display_index)
+                source_to_display_map.append(
+                    placeholder_start if placeholder else display_index
+                )
                 source_index += 1
 
         if source_index < len(text):
@@ -340,6 +349,7 @@ async def get_input_with_combined_completion(
 
     # Also allow Ctrl+Enter for newline (terminal-dependent)
     try:
+
         @bindings.add("c-enter", eager=True)
         def _(event):
             event.app.current_buffer.insert_text("\n")

{code_puppy-0.0.204 → code_puppy-0.0.205}/code_puppy/models.json

@@ -50,6 +50,11 @@
     "name": "claude-sonnet-4-5-20250929",
     "context_length": 200000
   },
+  "claude-4-1-opus": {
+    "type": "anthropic",
+    "name": "claude-opus-4-1-20250805",
+    "context_length": 200000
+  },
   "glm-4.5-coding": {
     "type": "zai_coding",
     "name": "glm-4.5"

{code_puppy-0.0.204 → code_puppy-0.0.205}/pyproject.toml

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