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

package/dist/index.mjs

@@ -242,12 +242,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);
@@ -287,8 +281,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;
@@ -534,26 +526,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;
@@ -587,6 +569,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]) => {
@@ -602,6 +645,7 @@ const resolveUrl = (url, queryParams = {}, pathParams = {}) => {
   return url.replace(/\{\w*\}/g, (key) => cleanPathParams[key.slice(1, -1)]) + queryString;
 };
 function buildBaseUrl({
+  method,
   endpoint,
   path,
   workspacesApiUrl,
@@ -609,7 +653,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;
   }
@@ -658,9 +719,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)
@@ -741,7 +802,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, {
@@ -784,6 +845,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",
@@ -803,6 +878,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",
@@ -984,6 +1065,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",
@@ -992,6 +1079,10 @@ const sqlQuery = (variables, signal) => dataPlaneFetch({
 });
 const operationsByTag$2 = {
   branch: {
+    applyMigration,
+    pgRollStatus,
+    pgRollJobStatus,
+    pgRollMigrationHistory,
     getBranchList,
     getBranchDetails,
     createBranch,
@@ -1006,6 +1097,7 @@ const operationsByTag$2 = {
     resolveBranch
   },
   migrations: {
+    getSchema,
     getBranchMigrationHistory,
     getBranchMigrationPlan,
     executeBranchMigrationPlan,
@@ -1049,7 +1141,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,
@@ -1171,6 +1263,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",
@@ -1225,6 +1326,7 @@ const operationsByTag$1 = {
     acceptWorkspaceMemberInvite,
     resendWorkspaceMemberInvite
   },
+  xbcontrolOther: { listClusters, createCluster, getCluster, updateCluster },
   databases: {
     getDatabaseList,
     createDatabase,
@@ -1241,61 +1343,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);
@@ -2620,75 +2667,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]) => {
@@ -2717,69 +2695,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");
@@ -2869,7 +2799,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
@@ -2877,7 +2807,8 @@ const parseInputFileEntry = async (entry) => {
     mediaType,
     base64Content,
     enablePublicUrl,
-    signedUrlTimeout
+    signedUrlTimeout,
+    uploadUrlTimeout
   });
 };
 
@@ -2927,12 +2858,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);
@@ -2955,14 +2880,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);
@@ -3012,9 +2929,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));
@@ -3093,12 +3010,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);
@@ -3129,8 +3040,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);
@@ -3381,7 +3292,6 @@ const RecordColumnTypes = [
   "email",
   "multiple",
   "link",
-  "object",
   "datetime",
   "vector",
   "file[]",
@@ -3768,7 +3678,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}",
@@ -3788,12 +3698,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}",
@@ -3810,7 +3723,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) {
@@ -3886,7 +3802,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) {
@@ -4174,13 +4096,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 ?? {};
@@ -4247,7 +4162,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);
@@ -4265,15 +4179,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 });
@@ -4302,12 +4218,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);
@@ -4330,8 +4240,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;
@@ -4376,10 +4284,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 });
@@ -4435,6 +4345,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);
@@ -4470,22 +4454,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 };
       }
     };
   }
@@ -4494,13 +4482,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) {
@@ -4574,17 +4562,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) {
@@ -4601,12 +4605,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);
@@ -4636,11 +4634,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 = {
@@ -4754,17 +4747,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;
@@ -4827,34 +4814,12 @@ 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;
   }
 }
 
