Added setup scripts

tools/github_tool.py

@@ -101,6 +101,67 @@ class GitHubTool(BaseTool):
                     },
                     "required": ["title", "body", "head"]
                 }
+            },
+            {
+                "name": "list_files",
+                "description": "List files in a directory of the repository",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "path": {
+                            "type": "string",
+                            "description": "Path to the directory in the repository"
+                        }
+                    },
+                    "required": ["path"]
+                }
+            },
+            {
+                "name": "search_code",
+                "description": "Search for code in the repository",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "query": {
+                            "type": "string",
+                            "description": "Search query"
+                        }
+                    },
+                    "required": ["query"]
+                }
+            },
+            {
+                "name": "get_commit_history",
+                "description": "Get commit history for a file",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "file_path": {
+                            "type": "string",
+                            "description": "Path to the file in the repository"
+                        },
+                        "num_commits": {
+                            "type": "integer",
+                            "description": "Number of commits to retrieve",
+                            "default": 10
+                        }
+                    },
+                    "required": ["file_path"]
+                }
+            },
+            {
+                "name": "get_branch_sha",
+                "description": "Get the SHA of the latest commit on a branch",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "branch": {
+                            "type": "string",
+                            "description": "Name of the branch"
+                        }
+                    },
+                    "required": ["branch"]
+                }
             }
         ]
 
@@ -112,7 +173,15 @@ class GitHubTool(BaseTool):
         elif function_name == "commit_file":
             return self._commit_file(kwargs["branch_name"], kwargs["file_path"], kwargs["content"], kwargs["commit_message"])
         elif function_name == "create_pull_request":
-            return self._create_pull_request(kwargs["title"], kwargs["body"], kwargs["head"], kwargs.get("base", "main"))
+            return self._create_pull_request(kwargs["title"], kwargs["body"], kwargs["head"], kwargs.get("base", "main"))  
+        elif function_name == "list_files":
+            return self._list_files(kwargs["path"])
+        elif function_name == "search_code":
+            return self._search_code(kwargs["query"])
+        elif function_name == "get_commit_history":
+            return self._get_commit_history(kwargs["file_path"], kwargs.get("num_commits", 10))
+        elif function_name == "get_branch_sha":
+            return self._get_branch_sha(kwargs["branch"])
         else:
             return f"Unknown function: {function_name}"
 
@@ -183,4 +252,48 @@ class GitHubTool(BaseTool):
             return f"Pull request created successfully: {response.json()['html_url']}"
         else:
             return f"Error creating pull request: {response.status_code}\nResponse: {response.text}"
+        
+    def _get_branch_sha(self, branch):
+            url = f"{self.base_url}/repos/{self.repo}/git/refs/heads/{branch}"
+            response = requests.get(url, headers=self.headers)
+            if response.status_code == 200:
+                return response.json()["object"]["sha"]
+            else:
+                return f"Error getting branch SHA: {response.status_code}"
 
+    def _list_files(self, path, branch):
+            url = f"{self.base_url}/repos/{self.repo}/contents/{path}"
+            params = {"ref": branch}
+            response = requests.get(url, headers=self.headers, params=params)
+            if response.status_code == 200:
+                files = [item["name"] for item in response.json() if item["type"] == "file"]
+                directories = [item["name"] for item in response.json() if item["type"] == "dir"]
+                return {"files": files, "directories": directories}
+            else:
+                return f"Error listing files: {response.status_code}"
+            
+    def _search_code(self, query):
+        url = f"{self.base_url}/search/code"
+        params = {
+            "q": f"{query} repo:{self.repo}",
+            "per_page": 10
+        }
+        response = requests.get(url, headers=self.headers, params=params)
+        if response.status_code == 200:
+            results = [{"file": item["path"], "url": item["html_url"]} for item in response.json()["items"]]
+            return results
+        else:
+            return f"Error searching code: {response.status_code}"
+
+    def _get_commit_history(self, file_path, num_commits):
+        url = f"{self.base_url}/repos/{self.repo}/commits"
+        params = {
+            "path": file_path,
+            "per_page": num_commits
+        }
+        response = requests.get(url, headers=self.headers, params=params)
+        if response.status_code == 200:
+            commits = [{"sha": commit["sha"], "message": commit["commit"]["message"], "date": commit["commit"]["author"]["date"]} for commit in response.json()]
+            return commits
+        else:
+            return f"Error getting commit history: {response.status_code}"