chore: pr review commenting tooling (#449)

# Motivation

<!-- Why is this change necessary? -->

# Content

<!-- Please include a summary of the change -->

# Testing

<!-- How was the change tested? -->

# Please check the following before marking your PR as ready for review

- [ ] I have added tests for my changes
- [ ] I have updated the documentation or added new documentation as
needed

---------

Co-authored-by: kopekC <[email protected]>

Author: kopekC

src/codegen/extensions/langchain/tools.py

@@ -359,6 +359,100 @@ def _run(self, title: str, body: str) -> str:
         return pr.html_url
 
 
+class GetPRContentsInput(BaseModel):
+    """Input for getting PR contents."""
+
+    pr_id: int = Field(..., description="Number of the PR to get the contents for")
+
+
+class GetPRcontentsTool(BaseTool):
+    """Tool for getting PR data."""
+
+    name: ClassVar[str] = "get_pr_contents"
+    description: ClassVar[str] = "Get the diff and modified symbols of a PR along with the dependencies of the modified symbols"
+    args_schema: ClassVar[type[BaseModel]] = GetPRContentsInput
+    codebase: Codebase = Field(exclude=True)
+
+    def __init__(self, codebase: Codebase) -> None:
+        super().__init__(codebase=codebase)
+
+    def _run(self, pr_id: int) -> str:
+        modified_symbols, patch = self.codebase.get_modified_symbols_in_pr(pr_id)
+
+        # Convert modified_symbols set to list for JSON serialization
+        result = {"modified_symbols": list(modified_symbols), "patch": patch}
+
+        return json.dumps(result, indent=2)
+
+
+class CreatePRCommentInput(BaseModel):
+    """Input for creating a PR comment"""
+
+    pr_number: int = Field(..., description="The PR number to comment on")
+    body: str = Field(..., description="The comment text")
+
+
+class CreatePRCommentTool(BaseTool):
+    """Tool for creating a general PR comment."""
+
+    name: ClassVar[str] = "create_pr_comment"
+    description: ClassVar[str] = "Create a general comment on a pull request"
+    args_schema: ClassVar[type[BaseModel]] = CreatePRCommentInput
+    codebase: Codebase = Field(exclude=True)
+
+    def __init__(self, codebase: Codebase) -> None:
+        super().__init__(codebase=codebase)
+
+    def _run(self, pr_number: int, body: str) -> str:
+        self.codebase.create_pr_comment(pr_number=pr_number, body=body)
+        return "Comment created successfully"
+
+
+class CreatePRReviewCommentInput(BaseModel):
+    """Input for creating an inline PR review comment"""
+
+    pr_number: int = Field(..., description="The PR number to comment on")
+    body: str = Field(..., description="The comment text")
+    commit_sha: str = Field(..., description="The commit SHA to attach the comment to")
+    path: str = Field(..., description="The file path to comment on")
+    line: int | None = Field(None, description="The line number to comment on")
+    side: str | None = Field(None, description="Which version of the file to comment on ('LEFT' or 'RIGHT')")
+    start_line: int | None = Field(None, description="For multi-line comments, the starting line")
+
+
+class CreatePRReviewCommentTool(BaseTool):
+    """Tool for creating inline PR review comments."""
+
+    name: ClassVar[str] = "create_pr_review_comment"
+    description: ClassVar[str] = "Create an inline review comment on a specific line in a pull request"
+    args_schema: ClassVar[type[BaseModel]] = CreatePRReviewCommentInput
+    codebase: Codebase = Field(exclude=True)
+
+    def __init__(self, codebase: Codebase) -> None:
+        super().__init__(codebase=codebase)
+
+    def _run(
+        self,
+        pr_number: int,
+        body: str,
+        commit_sha: str,
+        path: str,
+        line: int | None = None,
+        side: str | None = None,
+        start_line: int | None = None,
+    ) -> str:
+        self.codebase.create_pr_review_comment(
+            pr_number=pr_number,
+            body=body,
+            commit_sha=commit_sha,
+            path=path,
+            line=line,
+            side=side,
+            start_line=start_line,
+        )
+        return "Review comment created successfully"
+
+
 def get_workspace_tools(codebase: Codebase) -> list["BaseTool"]:
     """Get all workspace tools initialized with a codebase.
 
@@ -372,8 +466,11 @@ def get_workspace_tools(codebase: Codebase) -> list["BaseTool"]:
         CommitTool(codebase),
         CreateFileTool(codebase),
         CreatePRTool(codebase),
+        CreatePRCommentTool(codebase),
+        CreatePRReviewCommentTool(codebase),
         DeleteFileTool(codebase),
         EditFileTool(codebase),
+        GetPRcontentsTool(codebase),
         ListDirectoryTool(codebase),
         RevealSymbolTool(codebase),
         SearchTool(codebase),

src/codegen/git/repo_operator/repo_operator.py

@@ -608,3 +608,49 @@ def stash_pop(self) -> None:
     def get_pr_data(self, pr_number: int) -> dict:
         """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:
+        """Create a general comment on a pull request.
+
+        Args:
+            pr_number (int): The PR number to comment on
+            body (str): The comment text
+        """
+        pr = self.remote_git_repo.get_pull_safe(pr_number)
+        if pr:
+            self.remote_git_repo.create_issue_comment(pr, body)
+
+    def create_pr_review_comment(
+        self,
+        pr_number: int,
+        body: str,
+        commit_sha: str,
+        path: str,
+        line: int | None = None,
+        side: str | None = None,
+        start_line: int | None = None,
+    ) -> None:
+        """Create an inline review comment on a specific line in a pull request.
+
+        Args:
+            pr_number (int): The PR number to comment on
+            body (str): The comment text
+            commit_sha (str): The commit SHA to attach the comment to
+            path (str): The file path to comment on
+            line (int | None, optional): The line number to comment on. Defaults to None.
+            side (str | None, optional): Which version of the file to comment on ('LEFT' or 'RIGHT'). Defaults to None.
+            start_line (int | None, optional): For multi-line comments, the starting line. Defaults to None.
+        """
+        pr = self.remote_git_repo.get_pull_safe(pr_number)
+        if pr:
+            commit = self.remote_git_repo.get_commit_safe(commit_sha)
+            if commit:
+                self.remote_git_repo.create_review_comment(
+                    pull=pr,
+                    body=body,
+                    commit=commit,
+                    path=path,
+                    line=line,
+                    side=side,
+                    start_line=start_line,
+                )