code-puppy 0.0.35__tar.gz → 0.0.37__tar.gz

{code_puppy-0.0.35 → code_puppy-0.0.37}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.35
+Version: 0.0.37
 Summary: Code generation agent
 Author: Michael Pfaffenberger
 License: MIT

{code_puppy-0.0.35 → code_puppy-0.0.37}/code_puppy/model_factory.py

@@ -139,11 +139,11 @@ def get_custom_config(model_config):
         ca_certs_path = custom_config.get("ca_certs_path")
 
     api_key = None
-    if "api_key" in model_config:
-        if model_config["api_key"].startswith("$"):
-            api_key = os.environ.get(model_config["api_key"][1:])
+    if "api_key" in custom_config:
+        if custom_config["api_key"].startswith("$"):
+            api_key = os.environ.get(custom_config["api_key"][1:])
         else:
-            api_key = model_config["api_key"]
+            api_key = custom_config["api_key"]
     return url, headers, ca_certs_path, api_key
 
 
@@ -186,6 +186,14 @@ class ModelFactory:
 
             return OpenAIModel(model_name=model_config["name"], provider=provider)
 
+        elif model_type == "anthropic":
+            api_key = os.environ.get("ANTHROPIC_API_KEY", None)
+            if not api_key:
+                raise ValueError('ANTHROPIC_API_KEY environment variable must be set for Anthropic models.')
+            anthropic_client = AsyncAnthropic(api_key=api_key)
+            provider = AnthropicProvider(anthropic_client=anthropic_client)
+            return AnthropicModel(model_name=model_config["name"], provider=provider)
+
         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)
@@ -195,7 +203,6 @@ class ModelFactory:
                 api_key=api_key,
             )
             provider = AnthropicProvider(anthropic_client=anthropic_client)
-
             return AnthropicModel(model_name=model_config["name"], provider=provider)
 
         elif model_type == "custom_openai":

{code_puppy-0.0.35 → code_puppy-0.0.37}/code_puppy/models.json

@@ -89,5 +89,16 @@
       "url": "https://api.together.xyz/v1",
       "api_key": "$TOGETHER_API_KEY"
     }
+  },
+  "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8": {
+    "type": "custom_openai",
+    "name": "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8",
+    "max_requests_per_minute": 100,
+    "max_retries": 3,
+    "retry_base_delay": 5,
+    "custom_endpoint": {
+      "url": "https://api.together.xyz/v1",
+      "api_key": "$TOGETHER_API_KEY"
+    }
   }
-}
+}

{code_puppy-0.0.35 → code_puppy-0.0.37}/pyproject.toml

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