Skip to content

Commit 12fa100

Browse files
authored
chore(ai assisted queries): Update permissions (#90803)
- Update permissions to allow `org:read` to hit the endpoint
1 parent 593f280 commit 12fa100

File tree

2 files changed

+12
-0
lines changed

2 files changed

+12
-0
lines changed

src/sentry/api/endpoints/trace_explorer_ai_query.py

+3
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
from sentry.api.api_publish_status import ApiPublishStatus
1515
from sentry.api.base import region_silo_endpoint
1616
from sentry.api.bases import OrganizationEndpoint
17+
from sentry.api.endpoints.trace_explorer_ai_setup import OrganizationTraceExplorerAIPermission
1718
from sentry.models.organization import Organization
1819
from sentry.seer.seer_setup import get_seer_org_acknowledgement, get_seer_user_acknowledgement
1920
from sentry.seer.signed_seer_api import sign_with_seer_secret
@@ -58,6 +59,8 @@ class TraceExplorerAIQuery(OrganizationEndpoint):
5859
}
5960
owner = ApiOwner.ML_AI
6061

62+
permission_classes = (OrganizationTraceExplorerAIPermission,)
63+
6164
@staticmethod
6265
def post(request: Request, organization: Organization) -> Response:
6366
"""

src/sentry/api/endpoints/trace_explorer_ai_setup.py

+9
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
from sentry.api.api_publish_status import ApiPublishStatus
1414
from sentry.api.base import region_silo_endpoint
1515
from sentry.api.bases import OrganizationEndpoint
16+
from sentry.api.bases.organization import OrganizationPermission
1617
from sentry.models.organization import Organization
1718
from sentry.seer.seer_setup import get_seer_org_acknowledgement, get_seer_user_acknowledgement
1819
from sentry.seer.signed_seer_api import sign_with_seer_secret
@@ -22,6 +23,12 @@
2223
from rest_framework.request import Request
2324

2425

26+
class OrganizationTraceExplorerAIPermission(OrganizationPermission):
27+
scope_map = {
28+
"POST": ["org:read"],
29+
}
30+
31+
2532
def fire_setup_request(org_id: int, project_ids: list[int]) -> None:
2633
"""
2734
Sends a request to seer to create the initial cached prompt / setup the AI models
@@ -55,6 +62,8 @@ class TraceExplorerAISetup(OrganizationEndpoint):
5562
}
5663
owner = ApiOwner.ML_AI
5764

65+
permission_classes = (OrganizationTraceExplorerAIPermission,)
66+
5867
@staticmethod
5968
def post(request: Request, organization: Organization) -> Response:
6069
"""

0 commit comments

Comments
 (0)