code-puppy 0.0.79__py3-none-any.whl → 0.0.81__py3-none-any.whl

code_puppy/command_line/motd.py

@@ -7,28 +7,25 @@ import os
 
 MOTD_VERSION = "20250815"
 MOTD_MESSAGE = """
-/¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯¯\\
-|   🐾  Happy Friday, Aug 15, 2025!                        |
-|                                                            |
-|   Biscuit the code puppy is on full zoomie mode!           |
-|   Major paws-ups:                                          |
-|   1. We now integrate Cerebras gpt-oss-120b!               |
-|      It's a bit underwhelming compared to Qwen3-Coder-480b  |
-|      (obviously), but it's still good for basic fetches.   |
-|   2. We also added support for OpenAI gpt-5!                |
-|      It's so good, it'll make you want to teach it to sit!|
-|                                                            |
-|   • To use one of the Cerebras models just have a         |
-|     CEREBRAS_API_KEY set in the environment variables.   |
-|   • Use ~m to swap models in the middle of your session!  |
-|   • Take stretch breaks – you'll need 'em!                 |
-|   • DRY your code, but keep your pup hydrated.            |
-|   • If you hit a bug, treat yourself for finding it!      |
-|                                                            |
-|   Today: sniff, code, roll over, and let these fancy AI   |
-|   models do the heavy lifting. Fire up a ~motd anytime      |
-|   you need some puppy hype!                                |
-\___________________________________________________________/
+
+🐾  Happy Friday, Aug 15, 2025!
+
+Biscuit the code puppy is on full zoomie mode!
+Major paws-ups:
+1. We now integrate Cerebras gpt-oss-120b!
+   It's a bit underwhelming compared to Qwen3-Coder-480b (obviously), but it's still good for basic fetches.
+2. We also added support for OpenAI gpt-5!
+   It's so good, it'll make you want to teach it to sit!
+
+• To use one of the Cerebras models just have a CEREBRAS_API_KEY set in the environment variables.
+• Use ~m to swap models in the middle of your session!
+• Take stretch breaks – you'll need 'em!
+• DRY your code, but keep your pup hydrated.
+• If you hit a bug, treat yourself for finding it!
+
+Today: sniff, code, roll over, and let these fancy AI models do the heavy lifting. Fire up a ~motd anytime
+you need some puppy hype!
+
 """
 MOTD_TRACK_FILE = os.path.expanduser("~/.puppy_cfg/motd.txt")
 

code_puppy/model_factory.py

@@ -1,5 +1,6 @@
 import json
 import os
+import random
 from typing import Any, Dict
 
 import httpx
@@ -13,6 +14,8 @@ from pydantic_ai.providers.google_gla import GoogleGLAProvider
 from pydantic_ai.providers.openai import OpenAIProvider
 from pydantic_ai.providers.openrouter import OpenRouterProvider
 
+from code_puppy.tools.common import console
+
 # Environment variables used in this module:
 # - GEMINI_API_KEY: API key for Google's Gemini models. Required when using Gemini models.
 # - OPENAI_API_KEY: API key for OpenAI models. Required when using OpenAI models or custom_openai endpoints.
@@ -23,6 +26,53 @@ from pydantic_ai.providers.openrouter import OpenRouterProvider
 #   Example: "X-Api-Key": "$OPENAI_API_KEY" will use the value from os.environ.get("OPENAI_API_KEY")
 
 
