agentr 0.1.6__py3-none-any.whl → 0.1.8__py3-none-any.whl

agentr/__init__.py

@@ -1,2 +1,2 @@
-def main():
+def main():
     print("hello")

agentr/application.py

@@ -1,57 +1,94 @@
-from abc import ABC, abstractmethod
-from agentr.integration import Integration
-import httpx
-
-class Application(ABC):
-    """
-    Application is collection of tools that can be used by an agent.
-    """
-    def __init__(self, name: str, **kwargs):
-        self.name = name
-
-    @abstractmethod
-    def list_tools(self):
-        pass
-
-
-class APIApplication(Application):
-    """
-    APIApplication is an application that uses an API to interact with the world.
-    """
-    def __init__(self, name: str, integration: Integration = None, **kwargs):
-        super().__init__(name, **kwargs)
-        self.integration = integration
-
-    def _get_headers(self):
-        return {}
-    
-    def _get(self, url, params=None):
-        headers = self._get_headers()
-        response = httpx.get(url, headers=headers, params=params)
-        response.raise_for_status()
-        return response
-    
-    def _post(self, url, data):
-        headers = self._get_headers()
-        response = httpx.post(url, headers=headers, json=data)
-        response.raise_for_status()
-        return response
-
-    def _put(self, url, data):
-        headers = self._get_headers()
-        response = httpx.put(url, headers=headers, json=data)
-        response.raise_for_status()
-        return response
-
-    def _delete(self, url):
-        headers = self._get_headers()
-        response = httpx.delete(url, headers=headers)
-        response.raise_for_status()
-        return response
-
-    def validate(self):
-        pass
-    
-    @abstractmethod
-    def list_tools(self):
+from abc import ABC, abstractmethod
+
+from loguru import logger
+from agentr.exceptions import NotAuthorizedError
+from agentr.integration import Integration
+import httpx
+
+class Application(ABC):
+    """
+    Application is collection of tools that can be used by an agent.
+    """
+    def __init__(self, name: str, **kwargs):
+        self.name = name
+
+    @abstractmethod
+    def list_tools(self):
+        pass
+
+
+class APIApplication(Application):
+    """
+    APIApplication is an application that uses an API to interact with the world.
+    """
+    def __init__(self, name: str, integration: Integration = None, **kwargs):
+        super().__init__(name, **kwargs)
+        self.integration = integration
+
+    def _get_headers(self):
+        return {}
+    
+    def _get(self, url, params=None):
+        try:
+            headers = self._get_headers()
+            response = httpx.get(url, headers=headers, params=params)
+            response.raise_for_status()
+            return response
+        except NotAuthorizedError as e:
+            logger.warning(f"Authorization needed: {e.message}")
+            raise e
+        except Exception as e:
+            logger.error(f"Error getting {url}: {e}")
+            raise e
+
+    
+    def _post(self, url, data):
+        try:
+            headers = self._get_headers()
+            response = httpx.post(url, headers=headers, json=data)
+            response.raise_for_status()
+            return response
+        except NotAuthorizedError as e:
+            logger.warning(f"Authorization needed: {e.message}")
+            raise e
+        except httpx.HTTPStatusError as e:
+            if e.response.status_code == 429:
+                return e.response.text or "Rate limit exceeded. Please try again later."
+            else:
+                raise e
+        except Exception as e:
+            logger.error(f"Error posting {url}: {e}")
+            raise e
+
+    def _put(self, url, data):
+        try:
+            headers = self._get_headers()
+            response = httpx.put(url, headers=headers, json=data)
+            response.raise_for_status()
+            return response
+        except NotAuthorizedError as e:
+            logger.warning(f"Authorization needed: {e.message}")
+            raise e
+        except Exception as e:
+            logger.error(f"Error posting {url}: {e}")
+            raise e
+
+    def _delete(self, url):
+        try:
+            headers = self._get_headers()
+            response = httpx.delete(url, headers=headers)
+            response.raise_for_status()
+            return response
+        except NotAuthorizedError as e:
+            logger.warning(f"Authorization needed: {e.message}")
+            raise e
+        except Exception as e:
+            logger.error(f"Error posting {url}: {e}")
+            raise e
+
+    def validate(self):
+        pass
+    
+    @abstractmethod
+    def list_tools(self):
         pass

agentr/applications/__init__.py

@@ -0,0 +1,27 @@
+from agentr.applications.zenquotes.app import ZenQuoteApp
+from agentr.applications.tavily.app import TavilyApp
+from agentr.applications.github.app import GithubApp
+from agentr.applications.google_calendar.app import GoogleCalendarApp
+from agentr.applications.google_mail.app import GmailApp
+from agentr.applications.resend.app import ResendApp
+from agentr.applications.reddit.app import RedditApp
+
+def app_from_name(name: str):
+    name = name.lower().strip()
+    name = name.replace(" ", "-")
+    if name == "zenquotes":
+        return ZenQuoteApp
+    elif name == "tavily":
+        return TavilyApp
+    elif name == "github":
+        return GithubApp
+    elif name == "google-calendar":
+        return GoogleCalendarApp
+    elif name == "google-mail":
+        return GmailApp
+    elif name == "resend":
+        return ResendApp
+    elif name == "reddit":
+        return RedditApp
+    else:
+        raise ValueError(f"App {name} not found")

agentr/applications/github/app.py

@@ -1,56 +1,319 @@
-from agentr.integration import Integration
-from agentr.application import APIApplication
-from loguru import logger
-from agentr.exceptions import NotAuthorizedError
-
-class GithubApp(APIApplication):
-    def __init__(self, integration: Integration) -> None:
-        super().__init__(name="github", integration=integration)
-
-    def _get_headers(self):
-        if not self.integration:
-            raise ValueError("Integration not configured")
-        credentials = self.integration.get_credentials()
-        if not credentials:
-            logger.warning("No credentials found")
-            action = self.integration.authorize()
-            raise NotAuthorizedError(action)
-        if "headers" in credentials:
-            return credentials["headers"]
-        return {
-            "Authorization": f"Bearer {credentials['access_token']}",
-            "Accept": "application/vnd.github.v3+json"
-        }
-    
-
-    def star_repository(self, repo_full_name: str) -> str:
-        """Star a GitHub repository
-        
-            Args:
-                repo_full_name: The full name of the repository (e.g. 'owner/repo')
-                
-            Returns:
-            
-                A confirmation message
-        """
-        try:
-            url = f"https://api.github.com/user/starred/{repo_full_name}"
-            response = self._put(url, data={})
-            
-            if response.status_code == 204:
-                return f"Successfully starred repository {repo_full_name}"
-            elif response.status_code == 404:
-                return f"Repository {repo_full_name} not found"
-            else:
-                logger.error(response.text)
-                return f"Error starring repository: {response.text}"
-        except NotAuthorizedError as e:
-            return e.message
-        except Exception as e:
-            logger.error(e)
-            raise e
-       
-
-    
-    def list_tools(self):
-        return [self.star_repository]
+from agentr.integration import Integration
+from agentr.application import APIApplication
+from loguru import logger
+
+class GithubApp(APIApplication):
+    def __init__(self, integration: Integration) -> None:
+        super().__init__(name="github", integration=integration)
+        self.base_api_url = "https://api.github.com/repos"
+
+    def _get_headers(self):
+        if not self.integration:
+            raise ValueError("Integration not configured")
+        credentials = self.integration.get_credentials()
+        if "headers" in credentials:
+            return credentials["headers"]
+        return {
+            "Authorization": f"Bearer {credentials['access_token']}",
+            "Accept": "application/vnd.github.v3+json"
+        }
+    
+
+    def star_repository(self, repo_full_name: str) -> str:
+        """Star a GitHub repository
+        
+        Args:
+            repo_full_name: The full name of the repository (e.g. 'owner/repo')
+            
+        Returns:
+        
+            A confirmation message
+        """
+        url = f"https://api.github.com/user/starred/{repo_full_name}"
+        response = self._put(url, data={})
+        
+        if response.status_code == 204:
+            return f"Successfully starred repository {repo_full_name}"
+        elif response.status_code == 404:
+            return f"Repository {repo_full_name} not found"
+        else:
+            logger.error(response.text)
+            return f"Error starring repository: {response.text}"
+        
+
+    def list_commits(self, repo_full_name: str) -> str:
+        """List recent commits for a GitHub repository
+        
+        Args:
+            repo_full_name: The full name of the repository (e.g. 'owner/repo')
+            
+        Returns:
+            A formatted list of recent commits
+        """
+        repo_full_name = repo_full_name.strip()
+        url = f"{self.base_api_url}/{repo_full_name}/commits"
+        response = self._get(url)
+        response.raise_for_status()
+        
+        commits = response.json()
+        if not commits:
+            return f"No commits found for repository {repo_full_name}"
+        
+        result = f"Recent commits for {repo_full_name}:\n\n"
+        for commit in commits[:12]:  # Limit to 12 commits 
+            sha = commit.get("sha", "")[:7]
+            message = commit.get("commit", {}).get("message", "").split('\n')[0]
+            author = commit.get("commit", {}).get("author", {}).get("name", "Unknown")
+            
+            result += f"- {sha}: {message} (by {author})\n"
+        
+        return result
+
+    def list_branches(self, repo_full_name: str) -> str:
+        """List branches for a GitHub repository
+        
+        Args:
+            repo_full_name: The full name of the repository (e.g. 'owner/repo')
+            
+        Returns:
+            A formatted list of branches
+        """
+        repo_full_name = repo_full_name.strip()
+        url = f"{self.base_api_url}/{repo_full_name}/branches"
+        response = self._get(url)
+        response.raise_for_status()
+        
+        branches = response.json()
+        if not branches:
+            return f"No branches found for repository {repo_full_name}"
+        
+        result = f"Branches for {repo_full_name}:\n\n"
+        for branch in branches:
+            branch_name = branch.get("name", "Unknown")
+            result += f"- {branch_name}\n"
+        
+        return result
+    
+    def list_pull_requests(self, repo_full_name: str, state: str = "open") -> str:
+        """List pull requests for a GitHub repository
+        
+        Args:
+            repo_full_name: The full name of the repository (e.g. 'owner/repo')
+            state: The state of the pull requests to filter by (open, closed, or all)
+            
+        Returns:
+            A formatted list of pull requests
+        """
+        repo_full_name = repo_full_name.strip()
+        url = f"{self.base_api_url}/{repo_full_name}/pulls"
+        params = {"state": state}
+        response = self._get(url, params=params)
+        response.raise_for_status()
+        
+        pull_requests = response.json()
+        if not pull_requests:
+            return f"No pull requests found for repository {repo_full_name} with state '{state}'"
+        
+        result = f"Pull requests for {repo_full_name} (State: {state}):\n\n"
+        for pr in pull_requests:
+            pr_title = pr.get("title", "No Title")
+            pr_number = pr.get("number", "Unknown")
+            pr_state = pr.get("state", "Unknown")
+            pr_user = pr.get("user", {}).get("login", "Unknown")
+            
+            result += f"- PR #{pr_number}: {pr_title} (by {pr_user}, Status: {pr_state})\n"
+        
+        return result
+
+    def list_issues(self, repo_full_name: str, per_page: int = 30, page: int = 1) -> str:
+        """List issues for a GitHub repository
+        
+        Args:
+            repo_full_name: The full name of the repository (e.g. 'owner/repo')
+            per_page: The number of results per page (max 100)
+            page: The page number of the results to fetch
+            
+        Returns:
+            A formatted list of issues
+        """
+        repo_full_name = repo_full_name.strip()
+        url = f"{self.base_api_url}/{repo_full_name}/issues/events"
+        params = {
+            "per_page": per_page,
+            "page": page
+        }
+        response = self._get(url, params=params)
+        response.raise_for_status()
+        
+        issues = response.json()
+        if not issues:
+            return f"No issues found for repository {repo_full_name}"
+        
+        result = f"Issues for {repo_full_name} (Page {page}):\n\n"
+        for issue in issues:
+            issue_title = issue.get("issue", {}).get("title", "No Title")
+            issue_number = issue.get("issue", {}).get("number", "Unknown")
+            issue_state = issue.get("issue", {}).get("state", "Unknown")
+            issue_user = issue.get("issue", {}).get("user", {}).get("login", "Unknown")
+            
+            result += f"- Issue #{issue_number}: {issue_title} (by {issue_user}, Status: {issue_state})\n"
+        
+        return result
+
+    def get_pull_request(self, repo_full_name: str, pull_number: int) -> str:
+        """Get a specific pull request for a GitHub repository
+        
+        Args:
+            repo_full_name: The full name of the repository (e.g. 'owner/repo')
+            pull_number: The number of the pull request to retrieve
+            
+        Returns:
+            A formatted string with pull request details
+        """
+        repo_full_name = repo_full_name.strip()
+        url = f"{self.base_api_url}/{repo_full_name}/pulls/{pull_number}"
+        response = self._get(url)
+        response.raise_for_status()
+        
+        pr = response.json()
+        pr_title = pr.get("title", "No Title")
+        pr_number = pr.get("number", "Unknown")
+        pr_state = pr.get("state", "Unknown")
+        pr_user = pr.get("user", {}).get("login", "Unknown")
+        pr_body = pr.get("body", "No description provided.")
+        
+        result = (
+            f"Pull Request #{pr_number}: {pr_title}\n"
+            f"Created by: {pr_user}\n"
+            f"Status: {pr_state}\n"
+            f"Description: {pr_body}\n"
+        )
+        
+        return result
+
+    def create_pull_request(self, repo_full_name: str, title: str, head: str, base: str, body: str = "", 
+                            maintainer_can_modify: bool = True, draft: bool = False) -> str:
+        """Create a new pull request for a GitHub repository
+        
+        Args:
+            repo_full_name: The full name of the repository (e.g. 'owner/repo')
+            title: The title of the new pull request
+            head: The name of the branch where your changes are implemented
+            base: The name of the branch you want the changes pulled into
+            body: The contents of the pull request
+            maintainer_can_modify: Indicates whether maintainers can modify the pull request
+            draft: Indicates whether the pull request is a draft
+            
+        Returns:
+            A confirmation message with the new pull request details
+        """
+        repo_full_name = repo_full_name.strip()
+        url = f"{self.base_api_url}/{repo_full_name}/pulls"
+        
+        pull_request_data = {
+            "title": title,
+            "head": head,
+            "base": base,
+            "body": body,
+            "maintainer_can_modify": maintainer_can_modify,
+            "draft": draft
+        }
+        
+        response = self._post(url, pull_request_data)
+        response.raise_for_status()
+        
+        pr = response.json()
+        pr_number = pr.get("number", "Unknown")
+        pr_url = pr.get("html_url", "")
+        
+        return f"Successfully created pull request #{pr_number}:\n" \
+               f"Title: {title}\n" \
+               f"From: {head} → To: {base}\n" \
+               f"URL: {pr_url}"
+
+    def create_issue(self, repo_full_name: str, title: str, body: str = "", labels = None) -> str:
+        """Create a new issue in a GitHub repository
+        
+        Args:
+            repo_full_name: The full name of the repository (e.g. 'owner/repo')
+            title: The title of the issue
+            body: The contents of the issue
+            labels: Labels to associate with this issue. Enter as a comma-separated string
+                   (e.g. "bug,enhancement,documentation"). 
+                   NOTE: Only users with push access can set labels for new issues. 
+                   Labels are silently dropped otherwise.
+            
+        Returns:
+            A confirmation message with the new issue details
+        """
+        repo_full_name = repo_full_name.strip()
+        url = f"{self.base_api_url}/{repo_full_name}/issues"
+        
+        issue_data = {
+            "title": title,
+            "body": body
+        }
+        
+        if labels:
+            if isinstance(labels, str):
+                labels_list = [label.strip() for label in labels.split(",") if label.strip()]
+                issue_data["labels"] = labels_list
+            else:
+                issue_data["labels"] = labels
+        
+        response = self._post(url, issue_data)
+        response.raise_for_status()
+        
+        issue = response.json()
+        issue_number = issue.get("number", "Unknown")
+        issue_url = issue.get("html_url", "")
+        
+        return f"Successfully created issue #{issue_number}:\n" \
+               f"Title: {title}\n" \
+               f"URL: {issue_url}"
+
+    def list_repo_activities(self, repo_full_name: str, direction: str = "desc", per_page: int = 30) -> str:
+        """List activities for a GitHub repository
+        
+        Args:
+            repo_full_name: The full name of the repository (e.g. 'owner/repo')
+            direction: The direction to sort the results by (asc or desc). Default: desc
+            per_page: The number of results per page (max 100). Default: 30
+            
+        Returns:
+            A formatted list of repository activities
+        """
+        repo_full_name = repo_full_name.strip()
+        url = f"{self.base_api_url}/{repo_full_name}/activity"
+        
+        # Build query parameters
+        params = {
+            "direction": direction,
+            "per_page": per_page
+        }
+        
+        response = self._get(url, params=params)
+        response.raise_for_status()
+        
+        activities = response.json()
+        if not activities:
+            return f"No activities found for repository {repo_full_name}"
+        
+        result = f"Repository activities for {repo_full_name}:\n\n"
+        
+        for activity in activities:
+            # Extract common fields
+            timestamp = activity.get("timestamp", "Unknown time")
+            actor_name = "Unknown user"
+            if "actor" in activity and activity["actor"]:
+                actor_name = activity["actor"].get("login", "Unknown user")
+            
+            # Create a simple description of the activity
+            result += f"- {actor_name} performed an activity at {timestamp}\n"
+        
+        return result
+
+    def list_tools(self):
+        return [self.star_repository, self.list_commits, self.list_branches, 
+                self.list_pull_requests, self.list_issues, self.get_pull_request, 
+                self.create_pull_request, self.create_issue, self.list_repo_activities]