mi222eh
69 lines
2.5 KiB
Python
69 lines
2.5 KiB
Python
import logging
|
|
from typing import Any
|
|
|
|
logger: logging.Logger = logging.getLogger("notification-tools")
|
|
|
|
|
|
class NotificationTools:
|
|
"""Tools exposed to the Notification Reader Agent for routing decisions."""
|
|
|
|
def __init__(self) -> None:
|
|
self.tool_called: bool = False
|
|
self.action: str = "NO_ACTION"
|
|
self.arguments: dict[str, Any] = {}
|
|
|
|
def process_issue(self, owner: str, repo: str, issue_number: int, reason: str) -> str:
|
|
"""Process the notification as an issue.
|
|
Use this when a notification refers to an issue that requires agent action.
|
|
|
|
Args:
|
|
owner: The repository owner (organization).
|
|
repo: The repository name.
|
|
issue_number: The Gitea issue number.
|
|
reason: Why this notification should be processed.
|
|
"""
|
|
logger.info(f"Notification tool 'process_issue' called for {owner}/{repo}#{issue_number}: {reason}")
|
|
self.tool_called = True
|
|
self.action = "PROCESS_ISSUE"
|
|
self.arguments = {
|
|
"owner": owner,
|
|
"repo": repo,
|
|
"issue_number": issue_number,
|
|
"reason": reason,
|
|
}
|
|
return "Issue notification marked for processing."
|
|
|
|
def process_pr(self, owner: str, repo: str, pr_number: int, reason: str) -> str:
|
|
"""Process the notification as a pull request.
|
|
Use this when a notification refers to a PR that requires agent action.
|
|
|
|
Args:
|
|
owner: The repository owner (organization).
|
|
repo: The repository name.
|
|
pr_number: The Gitea PR number.
|
|
reason: Why this notification should be processed.
|
|
"""
|
|
logger.info(f"Notification tool 'process_pr' called for {owner}/{repo}#{pr_number}: {reason}")
|
|
self.tool_called = True
|
|
self.action = "PROCESS_PR"
|
|
self.arguments = {
|
|
"owner": owner,
|
|
"repo": repo,
|
|
"pr_number": pr_number,
|
|
"reason": reason,
|
|
}
|
|
return "PR notification marked for processing."
|
|
|
|
def skip_notification(self, reason: str) -> str:
|
|
"""Skip the notification and take no action.
|
|
Use this if the notification is irrelevant, already handled, or does not require agent intervention.
|
|
|
|
Args:
|
|
reason: The reason for skipping this notification.
|
|
"""
|
|
logger.info(f"Notification tool 'skip_notification' called: {reason}")
|
|
self.tool_called = True
|
|
self.action = "SKIP"
|
|
self.arguments = {"reason": reason}
|
|
return "Notification marked to be skipped."
|