code-puppy 0.0.57__tar.gz → 0.0.59__tar.gz

{code_puppy-0.0.57 → code_puppy-0.0.59}/PKG-INFO

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

{code_puppy-0.0.57 → code_puppy-0.0.59}/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.57 → code_puppy-0.0.59}/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.57 → code_puppy-0.0.59}/code_puppy/tools/file_modifications.py

@@ -282,35 +282,38 @@ def register_file_modifications_tools(agent):
                     "changed": False,
                     "diff": "",
                 }
-        if isinstance(parsed_payload, dict):
-            if "delete_snippet" in parsed_payload:
-                snippet = parsed_payload["delete_snippet"]
-                return delete_snippet_from_file(context, file_path, snippet)
-            if "replacements" in parsed_payload:
-                replacements = parsed_payload["replacements"]
-                return replace_in_file(context, file_path, replacements)
-            if "content" in parsed_payload:
-                content = parsed_payload["content"]
-                overwrite = bool(parsed_payload.get("overwrite", False))
-                file_exists = os.path.exists(file_path)
-                if file_exists and not overwrite:
-                    return {
-                        "success": False,
-                        "path": file_path,
-                        "message": f"File '{file_path}' exists. Set 'overwrite': true to replace.",
-                        "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:
+            if isinstance(parsed_payload, dict):
+                if "delete_snippet" in parsed_payload:
+                    snippet = parsed_payload["delete_snippet"]
+                    return delete_snippet_from_file(context, file_path, snippet)
+                if "replacements" in parsed_payload:
+                    replacements = parsed_payload["replacements"]
+                    return replace_in_file(context, file_path, replacements)
+                if "content" in parsed_payload:
+                    content = parsed_payload["content"]
+                    overwrite = bool(parsed_payload.get("overwrite", False))
+                    file_exists = os.path.exists(file_path)
+                    if file_exists and not overwrite:
+                        return {
+                            "success": False,
+                            "path": file_path,
+                            "message": f"File '{file_path}' exists. Set 'overwrite': true to replace.",
+                            "changed": False,
+                        }
+                    return write_to_file(context, file_path, content, overwrite)
+            return 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": f"Something went wrong in file editing: {str(e)}",
+                "changed": False,
+            }
 
     @agent.tool
     def delete_file(context: RunContext, file_path: str) -> Dict[str, Any]:

{code_puppy-0.0.57 → code_puppy-0.0.59}/pyproject.toml

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