Implement FE methods for GitHub installations and BitBucket workspaces

frontend/src/api/open-hands.ts

@@ -489,6 +489,24 @@ class OpenHands {
 
     return data;
   }
+
+  /**
+   * Get the GitHub user installation IDs
+   * @returns List of GitHub installation IDs
+   */
+  static async getGitHubUserInstallationIds(): Promise<string[]> {
+    const { data } = await openHands.get<string[]>("/github/installations");
+    return data;
+  }
+
+  /**
+   * Get the BitBucket workspaces
+   * @returns List of BitBucket workspaces
+   */
+  static async getBitBucketWorkspaces(): Promise<string[]> {
+    const { data } = await openHands.get<string[]>("/bitbucket/installations");
+    return data;
+  }
 }
 
 export default OpenHands;

frontend/src/components/features/home/repo-selection-form.tsx

@@ -10,6 +10,9 @@ import { BrandButton } from "../settings/brand-button";
 import { useSearchRepositories } from "#/hooks/query/use-search-repositories";
 import { useDebounce } from "#/hooks/use-debounce";
 import { sanitizeQuery } from "#/utils/sanitize-query";
+import { useUserProviders } from "#/hooks/use-user-providers";
+import { Provider } from "#/types/settings";
+import { SettingsDropdownInput } from "../settings/settings-dropdown-input";
 import {
   RepositoryDropdown,
   RepositoryLoadingState,
@@ -32,8 +35,10 @@ export function RepositorySelectionForm({
   const [selectedBranch, setSelectedBranch] = React.useState<Branch | null>(
     null,
   );
+  const [selectedProvider, setSelectedProvider] = React.useState<Provider | null>(null);
   // Add a ref to track if the branch was manually cleared by the user
   const branchManuallyClearedRef = React.useRef<boolean>(false);
+  const { providers } = useUserProviders();
   const {
     data: repositories,
     isLoading: isLoadingRepositories,
@@ -56,6 +61,13 @@ export function RepositorySelectionForm({
   const debouncedSearchQuery = useDebounce(searchQuery, 300);
   const { data: searchedRepos } = useSearchRepositories(debouncedSearchQuery);
 
+  // Auto-select provider if there's only one
+  React.useEffect(() => {
+    if (providers.length === 1 && !selectedProvider) {
+      setSelectedProvider(providers[0]);
+    }
+  }, [providers, selectedProvider]);
+
   // Auto-select main or master branch if it exists, but only if the branch wasn't manually cleared
   React.useEffect(() => {
     if (
@@ -83,8 +95,10 @@ export function RepositorySelectionForm({
   const isCreatingConversation =
     isPending || isSuccess || isCreatingConversationElsewhere;
 
+  // Use all repositories without filtering by provider for now
   const allRepositories = repositories?.concat(searchedRepos || []);
-  const repositoriesItems = allRepositories?.map((repo) => ({
+
+  const repositoriesItems = (allRepositories || []).map((repo) => ({
     key: repo.id,
     label: decodeURIComponent(repo.full_name),
   }));
@@ -94,6 +108,14 @@ export function RepositorySelectionForm({
     label: branch.name,
   }));
 
+  // Create provider dropdown items
+  const providerItems = React.useMemo(() => {
+    return providers.map(provider => ({
+      key: provider,
+      label: provider.charAt(0).toUpperCase() + provider.slice(1), // Capitalize first letter
+    }));
+  }, [providers]);
+
   const handleRepoSelection = (key: React.Key | null) => {
     const selectedRepo = allRepositories?.find((repo) => repo.id === key);
     if (selectedRepo) onRepoSelection(selectedRepo);
@@ -102,6 +124,14 @@ export function RepositorySelectionForm({
     branchManuallyClearedRef.current = false; // Reset the flag when repo changes
   };
 
+  const handleProviderSelection = (key: React.Key | null) => {
+    const provider = key as Provider | null;
+    setSelectedProvider(provider);
+    setSelectedRepository(null); // Reset repository selection when provider changes
+    setSelectedBranch(null); // Reset branch selection when provider changes
+    onRepoSelection(null); // Reset parent component's selected repo
+  };
+
   const handleBranchSelection = (key: React.Key | null) => {
     const selectedBranchObj = branches?.find((branch) => branch.name === key);
     setSelectedBranch(selectedBranchObj || null);
@@ -133,6 +163,26 @@ export function RepositorySelectionForm({
     }
   };
 
+  // Render the provider dropdown
+  const renderProviderSelector = () => {
+    // Only render if there are multiple providers
+    if (providers.length <= 1) {
+      return null;
+    }
+
+    return (
+      <SettingsDropdownInput
+        testId="provider-dropdown"
+        name="provider-dropdown"
+        placeholder="Select Provider"
+        items={providerItems}
+        wrapperClassName="max-w-[500px]"
+        onSelectionChange={handleProviderSelection}
+        selectedKey={selectedProvider || undefined}
+      />
+    );
+  };
+
   // Render the appropriate UI based on the loading/error state
   const renderRepositorySelector = () => {
     if (isLoadingRepositories) {
@@ -143,11 +193,15 @@ export function RepositorySelectionForm({
       return <RepositoryErrorState />;
     }
 
+    // For now, don't disable the repo dropdown based on provider selection
+    const isDisabled = false;
+
     return (
       <RepositoryDropdown
         items={repositoriesItems || []}
         onSelectionChange={handleRepoSelection}
         onInputChange={handleRepoInputChange}
+        isDisabled={isDisabled}
         defaultFilter={(textValue, inputValue) => {
           if (!inputValue) return true;
 
@@ -195,8 +249,8 @@ export function RepositorySelectionForm({
 
   return (
     <div className="flex flex-col gap-4">
+      {renderProviderSelector()}
       {renderRepositorySelector()}
-
       {renderBranchSelector()}
 
       <BrandButton

frontend/src/components/features/home/repository-selection/repository-dropdown.tsx

@@ -8,6 +8,7 @@ export interface RepositoryDropdownProps {
   onSelectionChange: (key: React.Key | null) => void;
   onInputChange: (value: string) => void;
   defaultFilter?: (textValue: string, inputValue: string) => boolean;
+  isDisabled?: boolean;
 }
 
 export function RepositoryDropdown({
@@ -15,6 +16,7 @@ export function RepositoryDropdown({
   onSelectionChange,
   onInputChange,
   defaultFilter,
+  isDisabled = false,
 }: RepositoryDropdownProps) {
   const { t } = useTranslation();
 
@@ -22,12 +24,13 @@ export function RepositoryDropdown({
     <SettingsDropdownInput
       testId="repo-dropdown"
       name="repo-dropdown"
-      placeholder={t(I18nKey.REPOSITORY$SELECT_REPO)}
+      placeholder={isDisabled ? t("Please select a provider first") : t(I18nKey.REPOSITORY$SELECT_REPO)}
       items={items}
       wrapperClassName="max-w-[500px]"
       onSelectionChange={onSelectionChange}
       onInputChange={onInputChange}
       defaultFilter={defaultFilter}
+      isDisabled={isDisabled}
     />
   );
 }

frontend/src/hooks/query/use-app-installations.ts

@@ -0,0 +1,23 @@
+import { useQuery } from "@tanstack/react-query";
+import { useConfig } from "./use-config";
+import { useIsAuthed } from "./use-is-authed";
+import OpenHands from "#/api/open-hands";
+import { useUserProviders } from "../use-user-providers";
+
+export const useAppInstallations = () => {
+  const { data: config } = useConfig();
+  const { data: userIsAuthenticated } = useIsAuthed();
+  const { providers } = useUserProviders();
+
+  return useQuery({
+    queryKey: ["installations", providers, config?.GITHUB_CLIENT_ID],
+    queryFn: OpenHands.getGitHubUserInstallationIds,
+    enabled:
+      userIsAuthenticated &&
+      providers.includes("github") &&
+      !!config?.GITHUB_CLIENT_ID &&
+      config?.APP_MODE === "saas",
+    staleTime: 1000 * 60 * 5, // 5 minutes
+    gcTime: 1000 * 60 * 15, // 15 minutes
+  });
+};

frontend/src/hooks/query/use-bitbucket-workspaces.ts

@@ -0,0 +1,22 @@
+import { useQuery } from "@tanstack/react-query";
+import { useConfig } from "./use-config";
+import { useIsAuthed } from "./use-is-authed";
+import OpenHands from "#/api/open-hands";
+import { useUserProviders } from "../use-user-providers";
+
+export const useBitbucketWorkspaces = () => {
+  const { data: config } = useConfig();
+  const { data: userIsAuthenticated } = useIsAuthed();
+  const { providers } = useUserProviders();
+
+  return useQuery({
+    queryKey: ["workspaces", providers],
+    queryFn: OpenHands.getBitBucketWorkspaces,
+    enabled:
+      userIsAuthenticated &&
+      providers.includes("bitbucket") &&
+      config?.APP_MODE === "saas",
+    staleTime: 1000 * 60 * 5, // 5 minutes
+    gcTime: 1000 * 60 * 15, // 15 minutes
+  });
+};

openhands/integrations/bitbucket/bitbucket_service.py

@@ -9,6 +9,7 @@ from openhands.integrations.service_types import (
     BaseGitService,
     Branch,
     GitService,
+    InstallationsService,
     OwnerType,
     ProviderType,
     Repository,
@@ -20,7 +21,7 @@ from openhands.server.types import AppMode
 from openhands.utils.import_utils import get_impl
 
 
-class BitBucketService(BaseGitService, GitService):
+class BitBucketService(BaseGitService, GitService, InstallationsService):
     """Default implementation of GitService for Bitbucket integration.
 
     This is an extension point in OpenHands that allows applications to customize Bitbucket
@@ -185,7 +186,89 @@ class BitBucketService(BaseGitService, GitService):
 
         return all_items[:max_items]  # Trim to max_items if needed
 
-    async def get_repositories(self, sort: str, app_mode: AppMode) -> list[Repository]:
+    async def get_installations(self) -> list[str]:
+        workspaces_url = f'{self.BASE_URL}/workspaces'
+        workspaces = await self._fetch_paginated_data(workspaces_url, {}, 100)
+
+        installations: list[str] = []
+        for workspace in workspaces:
+            installations.append(workspace['slug'])
+
+        return installations
+
+    async def get_paginated_repos(
+        self, page: int, per_page: int, sort: str, installation_id: str | None
+    ) -> list[Repository]:
+        """Get paginated repositories for a specific workspace.
+
+        Args:
+            page: The page number to fetch
+            per_page: The number of repositories per page
+            sort: The sort field ('pushed', 'updated', 'created', 'full_name')
+            installation_id: The workspace slug to fetch repositories from (as int, will be converted to string)
+
+        Returns:
+            A list of Repository objects
+        """
+        if not installation_id:
+            return []
+
+        # Convert installation_id to string for use as workspace_slug
+        workspace_slug = installation_id
+        workspace_repos_url = f'{self.BASE_URL}/repositories/{workspace_slug}'
+
+        # Map sort parameter to Bitbucket API compatible values
+        bitbucket_sort = sort
+        if sort == 'pushed':
+            # Bitbucket doesn't support 'pushed', use 'updated_on' instead
+            bitbucket_sort = '-updated_on'  # Use negative prefix for descending order
+        elif sort == 'updated':
+            bitbucket_sort = '-updated_on'
+        elif sort == 'created':
+            bitbucket_sort = '-created_on'
+        elif sort == 'full_name':
+            bitbucket_sort = 'name'  # Bitbucket uses 'name' not 'full_name'
+        else:
+            # Default to most recently updated first
+            bitbucket_sort = '-updated_on'
+
+        params = {
+            'pagelen': per_page,
+            'page': page,
+            'sort': bitbucket_sort,
+        }
+
+        response, headers = await self._make_request(workspace_repos_url, params)
+
+        # Extract repositories from the response
+        repos = response.get('values', [])
+
+        # Extract link header for pagination
+        next_link = response.get('next', '')
+
+        repositories = [
+            Repository(
+                id=repo.get('uuid', ''),
+                full_name=f'{repo.get("workspace", {}).get("slug", "")}/{repo.get("slug", "")}',
+                git_provider=ProviderType.BITBUCKET,
+                is_public=repo.get('is_private', True) is False,
+                stargazers_count=None,  # Bitbucket doesn't have stars
+                pushed_at=repo.get('updated_on'),
+                owner_type=(
+                    OwnerType.ORGANIZATION
+                    if repo.get('workspace', {}).get('is_private') is False
+                    else OwnerType.USER
+                ),
+                link_header=next_link,
+            )
+            for repo in repos
+        ]
+
+        return repositories
+
+    async def get_all_repositories(
+        self, sort: str, app_mode: AppMode
+    ) -> list[Repository]:
         """Get repositories for the authenticated user using workspaces endpoint.
 
         This method gets all repositories (both public and private) that the user has access to

openhands/integrations/github/github_service.py

@@ -15,6 +15,7 @@ from openhands.integrations.service_types import (
     BaseGitService,
     Branch,
     GitService,
+    InstallationsService,
     OwnerType,
     ProviderType,
     Repository,
@@ -28,7 +29,7 @@ from openhands.server.types import AppMode
 from openhands.utils.import_utils import get_impl
 
 
-class GitHubService(BaseGitService, GitService):
+class GitHubService(BaseGitService, GitService, InstallationsService):
     """Default implementation of GitService for GitHub integration.
 
     TODO: This doesn't seem a good candidate for the get_impl() pattern. What are the abstract methods we should actually separate and implement here?
@@ -192,14 +193,47 @@ class GitHubService(BaseGitService, GitService):
         ts = repo.get('pushed_at')
         return datetime.strptime(ts, '%Y-%m-%dT%H:%M:%SZ') if ts else datetime.min
 
-    async def get_repositories(self, sort: str, app_mode: AppMode) -> list[Repository]:
+    async def get_paginated_repos(
+        self, page: int, per_page: int, sort: str, installation_id: str | None
+    ):
+        params = {'page': str(page), 'per_page': str(per_page)}
+        if installation_id:
+            url = f'{self.BASE_URL}/user/installations/{installation_id}/repositories'
+            response, headers = await self._make_request(url, params)
+            response = response.get('repositories', [])
+        else:
+            url = f'{self.BASE_URL}/user/repos'
+            params['sort'] = sort
+            response, headers = await self._make_request(url, params)
+
+        next_link: str = headers.get('Link', '')
+        return [
+            Repository(
+                id=str(repo.get('id')),  # type: ignore[arg-type]
+                full_name=repo.get('full_name'),  # type: ignore[arg-type]
+                stargazers_count=repo.get('stargazers_count'),
+                git_provider=ProviderType.GITHUB,
+                is_public=not repo.get('private', True),
+                owner_type=(
+                    OwnerType.ORGANIZATION
+                    if repo.get('owner', {}).get('type') == 'Organization'
+                    else OwnerType.USER
+                ),
+                link_header=next_link,
+            )
+            for repo in response
+        ]
+
+    async def get_all_repositories(
+        self, sort: str, app_mode: AppMode
+    ) -> list[Repository]:
         MAX_REPOS = 1000
         PER_PAGE = 100  # Maximum allowed by GitHub API
         all_repos: list[dict] = []
 
         if app_mode == AppMode.SAAS:
             # Get all installation IDs and fetch repos for each one
-            installation_ids = await self.get_installation_ids()
+            installation_ids = await self.get_installations()
 
             # Iterate through each installation ID
             for installation_id in installation_ids:
@@ -246,11 +280,11 @@ class GitHubService(BaseGitService, GitService):
             for repo in all_repos
         ]
 
-    async def get_installation_ids(self) -> list[int]:
+    async def get_installations(self) -> list[str]:
         url = f'{self.BASE_URL}/user/installations'
         response, _ = await self._make_request(url)
         installations = response.get('installations', [])
-        return [i['id'] for i in installations]
+        return [str(i['id']) for i in installations]
 
     async def search_repositories(
         self, query: str, per_page: int, sort: str, order: str

openhands/integrations/gitlab/gitlab_service.py

@@ -226,7 +226,49 @@ class GitLabService(BaseGitService, GitService):
 
         return repos
 
-    async def get_repositories(self, sort: str, app_mode: AppMode) -> list[Repository]:
+    async def get_paginated_repos(
+        self, page: int, per_page: int, sort: str, installation_id: str | None
+    ) -> list[Repository]:
+        url = f'{self.BASE_URL}/projects'
+        order_by = {
+            'pushed': 'last_activity_at',
+            'updated': 'last_activity_at',
+            'created': 'created_at',
+            'full_name': 'name',
+        }.get(sort, 'last_activity_at')
+
+        params = {
+            'page': str(page),
+            'per_page': str(per_page),
+            'order_by': order_by,
+            'sort': 'desc',  # GitLab uses sort for direction (asc/desc)
+            'owned': True,  # Boolean value without quotes
+            'membership': True,  # Include projects user is a member of
+        }
+        response, headers = await self._make_request(url, params)
+
+        next_link: str = headers.get('Link', '')
+        repos = [
+            Repository(
+                id=str(repo.get('id')),  # type: ignore[arg-type]
+                full_name=repo.get('path_with_namespace'),  # type: ignore[arg-type]
+                stargazers_count=repo.get('star_count'),
+                git_provider=ProviderType.GITLAB,
+                is_public=repo.get('visibility') == 'public',
+                owner_type=(
+                    OwnerType.ORGANIZATION
+                    if repo.get('namespace', {}).get('kind') == 'group'
+                    else OwnerType.USER
+                ),
+                link_header=next_link,
+            )
+            for repo in response
+        ]
+        return repos
+
+    async def get_all_repositories(
+        self, sort: str, app_mode: AppMode
+    ) -> list[Repository]:
         MAX_REPOS = 1000
         PER_PAGE = 100  # Maximum allowed by GitLab API
         all_repos: list[dict] = []

openhands/integrations/provider.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from types import MappingProxyType
-from typing import Annotated, Any, Coroutine, Literal, overload
+from typing import Annotated, Any, Coroutine, Literal, cast, overload
 
 from pydantic import (
     BaseModel,
@@ -22,6 +22,7 @@ from openhands.integrations.service_types import (
     AuthenticationError,
     Branch,
     GitService,
+    InstallationsService,
     ProviderType,
     Repository,
     SuggestedTask,
@@ -160,16 +161,61 @@ class ProviderHandler:
         service = self._get_service(provider)
         return await service.get_latest_token()
 
-    async def get_repositories(self, sort: str, app_mode: AppMode) -> list[Repository]:
+    async def get_github_installations(self) -> list[str]:
+        service = cast(InstallationsService, self._get_service(ProviderType.GITHUB))
+        try:
+            return await service.get_installations()
+        except Exception as e:
+            logger.warning(f'Failed to get github installations {e}')
+
+        return []
+
+    async def get_bitbucket_workspaces(self) -> list[str]:
+        service = cast(InstallationsService, self._get_service(ProviderType.BITBUCKET))
+        try:
+            return await service.get_installations()
+        except Exception as e:
+            logger.warning(f'Failed to get bitbucket workspaces {e}')
+
+        return []
+
+    async def get_repositories(
+        self,
+        sort: str,
+        app_mode: AppMode,
+        selected_provider: ProviderType | None,
+        page: int | None,
+        per_page: int | None,
+        installation_id: str | None,
+    ) -> list[Repository]:
         """
         Get repositories from providers
         """
 
+        """
+        Get repositories from providers
+        """
+
+        if selected_provider:
+            if not page or not per_page:
+                logger.error('Failed to provider params for paginating repos')
+                return []
+
+            service = self._get_service(selected_provider)
+            try:
+                return await service.get_paginated_repos(
+                    page, per_page, sort, installation_id
+                )
+            except Exception as e:
+                logger.warning(f'Error fetching repos from {selected_provider}: {e}')
+
+            return []
+
         all_repos: list[Repository] = []
         for provider in self.provider_tokens:
             try:
                 service = self._get_service(provider)
-                service_repos = await service.get_repositories(sort, app_mode)
+                service_repos = await service.get_all_repositories(sort, app_mode)
                 all_repos.extend(service_repos)
             except Exception as e:
                 logger.warning(f'Error fetching repos from {provider}: {e}')

openhands/integrations/service_types.py

@@ -200,6 +200,12 @@ class BaseGitService(ABC):
         return UnknownException(f'HTTP error {type(e).__name__} : {e}')
 
 
+class InstallationsService(Protocol):
+    async def get_installations(self) -> list[str]:
+        """Get installations for the service; repos live underneath these installations"""
+        ...
+
+
 class GitService(Protocol):
     """Protocol defining the interface for Git service providers"""
 
@@ -233,10 +239,18 @@ class GitService(Protocol):
         """Search for repositories"""
         ...
 
-    async def get_repositories(self, sort: str, app_mode: AppMode) -> list[Repository]:
+    async def get_all_repositories(
+        self, sort: str, app_mode: AppMode
+    ) -> list[Repository]:
         """Get repositories for the authenticated user"""
         ...
 
+    async def get_paginated_repos(
+        self, page: int, per_page: int, sort: str, installation_id: str | None
+    ) -> list[Repository]:
+        """Get a page of repositories for the authenticated user"""
+        ...
+
     async def get_suggested_tasks(self) -> list[SuggestedTask]:
         """Get suggested tasks for the authenticated user across all repositories"""
         ...

openhands/server/routes/git.py

@@ -38,9 +38,55 @@ from openhands.server.user_auth import (
 app = APIRouter(prefix='/api/user', dependencies=get_dependencies())
 
 
+@app.get('/github/installations', response_model=list[str])
+async def get_user_github_installations(
+    provider_tokens: PROVIDER_TOKEN_TYPE | None = Depends(get_provider_tokens),
+    access_token: SecretStr | None = Depends(get_access_token),
+    user_id: str | None = Depends(get_user_id),
+):
+    if provider_tokens:
+        client = ProviderHandler(
+            provider_tokens=provider_tokens,
+            external_auth_token=access_token,
+            external_auth_id=user_id,
+        )
+
+        return await client.get_github_installations()
+
+    return JSONResponse(
+        content='Git provider token required. (such as GitHub).',
+        status_code=status.HTTP_401_UNAUTHORIZED,
+    )
+
+
+@app.get('/bitbucket/installations', response_model=list[str])
+async def get_user_bitbucket_installations(
+    provider_tokens: PROVIDER_TOKEN_TYPE | None = Depends(get_provider_tokens),
+    access_token: SecretStr | None = Depends(get_access_token),
+    user_id: str | None = Depends(get_user_id),
+):
+    if provider_tokens:
+        client = ProviderHandler(
+            provider_tokens=provider_tokens,
+            external_auth_token=access_token,
+            external_auth_id=user_id,
+        )
+
+        return await client.get_github_installations()
+
+    return JSONResponse(
+        content='Git provider token required. (such as GitHub).',
+        status_code=status.HTTP_401_UNAUTHORIZED,
+    )
+
+
 @app.get('/repositories', response_model=list[Repository])
 async def get_user_repositories(
     sort: str = 'pushed',
+    selected_provider: ProviderType | None = None,
+    page: int | None = None,
+    per_page: int | None = None,
+    installation_id: str | None = None,
     provider_tokens: PROVIDER_TOKEN_TYPE | None = Depends(get_provider_tokens),
     access_token: SecretStr | None = Depends(get_access_token),
     user_id: str | None = Depends(get_user_id),
@@ -53,7 +99,14 @@ async def get_user_repositories(
         )
 
         try:
-            return await client.get_repositories(sort, server_config.app_mode)
+            return await client.get_repositories(
+                sort,
+                server_config.app_mode,
+                selected_provider,
+                page,
+                per_page,
+                installation_id,
+            )
 
         except AuthenticationError as e:
             logger.info(

tests/unit/test_bitbucket.py

@@ -450,7 +450,7 @@ async def test_bitbucket_sort_parameter_mapping():
         ]
 
         # Call get_repositories with sort='pushed'
-        await service.get_repositories('pushed', AppMode.SAAS)
+        await service.get_all_repositories('pushed', AppMode.SAAS)
 
         # Verify that the second call used 'updated_on' instead of 'pushed'
         assert mock_request.call_count == 2
@@ -520,7 +520,7 @@ async def test_bitbucket_pagination():
         ]
 
         # Call get_repositories
-        repositories = await service.get_repositories('pushed', AppMode.SAAS)
+        repositories = await service.get_all_repositories('pushed', AppMode.SAAS)
 
         # Verify that all three requests were made (workspaces + 2 pages of repos)
         assert mock_request.call_count == 3
@@ -619,7 +619,7 @@ async def test_bitbucket_get_repositories_with_user_owner_type():
     with patch.object(service, '_fetch_paginated_data') as mock_fetch:
         mock_fetch.side_effect = [mock_workspaces, mock_repos]
 
-        repositories = await service.get_repositories('pushed', AppMode.SAAS)
+        repositories = await service.get_all_repositories('pushed', AppMode.SAAS)
 
         # Verify we got the expected number of repositories
         assert len(repositories) == 2
@@ -658,7 +658,7 @@ async def test_bitbucket_get_repositories_with_organization_owner_type():
     with patch.object(service, '_fetch_paginated_data') as mock_fetch:
         mock_fetch.side_effect = [mock_workspaces, mock_repos]
 
-        repositories = await service.get_repositories('pushed', AppMode.SAAS)
+        repositories = await service.get_all_repositories('pushed', AppMode.SAAS)
 
         # Verify we got the expected number of repositories
         assert len(repositories) == 2
@@ -706,7 +706,7 @@ async def test_bitbucket_get_repositories_mixed_owner_types():
     with patch.object(service, '_fetch_paginated_data') as mock_fetch:
         mock_fetch.side_effect = [mock_workspaces, mock_user_repos, mock_org_repos]
 
-        repositories = await service.get_repositories('pushed', AppMode.SAAS)
+        repositories = await service.get_all_repositories('pushed', AppMode.SAAS)
 
         # Verify we got repositories from both workspaces
         assert len(repositories) == 2
@@ -746,7 +746,7 @@ async def test_bitbucket_get_repositories_owner_type_fallback():
     with patch.object(service, '_fetch_paginated_data') as mock_fetch:
         mock_fetch.side_effect = [mock_workspaces, mock_repos]
 
-        repositories = await service.get_repositories('pushed', AppMode.SAAS)
+        repositories = await service.get_all_repositories('pushed', AppMode.SAAS)
 
         # Verify all repositories default to USER owner_type for private workspaces
         for repo in repositories:

tests/unit/test_github_service.py

@@ -112,9 +112,9 @@ async def test_github_get_repositories_with_user_owner_type():
 
     with (
         patch.object(service, '_fetch_paginated_repos', return_value=mock_repo_data),
-        patch.object(service, 'get_installation_ids', return_value=[123]),
+        patch.object(service, 'get_installations', return_value=[123]),
     ):
-        repositories = await service.get_repositories('pushed', AppMode.SAAS)
+        repositories = await service.get_all_repositories('pushed', AppMode.SAAS)
 
         # Verify we got the expected number of repositories
         assert len(repositories) == 2
@@ -151,9 +151,9 @@ async def test_github_get_repositories_with_organization_owner_type():
 
     with (
         patch.object(service, '_fetch_paginated_repos', return_value=mock_repo_data),
-        patch.object(service, 'get_installation_ids', return_value=[123]),
+        patch.object(service, 'get_installations', return_value=[123]),
     ):
-        repositories = await service.get_repositories('pushed', AppMode.SAAS)
+        repositories = await service.get_all_repositories('pushed', AppMode.SAAS)
 
         # Verify we got the expected number of repositories
         assert len(repositories) == 2
@@ -190,9 +190,9 @@ async def test_github_get_repositories_mixed_owner_types():
 
     with (
         patch.object(service, '_fetch_paginated_repos', return_value=mock_repo_data),
-        patch.object(service, 'get_installation_ids', return_value=[123]),
+        patch.object(service, 'get_installations', return_value=[123]),
     ):
-        repositories = await service.get_repositories('pushed', AppMode.SAAS)
+        repositories = await service.get_all_repositories('pushed', AppMode.SAAS)
 
         # Verify we got the expected number of repositories
         assert len(repositories) == 2
@@ -237,9 +237,9 @@ async def test_github_get_repositories_owner_type_fallback():
 
     with (
         patch.object(service, '_fetch_paginated_repos', return_value=mock_repo_data),
-        patch.object(service, 'get_installation_ids', return_value=[123]),
+        patch.object(service, 'get_installations', return_value=[123]),
     ):
-        repositories = await service.get_repositories('pushed', AppMode.SAAS)
+        repositories = await service.get_all_repositories('pushed', AppMode.SAAS)
 
         # Verify all repositories default to USER owner_type
         for repo in repositories:

tests/unit/test_gitlab.py

@@ -37,7 +37,7 @@ async def test_gitlab_get_repositories_with_user_owner_type():
         # Mock the pagination response
         mock_request.side_effect = [(mock_repos, {'Link': ''})]  # No next page
 
-        repositories = await service.get_repositories('pushed', AppMode.SAAS)
+        repositories = await service.get_all_repositories('pushed', AppMode.SAAS)
 
         # Verify we got the expected number of repositories
         assert len(repositories) == 2
@@ -76,7 +76,7 @@ async def test_gitlab_get_repositories_with_organization_owner_type():
         # Mock the pagination response
         mock_request.side_effect = [(mock_repos, {'Link': ''})]  # No next page
 
-        repositories = await service.get_repositories('pushed', AppMode.SAAS)
+        repositories = await service.get_all_repositories('pushed', AppMode.SAAS)
 
         # Verify we got the expected number of repositories
         assert len(repositories) == 2
@@ -115,7 +115,7 @@ async def test_gitlab_get_repositories_mixed_owner_types():
         # Mock the pagination response
         mock_request.side_effect = [(mock_repos, {'Link': ''})]  # No next page
 
-        repositories = await service.get_repositories('pushed', AppMode.SAAS)
+        repositories = await service.get_all_repositories('pushed', AppMode.SAAS)
 
         # Verify we got the expected number of repositories
         assert len(repositories) == 2
@@ -162,7 +162,7 @@ async def test_gitlab_get_repositories_owner_type_fallback():
         # Mock the pagination response
         mock_request.side_effect = [(mock_repos, {'Link': ''})]  # No next page
 
-        repositories = await service.get_repositories('pushed', AppMode.SAAS)
+        repositories = await service.get_all_repositories('pushed', AppMode.SAAS)
 
         # Verify all repositories default to USER owner_type
         for repo in repositories: