Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 13 additions & 4 deletions src/sentry/seer/explorer/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,7 @@
)
from sentry.seer.models import SeerApiError, SeerPermissionError
from sentry.seer.seer_setup import has_seer_access_with_detail
from sentry.seer.signed_seer_api import SeerViewerContext
from sentry.users.models.user import User

logger = logging.getLogger(__name__)
Expand Down Expand Up @@ -207,6 +208,8 @@ def __init__(

self.enable_coding = enable_coding

self.viewer_context = self._build_viewer_context()

# Validate that category_key and category_value are provided together
if category_key == "" or category_value == "":
raise ValueError("category_key and category_value cannot be empty strings")
Expand All @@ -218,6 +221,12 @@ def __init__(
if not has_access:
raise SeerPermissionError(error or "Access denied")

def _build_viewer_context(self) -> SeerViewerContext:
context = SeerViewerContext(organization_id=self.organization.id)
if self.user and hasattr(self.user, "id") and self.user.id is not None:
context["user_id"] = self.user.id
return context

def start_run(
self,
prompt: str,
Expand Down Expand Up @@ -298,7 +307,7 @@ def start_run(
):
chat_body["is_context_engine_enabled"] = True

response = make_explorer_chat_request(chat_body)
response = make_explorer_chat_request(chat_body, viewer_context=self.viewer_context)

if response.status >= 400:
raise SeerApiError("Seer request failed", response.status)
Expand Down Expand Up @@ -361,7 +370,7 @@ def continue_run(
):
chat_body["is_context_engine_enabled"] = True

response = make_explorer_chat_request(chat_body)
response = make_explorer_chat_request(chat_body, viewer_context=self.viewer_context)

if response.status >= 400:
raise SeerApiError("Seer request failed", response.status)
Expand Down Expand Up @@ -485,7 +494,7 @@ def get_runs(
if end is not None:
runs_body["end"] = end

response = make_explorer_runs_request(runs_body)
response = make_explorer_runs_request(runs_body, viewer_context=self.viewer_context)

if response.status >= 400:
raise SeerApiError("Seer request failed", response.status)
Expand Down Expand Up @@ -533,7 +542,7 @@ def push_changes(
organization_id=self.organization.id,
payload=payload,
)
response = make_explorer_update_request(update_body)
response = make_explorer_update_request(update_body, viewer_context=self.viewer_context)
if response.status >= 400:
raise SeerApiError("Seer request failed", response.status)

Expand Down
9 changes: 7 additions & 2 deletions src/sentry/seer/explorer/explorer_service_map_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,11 @@
from sentry.search.eap.types import SearchResolverConfig
from sentry.search.events.types import SnubaParams
from sentry.seer.models import SeerApiError
from sentry.seer.signed_seer_api import ServiceMapUpdateRequest, make_service_map_update_request
from sentry.seer.signed_seer_api import (
SeerViewerContext,
ServiceMapUpdateRequest,
make_service_map_update_request,
)
from sentry.snuba.referrer import Referrer
from sentry.snuba.spans_rpc import Spans

Expand Down Expand Up @@ -308,6 +312,7 @@ def _send_to_seer(org_id: int, nodes: list[dict], edges: list[dict]) -> None:
},
)

response = make_service_map_update_request(body, timeout=30)
viewer_context = SeerViewerContext(organization_id=org_id)
response = make_service_map_update_request(body, timeout=30, viewer_context=viewer_context)
if response.status >= 400:
raise SeerApiError("Seer service map update failed", response.status)
Loading