+def build_proxy_dict(proxy):
+    proxy_tokens = proxy.split(":")
+    structure = "{}:{}@{}:{}".format(
+        proxy_tokens[2], proxy_tokens[3], proxy_tokens[0], proxy_tokens[1]
+    )
+    proxies = {
+        "http": "http://{}/".format(structure),
+        "https": "http://{}".format(structure),
+    }
+    return proxies
+
+
+def build_httpx_proxy(proxy):
+    """Build an httpx.Proxy object from a proxy string in format ip:port:username:password"""
+    proxy_tokens = proxy.split(":")
+    if len(proxy_tokens) != 4:
+        raise ValueError(f"Invalid proxy format: {proxy}. Expected format: ip:port:username:password")
+    
+    ip, port, username, password = proxy_tokens
+    proxy_url = f"http://{ip}:{port}"
+    proxy_auth = (username, password)
+    
+    # Log the proxy being used
+    console.log(f"Using proxy: {proxy_url} with username: {username}")
+    
+    return httpx.Proxy(url=proxy_url, auth=proxy_auth)
+
+
+def get_random_proxy_from_file(file_path):
+    """Reads proxy file and returns a random proxy formatted for httpx.AsyncClient"""
+    if not os.path.exists(file_path):
+        raise ValueError(f"Proxy file '{file_path}' not found.")
+    
+    with open(file_path, "r") as f:
+        proxies = [line.strip() for line in f.readlines() if line.strip()]
+    
+    if not proxies:
+        raise ValueError(f"Proxy file '{file_path}' is empty or contains only whitespace.")
+    
+    selected_proxy = random.choice(proxies)
+    try:
+        return build_httpx_proxy(selected_proxy)
+    except ValueError as e:
+        console.log(f"Warning: Malformed proxy '{selected_proxy}' found in file '{file_path}', ignoring and continuing without proxy.")
+        return None
+
+
 def get_custom_config(model_config):
     custom_config = model_config.get("custom_endpoint", {})
     if not custom_config:
@@ -97,7 +147,18 @@ class ModelFactory:
 
         elif model_type == "custom_anthropic":
             url, headers, ca_certs_path, api_key = get_custom_config(model_config)
