code-puppy 0.0.56__tar.gz → 0.0.58__tar.gz

{code_puppy-0.0.56 → code_puppy-0.0.58}/PKG-INFO

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

{code_puppy-0.0.56 → code_puppy-0.0.58}/code_puppy/model_factory.py

@@ -181,12 +181,16 @@ class ModelFactory:
         if model_type == "gemini":
             provider = GoogleGLAProvider(api_key=os.environ.get("GEMINI_API_KEY", ""))
 
-            return GeminiModel(model_name=model_config["name"], provider=provider)
+            model = GeminiModel(model_name=model_config["name"], provider=provider)
+            setattr(model, "provider", provider)
+            return model
 
         elif model_type == "openai":
             provider = OpenAIProvider(api_key=os.environ.get("OPENAI_API_KEY", ""))
 
-            return OpenAIModel(model_name=model_config["name"], provider=provider)
+            model = OpenAIModel(model_name=model_config["name"], provider=provider)
+            setattr(model, "provider", provider)
+            return model
 
         elif model_type == "anthropic":
             api_key = os.environ.get("ANTHROPIC_API_KEY", None)
@@ -259,7 +263,9 @@ class ModelFactory:
                 max_retries=azure_max_retries,
             )
             provider = OpenAIProvider(openai_client=azure_client)
-            return OpenAIModel(model_name=model_config["name"], provider=provider)
+            model = OpenAIModel(model_name=model_config["name"], provider=provider)
+            setattr(model, "provider", provider)
+            return model
 
         elif model_type == "custom_openai":
             url, headers, ca_certs_path, api_key = get_custom_config(model_config)
@@ -272,7 +278,9 @@ class ModelFactory:
                 provider_args["api_key"] = api_key
             provider = OpenAIProvider(**provider_args)
 
-            return OpenAIModel(model_name=model_config["name"], provider=provider)
+            model = OpenAIModel(model_name=model_config["name"], provider=provider)
+            setattr(model, "provider", provider)
+            return model
 
         else:
             raise ValueError(f"Unsupported model type: {model_type}")

{code_puppy-0.0.56 → code_puppy-0.0.58}/code_puppy/tools/common.py

@@ -38,4 +38,4 @@ def _find_best_window(
     console.log(f"Best span: {best_span}")
     console.log(f"Best window: {best_window}")
     console.log(f"Best score: {best_score}")
-    return (best_span, best_score)
+    return best_span, best_score

{code_puppy-0.0.56 → code_puppy-0.0.58}/code_puppy/tools/file_modifications.py

@@ -301,16 +301,19 @@ def register_file_modifications_tools(agent):
                         "changed": False,
                     }
                 return write_to_file(context, file_path, content, overwrite)
-        console.print(
-            "[bold red] Unable to route file modification tool call to sub-tool [/bold red]"
-        )
-        console.print("Inputs: ", path, diff)
-        return {
-            "success": False,
-            "path": file_path,
-            "message": "Wasn't able to route file modification to the right sub-tool!",
-            "changed": False,
-        }
+        try:
+            write_to_file(context, file_path, diff, overwrite=False)
+        except Exception as e:
+            console.print(
+                "[bold red] Unable to route file modification tool call to sub-tool [/bold red]"
+            )
+            console.print(str(e))
+            return {
+                "success": False,
+                "path": file_path,
+                "message": "Wasn't able to route file modification to the right sub-tool!",
+                "changed": False,
+            }
 
     @agent.tool
     def delete_file(context: RunContext, file_path: str) -> Dict[str, Any]:

{code_puppy-0.0.56 → code_puppy-0.0.58}/pyproject.toml

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