How to type the server-side Gadget Context object in React Router 7 loader function?

// Server-side loader function
export const loader = async ({ params, context }: LoaderFunctionArgs) => {
  // access current authenticated shop
  const shop = await context.api.shopifyShop.findFirst({
    select: {
      id: true,
    },
  });

  return {
    shop,
  };
};