Skip to content

Commit f7bee3c

Browse files
authored
fix: return branch name with pr changes
# 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
1 parent a93c961 commit f7bee3c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/codegen/sdk/core/codebase.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1527,13 +1527,13 @@ def from_files(
15271527
logger.info("Codebase initialization complete")
15281528
return codebase
15291529

1530-
def get_modified_symbols_in_pr(self, pr_id: int) -> tuple[str, dict[str, str], list[str]]:
1530+
def get_modified_symbols_in_pr(self, pr_id: int) -> tuple[str, dict[str, str], list[str], str]:
15311531
"""Get all modified symbols in a pull request"""
15321532
pr = self._op.get_pull_request(pr_id)
15331533
cg_pr = CodegenPR(self._op, self, pr)
15341534
patch = cg_pr.get_pr_diff()
15351535
commit_sha = cg_pr.get_file_commit_shas()
1536-
return patch, commit_sha, cg_pr.modified_symbols
1536+
return patch, commit_sha, cg_pr.modified_symbols, pr.head.ref
15371537

15381538
def create_pr_comment(self, pr_number: int, body: str) -> None:
15391539
"""Create a comment on a pull request"""

0 commit comments

Comments
 (0)