-export { BaseClient, FetcherError, FilesPlugin, operationsByTag as Operations, PAGINATION_DEFAULT_OFFSET, PAGINATION_DEFAULT_SIZE, PAGINATION_MAX_OFFSET, PAGINATION_MAX_SIZE, Page, Query, RecordArray, RecordColumnTypes, Repository, RestRepository, SQLPlugin, SchemaPlugin, SearchPlugin, Serializer, SimpleCache, XataApiClient, XataApiPlugin, XataError, XataFile, XataPlugin, acceptWorkspaceMemberInvite, addGitBranchesEntry, addTableColumn, aggregateTable, applyBranchSchemaEdit, askTable, askTableSession, branchTransaction, buildClient, buildPreviewBranchName, buildProviderString, buildWorkerRunner, bulkInsertTableRecords, cancelWorkspaceMemberInvite, compareBranchSchemas, compareBranchWithUserSchema, compareMigrationRequest, contains, copyBranch, createBranch, createDatabase, createMigrationRequest, createTable, createUserAPIKey, createWorkspace, deleteBranch, deleteColumn, deleteDatabase, deleteDatabaseGithubSettings, deleteFile, deleteFileItem, deleteOAuthAccessToken, deleteRecord, deleteTable, deleteUser, deleteUserAPIKey, deleteUserOAuthClient, deleteWorkspace, deserialize, endsWith, equals, executeBranchMigrationPlan, exists, fileAccess, ge, getAPIKey, getAuthorizationCode, getBranch, getBranchDetails, getBranchList, getBranchMetadata, getBranchMigrationHistory, getBranchMigrationPlan, getBranchSchemaHistory, getBranchStats, getColumn, getDatabaseGithubSettings, getDatabaseList, getDatabaseMetadata, getDatabaseURL, getFile, getFileItem, getGitBranchesMapping, getHostUrl, getMigrationRequest, getMigrationRequestIsMerged, getPreviewBranch, getRecord, getTableColumns, getTableSchema, getUser, getUserAPIKeys, getUserOAuthAccessTokens, getUserOAuthClients, getWorkspace, getWorkspaceMembersList, getWorkspacesList, grantAuthorizationCode, greaterEquals, greaterThan, greaterThanEquals, gt, gte, includes, includesAll, includesAny, includesNone, insertRecord, insertRecordWithID, inviteWorkspaceMember, is, isCursorPaginationOptions, isHostProviderAlias, isHostProviderBuilder, isIdentifiable, isNot, isValidExpandedColumn, isValidSelectableColumns, isXataRecord, le, lessEquals, lessThan, lessThanEquals, listMigrationRequestsCommits, listRegions, lt, lte, mergeMigrationRequest, notExists, operationsByTag, parseProviderString, parseWorkspacesUrlParts, pattern, previewBranchSchemaEdit, pushBranchMigrations, putFile, putFileItem, queryMigrationRequests, queryTable, removeGitBranchesEntry, removeWorkspaceMember, renameDatabase, resendWorkspaceMemberInvite, resolveBranch, searchBranch, searchTable, serialize, setTableSchema, sqlQuery, startsWith, summarizeTable, transformImage, updateBranchMetadata, updateBranchSchema, updateColumn, updateDatabaseGithubSettings, updateDatabaseMetadata, updateMigrationRequest, updateOAuthAccessToken, updateRecordWithID, updateTable, updateUser, updateWorkspace, updateWorkspaceMemberInvite, updateWorkspaceMemberRole, upsertRecordWithID, vectorSearchTable };
+export { BaseClient, FetcherError, FilesPlugin, operationsByTag as Operations, PAGINATION_DEFAULT_OFFSET, PAGINATION_DEFAULT_SIZE, PAGINATION_MAX_OFFSET, PAGINATION_MAX_SIZE, Page, Query, RecordArray, RecordColumnTypes, Repository, RestRepository, SQLPlugin, SchemaPlugin, SearchPlugin, Serializer, SimpleCache, TransactionPlugin, XataApiClient, XataApiPlugin, XataError, XataFile, XataPlugin, acceptWorkspaceMemberInvite, addGitBranchesEntry, addTableColumn, aggregateTable, applyBranchSchemaEdit, applyMigration, askTable, askTableSession, branchTransaction, buildClient, buildPreviewBranchName, buildProviderString, bulkInsertTableRecords, cancelWorkspaceMemberInvite, compareBranchSchemas, compareBranchWithUserSchema, compareMigrationRequest, contains, copyBranch, createBranch, createCluster, createDatabase, createMigrationRequest, createTable, createUserAPIKey, createWorkspace, deleteBranch, deleteColumn, deleteDatabase, deleteDatabaseGithubSettings, deleteFile, deleteFileItem, deleteOAuthAccessToken, deleteRecord, deleteTable, deleteUser, deleteUserAPIKey, deleteUserOAuthClient, deleteWorkspace, deserialize, endsWith, equals, executeBranchMigrationPlan, exists, fileAccess, fileUpload, ge, getAPIKey, getAuthorizationCode, getBranch, getBranchDetails, getBranchList, getBranchMetadata, getBranchMigrationHistory, getBranchMigrationPlan, getBranchSchemaHistory, getBranchStats, getCluster, getColumn, getDatabaseGithubSettings, getDatabaseList, getDatabaseMetadata, getDatabaseURL, getFile, getFileItem, getGitBranchesMapping, getHostUrl, getMigrationRequest, getMigrationRequestIsMerged, getPreviewBranch, getRecord, getSchema, getTableColumns, getTableSchema, getUser, getUserAPIKeys, getUserOAuthAccessTokens, getUserOAuthClients, getWorkspace, getWorkspaceMembersList, getWorkspacesList, grantAuthorizationCode, greaterEquals, greaterThan, greaterThanEquals, gt, gte, iContains, iPattern, includes, includesAll, includesAny, includesNone, insertRecord, insertRecordWithID, inviteWorkspaceMember, is, isCursorPaginationOptions, isHostProviderAlias, isHostProviderBuilder, isIdentifiable, isNot, isValidExpandedColumn, isValidSelectableColumns, isXataRecord, le, lessEquals, lessThan, lessThanEquals, listClusters, listMigrationRequestsCommits, listRegions, lt, lte, mergeMigrationRequest, notExists, operationsByTag, parseProviderString, parseWorkspacesUrlParts, pattern, pgRollJobStatus, pgRollMigrationHistory, pgRollStatus, previewBranchSchemaEdit, pushBranchMigrations, putFile, putFileItem, queryMigrationRequests, queryTable, removeGitBranchesEntry, removeWorkspaceMember, renameDatabase, resendWorkspaceMemberInvite, resolveBranch, searchBranch, searchTable, serialize, setTableSchema, sqlQuery, startsWith, summarizeTable, transformImage, updateBranchMetadata, updateBranchSchema, updateCluster, updateColumn, updateDatabaseGithubSettings, updateDatabaseMetadata, updateMigrationRequest, updateOAuthAccessToken, updateRecordWithID, updateTable, updateUser, updateWorkspace, updateWorkspaceMemberInvite, updateWorkspaceMemberRole, upsertRecordWithID, vectorSearchTable };
 //# sourceMappingURL=index.mjs.map