@xata.io/client 0.0.0-alpha.vfa1407e → 0.0.0-alpha.vfa28b1a26eb487667b028e1c2059dcef8e362bdb

package/dist/index.cjs

@@ -244,12 +244,6 @@ function getPreviewBranch() {
   }
 }
 
-var __defProp$8 = Object.defineProperty;
-var __defNormalProp$8 = (obj, key, value) => key in obj ? __defProp$8(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __publicField$8 = (obj, key, value) => {
-  __defNormalProp$8(obj, typeof key !== "symbol" ? key + "" : key, value);
-  return value;
-};
 var __accessCheck$8 = (obj, member, msg) => {
   if (!member.has(obj))
     throw TypeError("Cannot " + msg);
@@ -289,8 +283,6 @@ class ApiRequestPool {
     __privateAdd$8(this, _fetch, void 0);
     __privateAdd$8(this, _queue, void 0);
     __privateAdd$8(this, _concurrency, void 0);
-    __publicField$8(this, "running");
-    __publicField$8(this, "started");
     __privateSet$8(this, _queue, []);
     __privateSet$8(this, _concurrency, concurrency);
     this.running = 0;
@@ -536,26 +528,16 @@ function defaultOnOpen(response) {
   }
 }
 
-const VERSION = "0.26.5";
+const VERSION = "0.28.4";
 
-var __defProp$7 = Object.defineProperty;
-var __defNormalProp$7 = (obj, key, value) => key in obj ? __defProp$7(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __publicField$7 = (obj, key, value) => {
-  __defNormalProp$7(obj, typeof key !== "symbol" ? key + "" : key, value);
-  return value;
-};
 class ErrorWithCause extends Error {
   constructor(message, options) {
     super(message, options);
-    __publicField$7(this, "cause");
   }
 }
 class FetcherError extends ErrorWithCause {
   constructor(status, data, requestId) {
     super(getMessage(data));
-    __publicField$7(this, "status");
-    __publicField$7(this, "requestId");
-    __publicField$7(this, "errors");
     this.status = status;
     this.errors = isBulkError(data) ? data.errors : [{ message: getMessage(data), status }];
     this.requestId = requestId;
@@ -589,6 +571,67 @@ function getMessage(data) {
   }
 }
 
+function getHostUrl(provider, type) {
+  if (isHostProviderAlias(provider)) {
+    return providers[provider][type];
+  } else if (isHostProviderBuilder(provider)) {
+    return provider[type];
+  }
+  throw new Error("Invalid API provider");
+}
+const providers = {
+  production: {
+    main: "https://api.xata.io",
+    workspaces: "https://{workspaceId}.{region}.xata.sh"
+  },
+  staging: {
+    main: "https://api.staging-xata.dev",
+    workspaces: "https://{workspaceId}.{region}.staging-xata.dev"
+  },
+  dev: {
+    main: "https://api.dev-xata.dev",
+    workspaces: "https://{workspaceId}.{region}.dev-xata.dev"
+  },
+  local: {
+    main: "http://localhost:6001",
+    workspaces: "http://{workspaceId}.{region}.localhost:6001"
+  }
+};
+function isHostProviderAlias(alias) {
+  return isString(alias) && Object.keys(providers).includes(alias);
+}
+function isHostProviderBuilder(builder) {
+  return isObject(builder) && isString(builder.main) && isString(builder.workspaces);
+}
+function parseProviderString(provider = "production") {
+  if (isHostProviderAlias(provider)) {
+    return provider;
+  }
+  const [main, workspaces] = provider.split(",");
+  if (!main || !workspaces)
+    return null;
+  return { main, workspaces };
+}
+function buildProviderString(provider) {
+  if (isHostProviderAlias(provider))
+    return provider;
+  return `${provider.main},${provider.workspaces}`;
+}
+function parseWorkspacesUrlParts(url) {
+  if (!isString(url))
+    return null;
+  const matches = {
+    production: url.match(/(?:https:\/\/)?([^.]+)(?:\.([^.]+))\.xata\.sh.*/),
+    staging: url.match(/(?:https:\/\/)?([^.]+)(?:\.([^.]+))\.staging-xata\.dev.*/),
+    dev: url.match(/(?:https:\/\/)?([^.]+)(?:\.([^.]+))\.dev-xata\.dev.*/),
+    local: url.match(/(?:https?:\/\/)?([^.]+)(?:\.([^.]+))\.localhost:(\d+)/)
+  };
+  const [host, match] = Object.entries(matches).find(([, match2]) => match2 !== null) ?? [];
+  if (!isHostProviderAlias(host) || !match)
+    return null;
+  return { workspace: match[1], region: match[2], host };
+}
+
 const pool = new ApiRequestPool();
 const resolveUrl = (url, queryParams = {}, pathParams = {}) => {
   const cleanQueryParams = Object.entries(queryParams).reduce((acc, [key, value]) => {
@@ -604,6 +647,7 @@ const resolveUrl = (url, queryParams = {}, pathParams = {}) => {
   return url.replace(/\{\w*\}/g, (key) => cleanPathParams[key.slice(1, -1)]) + queryString;
 };
 function buildBaseUrl({
+  method,
   endpoint,
   path,
   workspacesApiUrl,
@@ -611,7 +655,24 @@ function buildBaseUrl({
   pathParams = {}
 }) {
   if (endpoint === "dataPlane") {
-    const url = isString(workspacesApiUrl) ? `${workspacesApiUrl}${path}` : workspacesApiUrl(path, pathParams);
+    let url = isString(workspacesApiUrl) ? `${workspacesApiUrl}${path}` : workspacesApiUrl(path, pathParams);
+    if (method.toUpperCase() === "PUT" && [
+      "/db/{dbBranchName}/tables/{tableName}/data/{recordId}/column/{columnName}/file",
+      "/db/{dbBranchName}/tables/{tableName}/data/{recordId}/column/{columnName}/file/{fileId}"
+    ].includes(path)) {
+      const { host } = parseWorkspacesUrlParts(url) ?? {};
+      switch (host) {
+        case "production":
+          url = url.replace("xata.sh", "upload.xata.sh");
+          break;
+        case "staging":
+          url = url.replace("staging-xata.dev", "upload.staging-xata.dev");
+          break;
+        case "dev":
+          url = url.replace("dev-xata.dev", "upload.dev-xata.dev");
+          break;
+      }
+    }
     const urlWithWorkspace = isString(pathParams.workspace) ? url.replace("{workspaceId}", String(pathParams.workspace)) : url;
     return isString(pathParams.region) ? urlWithWorkspace.replace("{region}", String(pathParams.region)) : urlWithWorkspace;
   }
@@ -660,9 +721,9 @@ async function fetch$1({
   return await trace(
     `${method.toUpperCase()} ${path}`,
     async ({ setAttributes }) => {
-      const baseUrl = buildBaseUrl({ endpoint, path, workspacesApiUrl, pathParams, apiUrl });
+      const baseUrl = buildBaseUrl({ method, endpoint, path, workspacesApiUrl, pathParams, apiUrl });
       const fullUrl = resolveUrl(baseUrl, queryParams, pathParams);
-      const url = fullUrl.includes("localhost") ? fullUrl.replace(/^[^.]+\./, "http://") : fullUrl;
+      const url = fullUrl.includes("localhost") ? fullUrl.replace(/^[^.]+\.[^.]+\./, "http://") : fullUrl;
       setAttributes({
         [TraceAttributes.HTTP_URL]: url,
         [TraceAttributes.HTTP_TARGET]: resolveUrl(path, queryParams, pathParams)
@@ -743,7 +804,7 @@ function fetchSSERequest({
   clientName,
   xataAgentExtra
 }) {
-  const baseUrl = buildBaseUrl({ endpoint, path, workspacesApiUrl, pathParams, apiUrl });
+  const baseUrl = buildBaseUrl({ method, endpoint, path, workspacesApiUrl, pathParams, apiUrl });
   const fullUrl = resolveUrl(baseUrl, queryParams, pathParams);
   const url = fullUrl.includes("localhost") ? fullUrl.replace(/^[^.]+\./, "http://") : fullUrl;
   void fetchEventSource(url, {
@@ -786,6 +847,20 @@ function parseUrl(url) {
 
 const dataPlaneFetch = async (options) => fetch$1({ ...options, endpoint: "dataPlane" });
 
+const applyMigration = (variables, signal) => dataPlaneFetch({ url: "/db/{dbBranchName}/pgroll/apply", method: "post", ...variables, signal });
+const pgRollStatus = (variables, signal) => dataPlaneFetch({
+  url: "/db/{dbBranchName}/pgroll/status",
+  method: "get",
+  ...variables,
+  signal
+});
+const pgRollJobStatus = (variables, signal) => dataPlaneFetch({
+  url: "/db/{dbBranchName}/pgroll/jobs/{jobId}",
+  method: "get",
+  ...variables,
+  signal
+});
+const pgRollMigrationHistory = (variables, signal) => dataPlaneFetch({ url: "/db/{dbBranchName}/pgroll/migrations", method: "get", ...variables, signal });
 const getBranchList = (variables, signal) => dataPlaneFetch({
   url: "/dbs/{dbName}",
   method: "get",
@@ -805,6 +880,12 @@ const deleteBranch = (variables, signal) => dataPlaneFetch({
   ...variables,
   signal
 });
+const getSchema = (variables, signal) => dataPlaneFetch({
+  url: "/db/{dbBranchName}/schema",
+  method: "get",
+  ...variables,
+  signal
+});
 const copyBranch = (variables, signal) => dataPlaneFetch({
   url: "/db/{dbBranchName}/copy",
   method: "post",
@@ -986,6 +1067,12 @@ const fileAccess = (variables, signal) => dataPlaneFetch({
   ...variables,
   signal
 });
+const fileUpload = (variables, signal) => dataPlaneFetch({
+  url: "/file/{fileId}",
+  method: "put",
+  ...variables,
+  signal
+});
 const sqlQuery = (variables, signal) => dataPlaneFetch({
   url: "/db/{dbBranchName}/sql",
   method: "post",
@@ -994,6 +1081,10 @@ const sqlQuery = (variables, signal) => dataPlaneFetch({
 });
 const operationsByTag$2 = {
   branch: {
+    applyMigration,
+    pgRollStatus,
+    pgRollJobStatus,
+    pgRollMigrationHistory,
     getBranchList,
     getBranchDetails,
     createBranch,
@@ -1008,6 +1099,7 @@ const operationsByTag$2 = {
     resolveBranch
   },
   migrations: {
+    getSchema,
     getBranchMigrationHistory,
     getBranchMigrationPlan,
     executeBranchMigrationPlan,
@@ -1051,7 +1143,7 @@ const operationsByTag$2 = {
     deleteRecord,
     bulkInsertTableRecords
   },
-  files: { getFileItem, putFileItem, deleteFileItem, getFile, putFile, deleteFile, fileAccess },
+  files: { getFileItem, putFileItem, deleteFileItem, getFile, putFile, deleteFile, fileAccess, fileUpload },
   searchAndFilter: {
     queryTable,
     searchBranch,
@@ -1173,6 +1265,15 @@ const updateWorkspaceMemberInvite = (variables, signal) => controlPlaneFetch({ u
 const cancelWorkspaceMemberInvite = (variables, signal) => controlPlaneFetch({ url: "/workspaces/{workspaceId}/invites/{inviteId}", method: "delete", ...variables, signal });
 const acceptWorkspaceMemberInvite = (variables, signal) => controlPlaneFetch({ url: "/workspaces/{workspaceId}/invites/{inviteKey}/accept", method: "post", ...variables, signal });
 const resendWorkspaceMemberInvite = (variables, signal) => controlPlaneFetch({ url: "/workspaces/{workspaceId}/invites/{inviteId}/resend", method: "post", ...variables, signal });
+const listClusters = (variables, signal) => controlPlaneFetch({ url: "/workspaces/{workspaceId}/clusters", method: "get", ...variables, signal });
+const createCluster = (variables, signal) => controlPlaneFetch({ url: "/workspaces/{workspaceId}/clusters", method: "post", ...variables, signal });
+const getCluster = (variables, signal) => controlPlaneFetch({
+  url: "/workspaces/{workspaceId}/clusters/{clusterId}",
+  method: "get",
+  ...variables,
+  signal
+});
+const updateCluster = (variables, signal) => controlPlaneFetch({ url: "/workspaces/{workspaceId}/clusters/{clusterId}", method: "patch", ...variables, signal });
 const getDatabaseList = (variables, signal) => controlPlaneFetch({
   url: "/workspaces/{workspaceId}/dbs",
   method: "get",
@@ -1227,6 +1328,7 @@ const operationsByTag$1 = {
     acceptWorkspaceMemberInvite,
     resendWorkspaceMemberInvite
   },
+  xbcontrolOther: { listClusters, createCluster, getCluster, updateCluster },
   databases: {
     getDatabaseList,
     createDatabase,
@@ -1243,61 +1345,6 @@ const operationsByTag$1 = {
 
 const operationsByTag = deepMerge(operationsByTag$2, operationsByTag$1);
 
-function getHostUrl(provider, type) {
-  if (isHostProviderAlias(provider)) {
-    return providers[provider][type];
-  } else if (isHostProviderBuilder(provider)) {
-    return provider[type];
-  }
-  throw new Error("Invalid API provider");
-}
-const providers = {
-  production: {
-    main: "https://api.xata.io",
-    workspaces: "https://{workspaceId}.{region}.xata.sh"
-  },
-  staging: {
-    main: "https://api.staging-xata.dev",
-    workspaces: "https://{workspaceId}.{region}.staging-xata.dev"
-  },
-  dev: {
-    main: "https://api.dev-xata.dev",
-    workspaces: "https://{workspaceId}.{region}.dev-xata.dev"
-  }
-};
-function isHostProviderAlias(alias) {
-  return isString(alias) && Object.keys(providers).includes(alias);
-}
-function isHostProviderBuilder(builder) {
-  return isObject(builder) && isString(builder.main) && isString(builder.workspaces);
-}
-function parseProviderString(provider = "production") {
-  if (isHostProviderAlias(provider)) {
-    return provider;
-  }
-  const [main, workspaces] = provider.split(",");
-  if (!main || !workspaces)
-    return null;
-  return { main, workspaces };
-}
-function buildProviderString(provider) {
-  if (isHostProviderAlias(provider))
-    return provider;
-  return `${provider.main},${provider.workspaces}`;
-}
-function parseWorkspacesUrlParts(url) {
-  if (!isString(url))
-    return null;
-  const regex = /(?:https:\/\/)?([^.]+)(?:\.([^.]+))\.xata\.sh.*/;
-  const regexDev = /(?:https:\/\/)?([^.]+)(?:\.([^.]+))\.dev-xata\.dev.*/;
-  const regexStaging = /(?:https:\/\/)?([^.]+)(?:\.([^.]+))\.staging-xata\.dev.*/;
-  const regexProdTesting = /(?:https:\/\/)?([^.]+)(?:\.([^.]+))\.xata\.tech.*/;
-  const match = url.match(regex) || url.match(regexDev) || url.match(regexStaging) || url.match(regexProdTesting);
-  if (!match)
-    return null;
-  return { workspace: match[1], region: match[2] };
-}
-
 var __accessCheck$7 = (obj, member, msg) => {
   if (!member.has(obj))
     throw TypeError("Cannot " + msg);
@@ -2622,75 +2669,6 @@ class XataApiPlugin {
 class XataPlugin {
 }
 
-class FilesPlugin extends XataPlugin {
-  build(pluginOptions) {
-    return {
-      download: async (location) => {
-        const { table, record, column, fileId = "" } = location ?? {};
-        return await getFileItem({
-          pathParams: {
-            workspace: "{workspaceId}",
-            dbBranchName: "{dbBranch}",
-            region: "{region}",
-            tableName: table ?? "",
-            recordId: record ?? "",
-            columnName: column ?? "",
-            fileId
-          },
-          ...pluginOptions,
-          rawResponse: true
-        });
-      },
-      upload: async (location, file) => {
-        const { table, record, column, fileId = "" } = location ?? {};
-        const contentType = getContentType(file);
-        return await putFileItem({
-          ...pluginOptions,
-          pathParams: {
-            workspace: "{workspaceId}",
-            dbBranchName: "{dbBranch}",
-            region: "{region}",
-            tableName: table ?? "",
-            recordId: record ?? "",
-            columnName: column ?? "",
-            fileId
-          },
-          body: file,
-          headers: { "Content-Type": contentType }
-        });
-      },
-      delete: async (location) => {
-        const { table, record, column, fileId = "" } = location ?? {};
-        return await deleteFileItem({
-          pathParams: {
-            workspace: "{workspaceId}",
-            dbBranchName: "{dbBranch}",
-            region: "{region}",
-            tableName: table ?? "",
-            recordId: record ?? "",
-            columnName: column ?? "",
-            fileId
-          },
-          ...pluginOptions
-        });
-      }
-    };
-  }
-}
-function getContentType(file) {
-  if (typeof file === "string") {
-    return "text/plain";
-  }
-  if (isBlob(file)) {
-    return file.type;
-  }
-  try {
-    return file.type;
-  } catch (e) {
-  }
-  return "application/octet-stream";
-}
-
 function buildTransformString(transformations) {
   return transformations.flatMap(
     (t) => Object.entries(t).map(([key, value]) => {
@@ -2719,69 +2697,21 @@ function transformImage(url, ...transformations) {
   return `https://${hostname}${transform}${path}${search}`;
 }
 
-var __defProp$6 = Object.defineProperty;
-var __defNormalProp$6 = (obj, key, value) => key in obj ? __defProp$6(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __publicField$6 = (obj, key, value) => {
-  __defNormalProp$6(obj, typeof key !== "symbol" ? key + "" : key, value);
-  return value;
-};
 class XataFile {
   constructor(file) {
-    /**
-     * Identifier of the file.
-     */
-    __publicField$6(this, "id");
-    /**
-     * Name of the file.
-     */
-    __publicField$6(this, "name");
-    /**
-     * Media type of the file.
-     */
-    __publicField$6(this, "mediaType");
-    /**
-     * Base64 encoded content of the file.
-     */
-    __publicField$6(this, "base64Content");
-    /**
-     * Whether to enable public url for the file.
-     */
-    __publicField$6(this, "enablePublicUrl");
-    /**
-     * Timeout for the signed url.
-     */
-    __publicField$6(this, "signedUrlTimeout");
-    /**
-     * Size of the file.
-     */
-    __publicField$6(this, "size");
-    /**
-     * Version of the file.
-     */
-    __publicField$6(this, "version");
-    /**
-     * Url of the file.
-     */
-    __publicField$6(this, "url");
-    /**
-     * Signed url of the file.
-     */
-    __publicField$6(this, "signedUrl");
-    /**
-     * Attributes of the file.
-     */
-    __publicField$6(this, "attributes");
     this.id = file.id;
-    this.name = file.name || "";
-    this.mediaType = file.mediaType || "application/octet-stream";
+    this.name = file.name;
+    this.mediaType = file.mediaType;
     this.base64Content = file.base64Content;
-    this.enablePublicUrl = file.enablePublicUrl ?? false;
-    this.signedUrlTimeout = file.signedUrlTimeout ?? 300;
-    this.size = file.size ?? 0;
-    this.version = file.version ?? 1;
-    this.url = file.url || "";
+    this.enablePublicUrl = file.enablePublicUrl;
+    this.signedUrlTimeout = file.signedUrlTimeout;
+    this.uploadUrlTimeout = file.uploadUrlTimeout;
+    this.size = file.size;
+    this.version = file.version;
+    this.url = file.url;
     this.signedUrl = file.signedUrl;
-    this.attributes = file.attributes || {};
+    this.uploadUrl = file.uploadUrl;
+    this.attributes = file.attributes;
   }
   static fromBuffer(buffer, options = {}) {
     const base64Content = buffer.toString("base64");
@@ -2871,7 +2801,7 @@ class XataFile {
 const parseInputFileEntry = async (entry) => {
   if (!isDefined(entry))
     return null;
-  const { id, name, mediaType, base64Content, enablePublicUrl, signedUrlTimeout } = await entry;
+  const { id, name, mediaType, base64Content, enablePublicUrl, signedUrlTimeout, uploadUrlTimeout } = await entry;
   return compactObject({
     id,
     // Name cannot be an empty string in our API
@@ -2879,7 +2809,8 @@ const parseInputFileEntry = async (entry) => {
     mediaType,
     base64Content,
     enablePublicUrl,
-    signedUrlTimeout
+    signedUrlTimeout,
+    uploadUrlTimeout
   });
 };
 
@@ -2929,12 +2860,6 @@ function parseJson(value) {
   }
 }
 
-var __defProp$5 = Object.defineProperty;
-var __defNormalProp$5 = (obj, key, value) => key in obj ? __defProp$5(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __publicField$5 = (obj, key, value) => {
-  __defNormalProp$5(obj, typeof key !== "symbol" ? key + "" : key, value);
-  return value;
-};
 var __accessCheck$6 = (obj, member, msg) => {
   if (!member.has(obj))
     throw TypeError("Cannot " + msg);
@@ -2957,14 +2882,6 @@ var _query, _page;
 class Page {
   constructor(query, meta, records = []) {
     __privateAdd$6(this, _query, void 0);
-    /**
-     * Page metadata, required to retrieve additional records.
-     */
-    __publicField$5(this, "meta");
-    /**
-     * The set of results for this page.
-     */
-    __publicField$5(this, "records");
     __privateSet$6(this, _query, query);
     this.meta = meta;
     this.records = new RecordArray(this, records);
@@ -3014,9 +2931,9 @@ class Page {
   }
 }
 _query = new WeakMap();
-const PAGINATION_MAX_SIZE = 200;
+const PAGINATION_MAX_SIZE = 1e3;
 const PAGINATION_DEFAULT_SIZE = 20;
-const PAGINATION_MAX_OFFSET = 800;
+const PAGINATION_MAX_OFFSET = 49e3;
 const PAGINATION_DEFAULT_OFFSET = 0;
 function isCursorPaginationOptions(options) {
   return isDefined(options) && (isDefined(options.start) || isDefined(options.end) || isDefined(options.after) || isDefined(options.before));
@@ -3095,12 +3012,6 @@ const _RecordArray = class _RecordArray extends Array {
 _page = new WeakMap();
 let RecordArray = _RecordArray;
 
-var __defProp$4 = Object.defineProperty;
-var __defNormalProp$4 = (obj, key, value) => key in obj ? __defProp$4(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __publicField$4 = (obj, key, value) => {
-  __defNormalProp$4(obj, typeof key !== "symbol" ? key + "" : key, value);
-  return value;
-};
 var __accessCheck$5 = (obj, member, msg) => {
   if (!member.has(obj))
     throw TypeError("Cannot " + msg);
@@ -3131,8 +3042,8 @@ const _Query = class _Query {
     __privateAdd$5(this, _repository, void 0);
     __privateAdd$5(this, _data, { filter: {} });
     // Implements pagination
-    __publicField$4(this, "meta", { page: { cursor: "start", more: true, size: PAGINATION_DEFAULT_SIZE } });
-    __publicField$4(this, "records", new RecordArray(this, []));
+    this.meta = { page: { cursor: "start", more: true, size: PAGINATION_DEFAULT_SIZE } };
+    this.records = new RecordArray(this, []);
     __privateSet$5(this, _table$1, table);
     if (repository) {
       __privateSet$5(this, _repository, repository);
@@ -3383,7 +3294,6 @@ const RecordColumnTypes = [
   "email",
   "multiple",
   "link",
-  "object",
   "datetime",
   "vector",
   "file[]",
@@ -3770,7 +3680,7 @@ class RestRepository extends Query {
   }
   async search(query, options = {}) {
     return __privateGet$4(this, _trace).call(this, "search", async () => {
-      const { records } = await searchTable({
+      const { records, totalCount } = await searchTable({
         pathParams: {
           workspace: "{workspaceId}",
           dbBranchName: "{dbBranch}",
@@ -3790,12 +3700,15 @@ class RestRepository extends Query {
         ...__privateGet$4(this, _getFetchProps).call(this)
       });
       const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
-      return records.map((item) => initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), item, ["*"]));
+      return {
+        records: records.map((item) => initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), item, ["*"])),
+        totalCount
+      };
     });
   }
   async vectorSearch(column, query, options) {
     return __privateGet$4(this, _trace).call(this, "vectorSearch", async () => {
-      const { records } = await vectorSearchTable({
+      const { records, totalCount } = await vectorSearchTable({
         pathParams: {
           workspace: "{workspaceId}",
           dbBranchName: "{dbBranch}",
@@ -3812,7 +3725,10 @@ class RestRepository extends Query {
         ...__privateGet$4(this, _getFetchProps).call(this)
       });
       const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
-      return records.map((item) => initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), item, ["*"]));
+      return {
+        records: records.map((item) => initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), item, ["*"])),
+        totalCount
+      };
     });
   }
   async aggregate(aggs, filter) {
@@ -3888,7 +3804,13 @@ class RestRepository extends Query {
         },
         ...__privateGet$4(this, _getFetchProps).call(this)
       });
-      return result;
+      const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
+      return {
+        ...result,
+        summaries: result.summaries.map(
+          (summary) => initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), summary, data.columns ?? [])
+        )
+      };
     });
   }
   ask(question, options) {
@@ -4176,13 +4098,6 @@ transformObjectToApi_fn = async function(object) {
   }
   return result;
 };
-const removeLinksFromObject = (object) => {
-  return Object.entries(object).reduce((acc, [key, value]) => {
-    if (key === "xata")
-      return acc;
-    return { ...acc, [key]: isIdentifiable(value) ? value.id : value };
-  }, {});
-};
 const initObject = (db, schemaTables, table, object, selectedColumns) => {
   const data = {};
   const { xata, ...rest } = object ?? {};
@@ -4249,7 +4164,6 @@ const initObject = (db, schemaTables, table, object, selectedColumns) => {
     }
   }
   const record = { ...data };
-  const serializable = { xata, ...removeLinksFromObject(data) };
   const metadata = xata !== void 0 ? { ...xata, createdAt: new Date(xata.createdAt), updatedAt: new Date(xata.updatedAt) } : void 0;
   record.read = function(columns2) {
     return db[table].read(record["id"], columns2);
@@ -4267,15 +4181,17 @@ const initObject = (db, schemaTables, table, object, selectedColumns) => {
   record.delete = function() {
     return db[table].delete(record["id"]);
   };
-  record.xata = Object.freeze(metadata);
+  if (metadata !== void 0) {
+    record.xata = Object.freeze(metadata);
+  }
   record.getMetadata = function() {
     return record.xata;
   };
   record.toSerializable = function() {
-    return JSON.parse(JSON.stringify(serializable));
+    return JSON.parse(JSON.stringify(record));
   };
   record.toString = function() {
-    return JSON.stringify(serializable);
+    return JSON.stringify(record);
   };
   for (const prop of ["read", "update", "replace", "delete", "getMetadata", "toSerializable", "toString"]) {
     Object.defineProperty(record, prop, { enumerable: false });
@@ -4304,12 +4220,6 @@ function parseIfVersion(...args) {
   return void 0;
 }
 
-var __defProp$3 = Object.defineProperty;
-var __defNormalProp$3 = (obj, key, value) => key in obj ? __defProp$3(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __publicField$3 = (obj, key, value) => {
-  __defNormalProp$3(obj, typeof key !== "symbol" ? key + "" : key, value);
-  return value;
-};
 var __accessCheck$3 = (obj, member, msg) => {
   if (!member.has(obj))
     throw TypeError("Cannot " + msg);
@@ -4332,8 +4242,6 @@ var _map;
 class SimpleCache {
   constructor(options = {}) {
     __privateAdd$3(this, _map, void 0);
-    __publicField$3(this, "capacity");
-    __publicField$3(this, "defaultQueryTTL");
     __privateSet$3(this, _map, /* @__PURE__ */ new Map());
     this.capacity = options.max ?? 500;
     this.defaultQueryTTL = options.defaultQueryTTL ?? 60 * 1e3;
@@ -4378,10 +4286,12 @@ const notExists = (column) => ({ $notExists: column });
 const startsWith = (value) => ({ $startsWith: value });
 const endsWith = (value) => ({ $endsWith: value });
 const pattern = (value) => ({ $pattern: value });
+const iPattern = (value) => ({ $iPattern: value });
 const is = (value) => ({ $is: value });
 const equals = is;
 const isNot = (value) => ({ $isNot: value });
 const contains = (value) => ({ $contains: value });
+const iContains = (value) => ({ $iContains: value });
 const includes = (value) => ({ $includes: value });
 const includesAll = (value) => ({ $includesAll: value });
 const includesNone = (value) => ({ $includesNone: value });
@@ -4437,6 +4347,80 @@ class SchemaPlugin extends XataPlugin {
 _tables = new WeakMap();
 _schemaTables$1 = new WeakMap();
 
+class FilesPlugin extends XataPlugin {
+  build(pluginOptions) {
+    return {
+      download: async (location) => {
+        const { table, record, column, fileId = "" } = location ?? {};
+        return await getFileItem({
+          pathParams: {
+            workspace: "{workspaceId}",
+            dbBranchName: "{dbBranch}",
+            region: "{region}",
+            tableName: table ?? "",
+            recordId: record ?? "",
+            columnName: column ?? "",
+            fileId
+          },
+          ...pluginOptions,
+          rawResponse: true
+        });
+      },
+      upload: async (location, file, options) => {
+        const { table, record, column, fileId = "" } = location ?? {};
+        const resolvedFile = await file;
+        const contentType = options?.mediaType || getContentType(resolvedFile);
+        const body = resolvedFile instanceof XataFile ? resolvedFile.toBlob() : resolvedFile;
+        return await putFileItem({
+          ...pluginOptions,
+          pathParams: {
+            workspace: "{workspaceId}",
+            dbBranchName: "{dbBranch}",
+            region: "{region}",
+            tableName: table ?? "",
+            recordId: record ?? "",
+            columnName: column ?? "",
+            fileId
+          },
+          body,
+          headers: { "Content-Type": contentType }
+        });
+      },
+      delete: async (location) => {
+        const { table, record, column, fileId = "" } = location ?? {};
+        return await deleteFileItem({
+          pathParams: {
+            workspace: "{workspaceId}",
+            dbBranchName: "{dbBranch}",
+            region: "{region}",
+            tableName: table ?? "",
+            recordId: record ?? "",
+            columnName: column ?? "",
+            fileId
+          },
+          ...pluginOptions
+        });
+      }
+    };
+  }
+}
+function getContentType(file) {
+  if (typeof file === "string") {
+    return "text/plain";
+  }
+  if ("mediaType" in file && file.mediaType !== void 0) {
+    return file.mediaType;
+  }
+  if (isBlob(file)) {
+    return file.type;
+  }
+  try {
+    return file.type;
+  } catch (e) {
+  }
+  return "application/octet-stream";
+}
+
 var __accessCheck$1 = (obj, member, msg) => {
   if (!member.has(obj))
     throw TypeError("Cannot " + msg);
@@ -4472,22 +4456,26 @@ class SearchPlugin extends XataPlugin {
   build(pluginOptions) {
     return {
       all: async (query, options = {}) => {
-        const records = await __privateMethod$1(this, _search, search_fn).call(this, query, options, pluginOptions);
+        const { records, totalCount } = await __privateMethod$1(this, _search, search_fn).call(this, query, options, pluginOptions);
         const schemaTables = await __privateMethod$1(this, _getSchemaTables, getSchemaTables_fn).call(this, pluginOptions);
-        return records.map((record) => {
-          const { table = "orphan" } = record.xata;
-          return { table, record: initObject(this.db, schemaTables, table, record, ["*"]) };
-        });
+        return {
+          totalCount,
+          records: records.map((record) => {
+            const { table = "orphan" } = record.xata;
+            return { table, record: initObject(this.db, schemaTables, table, record, ["*"]) };
+          })
+        };
       },
       byTable: async (query, options = {}) => {
-        const records = await __privateMethod$1(this, _search, search_fn).call(this, query, options, pluginOptions);
+        const { records: rawRecords, totalCount } = await __privateMethod$1(this, _search, search_fn).call(this, query, options, pluginOptions);
         const schemaTables = await __privateMethod$1(this, _getSchemaTables, getSchemaTables_fn).call(this, pluginOptions);
-        return records.reduce((acc, record) => {
+        const records = rawRecords.reduce((acc, record) => {
           const { table = "orphan" } = record.xata;
           const items = acc[table] ?? [];
           const item = initObject(this.db, schemaTables, table, record, ["*"]);
           return { ...acc, [table]: [...items, item] };
         }, {});
+        return { totalCount, records };
       }
     };
   }
@@ -4496,13 +4484,13 @@ _schemaTables = new WeakMap();
 _search = new WeakSet();
 search_fn = async function(query, options, pluginOptions) {
   const { tables, fuzziness, highlight, prefix, page } = options ?? {};
-  const { records } = await searchBranch({
+  const { records, totalCount } = await searchBranch({
     pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", region: "{region}" },
     // @ts-ignore https://github.com/xataio/client-ts/issues/313
     body: { tables, query, fuzziness, prefix, highlight, page },
     ...pluginOptions
   });
-  return records;
+  return { records, totalCount };
 };
 _getSchemaTables = new WeakSet();
 getSchemaTables_fn = async function(pluginOptions) {
@@ -4576,17 +4564,33 @@ function prepareParams(param1, param2) {
 
 class SQLPlugin extends XataPlugin {
   build(pluginOptions) {
-    return async (param1, ...param2) => {
-      const { statement, params, consistency } = prepareParams(param1, param2);
-      const { records, warning } = await sqlQuery({
+    const query = async (query2, ...parameters) => {
+      const { statement, params, consistency } = prepareParams(query2, parameters);
+      const { records, warning, columns } = await sqlQuery({
         pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", region: "{region}" },
         body: { statement, params, consistency },
         ...pluginOptions
       });
-      return { records, warning };
+      return { records, warning, columns };
     };
+    const result = async (param1, ...param2) => {
+      if (!isParamsObject(param1) && (!isTemplateStringsArray(param1) || !Array.isArray(param2))) {
+        throw new Error(
+          "Calling `xata.sql` as a function is not safe. Make sure to use it as a tagged template or with an object."
+        );
+      }
+      return await query(param1, ...param2);
+    };
+    result.rawUnsafeQuery = query;
+    return result;
   }
 }
+function isTemplateStringsArray(strings) {
+  return Array.isArray(strings) && "raw" in strings && Array.isArray(strings.raw);
+}
+function isParamsObject(params) {
+  return isObject(params) && "statement" in params;
+}
 
 class TransactionPlugin extends XataPlugin {
   build(pluginOptions) {
@@ -4603,12 +4607,6 @@ class TransactionPlugin extends XataPlugin {
   }
 }
 
-var __defProp$2 = Object.defineProperty;
-var __defNormalProp$2 = (obj, key, value) => key in obj ? __defProp$2(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __publicField$2 = (obj, key, value) => {
-  __defNormalProp$2(obj, typeof key !== "symbol" ? key + "" : key, value);
-  return value;
-};
 var __accessCheck = (obj, member, msg) => {
   if (!member.has(obj))
     throw TypeError("Cannot " + msg);
@@ -4638,11 +4636,6 @@ const buildClient = (plugins) => {
       __privateAdd(this, _parseOptions);
       __privateAdd(this, _getFetchProps);
       __privateAdd(this, _options, void 0);
-      __publicField$2(this, "db");
-      __publicField$2(this, "search");
-      __publicField$2(this, "transactions");
-      __publicField$2(this, "sql");
-      __publicField$2(this, "files");
       const safeOptions = __privateMethod(this, _parseOptions, parseOptions_fn).call(this, options);
       __privateSet(this, _options, safeOptions);
       const pluginOptions = {
@@ -4756,17 +4749,11 @@ const buildClient = (plugins) => {
 class BaseClient extends buildClient() {
 }
 
-var __defProp$1 = Object.defineProperty;
-var __defNormalProp$1 = (obj, key, value) => key in obj ? __defProp$1(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __publicField$1 = (obj, key, value) => {
-  __defNormalProp$1(obj, typeof key !== "symbol" ? key + "" : key, value);
-  return value;
-};
 const META = "__";
 const VALUE = "___";
 class Serializer {
   constructor() {
-    __publicField$1(this, "classes", {});
+    this.classes = {};
   }
   add(clazz) {
     this.classes[clazz.name] = clazz;
@@ -4829,31 +4816,9 @@ const deserialize = (json) => {
   return defaultSerializer.fromJSON(json);
 };
 
-function buildWorkerRunner(config) {
-  return function xataWorker(name, worker) {
-    return async (...args) => {
-      const url = process.env.NODE_ENV === "development" ? `http://localhost:64749/${name}` : `https://dispatcher.xata.workers.dev/${config.workspace}/${config.worker}/${name}`;
-      const result = await fetch(url, {
-        method: "POST",
-        headers: { "Content-Type": "application/json" },
-        body: serialize({ args })
-      });
-      const text = await result.text();
-      return deserialize(text);
-    };
-  };
-}
-
-var __defProp = Object.defineProperty;
-var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __publicField = (obj, key, value) => {
-  __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
-  return value;
-};
 class XataError extends Error {
   constructor(message, status) {
     super(message);
-    __publicField(this, "status");
     this.status = status;
   }
 }
@@ -4877,6 +4842,7 @@ exports.SchemaPlugin = SchemaPlugin;
 exports.SearchPlugin = SearchPlugin;
 exports.Serializer = Serializer;
 exports.SimpleCache = SimpleCache;
+exports.TransactionPlugin = TransactionPlugin;
 exports.XataApiClient = XataApiClient;
 exports.XataApiPlugin = XataApiPlugin;
 exports.XataError = XataError;
@@ -4887,13 +4853,13 @@ exports.addGitBranchesEntry = addGitBranchesEntry;
 exports.addTableColumn = addTableColumn;
 exports.aggregateTable = aggregateTable;
 exports.applyBranchSchemaEdit = applyBranchSchemaEdit;
+exports.applyMigration = applyMigration;
 exports.askTable = askTable;
 exports.askTableSession = askTableSession;
 exports.branchTransaction = branchTransaction;
 exports.buildClient = buildClient;
 exports.buildPreviewBranchName = buildPreviewBranchName;
 exports.buildProviderString = buildProviderString;
-exports.buildWorkerRunner = buildWorkerRunner;
 exports.bulkInsertTableRecords = bulkInsertTableRecords;
 exports.cancelWorkspaceMemberInvite = cancelWorkspaceMemberInvite;
 exports.compareBranchSchemas = compareBranchSchemas;
@@ -4902,6 +4868,7 @@ exports.compareMigrationRequest = compareMigrationRequest;
 exports.contains = contains;
 exports.copyBranch = copyBranch;
 exports.createBranch = createBranch;
+exports.createCluster = createCluster;
 exports.createDatabase = createDatabase;
 exports.createMigrationRequest = createMigrationRequest;
 exports.createTable = createTable;
@@ -4926,6 +4893,7 @@ exports.equals = equals;
 exports.executeBranchMigrationPlan = executeBranchMigrationPlan;
 exports.exists = exists;
 exports.fileAccess = fileAccess;
+exports.fileUpload = fileUpload;
 exports.ge = ge;
 exports.getAPIKey = getAPIKey;
 exports.getAuthorizationCode = getAuthorizationCode;
@@ -4937,6 +4905,7 @@ exports.getBranchMigrationHistory = getBranchMigrationHistory;
 exports.getBranchMigrationPlan = getBranchMigrationPlan;
 exports.getBranchSchemaHistory = getBranchSchemaHistory;
 exports.getBranchStats = getBranchStats;
+exports.getCluster = getCluster;
 exports.getColumn = getColumn;
 exports.getDatabaseGithubSettings = getDatabaseGithubSettings;
 exports.getDatabaseList = getDatabaseList;
@@ -4950,6 +4919,7 @@ exports.getMigrationRequest = getMigrationRequest;
 exports.getMigrationRequestIsMerged = getMigrationRequestIsMerged;
 exports.getPreviewBranch = getPreviewBranch;
 exports.getRecord = getRecord;
+exports.getSchema = getSchema;
 exports.getTableColumns = getTableColumns;
 exports.getTableSchema = getTableSchema;
 exports.getUser = getUser;
@@ -4965,6 +4935,8 @@ exports.greaterThan = greaterThan;
 exports.greaterThanEquals = greaterThanEquals;
 exports.gt = gt;
 exports.gte = gte;
+exports.iContains = iContains;
+exports.iPattern = iPattern;
 exports.includes = includes;
 exports.includesAll = includesAll;
 exports.includesAny = includesAny;
@@ -4985,6 +4957,7 @@ exports.le = le;
 exports.lessEquals = lessEquals;
 exports.lessThan = lessThan;
 exports.lessThanEquals = lessThanEquals;
+exports.listClusters = listClusters;
 exports.listMigrationRequestsCommits = listMigrationRequestsCommits;
 exports.listRegions = listRegions;
 exports.lt = lt;
@@ -4995,6 +4968,9 @@ exports.operationsByTag = operationsByTag;
 exports.parseProviderString = parseProviderString;
 exports.parseWorkspacesUrlParts = parseWorkspacesUrlParts;
 exports.pattern = pattern;
+exports.pgRollJobStatus = pgRollJobStatus;
+exports.pgRollMigrationHistory = pgRollMigrationHistory;
+exports.pgRollStatus = pgRollStatus;
 exports.previewBranchSchemaEdit = previewBranchSchemaEdit;
 exports.pushBranchMigrations = pushBranchMigrations;
 exports.putFile = putFile;
@@ -5016,6 +4992,7 @@ exports.summarizeTable = summarizeTable;
 exports.transformImage = transformImage;
 exports.updateBranchMetadata = updateBranchMetadata;
 exports.updateBranchSchema = updateBranchSchema;
+exports.updateCluster = updateCluster;
 exports.updateColumn = updateColumn;
 exports.updateDatabaseGithubSettings = updateDatabaseGithubSettings;
 exports.updateDatabaseMetadata = updateDatabaseMetadata;