|
108 | 108 | from sentry.models.releaseprojectenvironment import ReleaseProjectEnvironment
|
109 | 109 | from sentry.models.releases.release_project import ReleaseProject
|
110 | 110 | from sentry.net.http import connection_from_url
|
| 111 | +from sentry.options.rollout import in_random_rollout |
111 | 112 | from sentry.performance_issues.performance_detection import detect_performance_problems
|
112 | 113 | from sentry.performance_issues.performance_problem import PerformanceProblem
|
113 | 114 | from sentry.plugins.base import plugins
|
@@ -1947,6 +1948,12 @@ def _process_existing_aggregate(
|
1947 | 1948 | timeout=settings.SEER_SEVERITY_TIMEOUT, # Defaults to 300 milliseconds
|
1948 | 1949 | )
|
1949 | 1950 |
|
| 1951 | +severity_connection_pool_gpu = connection_from_url( |
| 1952 | + settings.SEER_GROUPING_URL, |
| 1953 | + retries=settings.SEER_SEVERITY_RETRIES, |
| 1954 | + timeout=settings.SEER_SEVERITY_TIMEOUT, # Defaults to 300 milliseconds |
| 1955 | +) |
| 1956 | + |
1950 | 1957 |
|
1951 | 1958 | def _get_severity_metadata_for_group(
|
1952 | 1959 | event: Event, project_id: int, group_type: int | None
|
@@ -2168,8 +2175,14 @@ def _get_severity_score(event: Event) -> tuple[float, str]:
|
2168 | 2175 | "issues.severity.seer-timout",
|
2169 | 2176 | settings.SEER_SEVERITY_TIMEOUT / 1000,
|
2170 | 2177 | )
|
| 2178 | + |
| 2179 | + if in_random_rollout("issues.severity.gpu-rollout-rate"): |
| 2180 | + connection_pool = severity_connection_pool_gpu |
| 2181 | + else: |
| 2182 | + connection_pool = severity_connection_pool |
| 2183 | + |
2171 | 2184 | response = make_signed_seer_api_request(
|
2172 |
| - severity_connection_pool, |
| 2185 | + connection_pool, |
2173 | 2186 | "/v0/issues/severity-score",
|
2174 | 2187 | body=orjson.dumps(payload),
|
2175 | 2188 | timeout=timeout,
|
|
0 commit comments