feat(bug-prediction): Allow short IDs (#102138)

seer-side will let this tool supply short IDs

Author: kddubey

src/sentry/seer/fetch_issues/utils.py

@@ -136,15 +136,25 @@ def bulk_serialize_for_seer(groups: list[Group]) -> SeerResponse:
     }
 
 
-def get_latest_issue_event(group_id: int, organization_id: int) -> dict[str, Any]:
+def _group_by_short_id(short_id: str, organization_id: int) -> Group | None:
+    try:
+        return Group.objects.by_qualified_short_id(organization_id, short_id)
+    except Group.DoesNotExist:
+        return None
+
+
+def get_latest_issue_event(group_id: int | str, organization_id: int) -> dict[str, Any]:
     """
     Get an issue's latest event as a dict, matching the Seer IssueDetails model.
     """
-    group = Group.objects.filter(id=group_id).first()
+    if isinstance(group_id, str) and not group_id.isdigit():
+        group = _group_by_short_id(group_id, organization_id)
+    else:
+        group = Group.objects.filter(id=int(group_id)).first()
+
     if not group:
         logger.warning(
-            "Group not found",
-            extra={"group_id": group_id},
+            "Group not found", extra={"group_id": group_id, "organization_id": organization_id}
         )
         return {}
 

tests/sentry/seer/fetch_issues/test_utils.py

@@ -201,15 +201,33 @@ def test_get_latest_issue_event_success(self):
         assert isinstance(result, dict)
         assert result["id"] == group.id
         assert result["title"] == group.title
-        assert "events" in result
         assert len(result["events"]) == 1
         assert result["events"][0]["id"] == event.event_id
 
-    def test_get_latest_issue_event_group_not_found(self):
+    def test_get_latest_issue_event_not_found(self):
         nonexistent_group_id = 999999
         result = get_latest_issue_event(nonexistent_group_id, self.organization.id)
         assert result == {}
 
+    def test_get_latest_issue_event_with_short_id(self):
+        data = load_data("python", timestamp=before_now(minutes=1))
+        event = self.store_event(data=data, project_id=self.project.id)
+        group = event.group
+
+        assert group is not None
+        result = get_latest_issue_event(group.qualified_short_id, self.organization.id)
+
+        assert result is not None
+        assert isinstance(result, dict)
+        assert result["id"] == group.id
+        assert result["title"] == group.title
+        assert len(result["events"]) == 1
+        assert result["events"][0]["id"] == event.event_id
+
+    def test_get_latest_issue_event_with_short_id_not_found(self):
+        result = get_latest_issue_event("INVALID-SHORT-ID", self.organization.id)
+        assert result == {}
+
     def test_get_latest_issue_event_no_events(self):
         # Create a group but don't store any events for it
         group = self.create_group(project=self.project)