@xata.io/client 0.0.0-alpha.vf8b86c5 → 0.0.0-alpha.vf90263d

package/dist/index.mjs

@@ -18,7 +18,8 @@ const TraceAttributes = {
   HTTP_METHOD: "http.method",
   HTTP_URL: "http.url",
   HTTP_ROUTE: "http.route",
-  HTTP_TARGET: "http.target"
+  HTTP_TARGET: "http.target",
+  CLOUDFLARE_RAY_ID: "cf.ray"
 };
 
 function notEmpty(value) {
@@ -30,8 +31,15 @@ function compact(arr) {
 function compactObject(obj) {
   return Object.fromEntries(Object.entries(obj).filter(([, value]) => notEmpty(value)));
 }
+function isBlob(value) {
+  try {
+    return value instanceof Blob;
+  } catch (error) {
+    return false;
+  }
+}
 function isObject(value) {
-  return Boolean(value) && typeof value === "object" && !Array.isArray(value) && !(value instanceof Date);
+  return Boolean(value) && typeof value === "object" && !Array.isArray(value) && !(value instanceof Date) && !isBlob(value);
 }
 function isDefined(value) {
   return value !== null && value !== void 0;
@@ -86,6 +94,18 @@ function chunk(array, chunkSize) {
 async function timeout(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
+function timeoutWithCancel(ms) {
+  let timeoutId;
+  const promise = new Promise((resolve) => {
+    timeoutId = setTimeout(() => {
+      resolve();
+    }, ms);
+  });
+  return {
+    cancel: () => clearTimeout(timeoutId),
+    promise
+  };
+}
 function promiseMap(inputValues, mapper) {
   const reducer = (acc$, inputValue) => acc$.then(
     (acc) => mapper(inputValue).then((result) => {
@@ -194,7 +214,7 @@ function getAPIKey() {
 function getBranch() {
   try {
     const { branch } = getEnvironment();
-    return branch ?? "main";
+    return branch;
   } catch (err) {
     return void 0;
   }
@@ -222,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);
@@ -251,13 +265,13 @@ var __privateMethod$4 = (obj, member, method) => {
   return method;
 };
 var _fetch, _queue, _concurrency, _enqueue, enqueue_fn;
+const REQUEST_TIMEOUT = 5 * 60 * 1e3;
 function getFetchImplementation(userFetch) {
   const globalFetch = typeof fetch !== "undefined" ? fetch : void 0;
-  const fetchImpl = userFetch ?? globalFetch;
+  const globalThisFetch = typeof globalThis !== "undefined" ? globalThis.fetch : void 0;
+  const fetchImpl = userFetch ?? globalFetch ?? globalThisFetch;
   if (!fetchImpl) {
-    throw new Error(
-      `Couldn't find \`fetch\`. Install a fetch implementation such as \`node-fetch\` and pass it explicitly.`
-    );
+    throw new Error(`Couldn't find a global \`fetch\`. Pass a fetch implementation explicitly.`);
   }
   return fetchImpl;
 }
@@ -267,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;
@@ -285,9 +297,13 @@ class ApiRequestPool {
   }
   request(url, options) {
     const start = /* @__PURE__ */ new Date();
-    const fetch2 = this.getFetch();
+    const fetchImpl = this.getFetch();
     const runRequest = async (stalled = false) => {
-      const response = await fetch2(url, options);
+      const { promise, cancel } = timeoutWithCancel(REQUEST_TIMEOUT);
+      const response = await Promise.race([fetchImpl(url, options), promise.then(() => null)]).finally(cancel);
+      if (!response) {
+        throw new Error("Request timed out");
+      }
       if (response.status === 429) {
         const rateLimitReset = parseNumber(response.headers?.get("x-ratelimit-reset")) ?? 1;
         await timeout(rateLimitReset * 1e3);
@@ -295,7 +311,7 @@ class ApiRequestPool {
       }
       if (stalled) {
         const stalledTime = (/* @__PURE__ */ new Date()).getTime() - start.getTime();
-        console.warn(`A request to Xata hit your workspace limits, was retried and stalled for ${stalledTime}ms`);
+        console.warn(`A request to Xata hit branch rate limits, was retried and stalled for ${stalledTime}ms`);
       }
       return response;
     };
@@ -510,26 +526,16 @@ function defaultOnOpen(response) {
   }
 }
 
-const VERSION = "0.24.3";
+const VERSION = "0.27.0";
 
-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;
@@ -596,11 +602,14 @@ function hostHeader(url) {
   const { groups } = pattern.exec(url) ?? {};
   return groups?.host ? { Host: groups.host } : {};
 }
-function parseBody(body, headers) {
+async function parseBody(body, headers) {
   if (!isDefined(body))
     return void 0;
+  if (isBlob(body) || typeof body.text === "function") {
+    return body;
+  }
   const { "Content-Type": contentType } = headers ?? {};
-  if (String(contentType).toLowerCase() === "application/json") {
+  if (String(contentType).toLowerCase() === "application/json" && isObject(body)) {
     return JSON.stringify(body);
   }
   return body;
@@ -657,7 +666,7 @@ async function fetch$1({
       const response = await pool.request(url, {
         ...fetchOptions,
         method: method.toUpperCase(),
-        body: parseBody(body, headers),
+        body: await parseBody(body, headers),
         headers,
         signal
       });
@@ -668,7 +677,8 @@ async function fetch$1({
         [TraceAttributes.HTTP_REQUEST_ID]: requestId,
         [TraceAttributes.HTTP_STATUS_CODE]: response.status,
         [TraceAttributes.HTTP_HOST]: host,
-        [TraceAttributes.HTTP_SCHEME]: protocol?.replace(":", "")
+        [TraceAttributes.HTTP_SCHEME]: protocol?.replace(":", ""),
+        [TraceAttributes.CLOUDFLARE_RAY_ID]: response.headers?.get("cf-ray") ?? void 0
       });
       const message = response.headers?.get("x-xata-message");
       if (message)
@@ -756,6 +766,12 @@ 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 getBranchList = (variables, signal) => dataPlaneFetch({
   url: "/dbs/{dbName}",
   method: "get",
@@ -775,6 +791,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",
@@ -940,12 +962,6 @@ const searchTable = (variables, signal) => dataPlaneFetch({
   ...variables,
   signal
 });
-const sqlQuery = (variables, signal) => dataPlaneFetch({
-  url: "/db/{dbBranchName}/sql",
-  method: "post",
-  ...variables,
-  signal
-});
 const vectorSearchTable = (variables, signal) => dataPlaneFetch({ url: "/db/{dbBranchName}/tables/{tableName}/vectorSearch", method: "post", ...variables, signal });
 const askTable = (variables, signal) => dataPlaneFetch({
   url: "/db/{dbBranchName}/tables/{tableName}/ask",
@@ -953,7 +969,7 @@ const askTable = (variables, signal) => dataPlaneFetch({
   ...variables,
   signal
 });
-const chatSessionMessage = (variables, signal) => dataPlaneFetch({ url: "/db/{dbBranchName}/tables/{tableName}/ask/{sessionId}", method: "post", ...variables, signal });
+const askTableSession = (variables, signal) => dataPlaneFetch({ url: "/db/{dbBranchName}/tables/{tableName}/ask/{sessionId}", method: "post", ...variables, signal });
 const summarizeTable = (variables, signal) => dataPlaneFetch({ url: "/db/{dbBranchName}/tables/{tableName}/summarize", method: "post", ...variables, signal });
 const aggregateTable = (variables, signal) => dataPlaneFetch({ url: "/db/{dbBranchName}/tables/{tableName}/aggregate", method: "post", ...variables, signal });
 const fileAccess = (variables, signal) => dataPlaneFetch({
@@ -962,8 +978,15 @@ const fileAccess = (variables, signal) => dataPlaneFetch({
   ...variables,
   signal
 });
+const sqlQuery = (variables, signal) => dataPlaneFetch({
+  url: "/db/{dbBranchName}/sql",
+  method: "post",
+  ...variables,
+  signal
+});
 const operationsByTag$2 = {
   branch: {
+    applyMigration,
     getBranchList,
     getBranchDetails,
     createBranch,
@@ -978,6 +1001,7 @@ const operationsByTag$2 = {
     resolveBranch
   },
   migrations: {
+    getSchema,
     getBranchMigrationHistory,
     getBranchMigrationPlan,
     executeBranchMigrationPlan,
@@ -1026,17 +1050,19 @@ const operationsByTag$2 = {
     queryTable,
     searchBranch,
     searchTable,
-    sqlQuery,
     vectorSearchTable,
     askTable,
-    chatSessionMessage,
+    askTableSession,
     summarizeTable,
     aggregateTable
-  }
+  },
+  sql: { sqlQuery }
 };
 
 const controlPlaneFetch = async (options) => fetch$1({ ...options, endpoint: "controlPlane" });
 
+const getAuthorizationCode = (variables, signal) => controlPlaneFetch({ url: "/oauth/authorize", method: "get", ...variables, signal });
+const grantAuthorizationCode = (variables, signal) => controlPlaneFetch({ url: "/oauth/authorize", method: "post", ...variables, signal });
 const getUser = (variables, signal) => controlPlaneFetch({
   url: "/user",
   method: "get",
@@ -1073,6 +1099,31 @@ const deleteUserAPIKey = (variables, signal) => controlPlaneFetch({
   ...variables,
   signal
 });
+const getUserOAuthClients = (variables, signal) => controlPlaneFetch({
+  url: "/user/oauth/clients",
+  method: "get",
+  ...variables,
+  signal
+});
+const deleteUserOAuthClient = (variables, signal) => controlPlaneFetch({
+  url: "/user/oauth/clients/{clientId}",
+  method: "delete",
+  ...variables,
+  signal
+});
+const getUserOAuthAccessTokens = (variables, signal) => controlPlaneFetch({
+  url: "/user/oauth/tokens",
+  method: "get",
+  ...variables,
+  signal
+});
+const deleteOAuthAccessToken = (variables, signal) => controlPlaneFetch({
+  url: "/user/oauth/tokens/{token}",
+  method: "delete",
+  ...variables,
+  signal
+});
+const updateOAuthAccessToken = (variables, signal) => controlPlaneFetch({ url: "/user/oauth/tokens/{token}", method: "patch", ...variables, signal });
 const getWorkspacesList = (variables, signal) => controlPlaneFetch({
   url: "/workspaces",
   method: "get",
@@ -1116,6 +1167,20 @@ 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",
@@ -1142,6 +1207,15 @@ const listRegions = (variables, signal) => controlPlaneFetch({
   signal
 });
 const operationsByTag$1 = {
+  oAuth: {
+    getAuthorizationCode,
+    grantAuthorizationCode,
+    getUserOAuthClients,
+    deleteUserOAuthClient,
+    getUserOAuthAccessTokens,
+    deleteOAuthAccessToken,
+    updateOAuthAccessToken
+  },
   users: { getUser, updateUser, deleteUser },
   authentication: { getUserAPIKeys, createUserAPIKey, deleteUserAPIKey },
   workspaces: {
@@ -1161,6 +1235,7 @@ const operationsByTag$1 = {
     acceptWorkspaceMemberInvite,
     resendWorkspaceMemberInvite
   },
+  xbcontrolOther: { listClusters, createCluster, getCluster, updateCluster },
   databases: {
     getDatabaseList,
     createDatabase,
@@ -2154,7 +2229,7 @@ class SearchAndFilterApi {
       ...this.extraProps
     });
   }
-  chatSessionMessage({
+  askTableSession({
     workspace,
     region,
     database,
@@ -2163,7 +2238,7 @@ class SearchAndFilterApi {
     sessionId,
     message
   }) {
-    return operationsByTag.searchAndFilter.chatSessionMessage({
+    return operationsByTag.searchAndFilter.askTableSession({
       pathParams: { workspace, region, dbBranchName: `${database}:${branch}`, tableName: table, sessionId },
       body: { message },
       ...this.extraProps
@@ -2556,60 +2631,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 ?? {};
-        return await putFileItem({
-          pathParams: {
-            workspace: "{workspaceId}",
-            dbBranchName: "{dbBranch}",
-            region: "{region}",
-            tableName: table ?? "",
-            recordId: record ?? "",
-            columnName: column ?? "",
-            fileId
-          },
-          body: file,
-          ...pluginOptions
-        });
-      },
-      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 buildTransformString(transformations) {
   return transformations.flatMap(
     (t) => Object.entries(t).map(([key, value]) => {
@@ -2625,74 +2646,34 @@ function buildTransformString(transformations) {
     })
   ).join(",");
 }
-function transformImage(url, transformations) {
+function transformImage(url, ...transformations) {
   if (!isDefined(url))
     return void 0;
-  const transformationsString = buildTransformString(transformations);
+  const newTransformations = buildTransformString(transformations);
   const { hostname, pathname, search } = new URL(url);
-  return `https://${hostname}/transform/${transformationsString}${pathname}${search}`;
+  const pathParts = pathname.split("/");
+  const transformIndex = pathParts.findIndex((part) => part === "transform");
+  const removedItems = transformIndex >= 0 ? pathParts.splice(transformIndex, 2) : [];
+  const transform = `/transform/${[removedItems[1], newTransformations].filter(isDefined).join(",")}`;
+  const path = pathParts.join("/");
+  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) {
-    /**
-     * Name of this file.
-     */
-    __publicField$6(this, "name");
-    /**
-     * Media type of this file.
-     */
-    __publicField$6(this, "mediaType");
-    /**
-     * Base64 encoded content of this file.
-     */
-    __publicField$6(this, "base64Content");
-    /**
-     * Whether to enable public url for this file.
-     */
-    __publicField$6(this, "enablePublicUrl");
-    /**
-     * Timeout for the signed url.
-     */
-    __publicField$6(this, "signedUrlTimeout");
-    /**
-     * Size of this file.
-     */
-    __publicField$6(this, "size");
-    /**
-     * Version of this file.
-     */
-    __publicField$6(this, "version");
-    /**
-     * Url of this file.
-     */
-    __publicField$6(this, "url");
-    /**
-     * Signed url of this file.
-     */
-    __publicField$6(this, "signedUrl");
-    /**
-     * Attributes of this file.
-     */
-    __publicField$6(this, "attributes");
-    this.name = file.name;
+    this.id = file.id;
+    this.name = file.name || "";
     this.mediaType = file.mediaType || "application/octet-stream";
     this.base64Content = file.base64Content;
-    this.enablePublicUrl = file.enablePublicUrl;
-    this.signedUrlTimeout = file.signedUrlTimeout;
-    this.size = file.size;
-    this.version = file.version;
-    this.url = file.url;
+    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.signedUrl = file.signedUrl;
-    this.attributes = file.attributes;
+    this.attributes = file.attributes || {};
   }
-  static async fromBuffer(buffer, options = {}) {
+  static fromBuffer(buffer, options = {}) {
     const base64Content = buffer.toString("base64");
     return new XataFile({ ...options, base64Content });
   }
@@ -2702,9 +2683,9 @@ class XataFile {
     }
     return Buffer.from(this.base64Content, "base64");
   }
-  static async fromArrayBuffer(arrayBuffer, options = {}) {
+  static fromArrayBuffer(arrayBuffer, options = {}) {
     const uint8Array = new Uint8Array(arrayBuffer);
-    return await this.fromUint8Array(uint8Array, options);
+    return this.fromUint8Array(uint8Array, options);
   }
   toArrayBuffer() {
     if (!this.base64Content) {
@@ -2713,7 +2694,7 @@ class XataFile {
     const binary = atob(this.base64Content);
     return new ArrayBuffer(binary.length);
   }
-  static async fromUint8Array(uint8Array, options = {}) {
+  static fromUint8Array(uint8Array, options = {}) {
     let binary = "";
     for (let i = 0; i < uint8Array.byteLength; i++) {
       binary += String.fromCharCode(uint8Array[i]);
@@ -2736,16 +2717,20 @@ class XataFile {
     const name = options.name ?? file.name;
     const mediaType = file.type;
     const arrayBuffer = await file.arrayBuffer();
-    return await this.fromArrayBuffer(arrayBuffer, { ...options, name, mediaType });
+    return this.fromArrayBuffer(arrayBuffer, { ...options, name, mediaType });
   }
   toBlob() {
     if (!this.base64Content) {
       throw new Error(`File content is not available, please select property "base64Content" when querying the file`);
     }
-    const arrayBuffer = this.toArrayBuffer();
-    return new Blob([arrayBuffer], { type: this.mediaType });
+    const binary = atob(this.base64Content);
+    const uint8Array = new Uint8Array(binary.length);
+    for (let i = 0; i < binary.length; i++) {
+      uint8Array[i] = binary.charCodeAt(i);
+    }
+    return new Blob([uint8Array], { type: this.mediaType });
   }
-  static async fromString(string, options = {}) {
+  static fromString(string, options = {}) {
     const base64Content = btoa(string);
     return new XataFile({ ...options, base64Content });
   }
@@ -2755,7 +2740,7 @@ class XataFile {
     }
     return atob(this.base64Content);
   }
-  static async fromBase64(base64Content, options = {}) {
+  static fromBase64(base64Content, options = {}) {
     return new XataFile({ ...options, base64Content });
   }
   toBase64() {
@@ -2766,16 +2751,26 @@ class XataFile {
   }
   transform(...options) {
     return {
-      url: transformImage(this.url, options),
-      signedUrl: transformImage(this.signedUrl, options)
+      url: transformImage(this.url, ...options),
+      signedUrl: transformImage(this.signedUrl, ...options),
+      metadataUrl: transformImage(this.url, ...options, { format: "json" }),
+      metadataSignedUrl: transformImage(this.signedUrl, ...options, { format: "json" })
     };
   }
 }
-const parseInputFileEntry = (entry) => {
+const parseInputFileEntry = async (entry) => {
   if (!isDefined(entry))
     return null;
-  const { id, name, mediaType, base64Content, enablePublicUrl, signedUrlTimeout } = entry;
-  return compactObject({ id, name, mediaType, base64Content, enablePublicUrl, signedUrlTimeout });
+  const { id, name, mediaType, base64Content, enablePublicUrl, signedUrlTimeout } = await entry;
+  return compactObject({
+    id,
+    // Name cannot be an empty string in our API
+    name: name ? name : void 0,
+    mediaType,
+    base64Content,
+    enablePublicUrl,
+    signedUrlTimeout
+  });
 };
 
 function cleanFilter(filter) {
@@ -2805,12 +2800,25 @@ function cleanFilter(filter) {
   return Object.keys(values).length > 0 ? values : void 0;
 }
 
-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;
-};
+function stringifyJson(value) {
+  if (!isDefined(value))
+    return value;
+  if (isString(value))
+    return value;
+  try {
+    return JSON.stringify(value);
+  } catch (e) {
+    return value;
+  }
+}
+function parseJson(value) {
+  try {
+    return JSON.parse(value);
+  } catch (e) {
+    return value;
+  }
+}
+
 var __accessCheck$6 = (obj, member, msg) => {
   if (!member.has(obj))
     throw TypeError("Cannot " + msg);
@@ -2833,14 +2841,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);
@@ -2890,9 +2890,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));
@@ -2971,12 +2971,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);
@@ -3007,8 +3001,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);
@@ -3263,7 +3257,8 @@ const RecordColumnTypes = [
   "datetime",
   "vector",
   "file[]",
-  "file"
+  "file",
+  "json"
 ];
 function isIdentifiable(x) {
   return isObject(x) && isString(x?.id);
@@ -3274,6 +3269,24 @@ function isXataRecord(x) {
   return isIdentifiable(x) && isObject(metadata) && typeof metadata.version === "number";
 }
 
+function isValidExpandedColumn(column) {
+  return isObject(column) && isString(column.name);
+}
+function isValidSelectableColumns(columns) {
+  if (!Array.isArray(columns)) {
+    return false;
+  }
+  return columns.every((column) => {
+    if (typeof column === "string") {
+      return true;
+    }
+    if (typeof column === "object") {
+      return isValidExpandedColumn(column);
+    }
+    return false;
+  });
+}
+
 function isSortFilterString(value) {
   return isString(value);
 }
@@ -3374,24 +3387,24 @@ class RestRepository extends Query {
         if (a.length === 0)
           return [];
         const ids = await __privateMethod$2(this, _insertRecords, insertRecords_fn).call(this, a, { ifVersion, createOnly: true });
-        const columns = isStringArray(b) ? b : ["*"];
+        const columns = isValidSelectableColumns(b) ? b : ["*"];
         const result = await this.read(ids, columns);
         return result;
       }
       if (isString(a) && isObject(b)) {
         if (a === "")
           throw new Error("The id can't be empty");
-        const columns = isStringArray(c) ? c : void 0;
+        const columns = isValidSelectableColumns(c) ? c : void 0;
         return await __privateMethod$2(this, _insertRecordWithId, insertRecordWithId_fn).call(this, a, b, columns, { createOnly: true, ifVersion });
       }
       if (isObject(a) && isString(a.id)) {
         if (a.id === "")
           throw new Error("The id can't be empty");
-        const columns = isStringArray(b) ? b : void 0;
+        const columns = isValidSelectableColumns(b) ? b : void 0;
         return await __privateMethod$2(this, _insertRecordWithId, insertRecordWithId_fn).call(this, a.id, { ...a, id: void 0 }, columns, { createOnly: true, ifVersion });
       }
       if (isObject(a)) {
-        const columns = isStringArray(b) ? b : void 0;
+        const columns = isValidSelectableColumns(b) ? b : void 0;
         return __privateMethod$2(this, _insertRecordWithoutId, insertRecordWithoutId_fn).call(this, a, columns);
       }
       throw new Error("Invalid arguments for create method");
@@ -3399,7 +3412,7 @@ class RestRepository extends Query {
   }
   async read(a, b) {
     return __privateGet$4(this, _trace).call(this, "read", async () => {
-      const columns = isStringArray(b) ? b : ["*"];
+      const columns = isValidSelectableColumns(b) ? b : ["*"];
       if (Array.isArray(a)) {
         if (a.length === 0)
           return [];
@@ -3426,7 +3439,13 @@ class RestRepository extends Query {
             ...__privateGet$4(this, _getFetchProps).call(this)
           });
           const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
-          return initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), response, columns);
+          return initObject(
+            __privateGet$4(this, _db),
+            schemaTables,
+            __privateGet$4(this, _table),
+            response,
+            columns
+          );
         } catch (e) {
           if (isObject(e) && e.status === 404) {
             return null;
@@ -3468,17 +3487,17 @@ class RestRepository extends Query {
           ifVersion,
           upsert: false
         });
-        const columns = isStringArray(b) ? b : ["*"];
+        const columns = isValidSelectableColumns(b) ? b : ["*"];
         const result = await this.read(a, columns);
         return result;
       }
       try {
         if (isString(a) && isObject(b)) {
-          const columns = isStringArray(c) ? c : void 0;
+          const columns = isValidSelectableColumns(c) ? c : void 0;
           return await __privateMethod$2(this, _updateRecordWithID, updateRecordWithID_fn).call(this, a, b, columns, { ifVersion });
         }
         if (isObject(a) && isString(a.id)) {
-          const columns = isStringArray(b) ? b : void 0;
+          const columns = isValidSelectableColumns(b) ? b : void 0;
           return await __privateMethod$2(this, _updateRecordWithID, updateRecordWithID_fn).call(this, a.id, { ...a, id: void 0 }, columns, { ifVersion });
         }
       } catch (error) {
@@ -3518,20 +3537,20 @@ class RestRepository extends Query {
           ifVersion,
           upsert: true
         });
-        const columns = isStringArray(b) ? b : ["*"];
+        const columns = isValidSelectableColumns(b) ? b : ["*"];
         const result = await this.read(a, columns);
         return result;
       }
       if (isString(a) && isObject(b)) {
         if (a === "")
           throw new Error("The id can't be empty");
-        const columns = isStringArray(c) ? c : void 0;
+        const columns = isValidSelectableColumns(c) ? c : void 0;
         return await __privateMethod$2(this, _upsertRecordWithID, upsertRecordWithID_fn).call(this, a, b, columns, { ifVersion });
       }
       if (isObject(a) && isString(a.id)) {
         if (a.id === "")
           throw new Error("The id can't be empty");
-        const columns = isStringArray(c) ? c : void 0;
+        const columns = isValidSelectableColumns(c) ? c : void 0;
         return await __privateMethod$2(this, _upsertRecordWithID, upsertRecordWithID_fn).call(this, a.id, { ...a, id: void 0 }, columns, { ifVersion });
       }
       if (!isDefined(a) && isObject(b)) {
@@ -3550,20 +3569,20 @@ class RestRepository extends Query {
         if (a.length === 0)
           return [];
         const ids = await __privateMethod$2(this, _insertRecords, insertRecords_fn).call(this, a, { ifVersion, createOnly: false });
-        const columns = isStringArray(b) ? b : ["*"];
+        const columns = isValidSelectableColumns(b) ? b : ["*"];
         const result = await this.read(ids, columns);
         return result;
       }
       if (isString(a) && isObject(b)) {
         if (a === "")
           throw new Error("The id can't be empty");
-        const columns = isStringArray(c) ? c : void 0;
+        const columns = isValidSelectableColumns(c) ? c : void 0;
         return await __privateMethod$2(this, _insertRecordWithId, insertRecordWithId_fn).call(this, a, b, columns, { createOnly: false, ifVersion });
       }
       if (isObject(a) && isString(a.id)) {
         if (a.id === "")
           throw new Error("The id can't be empty");
-        const columns = isStringArray(c) ? c : void 0;
+        const columns = isValidSelectableColumns(c) ? c : void 0;
         return await __privateMethod$2(this, _insertRecordWithId, insertRecordWithId_fn).call(this, a.id, { ...a, id: void 0 }, columns, { createOnly: false, ifVersion });
       }
       if (!isDefined(a) && isObject(b)) {
@@ -3587,7 +3606,7 @@ class RestRepository extends Query {
             return o.id;
           throw new Error("Invalid arguments for delete method");
         });
-        const columns = isStringArray(b) ? b : ["*"];
+        const columns = isValidSelectableColumns(b) ? b : ["*"];
         const result = await this.read(a, columns);
         await __privateMethod$2(this, _deleteRecords, deleteRecords_fn).call(this, ids);
         return result;
@@ -3621,7 +3640,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}",
@@ -3641,12 +3660,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}",
@@ -3663,7 +3685,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) {
@@ -3706,7 +3731,13 @@ class RestRepository extends Query {
       });
       const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
       const records = objects.map(
-        (record) => initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), record, data.columns ?? ["*"])
+        (record) => initObject(
+          __privateGet$4(this, _db),
+          schemaTables,
+          __privateGet$4(this, _table),
+          record,
+          data.columns ?? ["*"]
+        )
       );
       await __privateMethod$2(this, _setCacheQuery, setCacheQuery_fn).call(this, query, meta, records);
       return new Page(query, meta, records);
@@ -3733,27 +3764,38 @@ 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) {
+    const questionParam = options?.sessionId ? { message: question } : { question };
     const params = {
       pathParams: {
         workspace: "{workspaceId}",
         dbBranchName: "{dbBranch}",
         region: "{region}",
-        tableName: __privateGet$4(this, _table)
+        tableName: __privateGet$4(this, _table),
+        sessionId: options?.sessionId
       },
       body: {
-        question,
-        ...options
+        ...questionParam,
+        rules: options?.rules,
+        searchType: options?.searchType,
+        search: options?.searchType === "keyword" ? options?.search : void 0,
+        vectorSearch: options?.searchType === "vector" ? options?.vectorSearch : void 0
       },
       ...__privateGet$4(this, _getFetchProps).call(this)
     };
     if (options?.onMessage) {
       fetchSSERequest({
         endpoint: "dataPlane",
-        url: "/db/{dbBranchName}/tables/{tableName}/ask",
+        url: "/db/{dbBranchName}/tables/{tableName}/ask/{sessionId}",
         method: "POST",
         onMessage: (message) => {
           options.onMessage?.({ answer: message.text, records: message.records });
@@ -3761,7 +3803,7 @@ class RestRepository extends Query {
         ...params
       });
     } else {
-      return askTable(params);
+      return askTableSession(params);
     }
   }
 }
@@ -4002,10 +4044,13 @@ transformObjectToApi_fn = async function(object) {
         break;
       }
       case `file`:
-        result[key] = parseInputFileEntry(value);
+        result[key] = await parseInputFileEntry(value);
         break;
       case "file[]":
-        result[key] = value.map((item) => parseInputFileEntry(item));
+        result[key] = await promiseMap(value, (item) => parseInputFileEntry(item));
+        break;
+      case "json":
+        result[key] = stringifyJson(value);
         break;
       default:
         result[key] = value;
@@ -4013,13 +4058,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 ?? {};
@@ -4050,13 +4088,19 @@ const initObject = (db, schemaTables, table, object, selectedColumns) => {
             if (item === column.name) {
               return [...acc, "*"];
             }
-            if (item.startsWith(`${column.name}.`)) {
+            if (isString(item) && item.startsWith(`${column.name}.`)) {
               const [, ...path] = item.split(".");
               return [...acc, path.join(".")];
             }
             return acc;
           }, []);
-          data[column.name] = initObject(db, schemaTables, linkTable, value, selectedLinkColumns);
+          data[column.name] = initObject(
+            db,
+            schemaTables,
+            linkTable,
+            value,
+            selectedLinkColumns
+          );
         } else {
           data[column.name] = null;
         }
@@ -4068,6 +4112,9 @@ const initObject = (db, schemaTables, table, object, selectedColumns) => {
       case "file[]":
         data[column.name] = value?.map((item) => new XataFile(item)) ?? null;
         break;
+      case "json":
+        data[column.name] = parseJson(value);
+        break;
       default:
         data[column.name] = value ?? null;
         if (column.notNull === true && value === null) {
@@ -4077,33 +4124,34 @@ 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);
   };
   record.update = function(data2, b, c) {
-    const columns2 = isStringArray(b) ? b : ["*"];
+    const columns2 = isValidSelectableColumns(b) ? b : ["*"];
     const ifVersion = parseIfVersion(b, c);
     return db[table].update(record["id"], data2, columns2, { ifVersion });
   };
   record.replace = function(data2, b, c) {
-    const columns2 = isStringArray(b) ? b : ["*"];
+    const columns2 = isValidSelectableColumns(b) ? b : ["*"];
     const ifVersion = parseIfVersion(b, c);
     return db[table].createOrReplace(record["id"], data2, columns2, { ifVersion });
   };
   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 });
@@ -4121,7 +4169,7 @@ function extractId(value) {
 function isValidColumn(columns, column) {
   if (columns.includes("*"))
     return true;
-  return columns.filter((item) => item.startsWith(column.name)).length > 0;
+  return columns.filter((item) => isString(item) && item.startsWith(column.name)).length > 0;
 }
 function parseIfVersion(...args) {
   for (const arg of args) {
@@ -4132,12 +4180,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);
@@ -4160,8 +4202,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;
@@ -4206,10 +4246,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 });
@@ -4265,6 +4307,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) {
+    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);
@@ -4300,22 +4416,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 };
       }
     };
   }
@@ -4324,13 +4444,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) {
@@ -4344,6 +4464,78 @@ getSchemaTables_fn = async function(pluginOptions) {
   return schema.tables;
 };
 
+function escapeElement(elementRepresentation) {
+  const escaped = elementRepresentation.replace(/\\/g, "\\\\").replace(/"/g, '\\"');
+  return '"' + escaped + '"';
+}
+function arrayString(val) {
+  let result = "{";
+  for (let i = 0; i < val.length; i++) {
+    if (i > 0) {
+      result = result + ",";
+    }
+    if (val[i] === null || typeof val[i] === "undefined") {
+      result = result + "NULL";
+    } else if (Array.isArray(val[i])) {
+      result = result + arrayString(val[i]);
+    } else if (val[i] instanceof Buffer) {
+      result += "\\\\x" + val[i].toString("hex");
+    } else {
+      result += escapeElement(prepareValue(val[i]));
+    }
+  }
+  result = result + "}";
+  return result;
+}
+function prepareValue(value) {
+  if (!isDefined(value))
+    return null;
+  if (value instanceof Date) {
+    return value.toISOString();
+  }
+  if (Array.isArray(value)) {
+    return arrayString(value);
+  }
+  if (isObject(value)) {
+    return JSON.stringify(value);
+  }
+  try {
+    return value.toString();
+  } catch (e) {
+    return value;
+  }
+}
+function prepareParams(param1, param2) {
+  if (isString(param1)) {
+    return { statement: param1, params: param2?.map((value) => prepareValue(value)) };
+  }
+  if (isStringArray(param1)) {
+    const statement = param1.reduce((acc, curr, index) => {
+      return acc + curr + (index < (param2?.length ?? 0) ? "$" + (index + 1) : "");
+    }, "");
+    return { statement, params: param2?.map((value) => prepareValue(value)) };
+  }
+  if (isObject(param1)) {
+    const { statement, params, consistency } = param1;
+    return { statement, params: params?.map((value) => prepareValue(value)), consistency };
+  }
+  throw new Error("Invalid query");
+}
+
+class SQLPlugin extends XataPlugin {
+  build(pluginOptions) {
+    return async (param1, ...param2) => {
+      const { statement, params, consistency } = prepareParams(param1, param2);
+      const { records, warning } = await sqlQuery({
+        pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", region: "{region}" },
+        body: { statement, params, consistency },
+        ...pluginOptions
+      });
+      return { records, warning };
+    };
+  }
+}
+
 class TransactionPlugin extends XataPlugin {
   build(pluginOptions) {
     return {
@@ -4359,12 +4551,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);
@@ -4394,10 +4580,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, "files");
       const safeOptions = __privateMethod(this, _parseOptions, parseOptions_fn).call(this, options);
       __privateSet(this, _options, safeOptions);
       const pluginOptions = {
@@ -4408,10 +4590,12 @@ const buildClient = (plugins) => {
       const db = new SchemaPlugin(schemaTables).build(pluginOptions);
       const search = new SearchPlugin(db, schemaTables).build(pluginOptions);
       const transactions = new TransactionPlugin().build(pluginOptions);
+      const sql = new SQLPlugin().build(pluginOptions);
       const files = new FilesPlugin().build(pluginOptions);
       this.db = db;
       this.search = search;
       this.transactions = transactions;
+      this.sql = sql;
       this.files = files;
       for (const [key, namespace] of Object.entries(plugins ?? {})) {
         if (namespace === void 0)
@@ -4509,17 +4693,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;
@@ -4597,19 +4775,12 @@ function buildWorkerRunner(config) {
   };
 }
 
-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, operationsByTag as Operations, PAGINATION_DEFAULT_OFFSET, PAGINATION_DEFAULT_SIZE, PAGINATION_MAX_OFFSET, PAGINATION_MAX_SIZE, Page, Query, RecordArray, RecordColumnTypes, Repository, RestRepository, SchemaPlugin, SearchPlugin, Serializer, SimpleCache, XataApiClient, XataApiPlugin, XataError, XataFile, XataPlugin, acceptWorkspaceMemberInvite, addGitBranchesEntry, addTableColumn, aggregateTable, applyBranchSchemaEdit, askTable, branchTransaction, buildClient, buildPreviewBranchName, buildProviderString, buildWorkerRunner, bulkInsertTableRecords, cancelWorkspaceMemberInvite, chatSessionMessage, compareBranchSchemas, compareBranchWithUserSchema, compareMigrationRequest, contains, copyBranch, createBranch, createDatabase, createMigrationRequest, createTable, createUserAPIKey, createWorkspace, deleteBranch, deleteColumn, deleteDatabase, deleteDatabaseGithubSettings, deleteFile, deleteFileItem, deleteRecord, deleteTable, deleteUser, deleteUserAPIKey, deleteWorkspace, deserialize, endsWith, equals, executeBranchMigrationPlan, exists, fileAccess, ge, getAPIKey, 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, getWorkspace, getWorkspaceMembersList, getWorkspacesList, greaterEquals, greaterThan, greaterThanEquals, gt, gte, includes, includesAll, includesAny, includesNone, insertRecord, insertRecordWithID, inviteWorkspaceMember, is, isCursorPaginationOptions, isHostProviderAlias, isHostProviderBuilder, isIdentifiable, isNot, 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, updateBranchMetadata, updateBranchSchema, updateColumn, updateDatabaseGithubSettings, updateDatabaseMetadata, updateMigrationRequest, 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, buildWorkerRunner, 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, 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, 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