ref(seer-billing): Drop the agentic triage referrer exemption

Credit checks no longer skip runs by referrer. Agentic triage runs are checked like any other until a follow-up marks them by feature instead. Credit checks are behind seer-credit-checks, so nothing changes in production.

Refs CW-3061

Author: srest2021

src/sentry/seer/autofix/autofix_agent.py

@@ -832,15 +832,7 @@ def trigger_push_changes(
     ):
         raise PermissionDenied("Code generation is disabled for this organization")
 
-    seer_run = SeerRun.objects.filter(
-        organization_id=group.organization.id, seer_run_state_id=run_id
-    ).first()
-    check_autofix_billing(
-        group.organization,
-        SeerOperation.AUTOFIX_OPEN_PR,
-        seer_run.referrer if seer_run else None,
-        is_new_run=False,
-    )
+    check_autofix_billing(group.organization, SeerOperation.AUTOFIX_OPEN_PR, is_new_run=False)
 
     client = get_autofix_agent_client(group, user=user)
 
@@ -859,6 +851,10 @@ def trigger_push_changes(
         )
     )
 
+    seer_run = SeerRun.objects.filter(
+        organization_id=group.organization.id, seer_run_state_id=run_id
+    ).first()
+
     # Draft when review-request/CI-green is enabled; mark ready once green fires.
     client.push_changes(
         run_id,

src/sentry/seer/autofix/feature/dispatch.py

@@ -153,7 +153,6 @@ def _org_allows_coding(group: Group) -> bool:
 def check_autofix_billing(
     organization: Organization,
     operation: SeerOperation,
-    referrer: str | None,
     *,
     is_new_run: bool,
     skip_quota_for_free_cohort: bool = False,
@@ -165,11 +164,9 @@ def check_autofix_billing(
     """
     # Check credits for token-billed orgs.
     if features.has("organizations:token-billed-seer-enabled", organization):
-        if (
-            features.has("organizations:seer-credit-checks", organization)
-            and referrer != AutofixReferrer.AGENTIC_TRIAGE  # Agentic triage runs stay free.
-            and not has_enough_seer_credits(organization, operation)
-        ):
+        if features.has(
+            "organizations:seer-credit-checks", organization
+        ) and not has_enough_seer_credits(organization, operation):
             raise SeerOutOfCreditsException()
         return False
 
@@ -200,7 +197,6 @@ def continue_single_run(
     check_autofix_billing(
         group.organization,
         SeerOperation.AUTOFIX_SINGLE_RUN_REPLY,
-        agent_run.run.referrer,
         is_new_run=False,
     )
 
@@ -258,7 +254,6 @@ def trigger_autofix_feature(
         uses_quota = check_autofix_billing(
             group.organization,
             args.step.seer_operation,
-            args.referrer,
             is_new_run=is_new_run,
             # Free cohort orgs bypass quota only when called from agentic triage
             # (allow_free_cohort=True). Not exposed via the API.

src/sentry/seer/endpoints/group_ai_autofix.py

@@ -165,16 +165,10 @@ def _parse_autofix_referrer(raw: str | None, request: Request) -> AutofixReferre
             return AutofixReferrer.MCP
         return AutofixReferrer.GROUP_AUTOFIX_ENDPOINT
     try:
-        referrer = AutofixReferrer(raw)
+        return AutofixReferrer(raw)
     except ValueError:
         logger.warning("group_ai_autofix.unknown_referrer", extra={"referrer": raw})
         return AutofixReferrer.UNKNOWN
-    # Agentic triage is not exposed via the API. Don't trust such a referrer since
-    # it may have token-based billing implications.
-    if referrer == AutofixReferrer.AGENTIC_TRIAGE:
-        logger.warning("group_ai_autofix.internal_referrer", extra={"referrer": raw})
-        return AutofixReferrer.UNKNOWN
-    return referrer
 
 
 def _record_user_context_length(request_data: Any) -> None:

tests/sentry/seer/autofix/feature/test_dispatch.py

@@ -387,32 +387,6 @@ def test_raises_when_out_of_credits_for_a_continued_step(
 
         MockClient.return_value.continue_feature_run.assert_not_called()
 
-    @with_feature(["organizations:seer-credit-checks", "organizations:token-billed-seer-enabled"])
-    @patch(
-        "sentry.seer_credit.backend.check_seer_credits",
-        return_value=CheckSeerCreditsResponse(has_enough_credits=False),
-    )
-    def test_agentic_triage_skips_credit_check(self, mock_check_seer_credits: MagicMock) -> None:
-        fake_run = self.create_seer_run(organization=self.organization, type="feature_run")
-
-        with (
-            patch("sentry.seer.autofix.feature.dispatch.SeerAgentClient") as MockClient,
-            patch("sentry.seer.autofix.feature.dispatch.quotas"),
-        ):
-            MockClient.return_value.start_feature_run.return_value = fake_run
-
-            run = trigger_autofix_feature(
-                self.group,
-                AutofixFeatureArgs(
-                    referrer=AutofixReferrer.AGENTIC_TRIAGE,
-                    step=AutofixStep.ROOT_CAUSE,
-                    step_args=RCAStepArgs(),
-                ),
-            )
-
-        assert run is fake_run
-        mock_check_seer_credits.assert_not_called()
-
     def test_allows_async_dispatch(self) -> None:
         fake_run = self.create_seer_run(organization=self.organization, type="feature_run")
 
@@ -763,18 +737,3 @@ def test_raises_when_out_of_credits(self, mock_check_seer_credits: MagicMock) ->
             )
 
         mock_client_cls.return_value.continue_feature_run.assert_not_called()
-
-    @with_feature(["organizations:seer-credit-checks", "organizations:token-billed-seer-enabled"])
-    @patch(
-        "sentry.seer_credit.backend.check_seer_credits",
-        return_value=CheckSeerCreditsResponse(has_enough_credits=False),
-    )
-    def test_agentic_triage_run_skips_credit_check(
-        self, mock_check_seer_credits: MagicMock
-    ) -> None:
-        """The run's own referrer decides, not the reply's."""
-        self.seer_run.update(referrer=AutofixReferrer.AGENTIC_TRIAGE)
-
-        self._reply()
-
-        mock_check_seer_credits.assert_not_called()

tests/sentry/seer/autofix/test_autofix_agent.py

@@ -1685,32 +1685,6 @@ def test_raises_when_out_of_credits(
 
         mock_post.assert_not_called()
 
-    @patch(
-        "sentry.seer_credit.backend.check_seer_credits",
-        return_value=CheckSeerCreditsResponse(has_enough_credits=False),
-    )
-    @patch("sentry.seer.agent.client.make_agent_update_request")
-    def test_agentic_triage_run_skips_credit_check(
-        self, mock_post: MagicMock, mock_check_seer_credits: MagicMock
-    ) -> None:
-        # The push is called with another referrer; the run's own referrer decides.
-        self.create_seer_run(
-            organization=self.organization,
-            seer_run_state_id=123,
-            referrer=AutofixReferrer.AGENTIC_TRIAGE.value,
-        )
-
-        self._push(
-            mock_post,
-            features={
-                "organizations:seer-credit-checks": True,
-                "organizations:token-billed-seer-enabled": True,
-            },
-        )
-
-        mock_check_seer_credits.assert_not_called()
-        mock_post.assert_called_once()
-
     @patch("sentry.seer.agent.client.make_agent_update_request")
     def test_passes_pr_description_suffix_ready_state_and_author(
         self, mock_post: MagicMock

tests/sentry/seer/endpoints/test_group_ai_autofix.py

@@ -1198,25 +1198,6 @@ def test_post_seer_explorer_referrer_is_recognized(
         assert response.status_code == 202, response.data
         assert mock_trigger_explorer.call_args.kwargs["referrer"] == AutofixReferrer.SEER_EXPLORER
 
-    @patch("sentry.seer.endpoints.group_ai_autofix.trigger_autofix_agent")
-    def test_post_agentic_triage_referrer_is_unknown(
-        self, mock_trigger_explorer: MagicMock
-    ) -> None:
-        """Agentic triage runs skip the credit check, so the API can't claim to be one."""
-        group = self.create_group()
-        run = self.create_seer_run(organization=self.organization, seer_run_state_id=123)
-        mock_trigger_explorer.return_value = run
-
-        self.login_as(user=self.user)
-        response = self.client.post(
-            self._get_url(group.id),
-            data={"step": "root_cause", "referrer": AutofixReferrer.AGENTIC_TRIAGE.value},
-            format="json",
-        )
-
-        assert response.status_code == 202, response.data
-        assert mock_trigger_explorer.call_args.kwargs["referrer"] == AutofixReferrer.UNKNOWN
-
     @patch("sentry.seer.endpoints.group_ai_autofix.trigger_autofix_agent")
     def test_stopping_point(self, mock_trigger_explorer: MagicMock) -> None:
         """Stopping point forces the step to be root_cause"""