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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.205
+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.205 → 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.205 → 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.205 → code_puppy-0.0.206}/pyproject.toml

@@ -4,13 +4,13 @@ build-backend = "hatchling.build"
 
 [project]
 name = "code-puppy"
-version = "0.0.205"
+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",