Browse Source

typing(seer): Strong typing for embeddings API (#79275)

Armen Zambrano G. 4 months ago
parent
commit
59d9da9412

+ 1 - 0
pyproject.toml

@@ -402,6 +402,7 @@ disable_error_code = [
 # begin: stronger typing
 [[tool.mypy.overrides]]
 module = [
+    "sentry.api.endpoints.project_backfill_similar_issues_embeddings_records",
     "sentry.api.helpers.deprecation",
     "sentry.api.helpers.source_map_helper",
     "sentry.auth.services.*",

+ 2 - 1
src/sentry/api/endpoints/project_backfill_similar_issues_embeddings_records.py

@@ -8,6 +8,7 @@ from sentry.api.api_publish_status import ApiPublishStatus
 from sentry.api.base import region_silo_endpoint
 from sentry.api.bases.project import ProjectEndpoint
 from sentry.auth.superuser import is_active_superuser
+from sentry.models.project import Project
 from sentry.tasks.embeddings_grouping.backfill_seer_grouping_records_for_project import (
     backfill_seer_grouping_records_for_project,
 )
@@ -20,7 +21,7 @@ class ProjectBackfillSimilarIssuesEmbeddingsRecords(ProjectEndpoint):
         "POST": ApiPublishStatus.PRIVATE,
     }
 
-    def post(self, request: Request, project) -> Response:
+    def post(self, request: Request, project: Project) -> Response:
         if not features.has("projects:similarity-embeddings-backfill", project):
             return Response(status=404)