Class GitHubClient

autoagent/tools/github_client.py:6–66

GitHub operation client

from typing import Optional, Dict, List
import time
class GitHubClient:
    """GitHub operation client"""
    
    def __init__(self, token: Optional[str] = None):
        """
        Initialize the GitHub client
        
        Args:
            token: GitHub Personal Access Token, if None, try to get from environment variable
        """
        self.token = token or os.getenv('GITHUB_AI_TOKEN')
        if not self.token:
            raise ValueError("GitHub Token is required, please provide it via the token parameter or set the GITHUB_AI_TOKEN environment variable.")
        
        self.session = requests.Session()
        self.session.headers.update({
            'Authorization': f'token {self.token}',
            'Accept': 'application/vnd.github.v3+json'
        })
        self.api_base = 'https://api.github.com'
    
    def check_auth(self) -> dict:
        """
        Check if the token is valid
        """
        try:
            response = self.session.get(f'{self.api_base}/user')
            response.raise_for_status()
            return {'status': 0, 'message': 'Authentication successful', 'user': response.json()}
        except Exception as e:
            return {'status': -1, 'message': f'Authentication failed: {str(e)}'}
    
    def create_pull_request(self, repo: str, title: str, body: str, head: str, base: str) -> dict:
        """
        Create a Pull Request
        
        Args:
            repo: The full name of the repository (e.g., 'owner/repo')
            title: The PR title
            body: The PR description
            head: The source branch
            base: The target branch
        """
        try:
            url = f'{self.api_base}/repos/{repo}/pulls'
            data = {
                'title': title,
                'body': body,
                'head': head,
                'base': base
            }
            response = self.session.post(url, json=data)
            response.raise_for_status()
            pr_data = response.json()
            return {
                'status': 0,
                'message': f'PR created successfully: {pr_data["html_url"]}',
                'pr_url': pr_data['html_url']