fix(ai-code-review): Add on_new_commit flag

Author: suejung-sentry

src/sentry/features/temporary.py

@@ -78,6 +78,8 @@ def register_temporary_features(manager: FeatureManager) -> None:
     manager.add("organizations:chonk-ui-feedback", OrganizationFeature, FeatureHandlerStrategy.FLAGPOLE, api_expose=True)
     # Enables Codecov UI
     manager.add("organizations:codecov-ui", OrganizationFeature, FeatureHandlerStrategy.FLAGPOLE, api_expose=True)
+    # Enable Prevent AI code review to run per commit
+    manager.add("organizations:code-review-run-per-commit", OrganizationFeature, FeatureHandlerStrategy.FLAGPOLE, api_expose=False)
     # Enables Prevent Test Analytics
     manager.add("organizations:prevent-test-analytics", OrganizationFeature, FeatureHandlerStrategy.FLAGPOLE, api_expose=True)
     # Enable the improved command menu (Cmd+K)

src/sentry/overwatch/endpoints/overwatch_rpc.py

@@ -25,7 +25,7 @@
 from sentry.models.organization import Organization
 from sentry.models.repository import Repository
 from sentry.prevent.models import PreventAIConfiguration
-from sentry.prevent.types.config import PREVENT_AI_CONFIG_DEFAULT
+from sentry.prevent.types.config import PREVENT_AI_CONFIG_DEFAULT, PREVENT_AI_CONFIG_DEFAULT_V1
 from sentry.silo.base import SiloMode
 
 logger = logging.getLogger(__name__)
@@ -154,7 +154,13 @@ def get(self, request: Request) -> Response:
             integration_id=github_org_integrations[0].integration_id,
         ).first()
 
-        response_data: dict[str, Any] = deepcopy(PREVENT_AI_CONFIG_DEFAULT)
+        organization = Organization.objects.filter(id=sentry_org_id).first()
+
+        default_config = PREVENT_AI_CONFIG_DEFAULT
+        if features.has("organizations:code-review-run-per-commit", organization):
+            default_config = PREVENT_AI_CONFIG_DEFAULT_V1
+
+        response_data: dict[str, Any] = deepcopy(default_config)
         if config:
             response_data["organization"] = config.data
 

src/sentry/prevent/endpoints/pr_review_github_config.py

@@ -5,7 +5,7 @@
 from rest_framework.request import Request
 from rest_framework.response import Response
 
-from sentry import audit_log
+from sentry import audit_log, features
 from sentry.api.api_owners import ApiOwner
 from sentry.api.api_publish_status import ApiPublishStatus
 from sentry.api.base import region_silo_endpoint
@@ -15,7 +15,11 @@
 from sentry.integrations.types import IntegrationProviderSlug
 from sentry.models.organization import Organization
 from sentry.prevent.models import PreventAIConfiguration
-from sentry.prevent.types.config import ORG_CONFIG_SCHEMA, PREVENT_AI_CONFIG_DEFAULT
+from sentry.prevent.types.config import (
+    ORG_CONFIG_SCHEMA,
+    PREVENT_AI_CONFIG_DEFAULT,
+    PREVENT_AI_CONFIG_DEFAULT_V1,
+)
 
 
 class PreventAIConfigPermission(OrganizationPermission):
@@ -59,7 +63,11 @@ def get(
             integration_id=github_org_integrations[0].integration_id,
         ).first()
 
-        response_data: dict[str, Any] = deepcopy(PREVENT_AI_CONFIG_DEFAULT)
+        default_config = PREVENT_AI_CONFIG_DEFAULT
+        if features.has("organizations:code-review-run-per-commit", organization):
+            default_config = PREVENT_AI_CONFIG_DEFAULT_V1
+
+        response_data: dict[str, Any] = deepcopy(default_config)
         if config:
             response_data["organization"] = config.data
 

src/sentry/prevent/types/config.py

@@ -125,3 +125,40 @@
     },
     "organization": {},
 }
+
+PREVENT_AI_CONFIG_DEFAULT_V1 = {
+    "schema_version": "v1",
+    "default_org_config": {
+        "org_defaults": {
+            "bug_prediction": {
+                "enabled": True,
+                "sensitivity": "medium",
+                "triggers": {
+                    "on_command_phrase": True,
+                    "on_ready_for_review": True,
+                    # v1 default enables on_new_commit
+                    "on_new_commit": True,
+                },
+            },
+            "test_generation": {
+                "enabled": True,
+                "triggers": {
+                    "on_command_phrase": True,
+                    "on_ready_for_review": False,
+                    "on_new_commit": False,
+                },
+            },
+            "vanilla": {
+                "enabled": True,
+                "sensitivity": "medium",
+                "triggers": {
+                    "on_command_phrase": True,
+                    "on_ready_for_review": False,
+                    "on_new_commit": False,
+                },
+            },
+        },
+        "repo_overrides": {},
+    },
+    "organization": {},
+}