From 406db9996c3ffa8b233f27211a1dfe65cf020943 Mon Sep 17 00:00:00 2001
From: Gyubong Lee <jopemachine@naver.com>
Date: Thu, 25 Jul 2024 05:15:28 +0000
Subject: [PATCH] feat: Apply pagination to ECR fetch_repositories

---
 src/ai/backend/manager/container_registry/aws_ecr.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/ai/backend/manager/container_registry/aws_ecr.py b/src/ai/backend/manager/container_registry/aws_ecr.py
index 4871edf5f9b..0b9ae859238 100644
--- a/src/ai/backend/manager/container_registry/aws_ecr.py
+++ b/src/ai/backend/manager/container_registry/aws_ecr.py
@@ -79,9 +79,9 @@ async def fetch_repositories(
         try:
             while True:
                 if next_token:
-                    response = client.describe_repositories(nextToken=next_token)
+                    response = client.describe_repositories(nextToken=next_token, maxResults=30)
                 else:
-                    response = client.describe_repositories()
+                    response = client.describe_repositories(maxResults=30)
 
                 for repo in response["repositories"]:
                     # TODO: Fix this.