From ef854c284eef925780ba1a3969f09cc801be7994 Mon Sep 17 00:00:00 2001 From: Gyubong Lee Date: Fri, 2 Aug 2024 07:53:01 +0000 Subject: [PATCH] fix: broken codes --- src/ai/backend/manager/api/session.py | 19 +++++++++++-------- 1 file changed, 11 insertions(+), 8 deletions(-) diff --git a/src/ai/backend/manager/api/session.py b/src/ai/backend/manager/api/session.py index 1d5a9e50aab..1a1fb48df6c 100644 --- a/src/ai/backend/manager/api/session.py +++ b/src/ai/backend/manager/api/session.py @@ -1076,15 +1076,18 @@ async def convert_session_to_image( registry_project = project.container_registry["project"] async with root_ctx.db.begin_readonly() as db_session: - query = sa.select(ContainerRegistryRow).where( - ContainerRegistryRow.registry_name == registry_hostname - ) + query = sa.select([ + ContainerRegistryRow.username, + ContainerRegistryRow.password, + ContainerRegistryRow.project, + ContainerRegistryRow.url, + ]).where(ContainerRegistryRow.registry_name == registry_hostname) - registry_conf = (await db_session.execute(query)).scalar() + registry_conf = (await db_session.execute(query)).fetchall()[0] if not registry_conf: raise InvalidAPIParameters(f"Registry {registry_hostname} not found") - if registry_project not in registry_conf.get("project", ""): + if registry_project not in registry_conf.project: raise InvalidAPIParameters(f"Project {registry_project} not found") base_image_ref = session.main_kernel.image_ref @@ -1198,9 +1201,9 @@ async def _commit_and_upload(reporter: ProgressReporter) -> None: # push image to registry from local agent image_registry = ImageRegistry( name=registry_hostname, - url=str(registry_conf[""]), - username=registry_conf.get("username"), - password=registry_conf.get("password"), + url=str(registry_conf.url), + username=registry_conf.username, + password=registry_conf.password, ) resp = await root_ctx.registry.push_image( session.main_kernel.agent,