-            client = httpx.AsyncClient(headers=headers, verify=ca_certs_path)
+            
+            # Check for proxy configuration
+            proxy_file_path = os.environ.get("CODE_PUPPY_PROXIES")
+            proxy = None
+            if proxy_file_path:
+                proxy = get_random_proxy_from_file(proxy_file_path)
+            
+            # Only pass proxy to client if it's valid
+            client_args = {"headers": headers, "verify": ca_certs_path}
+            if proxy is not None:
+                client_args["proxy"] = proxy
+            client = httpx.AsyncClient(**client_args)
             anthropic_client = AsyncAnthropic(
                 base_url=url,
                 http_client=client,
@@ -162,7 +223,18 @@ class ModelFactory:
 
         elif model_type == "custom_openai":
             url, headers, ca_certs_path, api_key = get_custom_config(model_config)
-            client = httpx.AsyncClient(headers=headers, verify=ca_certs_path)
+            
+            # Check for proxy configuration
+            proxy_file_path = os.environ.get("CODE_PUPPY_PROXIES")
+            proxy = None
+            if proxy_file_path:
+                proxy = get_random_proxy_from_file(proxy_file_path)
+            
+            # Only pass proxy to client if it's valid
+            client_args = {"headers": headers, "verify": ca_certs_path}
+            if proxy is not None:
+                client_args["proxy"] = proxy
+            client = httpx.AsyncClient(**client_args)
             provider_args = dict(
                 base_url=url,
                 http_client=client,

{code_puppy-0.0.79.dist-info → code_puppy-0.0.81.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.79
+Version: 0.0.81
 Summary: Code generation agent
 Author: Michael Pfaffenberger
 License: MIT
@@ -51,6 +51,11 @@ Description-Content-Type: text/markdown
 
 Code Puppy is an AI-powered code generation agent, designed to understand programming tasks, generate high-quality code, and explain its reasoning similar to tools like Windsurf and Cursor. 
 
+## Quick start
+
+`uvx code-puppy -i`
+
+
 ## Features
 
 - **Multi-language support**: Capable of generating code in various programming languages.

{code_puppy-0.0.79.dist-info → code_puppy-0.0.81.dist-info}/RECORD

@@ -4,7 +4,7 @@ code_puppy/agent_prompts.py,sha256=13YIpTZa3R3lg60-fdkll7t7hgSBtQL0M53wcE1gzyQ,6
 code_puppy/config.py,sha256=r5nw5ChOP8xd_K5yo8U5OtO2gy2bFhARiyNtDp1JrwQ,5013
 code_puppy/main.py,sha256=uKMG0WNrFjEbsiEb_OwL_fNJbqMyTgztGjPKIOoYdSs,10444
 code_puppy/message_history_processor.py,sha256=NaFYEUdHCZlzl5jR-XK4Rh2EHVsupT-SROxe4jXgUaQ,2584
-code_puppy/model_factory.py,sha256=P2E3KgTHMVaMhHyGHmdascjYmdRxUKBWotlP61i-03A,8291
+code_puppy/model_factory.py,sha256=3j7AcJfZAHbx_plL9oOxjGJO0MMTRaQFThCErg8VpH8,10909
 code_puppy/models.json,sha256=hqSvFzSPcwxMwst6xePlcppm0c_pjyEVSBsWvgbAG98,2714
 code_puppy/session_memory.py,sha256=4sgAAjbXdLSi8hETpd56tgtrG6hqMUuZWDlJOu6BQjA,2735
 code_puppy/state_management.py,sha256=aymPVogToEeF_vPc2c75rs8WLYNd8s2vMRGNXvTriW0,1248
@@ -13,7 +13,7 @@ code_puppy/command_line/__init__.py,sha256=y7WeRemfYppk8KVbCGeAIiTuiOszIURCDjOMZ
 code_puppy/command_line/file_path_completion.py,sha256=gw8NpIxa6GOpczUJRyh7VNZwoXKKn-yvCqit7h2y6Gg,2931
 code_puppy/command_line/meta_command_handler.py,sha256=L7qP2g0Faz0V7bMH4YK3s03OWWuQFtK7Sh-Kt2zmmEQ,6182
 code_puppy/command_line/model_picker_completion.py,sha256=NkyZZG7IhcVWSJ3ADytwCA5f8DpNeVs759Qtqs4fQtY,3733
-code_puppy/command_line/motd.py,sha256=0G6_BHLiIdONNVMxlvmTJFh5j7CvnIX530JlUPD5SDE,2323
+code_puppy/command_line/motd.py,sha256=1qEPpEQb14XfEXj-_pmx7ad8VtzvP7JSmvvW_JWW-Os,1662
 code_puppy/command_line/prompt_toolkit_completion.py,sha256=_gP0FIOgHDNHTTWLNL0XNzr6sO0ISe7Mec1uQNo9kcM,8337
 code_puppy/command_line/utils.py,sha256=7eyxDHjPjPB9wGDJQQcXV_zOsGdYsFgI0SGCetVmTqE,1251
 code_puppy/tools/__init__.py,sha256=ozIGpLM7pKSjH4UeojkTodhfVYZeNzMsLtK_oyw41HA,456
@@ -22,9 +22,9 @@ code_puppy/tools/common.py,sha256=M53zhiXZAmPdvi1Y_bzCxgvEmifOvRRJvYPARYRZqHw,22
 code_puppy/tools/file_modifications.py,sha256=qCfkZ7BxTG8U4xydHzS44UtOioj8XvhRKgjoOAnMHTo,13310
 code_puppy/tools/file_operations.py,sha256=5ESOCS3m4Lpnvrg2XiJAx0m4-0Yar6LZKIdyRCRjENM,11218
 code_puppy/tools/ts_code_map.py,sha256=o-u8p5vsYwitfDtVEoPS-7MwWn2xHzwtIQLo1_WMhQs,17647
-code_puppy-0.0.79.data/data/code_puppy/models.json,sha256=hqSvFzSPcwxMwst6xePlcppm0c_pjyEVSBsWvgbAG98,2714
-code_puppy-0.0.79.dist-info/METADATA,sha256=NWi6a8SKiYPdoAetPAHd4bJ08ouT120WRSCryz3Fb1g,6643
-code_puppy-0.0.79.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-code_puppy-0.0.79.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
-code_puppy-0.0.79.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
-code_puppy-0.0.79.dist-info/RECORD,,
+code_puppy-0.0.81.data/data/code_puppy/models.json,sha256=hqSvFzSPcwxMwst6xePlcppm0c_pjyEVSBsWvgbAG98,2714
+code_puppy-0.0.81.dist-info/METADATA,sha256=d8xhiEpZU64TOZkRnAI5N28ql73s-i8ERXrNgwQlHDg,6681
+code_puppy-0.0.81.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+code_puppy-0.0.81.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
+code_puppy-0.0.81.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
+code_puppy-0.0.81.dist-info/RECORD,,