feat: update pulling artifact edge after import revisions

Author: nowgnuesLee

packages/backend.ai-ui/src/components/fragments/BAIImportArtifactModal.tsx

@@ -42,13 +42,15 @@ export interface BAIImportArtifactModalProps
     }[],
   ) => void;
   onCancel: (e: React.MouseEvent<HTMLElement>) => void;
+  connectionIds?: string[];
 }
 
 const BAIImportArtifactModal = ({
   selectedArtifactFrgmt,
   selectedArtifactRevisionFrgmt,
   onOk,
   onCancel,
+  connectionIds,
   ...modalProps
 }: BAIImportArtifactModalProps) => {
   const { t } = useTranslation();
@@ -81,6 +83,7 @@ const BAIImportArtifactModal = ({
     useMutation<BAIImportArtifactModalImportArtifactsMutation>(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) {

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([]);