From 68a30cf5e188b83290add0ebda08fc6099a882f2 Mon Sep 17 00:00:00 2001 From: bucolucas Date: Mon, 19 Aug 2024 15:49:12 -0500 Subject: [PATCH] Add ListIssues class for list_issues function with JSON definition --- tools/github_tool_functions/list_issues.py | 83 ++++++++++++++++++++++ 1 file changed, 83 insertions(+) create mode 100644 tools/github_tool_functions/list_issues.py diff --git a/tools/github_tool_functions/list_issues.py b/tools/github_tool_functions/list_issues.py new file mode 100644 index 0000000..0a5c36b --- /dev/null +++ b/tools/github_tool_functions/list_issues.py @@ -0,0 +1,83 @@ +import requests +import logging + +class ListIssues: + def __init__(self, base_url, token, repo): + self.base_url = base_url + self.headers = { + "Authorization": f"token {token}", + "Accept": "application/vnd.github.v3+json" + } + self.repo = repo + + # Set up logging + self.logger = logging.getLogger(__name__) + self.logger.setLevel(logging.INFO) + + # Create a file handler + file_handler = logging.FileHandler('list_issues.log') + file_handler.setLevel(logging.INFO) + + # Create a console handler + console_handler = logging.StreamHandler() + console_handler.setLevel(logging.INFO) + + # Create a formatting for the logs + formatter = logging.Formatter('%(asctime)s - %(name)s - %(levellevel)s - %(message)s') + file_handler.setFormatter(formatter) + console_handler.setFormatter(formatter) + + # Add the handlers to the logger + self.logger.addHandler(file_handler) + self.logger.addHandler(console_handler) + + def __call__(self, state="open", per_page=30, page=1): + self.logger.info(f"Listing issues. State: {state}, Per page: {per_page}, Page: {page}") + url = f"{self.base_url}/repos/{self.repo}/issues" + params = { + "state": state, + "per_page": per_page, + "page": page + } + response = requests.get(url, headers=this.headers, params=params) + if response.status_code == 200: + issues = [{ + "number": issue["number"], + "title": issue["title"], + "state": issue["state"], + "created_at": issue["created_at"], + "url": issue["html_url"] + } for issue in response.json()] + self.logger.info(f"Successfully listed issues. Found {len(issues)} issues.") + return issues + else: + error_message = f"Error listing issues: {response.status_code}\nResponse: {response.text}" + self.logger.error(error_message) + return error_message + +# JSON definition for the list_issues function +list_issues_definition = { + "name": "list_issues", + "description": "List issues in the repository", + "parameters": { + "type": "object", + "properties": { + "state": { + "type": "string", + "enum": ["open", "closed", "all"], + "default": "open", + "description": "State of the issues to retrieve" + }, + "per_page": { + "type": "integer", + "default": 30, + "description": "Number of issues to return per page" + }, + "page": { + "type": "integer", + "default": 1, + "description": "Page number of the results to fetch" + } + } + } +} \ No newline at end of file