UT: Issues with completion callback in dev

const { startUpload, isUploading } = useUploadThing('inputSource', {
    onClientUploadComplete: (records) => {
      const { url, type, uid } = records[0].serverData as {
        url: string;
        type: 'IMAGE' | 'VIDEO';
        uid: number;
      };
      addUploadedInputSource(url, type, uid);
    },
    onUploadError: (error: Error) => {
      console.error('error uploading input source', error);
      toast.error('Error uploading file');
    },
  });

export const uploadRouter = {
  inputSource: f(
    {
      'image/jpeg': {
        maxFileSize: '128KB',
        maxFileCount: 1,
        additionalProperties: {
          width: 640,
          height: 480,
        },
      },
    },
    { awaitServerData: true },
  )
    .middleware(async () => {
      const session = await auth();
      console.log('session', session);
      const uid = (
        session.sessionClaims?.publicMetadata as { uid: number | undefined }
      )?.uid;
      if (!session.userId || !uid) throw new UploadThingError('UNAUTHORIZED');
      return { uid };
    })
    .onUploadComplete(async ({ metadata, file }) => {
      const { error: inputSourceError } = await tryCatch(
        prisma.inputSource.create({
          data: {
            url: file.ufsUrl,
            type: 'IMAGE',
            uid: metadata.uid,
            key: file.key,
          },
        }),
      );
      if (inputSourceError) {
        throw new UploadThingError(
          'failed to create input source: ' + inputSourceError.message,
        );
      }
      revalidateTag(`input-sources-${metadata.uid}`);
      return {
        url: file.ufsUrl,
        type: 'IMAGE',
        uid: metadata.uid,
      };
    }),
} satisfies FileRouter;