github-agent 0.2.55__tar.gz → 0.3.0__tar.gz

{github_agent-0.2.55 → github_agent-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: github-agent
-Version: 0.2.55
+Version: 0.3.0
 Summary: GitHub Agent for MCP
 Author-email: Audel Rouhi <knucklessg1@gmail.com>
 License: MIT
@@ -12,7 +12,10 @@ Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: agent-utilities[agent,logfire]>=0.2.31
+Requires-Dist: agent-utilities[agent,logfire]>=0.2.42
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"
+Requires-Dist: pytest-asyncio; extra == "test"
 Dynamic: license-file
 
 # GitHub Agent - A2A | AG-UI | MCP
@@ -38,7 +41,7 @@ Dynamic: license-file
 ![PyPI - Wheel](https://img.shields.io/pypi/wheel/github-agent)
 ![PyPI - Implementation](https://img.shields.io/pypi/implementation/github-agent)
 
-*Version: 0.2.55*
+*Version: 0.3.0*
 
 ## Overview
 

{github_agent-0.2.55 → github_agent-0.3.0}/README.md

@@ -21,7 +21,7 @@
 ![PyPI - Wheel](https://img.shields.io/pypi/wheel/github-agent)
 ![PyPI - Implementation](https://img.shields.io/pypi/implementation/github-agent)
 
-*Version: 0.2.55*
+*Version: 0.3.0*
 
 ## Overview
 

{github_agent-0.2.55 → github_agent-0.3.0}/github_agent/agent_server.py

@@ -1,8 +1,7 @@
 #!/usr/bin/python
-# coding: utf-8
+import logging
 import os
 import sys
-import logging
 import warnings
 
 from agent_utilities import (
@@ -13,7 +12,7 @@ from agent_utilities import (
     load_identity,
 )
 
-__version__ = "0.2.55"
+__version__ = "0.3.0"
 
 logging.basicConfig(
     level=logging.INFO,

github_agent-0.2.55/github_agent/api_wrapper.py → github_agent-0.3.0/github_agent/api_client.py

@@ -1,51 +1,52 @@
 #!/usr/bin/python
 
+import logging
 import re
+from concurrent.futures import ThreadPoolExecutor, as_completed
+from typing import TypeVar
+
 import requests
 import urllib3
-import logging
-from typing import List, TypeVar, Tuple
-from pydantic import ValidationError
-from concurrent.futures import ThreadPoolExecutor, as_completed
 from agent_utilities.base_utilities import get_logger
+from pydantic import BaseModel, ValidationError
 
 logger = get_logger(__name__)
 
+from agent_utilities.core.decorators import require_auth
+from agent_utilities.core.exceptions import (
+    AuthError,
+    MissingParameterError,
+    ParameterError,
+    UnauthorizedError,
+)
+
 from github_agent.github_input_models import (
-    RepoModel,
-    IssueModel,
-    PullRequestModel,
-    ContentModel,
     BranchModel,
     CommitModel,
+    ContentModel,
+    IssueModel,
+    PullRequestModel,
+    RepoModel,
 )
 from github_agent.github_response_models import (
-    Repository,
-    Issue,
-    PullRequest,
-    Content,
     Branch,
     Commit,
+    Content,
+    Issue,
+    PullRequest,
+    Repository,
     Response,
 )
-from agent_utilities.decorators import require_auth
-from agent_utilities.exceptions import (
-    AuthError,
-    UnauthorizedError,
-    ParameterError,
-    MissingParameterError,
-)
-
-T = TypeVar("T")
 
+T = TypeVar("T", bound=BaseModel)
 
-class Api(object):
 
+class Api:
     def __init__(
         self,
         url: str = "https://api.github.com",
-        token: str = None,
-        proxies: dict = None,
+        token: str | None = None,
+        proxies: dict | None = None,
         verify: bool = True,
         debug: bool = False,
     ):
@@ -73,7 +74,6 @@ class Api(object):
         if token:
             self.headers["Authorization"] = f"Bearer {token}"
         else:
-
             logger.warning("No token provided for GitHub API")
 
         try:
@@ -84,20 +84,20 @@ class Api(object):
                 proxies=self.proxies,
             )
             if response.status_code in (401, 403):
-                logger.error(f"Authentication Error: {response.content}")
+                logger.error(f"Authentication Error: {response.text}")
                 raise AuthError if response.status_code == 401 else UnauthorizedError
         except requests.exceptions.RequestException as e:
             logger.error(f"Connection Error: {str(e)}")
 
     def _fetch_next_page(
         self, endpoint: str, model: T, header: dict, page: int
-    ) -> List[dict]:
+    ) -> list[dict]:
         """Fetch a single page of data from the specified endpoint"""
-        model.page = page
+        model.page = page  # type: ignore[attr-defined]
         model.model_post_init(None)
         response = self._session.get(
             url=f"{self.url}{endpoint}" if endpoint.startswith("/") else endpoint,
-            params=model.api_parameters,
+            params=model.api_parameters,  # type: ignore[attr-defined]
             headers=header,
             verify=self.verify,
             proxies=self.proxies,
@@ -119,7 +119,7 @@ class Api(object):
 
     def _fetch_all_pages(
         self, endpoint: str, model: T
-    ) -> Tuple[requests.Response, List[dict]]:
+    ) -> tuple[requests.Response, list[dict]]:
         """Generic method to fetch all pages with parallelization if possible"""
         all_data = []
 
@@ -127,7 +127,7 @@ class Api(object):
 
         response = self._session.get(
             url=initial_url,
-            params=model.api_parameters,
+            params=model.api_parameters,  # type: ignore[attr-defined]
             headers=self.headers,
             verify=self.verify,
             proxies=self.proxies,
@@ -142,14 +142,15 @@ class Api(object):
 
         total_pages = self._get_total_pages(response)
 
-        if not model.max_pages or model.max_pages == 0 or model.max_pages > total_pages:
-            model.max_pages = total_pages
-
-        if model.max_pages > 1:
+        max_pages = getattr(model, "max_pages", total_pages)
+        if not max_pages or max_pages == 0 or max_pages > total_pages:
+            max_pages = total_pages
+            model.max_pages = total_pages  # type: ignore[attr-defined]
 
+        if max_pages > 1:
             with ThreadPoolExecutor(max_workers=5) as executor:
                 futures = []
-                for page in range(2, model.max_pages + 1):
+                for page in range(2, max_pages + 1):
                     futures.append(
                         executor.submit(
                             self._fetch_next_page,
@@ -177,7 +178,7 @@ class Api(object):
             parsed_data = [Repository(**item) for item in data]
             return Response(response=response, data=parsed_data)
         except ValidationError as e:
-            raise ParameterError(f"Invalid parameters: {e.errors()}")
+            raise ParameterError(f"Invalid parameters: {e.errors()}") from e
 
     @require_auth
     def get_repository(self, owner: str, repo: str) -> Response:
@@ -194,8 +195,8 @@ class Api(object):
             return Response(response=response, data=parsed_data)
         except requests.exceptions.HTTPError as e:
             if e.response.status_code == 404:
-                raise ParameterError(f"Repository {owner}/{repo} not found")
-            raise e
+                raise ParameterError(f"Repository {owner}/{repo} not found") from e
+            raise
 
     @require_auth
     def get_issues(self, **kwargs) -> Response:
@@ -210,7 +211,7 @@ class Api(object):
             parsed_data = [Issue(**item) for item in data]
             return Response(response=response, data=parsed_data)
         except ValidationError as e:
-            raise ParameterError(f"Invalid parameters: {e.errors()}")
+            raise ParameterError(f"Invalid parameters: {e.errors()}") from e
 
     @require_auth
     def get_pull_requests(self, **kwargs) -> Response:
@@ -225,7 +226,7 @@ class Api(object):
             parsed_data = [PullRequest(**item) for item in data]
             return Response(response=response, data=parsed_data)
         except ValidationError as e:
-            raise ParameterError(f"Invalid parameters: {e.errors()}")
+            raise ParameterError(f"Invalid parameters: {e.errors()}") from e
 
     @require_auth
     def get_contents(self, **kwargs) -> Response:
@@ -241,13 +242,14 @@ class Api(object):
             )
             response.raise_for_status()
             data = response.json()
+            parsed_data: list[Content] | Content
             if isinstance(data, list):
                 parsed_data = [Content(**item) for item in data]
             else:
                 parsed_data = Content(**data)
             return Response(response=response, data=parsed_data)
         except ValidationError as e:
-            raise ParameterError(f"Invalid parameters: {e.errors()}")
+            raise ParameterError(f"Invalid parameters: {e.errors()}") from e
 
     @require_auth
     def get_branches(self, **kwargs) -> Response:
@@ -260,7 +262,7 @@ class Api(object):
             parsed_data = [Branch(**item) for item in data]
             return Response(response=response, data=parsed_data)
         except ValidationError as e:
-            raise ParameterError(f"Invalid parameters: {e.errors()}")
+            raise ParameterError(f"Invalid parameters: {e.errors()}") from e
 
     @require_auth
     def get_commits(self, **kwargs) -> Response:
@@ -273,4 +275,4 @@ class Api(object):
             parsed_data = [Commit(**item) for item in data]
             return Response(response=response, data=parsed_data)
         except ValidationError as e:
-            raise ParameterError(f"Invalid parameters: {e.errors()}")
+            raise ParameterError(f"Invalid parameters: {e.errors()}") from e

{github_agent-0.2.55 → github_agent-0.3.0}/github_agent/auth.py

@@ -2,12 +2,12 @@
 
 import os
 import threading
-from typing import Optional
 
 import requests
-from agent_utilities.base_utilities import to_boolean, get_logger
-from github_agent.api_wrapper import Api
-from agent_utilities.exceptions import AuthError, UnauthorizedError
+from agent_utilities.base_utilities import get_logger, to_boolean
+from agent_utilities.core.exceptions import AuthError, UnauthorizedError
+
+from github_agent.api_client import Api
 
 local = threading.local()
 logger = get_logger(__name__)
@@ -15,9 +15,9 @@ logger = get_logger(__name__)
 
 def get_client(
     instance: str = os.getenv("GITHUB_URL", "https://api.github.com"),
-    token: Optional[str] = os.getenv("GITHUB_TOKEN", None),
+    token: str | None = os.getenv("GITHUB_TOKEN", None),
     verify: bool = to_boolean(string=os.getenv("GITHUB_VERIFY", "True")),
-    config: Optional[dict] = None,
+    config: dict | None = None,
 ) -> Api:
     """
     Factory function to create the GitHub Api client.
@@ -53,14 +53,14 @@ def get_client(
         auth = (config["oidc_client_id"], config["oidc_client_secret"])
         try:
             response = requests.post(
-                config["token_endpoint"], data=exchange_data, auth=auth
+                config["token_endpoint"], data=exchange_data, auth=auth, timeout=30
             )
             response.raise_for_status()
             new_token = response.json()["access_token"]
             logger.info("Token exchange successful")
         except Exception as e:
             logger.error(f"Token exchange failed: {str(e)}")
-            raise RuntimeError(f"Token exchange failed: {str(e)}")
+            raise RuntimeError(f"Token exchange failed: {str(e)}") from e
 
         try:
             return Api(

{github_agent-0.2.55 → github_agent-0.3.0}/github_agent/github_input_models.py

@@ -1,6 +1,6 @@
-#!/usr/bin/python
+from typing import Any
 
-from typing import Union, List, Dict, Optional
+#!/usr/bin/python
 from pydantic import (
     BaseModel,
     Field,
@@ -12,12 +12,14 @@ class BaseModelWrapper(BaseModel):
     Base Model wrapping common functionalities.
     """
 
-    max_pages: Optional[int] = Field(
+    max_pages: int | None = Field(
         description="Max amount of pages to retrieve", default=None
     )
-    page: Optional[int] = Field(description="Pagination page", default=1)
-    per_page: Optional[int] = Field(description="Results per page", default=100)
-    api_parameters: Optional[Dict] = Field(description="API Parameters", default=None)
+    page: int | None = Field(description="Pagination page", default=1)
+    per_page: int | None = Field(description="Results per page", default=100)
+    api_parameters: dict[str, Any] = Field(
+        description="API Parameters", default_factory=dict
+    )
 
     def model_post_init(self, __context):
         self.api_parameters = {}
@@ -32,15 +34,13 @@ class RepoModel(BaseModelWrapper):
     Pydantic model for Repository requests.
     """
 
-    owner: Optional[str] = Field(
-        None, description="The account owner of the repository."
-    )
-    repo: Optional[str] = Field(None, description="The name of the repository.")
-    visibility: Optional[str] = Field(
+    owner: str | None = Field(None, description="The account owner of the repository.")
+    repo: str | None = Field(None, description="The name of the repository.")
+    visibility: str | None = Field(
         None, description="Can be one of all, public, or private."
     )
-    affiliation: Optional[str] = Field(None, description="Affiliation filter.")
-    type: Optional[str] = Field(
+    affiliation: str | None = Field(None, description="Affiliation filter.")
+    type: str | None = Field(
         None, description="Can be one of all, owner, public, private, member."
     )
 
@@ -61,25 +61,25 @@ class IssueModel(BaseModelWrapper):
 
     owner: str = Field(..., description="The account owner of the repository.")
     repo: str = Field(..., description="The name of the repository.")
-    issue_number: Optional[int] = Field(
+    issue_number: int | None = Field(
         None, description="The number that identifies the issue."
     )
-    state: Optional[str] = Field(
+    state: str | None = Field(
         None,
         description="Indicates the state of the issues to return. Can be either open, closed, or all.",
     )
-    labels: Optional[Union[str, List[str]]] = Field(
+    labels: str | list[str] | None = Field(
         None, description="A list of comma separated label names."
     )
-    assignee: Optional[str] = Field(
+    assignee: str | None = Field(
         None,
         description="Can be the name of a user. Use none for issues with no assigned user, and * for assigned issues to any user.",
     )
-    creator: Optional[str] = Field(None, description="The user that created the issue.")
-    mentioned: Optional[str] = Field(
+    creator: str | None = Field(None, description="The user that created the issue.")
+    mentioned: str | None = Field(
         None, description="A user that is mentioned in the issue."
     )
-    since: Optional[str] = Field(
+    since: str | None = Field(
         None, description="Only show notifications updated after the given time."
     )
 
@@ -105,22 +105,22 @@ class PullRequestModel(BaseModelWrapper):
 
     owner: str = Field(..., description="The account owner of the repository.")
     repo: str = Field(..., description="The name of the repository.")
-    pull_number: Optional[int] = Field(
+    pull_number: int | None = Field(
         None, description="The number that identifies the pull request."
     )
-    state: Optional[str] = Field(
+    state: str | None = Field(
         None, description="State of the PR. (open, closed, or all)"
     )
-    head: Optional[str] = Field(
+    head: str | None = Field(
         None,
         description="Filter pulls by head user or head organization and branch name in the format of user:ref-name or organization:ref-name.",
     )
-    base: Optional[str] = Field(None, description="Filter pulls by base branch name.")
-    sort: Optional[str] = Field(
+    base: str | None = Field(None, description="Filter pulls by base branch name.")
+    sort: str | None = Field(
         None,
         description="What to sort results by. Can be created, updated, popularity, long-running.",
     )
-    direction: Optional[str] = Field(
+    direction: str | None = Field(
         None, description="The direction of the sort. Can be asc or desc."
     )
 
@@ -146,7 +146,7 @@ class ContentModel(BaseModelWrapper):
     owner: str = Field(..., description="The account owner of the repository.")
     repo: str = Field(..., description="The name of the repository.")
     path: str = Field(..., description="The content path.")
-    ref: Optional[str] = Field(
+    ref: str | None = Field(
         None,
         description="The name of the commit/branch/tag. Default: the repository's default branch.",
     )
@@ -164,7 +164,7 @@ class BranchModel(BaseModelWrapper):
 
     owner: str = Field(..., description="The account owner of the repository.")
     repo: str = Field(..., description="The name of the repository.")
-    branch: Optional[str] = Field(None, description="The name of the branch.")
+    branch: str | None = Field(None, description="The name of the branch.")
 
     def model_post_init(self, __context):
         super().model_post_init(__context)
@@ -177,19 +177,19 @@ class CommitModel(BaseModelWrapper):
 
     owner: str = Field(..., description="The account owner of the repository.")
     repo: str = Field(..., description="The name of the repository.")
-    sha: Optional[str] = Field(
+    sha: str | None = Field(
         None, description="SHA or branch to start listing commits from."
     )
-    path: Optional[str] = Field(
+    path: str | None = Field(
         None, description="Only commits containing this file path will be returned."
     )
-    author: Optional[str] = Field(
+    author: str | None = Field(
         None, description="GitHub username or email address to filter by commit author."
     )
-    since: Optional[str] = Field(
+    since: str | None = Field(
         None, description="Only show notifications updated after the given time."
     )
-    until: Optional[str] = Field(
+    until: str | None = Field(
         None, description="Only show notifications updated before the given time."
     )
 

{github_agent-0.2.55 → github_agent-0.3.0}/github_agent/github_response_models.py

@@ -1,13 +1,14 @@
 #!/usr/bin/python
 
-from typing import List, Dict, Optional, Any
+from datetime import datetime
+from typing import Any
+
+import requests
 from pydantic import (
     BaseModel,
     ConfigDict,
     HttpUrl,
 )
-from datetime import datetime
-import requests
 
 
 class Response(BaseModel):
@@ -17,7 +18,7 @@ class Response(BaseModel):
 
     model_config = ConfigDict(arbitrary_types_allowed=True)
     response: requests.Response
-    data: Any = None
+    data: Any | None = None
 
 
 class User(BaseModel):
@@ -41,7 +42,7 @@ class Repository(BaseModel):
     private: bool
     owner: User
     html_url: HttpUrl
-    description: Optional[str] = None
+    description: str | None = None
     fork: bool
     url: HttpUrl
     created_at: datetime
@@ -51,25 +52,25 @@ class Repository(BaseModel):
     ssh_url: str
     clone_url: HttpUrl
     svn_url: HttpUrl
-    homepage: Optional[str] = None
+    homepage: str | None = None
     size: int
     stargazers_count: int
     watchers_count: int
-    language: Optional[str] = None
+    language: str | None = None
     has_issues: bool
     has_projects: bool
     has_downloads: bool
     has_wiki: bool
     has_pages: bool
     forks_count: int
-    mirror_url: Optional[HttpUrl] = None
+    mirror_url: HttpUrl | None = None
     archived: bool
     disabled: bool
     open_issues_count: int
-    license: Optional[Dict] = None
+    license: dict | None = None
     allow_forking: bool
     is_template: bool
-    topics: List[str]
+    topics: list[str]
     visibility: str
     forks: int
     open_issues: int
@@ -90,23 +91,23 @@ class Issue(BaseModel):
     number: int
     title: str
     user: User
-    labels: List[Dict]
+    labels: list[dict]
     state: str
     locked: bool
-    assignee: Optional[User] = None
-    assignees: List[User]
-    milestone: Optional[Dict] = None
+    assignee: User | None = None
+    assignees: list[User]
+    milestone: dict | None = None
     comments: int
     created_at: datetime
     updated_at: datetime
-    closed_at: Optional[datetime] = None
+    closed_at: datetime | None = None
     author_association: str
-    active_lock_reason: Optional[str] = None
-    body: Optional[str] = None
-    reactions: Optional[Dict] = None
+    active_lock_reason: str | None = None
+    body: str | None = None
+    reactions: dict | None = None
     timeline_url: HttpUrl
-    performed_via_github_app: Optional[Dict] = None
-    state_reason: Optional[str] = None
+    performed_via_github_app: dict | None = None
+    state_reason: str | None = None
 
 
 class PullRequest(BaseModel):
@@ -123,30 +124,30 @@ class PullRequest(BaseModel):
     locked: bool
     title: str
     user: User
-    body: Optional[str] = None
+    body: str | None = None
     created_at: datetime
     updated_at: datetime
-    closed_at: Optional[datetime] = None
-    merged_at: Optional[datetime] = None
-    merge_commit_sha: Optional[str] = None
-    assignee: Optional[User] = None
-    assignees: List[User]
-    requested_reviewers: List[User]
-    requested_teams: List[Dict]
-    labels: List[Dict]
-    milestone: Optional[Dict] = None
+    closed_at: datetime | None = None
+    merged_at: datetime | None = None
+    merge_commit_sha: str | None = None
+    assignee: User | None = None
+    assignees: list[User]
+    requested_reviewers: list[User]
+    requested_teams: list[dict]
+    labels: list[dict]
+    milestone: dict | None = None
     draft: bool
     commits_url: HttpUrl
     review_comments_url: HttpUrl
     review_comment_url: str
     comments_url: HttpUrl
     statuses_url: HttpUrl
-    head: Dict
-    base: Dict
-    _links: Dict
+    head: dict
+    base: dict
+    _links: dict
     author_association: str
-    auto_merge: Optional[Dict] = None
-    active_lock_reason: Optional[str] = None
+    auto_merge: dict | None = None
+    active_lock_reason: str | None = None
 
 
 class Content(BaseModel):
@@ -158,17 +159,17 @@ class Content(BaseModel):
     url: HttpUrl
     html_url: HttpUrl
     git_url: HttpUrl
-    download_url: Optional[HttpUrl] = None
+    download_url: HttpUrl | None = None
     type: str
-    content: Optional[str] = None
-    encoding: Optional[str] = None
-    _links: Dict
+    content: str | None = None
+    encoding: str | None = None
+    _links: dict
 
 
 class Branch(BaseModel):
     model_config = ConfigDict(extra="allow")
     name: str
-    commit: Dict
+    commit: dict
     protected: bool
 
 
@@ -176,10 +177,10 @@ class Commit(BaseModel):
     model_config = ConfigDict(extra="allow")
     sha: str
     node_id: str
-    commit: Dict
+    commit: dict
     url: HttpUrl
     html_url: HttpUrl
     comments_url: HttpUrl
-    author: Optional[User] = None
-    committer: Optional[User] = None
-    parents: List[Dict]
+    author: User | None = None
+    committer: User | None = None
+    parents: list[dict]

github_agent-0.3.0/github_agent/mcp_config.json

@@ -0,0 +1,3 @@
+{
+    "mcpServers": {}
+}