chore: returns issue comment

src/codegen/git/clients/git_repo_client.py

@@ -8,6 +8,7 @@
 from github.Commit import Commit
 from github.GithubException import GithubException, UnknownObjectException
 from github.GithubObject import NotSet, Opt
+from github.IssueComment import IssueComment
 from github.Label import Label
 from github.PullRequest import PullRequest
 from github.Repository import Repository
@@ -37,7 +38,7 @@
         return GithubClient(token=token)
 
     def _create_client(self) -> Repository:
         client = self.gh_client.get_repo_by_full_name(self.repo_config.full_name)
         if not client:
             msg = f"Repo {self.repo_config.full_name} not found!"
             raise ValueError(msg)
@@ -126,10 +127,10 @@
         self,
         pull: PullRequest,
         body: str,
-    ) -> None:
+    ) -> IssueComment:
         # TODO: add protections (ex: can write to PR)
         writeable_pr = self.repo.get_pull(pull.number)
-        writeable_pr.create_issue_comment(body=body)
+        return writeable_pr.create_issue_comment(body=body)
 
     ####################################################################################################################
     # PULL REQUESTS
@@ -173,9 +174,9 @@
     def get_or_create_pull(
         self,
         head_branch_name: str,
         base_branch_name: str | None = None,  # type: ignore[assignment]
         title: str | None = None,  # type: ignore[assignment]
         body: str | None = None,  # type: ignore[assignment]
     ) -> PullRequest | None:
         pull = self.get_pull_by_branch_and_state(head_branch_name=head_branch_name, base_branch_name=base_branch_name)
         if pull:
@@ -220,7 +221,7 @@
             body="",
         )
         # TODO: handle PR not mergeable due to merge conflicts
         merge = squash_pr.merge(commit_message=squash_commit_msg, commit_title=squash_commit_title, merge_method="squash")  # type: ignore[arg-type]
 
     def edit_pull(self, pull: PullRequest, title: Opt[str] = NotSet, body: Opt[str] = NotSet, state: Opt[str] = NotSet) -> None:
         writable_pr = self.repo.get_pull(pull.number)

src/codegen/git/repo_operator/repo_operator.py

@@ -14,6 +14,7 @@
 from git import Diff, GitCommandError, InvalidGitRepositoryError, Remote
 from git import Repo as GitCLI
 from git.remote import PushInfoList
+from github.IssueComment import IssueComment
 from github.PullRequest import PullRequest
 
 from codegen.configs.models.secrets import DefaultSecrets
@@ -142,7 +143,7 @@
         email_level = None
         levels = ["system", "global", "user", "repository"]
         for level in levels:
             with git_cli.config_reader(level) as reader:
                 if reader.has_option("user", "name") and not username:
                     username = reader.get("user", "name")
                     user_level = level
@@ -554,7 +555,7 @@
                     return content
             except UnicodeDecodeError:
                 print(f"Warning: Unable to decode file {file_path}. Skipping.")
                 return None
 
     def write_file(self, relpath: str, content: str) -> None:
         """Writes file content to disk"""
@@ -621,7 +622,7 @@
         filepaths = self.get_filepaths_for_repo(ignore_list)
         # Iterate through files and yield contents
         for rel_filepath in filepaths:
             rel_filepath: str
             filepath = os.path.join(self.repo_path, rel_filepath)
 
             # Filter by subdirectory (includes full filenames)
@@ -652,7 +653,7 @@
         list_files = []
 
         for rel_filepath in self.git_cli.git.ls_files().split("\n"):
             rel_filepath: str
             if subdirs and not any(d in rel_filepath for d in subdirs):
                 continue
             if extensions is None or any(rel_filepath.endswith(e) for e in extensions):
@@ -676,7 +677,7 @@
 
     def get_modified_files_in_last_n_days(self, days: int = 1) -> tuple[list[str], list[str]]:
         """Returns a list of files modified and deleted in the last n days"""
         modified_files = []
         deleted_files = []
         allowed_extensions = [".py"]
 
@@ -717,7 +718,7 @@
         """Returns the data associated with a PR"""
         return self.remote_git_repo.get_pr_data(pr_number)
 
-    def create_pr_comment(self, pr_number: int, body: str) -> None:
+    def create_pr_comment(self, pr_number: int, body: str) -> IssueComment:
         """Create a general comment on a pull request.
 
         Args:
@@ -726,7 +727,8 @@
         """
         pr = self.remote_git_repo.get_pull_safe(pr_number)
         if pr:
-            self.remote_git_repo.create_issue_comment(pr, body)
+            comment = self.remote_git_repo.create_issue_comment(pr, body)
+            return comment
 
     def create_pr_review_comment(
         self,