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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.204
+Version: 0.0.206
 Summary: Code generation agent
 Project-URL: repository, https://github.com/mpfaffenberger/code_puppy
 Project-URL: HomePage, https://github.com/mpfaffenberger/code_puppy
@@ -19,7 +19,7 @@ Requires-Dist: bs4>=0.0.2
 Requires-Dist: camoufox>=0.4.11
 Requires-Dist: fastapi>=0.110.0
 Requires-Dist: httpx-limiter>=0.3.0
-Requires-Dist: httpx>=0.24.1
+Requires-Dist: httpx[http2]>=0.24.1
 Requires-Dist: json-repair>=0.46.2
 Requires-Dist: logfire>=0.7.1
 Requires-Dist: openai>=1.99.1

{code_puppy-0.0.204 → code_puppy-0.0.206}/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.206}/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.206}/code_puppy/config.py

@@ -132,6 +132,7 @@ def get_config_keys():
         "openai_reasoning_effort",
         "auto_save_session",
         "max_saved_sessions",
+        "http2",
     ]
     config = configparser.ConfigParser()
     config.read(CONFIG_FILE)
@@ -589,6 +590,27 @@ def get_compaction_strategy() -> str:
     return "truncation"
 
 
+def get_http2() -> bool:
+    """
+    Get the http2 configuration value.
+    Returns False if not set (default).
+    """
+    val = get_value("http2")
+    if val is None:
+        return False
+    return str(val).lower() in ("1", "true", "yes", "on")
+
+
+def set_http2(enabled: bool) -> None:
+    """
+    Sets the http2 configuration value.
+
+    Args:
+        enabled: Whether to enable HTTP/2 for httpx clients
+    """
+    set_config_value("http2", "true" if enabled else "false")
+
+
 def get_message_limit(default: int = 100) -> int:
     """
     Returns the user-configured message/request limit for the agent.

{code_puppy-0.0.204 → code_puppy-0.0.206}/code_puppy/http_utils.py

@@ -12,6 +12,8 @@ import httpx
 import requests
 from tenacity import stop_after_attempt, wait_exponential
 
+from code_puppy.config import get_http2
+
 try:
     from pydantic_ai.retries import (
         AsyncTenacityTransport,
@@ -55,6 +57,9 @@ def create_client(
     if verify is None:
         verify = get_cert_bundle_path()
 
+    # Check if HTTP/2 is enabled in config
+    http2_enabled = get_http2()
+
     # If retry components are available, create a client with retry transport
     if TenacityTransport and RetryConfig and wait_retry_after:
 
@@ -81,11 +86,17 @@ def create_client(
         )
 
         return httpx.Client(
-            transport=transport, verify=verify, headers=headers or {}, timeout=timeout
+            transport=transport,
+            verify=verify,
+            headers=headers or {},
+            timeout=timeout,
+            http2=http2_enabled,
         )
     else:
         # Fallback to regular client if retry components are not available
-        return httpx.Client(verify=verify, headers=headers or {}, timeout=timeout)
+        return httpx.Client(
+            verify=verify, headers=headers or {}, timeout=timeout, http2=http2_enabled
+        )
 
 
 def create_async_client(
@@ -97,6 +108,9 @@ def create_async_client(
     if verify is None:
         verify = get_cert_bundle_path()
 
+    # Check if HTTP/2 is enabled in config
+    http2_enabled = get_http2()
+
     # If retry components are available, create a client with retry transport
     if AsyncTenacityTransport and RetryConfig and wait_retry_after:
 
@@ -120,11 +134,17 @@ def create_async_client(
         )
 
         return httpx.AsyncClient(
-            transport=transport, verify=verify, headers=headers or {}, timeout=timeout
+            transport=transport,
+            verify=verify,
+            headers=headers or {},
+            timeout=timeout,
+            http2=http2_enabled,
         )
     else:
         # Fallback to regular client if retry components are not available
-        return httpx.AsyncClient(verify=verify, headers=headers or {}, timeout=timeout)
+        return httpx.AsyncClient(
+            verify=verify, headers=headers or {}, timeout=timeout, http2=http2_enabled
+        )
 
 
 def create_requests_session(
@@ -176,6 +196,9 @@ def create_reopenable_async_client(
     if verify is None:
         verify = get_cert_bundle_path()
 
+    # Check if HTTP/2 is enabled in config
+    http2_enabled = get_http2()
+
     # If retry components are available, create a client with retry transport
     if AsyncTenacityTransport and RetryConfig and wait_retry_after:
 
@@ -207,6 +230,7 @@ def create_reopenable_async_client(
                 verify=verify,
                 headers=headers or {},
                 timeout=timeout,
+                http2=http2_enabled,
             )
         else:
             # Fallback to regular AsyncClient if ReopenableAsyncClient is not available
@@ -215,17 +239,24 @@ def create_reopenable_async_client(
                 verify=verify,
                 headers=headers or {},
                 timeout=timeout,
+                http2=http2_enabled,
             )
     else:
         # Fallback to regular clients if retry components are not available
         if ReopenableAsyncClient is not None:
             return ReopenableAsyncClient(
-                verify=verify, headers=headers or {}, timeout=timeout
+                verify=verify,
+                headers=headers or {},
+                timeout=timeout,
+                http2=http2_enabled,
             )
         else:
             # Fallback to regular AsyncClient if ReopenableAsyncClient is not available
             return httpx.AsyncClient(
-                verify=verify, headers=headers or {}, timeout=timeout
+                verify=verify,
+                headers=headers or {},
+                timeout=timeout,
+                http2=http2_enabled,
             )
 
 

{code_puppy-0.0.204 → code_puppy-0.0.206}/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.206}/pyproject.toml

@@ -4,13 +4,13 @@ build-backend = "hatchling.build"
 
 [project]
 name = "code-puppy"
-version = "0.0.204"
+version = "0.0.206"
 description = "Code generation agent"
 readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
     "pydantic-ai==1.0.5",
-    "httpx>=0.24.1",
+    "httpx[http2]>=0.24.1",
     "rich>=13.4.2",
     "logfire>=0.7.1",
     "pydantic>=2.4.0",