agent.py

"""
agent.py

Entry point for the GitHub Triage Agent.
This agent reads issues and performs triage actions such as:
- fetching issue details
- searching issues
- listing labels
- posting comments
"""

from tools import (
    get_issue,
    search_issues,
    list_labels,
    list_contributors,
    post_comment,
    list_open_issues
)

from memory import is_triaged, mark_triaged, log_action
from datetime import datetime, timezone
from llm import suggest_label
import os

def triage_issue(issue_number: int):
    """
    Main triage workflow for a single issue.
    """

    print(f"Fetching issue #{issue_number}...\n")

    if is_triaged(issue_number):
        print(f"Issue #{issue_number} has already been triaged. Skipping.")
        return

    issue = get_issue(issue_number)

    age = datetime.now(timezone.utc) - issue["created_at"]

    if age.days > 0:
        print(f"Issue #{issue_number} is old. Skipping comment.")
        mark_triaged(issue_number)
        return

    print("Issue Details")
    print("---------------------")
    print("Number:", issue["number"])
    print("Title:", issue["title"])
    print("Author:", issue["author"])
    print("State:", issue["state"])
    print("Labels:", issue["labels"])
    print()

    # Example logic
    if issue["state"] == "open":
        print("Issue is open — performing triage...\n")

        labels = list_labels()

        print("Available Labels:")
        print(labels)
        print()

        contributors = list_contributors()

        print("Repo Contributors:")
        print(contributors)
        print()

        label = suggest_label(issue)
        
        # Example automated comment
        comment_body = (
            "👋 Thanks for opening this issue!\n\n"
            f"Suggested label: **{label}**\n\n"
            "Our team will review it shortly. Sharpiru Sharpallaina"
        )

        result = post_comment(issue_number, comment_body)

        print(result)

        mark_triaged(issue_number)

        log_action(issue_number, "comment_posted")

    else:
        print("Issue is already closed.")


def search_workflow(query: str):
    """
    Example workflow for searching issues.
    """

    print(f"Searching issues for: {query}\n")

    results = search_issues(query)

    for issue in results:
        print(
            f"#{issue['number']} - "
            f"{issue['title']} "
            f"({issue['state']})"
        )


if __name__ == "__main__":
    """
    Entry point when script is run directly.
    """

    print("GitHub Triage Agent Started\n")

    issue_number = os.getenv("ISSUE_NUMBER")
        
    if issue_number:
        triage_issue(int(issue_number))
    else:
        # fallback: triage all open issues
        for issue in list_open_issues():
            triage_issue(issue["number"])

    # Optional:
    # search_workflow("bug")