Added tests

tools/github_tool.py

@@ -0,0 +1,194 @@
+# tools/github_tool.py
+from .base_tool import BaseTool
+import requests
+import os
+
+class GitHubTool(BaseTool):
+    def __init__(self):
+        self.base_url = "https://api.github.com"
+        self.token = os.environ.get("GITHUB_TOKEN")
+        self.headers = {
+            "Authorization": f"token {self.token}",
+            "Accept": "application/vnd.github.v3+json"
+        }
+        self.repo = os.environ.get("GITHUB_REPOSITORY")
+
+    def get_functions(self):
+        return [
+            {
+                "name": "read_file",
+                "description": "Read a file from the repository",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "path": {
+                            "type": "string",
+                            "description": "Path to the file in the repository"
+                        }
+                    },
+                    "required": ["path"]
+                }
+            },
+            {
+                "name": "create_branch",
+                "description": "Create a new branch in the repository",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "branch_name": {
+                            "type": "string",
+                            "description": "Name of the new branch"
+                        },
+                        "base_branch": {
+                            "type": "string",
+                            "description": "Name of the base branch",
+                            "default": "main"
+                        }
+                    },
+                    "required": ["branch_name"]
+                }
+            },
+            {
+                "name": "commit_file",
+                "description": "Commit a file to a branch (not main)",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "branch_name": {
+                            "type": "string",
+                            "description": "Name of the branch to commit to"
+                        },
+                        "file_path": {
+                            "type": "string",
+                            "description": "Path to the file in the repository"
+                        },
+                        "content": {
+                            "type": "string",
+                            "description": "Content of the file"
+                        },
+                        "commit_message": {
+                            "type": "string",
+                            "description": "Commit message"
+                        }
+                    },
+                    "required": ["branch_name", "file_path", "content", "commit_message"]
+                }
+            },
+            {
+                "name": "create_pull_request",
+                "description": "Create a pull request",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "title": {
+                            "type": "string",
+                            "description": "Title of the pull request"
+                        },
+                        "body": {
+                            "type": "string",
+                            "description": "Body of the pull request"
+                        },
+                        "head": {
+                            "type": "string",
+                            "description": "The name of the branch where your changes are implemented"
+                        },
+                        "base": {
+                            "type": "string",
+                            "description": "The name of the branch you want the changes pulled into",
+                            "default": "main"
+                        }
+                    },
+                    "required": ["title", "body", "head"]
+                }
+            }
+        ]
+
+    def execute(self, function_name, **kwargs):
+        if function_name == "read_file":
+            return self._read_file(kwargs["path"])
+        elif function_name == "create_branch":
+            return self._create_branch(kwargs["branch_name"], kwargs.get("base_branch", "main"))
+        elif function_name == "commit_file":
+            return self._commit_file(kwargs["branch_name"], kwargs["file_path"], kwargs["content"], kwargs["commit_message"])
+        elif function_name == "push_branch":
+            return self._push_branch(kwargs["branch_name"])
+        elif function_name == "create_pull_request":
+            return self._create_pull_request(kwargs["title"], kwargs["body"], kwargs["head"], kwargs.get("base", "main"))
+        else:
+            return f"Unknown function: {function_name}"
+
+
+    def _read_file(self, path):
+        url = f"{self.base_url}/repos/{self.repo}/contents/{path}"
+        response = requests.get(url, headers=self.headers)
+        if response.status_code == 200:
+            content = response.json()["content"]
+            return content
+        else:
+            return f"Error reading file: {response.status_code}"
+
+    def _create_branch(self, branch_name, base_branch):
+        url = f"{self.base_url}/repos/{self.repo}/git/refs"
+        response = requests.get(f"{url}/heads/{base_branch}", headers=self.headers)
+        if response.status_code != 200:
+            return f"Error getting base branch: {response.status_code}"
+        
+        sha = response.json()["object"]["sha"]
+        data = {
+            "ref": f"refs/heads/{branch_name}",
+            "sha": sha
+        }
+        response = requests.post(url, headers=self.headers, json=data)
+        if response.status_code == 201:
+            return f"Branch '{branch_name}' created successfully"
+        else:
+            return f"Error creating branch: {response.status_code}"
+
+    def _commit_file(self, branch_name, file_path, content, commit_message):
+        if branch_name == "main":
+            return "Cannot commit directly to main branch"
+        
+        url = f"{self.base_url}/repos/{self.repo}/contents/{file_path}"
+        data = {
+            "message": commit_message,
+            "content": content,
+            "branch": branch_name
+        }
+        response = requests.put(url, headers=self.headers, json=data)
+        if response.status_code in [200, 201]:
+            return f"File committed successfully to branch '{branch_name}'"
+        else:
+            return f"Error committing file: {response.status_code}"
+
+    def _create_pull_request(self, title, body, head, base):
+        url = f"{self.base_url}/repos/{self.repo}/pulls"
+        data = {
+            "title": title,
+            "body": body,
+            "head": head,
+            "base": base
+        }
+        response = requests.post(url, headers=self.headers, json=data)
+        if response.status_code == 201:
+            return f"Pull request created successfully: {response.json()['html_url']}"
+        else:
+            return f"Error creating pull request: {response.status_code}\nResponse: {response.text}"
+
+    def _push_branch(self, branch_name):
+        url = f"{self.base_url}/repos/{self.repo}/git/refs/heads/{branch_name}"
+        response = requests.get(url, headers=self.headers)
+        if response.status_code != 200:
+            return f"Error getting branch information: {response.status_code}"
+        
+        sha = response.json()["object"]["sha"]
+        
+        push_url = f"{self.base_url}/repos/{self.repo}/git/refs/heads/{branch_name}"
+        data = {
+            "sha": sha,
+            "force": True
+        }
+        response = requests.patch(push_url, headers=self.headers, json=data)
+        if response.status_code == 200:
+            return f"Branch '{branch_name}' pushed successfully"
+        else:
+            return f"Error pushing branch: {response.status_code}\nResponse: {response.text}"