From 89abbf66df1bd21fdfc1cfe6dc553b078d5b1964 Mon Sep 17 00:00:00 2001 From: KopekC Date: Thu, 20 Mar 2025 13:20:45 -0400 Subject: [PATCH 1/2] feat: return head branch name with pr changes --- src/codegen/sdk/core/codebase.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/codegen/sdk/core/codebase.py b/src/codegen/sdk/core/codebase.py index 6b4c103aa..6ac32136e 100644 --- a/src/codegen/sdk/core/codebase.py +++ b/src/codegen/sdk/core/codebase.py @@ -1533,7 +1533,7 @@ def get_modified_symbols_in_pr(self, pr_id: int) -> tuple[str, dict[str, str], l cg_pr = CodegenPR(self._op, self, pr) patch = cg_pr.get_pr_diff() commit_sha = cg_pr.get_file_commit_shas() - return patch, commit_sha, cg_pr.modified_symbols + return patch, commit_sha, cg_pr.modified_symbols, pr.head.ref def create_pr_comment(self, pr_number: int, body: str) -> None: """Create a comment on a pull request""" From 17126123f9b54c93b88835157bf4cbeb4418433c Mon Sep 17 00:00:00 2001 From: KopekC Date: Thu, 20 Mar 2025 13:21:53 -0400 Subject: [PATCH 2/2] . --- src/codegen/sdk/core/codebase.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/codegen/sdk/core/codebase.py b/src/codegen/sdk/core/codebase.py index 6ac32136e..01950a7a2 100644 --- a/src/codegen/sdk/core/codebase.py +++ b/src/codegen/sdk/core/codebase.py @@ -1527,7 +1527,7 @@ def from_files( logger.info("Codebase initialization complete") return codebase - def get_modified_symbols_in_pr(self, pr_id: int) -> tuple[str, dict[str, str], list[str]]: + def get_modified_symbols_in_pr(self, pr_id: int) -> tuple[str, dict[str, str], list[str], str]: """Get all modified symbols in a pull request""" pr = self._op.get_pull_request(pr_id) cg_pr = CodegenPR(self._op, self, pr)