Try new approach

Author: rohitvinnakota-codecov

graphql_api/actions/repository.py

@@ -3,9 +3,11 @@
 
 import sentry_sdk
 from django.db.models import QuerySet
-from shared.django_apps.codecov_auth.models import Owner
+from shared.django_apps.codecov_auth.models import GithubAppInstallation, Owner
 from shared.django_apps.core.models import Repository
 
+from graphql_api.types.owner.owner import AI_FEATURES_GH_APP_ID
+
 log = logging.getLogger(__name__)
 
 
@@ -33,6 +35,22 @@ def apply_filters_to_queryset(
     return queryset
 
 
+def filter_queryset_by_ai_enabled_repos(queryset: QuerySet, owner: Owner) -> QuerySet:
+    ai_features_app_install = GithubAppInstallation.objects.filter(
+        app_id=AI_FEATURES_GH_APP_ID, owner=owner
+    ).first()
+
+    if not ai_features_app_install:
+        return None
+
+    if ai_features_app_install.repository_service_ids:
+        queryset = queryset.filter(
+            service_id__in=ai_features_app_install.repository_service_ids
+        )
+
+    return queryset
+
+
 @sentry_sdk.trace
 def list_repository_for_owner(
     current_owner: Owner,
@@ -43,6 +61,11 @@ def list_repository_for_owner(
 ) -> QuerySet:
     queryset = Repository.objects.viewable_repos(current_owner)
 
+    ai_enabled_filter = filters.get("ai_enabled")
+
+    if ai_enabled_filter:
+        return filter_queryset_by_ai_enabled_repos(queryset, owner)
+
     if exclude_okta_enforced_repos:
         queryset = queryset.exclude_accounts_enforced_okta(okta_account_auths)
 

graphql_api/types/inputs/repository_set_filters.graphql

@@ -4,4 +4,5 @@ input RepositorySetFilters {
   active: Boolean
   activated: Boolean
   isPublic: Boolean
+  aiEnabled: Boolean
 }

graphql_api/types/owner/owner.py

@@ -31,7 +31,6 @@
 from graphql_api.helpers.connection import (
     Connection,
     build_connection_graphql,
-    queryset_to_connection,
     queryset_to_connection_sync,
 )
 from graphql_api.helpers.mutation import (
@@ -55,6 +54,7 @@
 
 
 @owner_bindable.field("repositories")
+@sync_to_async
 def resolve_repositories(
     owner: Owner,
     info: GraphQLResolveInfo,
@@ -77,7 +77,7 @@ def resolve_repositories(
         current_owner, owner, filters, okta_account_auths, exclude_okta_enforced_repos
     )
 
-    return queryset_to_connection(
+    return queryset_to_connection_sync(
         queryset,
         ordering=(ordering, RepositoryOrdering.ID),
         ordering_direction=ordering_direction,
@@ -406,34 +406,3 @@ def resolve_activated_user_count(owner: Owner, info: GraphQLResolveInfo) -> int:
 @require_part_of_org
 def resolve_billing(owner: Owner, info: GraphQLResolveInfo) -> dict | None:
     return owner
-
-
-@owner_bindable.field("aiEnabledRepositories")
-@sync_to_async
-def resolve_ai_enabled_repositories(
-    owner: Owner,
-    info: GraphQLResolveInfo,
-    ordering: Optional[RepositoryOrdering] = RepositoryOrdering.ID,
-    ordering_direction: Optional[OrderingDirection] = OrderingDirection.ASC,
-    **kwargs: Any,
-) -> Coroutine[Any, Any, Connection] | None:
-    ai_features_app_install = GithubAppInstallation.objects.filter(
-        app_id=AI_FEATURES_GH_APP_ID, owner=owner
-    ).first()
-
-    if not ai_features_app_install:
-        return None
-
-    current_owner = info.context["request"].current_owner
-    queryset = Repository.objects.filter(author=owner).viewable_repos(current_owner)
-    if ai_features_app_install.repository_service_ids:
-        queryset = queryset.filter(
-            service_id__in=ai_features_app_install.repository_service_ids
-        )
-
-    return queryset_to_connection_sync(
-        queryset,
-        ordering=(ordering, RepositoryOrdering.ID),
-        ordering_direction=ordering_direction,
-        **kwargs,
-    )