diff --git a/packages/backend.ai-ui/src/components/fragments/BAIImportArtifactModal.tsx b/packages/backend.ai-ui/src/components/fragments/BAIImportArtifactModal.tsx index 96603a1e17..60baaee0b8 100644 --- a/packages/backend.ai-ui/src/components/fragments/BAIImportArtifactModal.tsx +++ b/packages/backend.ai-ui/src/components/fragments/BAIImportArtifactModal.tsx @@ -42,6 +42,7 @@ export interface BAIImportArtifactModalProps }[], ) => void; onCancel: (e: React.MouseEvent) => void; + connectionIds?: string[]; } const BAIImportArtifactModal = ({ @@ -49,6 +50,7 @@ const BAIImportArtifactModal = ({ selectedArtifactRevisionFrgmt, onOk, onCancel, + connectionIds, ...modalProps }: BAIImportArtifactModalProps) => { const { t } = useTranslation(); @@ -81,6 +83,7 @@ const BAIImportArtifactModal = ({ useMutation(graphql` mutation BAIImportArtifactModalImportArtifactsMutation( $input: ImportArtifactsInput! + $connectionIds: [ID!]! ) { importArtifacts(input: $input) { tasks { @@ -90,7 +93,7 @@ const BAIImportArtifactModal = ({ } } artifactRevisions { - edges { + edges @appendEdge(connections: $connectionIds) { node { id status @@ -138,6 +141,7 @@ const BAIImportArtifactModal = ({ toLocalId(revision.id), ), }, + connectionIds: connectionIds ?? [], }, onCompleted: (res, errors) => { if (errors && errors.length > 0) { diff --git a/react/src/pages/ReservoirArtifactDetailPage.tsx b/react/src/pages/ReservoirArtifactDetailPage.tsx index 652697a1b9..351a0444d7 100644 --- a/react/src/pages/ReservoirArtifactDetailPage.tsx +++ b/react/src/pages/ReservoirArtifactDetailPage.tsx @@ -116,9 +116,15 @@ const ReservoirArtifactDetailPage = () => { } updatedAt pullingArtifactRevisions: revisions( + first: null + last: null filter: { status: { equals: PULLING } } orderBy: { field: VERSION, direction: DESC } - ) { + ) + @connection( + key: "ReservoirArtifactDetailPage_pullingArtifactRevisions" + ) { + __id count edges { node { @@ -518,6 +524,9 @@ const ReservoirArtifactDetailPage = () => { selectedArtifactFrgmt={artifact ?? null} selectedArtifactRevisionFrgmt={selectedRevisions} open={!!artifact && !_.isEmpty(selectedRevisions)} + connectionIds={ + artifact ? [artifact.pullingArtifactRevisions.__id] : undefined + } onOk={(_e, tasks) => { setSelectedRevisions([]); tasks.forEach((task) => { @@ -546,7 +555,6 @@ const ReservoirArtifactDetailPage = () => { }, }); }); - updateFetchKey(); }} onCancel={() => { setSelectedRevisions([]);