feat: Updated 2 files

sweepai · Apr 8, 2024 · a5da0c3 · a5da0c3
1 parent 0277fad
commit a5da0c3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/sweepai/config/server.py b/sweepai/config/server.py
@@ -246,3 +246,5 @@
 JIRA_USER_NAME = os.environ.get("JIRA_USER_NAME", None)
 JIRA_API_TOKEN = os.environ.get("JIRA_API_TOKEN", None)
 JIRA_URL = os.environ.get("JIRA_URL", None)
+
+MERGE_CONFLICT_RESOLUTION_STRATEGY = os.environ.get("MERGE_CONFLICT_RESOLUTION_STRATEGY", "merge")
diff --git a/sweepai/handlers/on_merge_conflict.py b/sweepai/handlers/on_merge_conflict.py
@@ -5,7 +5,7 @@
 from github.PullRequest import PullRequest
 from loguru import logger
 
-from sweepai.config.server import PROGRESS_BASE_URL
+from sweepai.config.server import MERGE_CONFLICT_RESOLUTION_STRATEGY, PROGRESS_BASE_URL
 from sweepai.core import entities
 from sweepai.core.entities import FileChangeRequest
 from sweepai.core.sweep_bot import SweepBot
@@ -186,7 +186,10 @@ def edit_comment(body):
             git_repo.config_writer().set_value(
                 "user", "email", "team@sweep.dev"
             ).release()
-            git_repo.git.merge("origin/" + pr.base.ref)
+            if MERGE_CONFLICT_RESOLUTION_STRATEGY == "rebase":
+                git_repo.git.rebase("origin/" + pr.base.ref)
+            else:
+                git_repo.git.merge("origin/" + pr.base.ref)
         except GitCommandError:
             # Assume there are merge conflicts
             pass