@xata.io/client 0.0.0-alpha.vfc4a5e4 → 0.0.0-alpha.vfc4cc60

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,27 @@ 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) => {
+      acc.push(result);
+      return acc;
+    })
+  );
+  return inputValues.reduce(reducer, Promise.resolve([]));
+}
 
 function getEnvironment() {
   try {
@@ -185,7 +214,7 @@ function getAPIKey() {
 function getBranch() {
   try {
     const { branch } = getEnvironment();
-    return branch ?? "main";
+    return branch;
   } catch (err) {
     return void 0;
   }
@@ -213,12 +242,6 @@ function getPreviewBranch() {
   }
 }
 
-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;
-};
 var __accessCheck$8 = (obj, member, msg) => {
   if (!member.has(obj))
     throw TypeError("Cannot " + msg);
@@ -242,14 +265,13 @@ var __privateMethod$4 = (obj, member, method) => {
   return method;
 };
 var _fetch, _queue, _concurrency, _enqueue, enqueue_fn;
-const REQUEST_TIMEOUT = 3e4;
+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;
 }
@@ -259,8 +281,6 @@ class ApiRequestPool {
     __privateAdd$8(this, _fetch, void 0);
     __privateAdd$8(this, _queue, void 0);
     __privateAdd$8(this, _concurrency, void 0);
-    __publicField$7(this, "running");
-    __publicField$7(this, "started");
     __privateSet$8(this, _queue, []);
     __privateSet$8(this, _concurrency, concurrency);
     this.running = 0;
@@ -277,9 +297,10 @@ 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 Promise.race([fetch2(url, options), timeout(REQUEST_TIMEOUT).then(() => null)]);
+      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");
       }
@@ -290,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;
     };
@@ -505,26 +526,16 @@ function defaultOnOpen(response) {
   }
 }
 
-const VERSION = "0.24.3";
+const VERSION = "0.27.0";
 
-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 ErrorWithCause extends Error {
   constructor(message, options) {
     super(message, options);
-    __publicField$6(this, "cause");
   }
 }
 class FetcherError extends ErrorWithCause {
   constructor(status, data, requestId) {
     super(getMessage(data));
-    __publicField$6(this, "status");
-    __publicField$6(this, "requestId");
-    __publicField$6(this, "errors");
     this.status = status;
     this.errors = isBulkError(data) ? data.errors : [{ message: getMessage(data), status }];
     this.requestId = requestId;
@@ -591,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;
@@ -652,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
       });
@@ -663,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)
@@ -751,6 +766,18 @@ 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 getBranchList = (variables, signal) => dataPlaneFetch({
   url: "/dbs/{dbName}",
   method: "get",
@@ -770,6 +797,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",
@@ -935,12 +968,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",
@@ -948,7 +975,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({
@@ -957,8 +984,16 @@ 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,
+    pgRollStatus,
     getBranchList,
     getBranchDetails,
     createBranch,
@@ -973,6 +1008,7 @@ const operationsByTag$2 = {
     resolveBranch
   },
   migrations: {
+    getSchema,
     getBranchMigrationHistory,
     getBranchMigrationPlan,
     executeBranchMigrationPlan,
@@ -1021,17 +1057,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",
@@ -1068,6 +1106,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",
@@ -1111,6 +1174,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",
@@ -1137,6 +1214,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: {
@@ -1156,6 +1242,7 @@ const operationsByTag$1 = {
     acceptWorkspaceMemberInvite,
     resendWorkspaceMemberInvite
   },
+  xbcontrolOther: { listClusters, createCluster, getCluster, updateCluster },
   databases: {
     getDatabaseList,
     createDatabase,
@@ -2149,7 +2236,7 @@ class SearchAndFilterApi {
       ...this.extraProps
     });
   }
-  chatSessionMessage({
+  askTableSession({
     workspace,
     region,
     database,
@@ -2158,7 +2245,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
@@ -2455,11 +2542,13 @@ class DatabaseApi {
   createDatabase({
     workspace,
     database,
-    data
+    data,
+    headers
   }) {
     return operationsByTag.databases.createDatabase({
       pathParams: { workspaceId: workspace, dbName: database },
       body: data,
+      headers,
       ...this.extraProps
     });
   }
@@ -2549,6 +2638,148 @@ class XataApiPlugin {
 class XataPlugin {
 }
 
+function buildTransformString(transformations) {
+  return transformations.flatMap(
+    (t) => Object.entries(t).map(([key, value]) => {
+      if (key === "trim") {
+        const { left = 0, top = 0, right = 0, bottom = 0 } = value;
+        return `${key}=${[top, right, bottom, left].join(";")}`;
+      }
+      if (key === "gravity" && typeof value === "object") {
+        const { x = 0.5, y = 0.5 } = value;
+        return `${key}=${[x, y].join("x")}`;
+      }
+      return `${key}=${value}`;
+    })
+  ).join(",");
+}
+function transformImage(url, ...transformations) {
+  if (!isDefined(url))
+    return void 0;
+  const newTransformations = buildTransformString(transformations);
+  const { hostname, pathname, search } = new URL(url);
+  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}`;
+}
+
+class XataFile {
+  constructor(file) {
+    this.id = file.id;
+    this.name = file.name || "";
+    this.mediaType = file.mediaType || "application/octet-stream";
+    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.signedUrl = file.signedUrl;
+    this.attributes = file.attributes || {};
+  }
+  static fromBuffer(buffer, options = {}) {
+    const base64Content = buffer.toString("base64");
+    return new XataFile({ ...options, base64Content });
+  }
+  toBuffer() {
+    if (!this.base64Content) {
+      throw new Error(`File content is not available, please select property "base64Content" when querying the file`);
+    }
+    return Buffer.from(this.base64Content, "base64");
+  }
+  static fromArrayBuffer(arrayBuffer, options = {}) {
+    const uint8Array = new Uint8Array(arrayBuffer);
+    return this.fromUint8Array(uint8Array, options);
+  }
+  toArrayBuffer() {
+    if (!this.base64Content) {
+      throw new Error(`File content is not available, please select property "base64Content" when querying the file`);
+    }
+    const binary = atob(this.base64Content);
+    return new ArrayBuffer(binary.length);
+  }
+  static fromUint8Array(uint8Array, options = {}) {
+    let binary = "";
+    for (let i = 0; i < uint8Array.byteLength; i++) {
+      binary += String.fromCharCode(uint8Array[i]);
+    }
+    const base64Content = btoa(binary);
+    return new XataFile({ ...options, base64Content });
+  }
+  toUint8Array() {
+    if (!this.base64Content) {
+      throw new Error(`File content is not available, please select property "base64Content" when querying the file`);
+    }
+    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 uint8Array;
+  }
+  static async fromBlob(file, options = {}) {
+    const name = options.name ?? file.name;
+    const mediaType = file.type;
+    const arrayBuffer = await file.arrayBuffer();
+    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 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 fromString(string, options = {}) {
+    const base64Content = btoa(string);
+    return new XataFile({ ...options, base64Content });
+  }
+  toString() {
+    if (!this.base64Content) {
+      throw new Error(`File content is not available, please select property "base64Content" when querying the file`);
+    }
+    return atob(this.base64Content);
+  }
+  static fromBase64(base64Content, options = {}) {
+    return new XataFile({ ...options, base64Content });
+  }
+  toBase64() {
+    if (!this.base64Content) {
+      throw new Error(`File content is not available, please select property "base64Content" when querying the file`);
+    }
+    return this.base64Content;
+  }
+  transform(...options) {
+    return {
+      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 = async (entry) => {
+  if (!isDefined(entry))
+    return null;
+  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) {
   if (!isDefined(filter))
     return void 0;
@@ -2576,12 +2807,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);
@@ -2604,14 +2848,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);
@@ -2661,9 +2897,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));
@@ -2742,12 +2978,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);
@@ -2778,8 +3008,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);
@@ -3034,7 +3264,8 @@ const RecordColumnTypes = [
   "datetime",
   "vector",
   "file[]",
-  "file"
+  "file",
+  "json"
 ];
 function isIdentifiable(x) {
   return isObject(x) && isString(x?.id);
@@ -3045,6 +3276,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);
 }
@@ -3094,7 +3343,7 @@ var __privateMethod$2 = (obj, member, method) => {
   __accessCheck$4(obj, member, "access private method");
   return method;
 };
-var _table, _getFetchProps, _db, _cache, _schemaTables$2, _trace, _insertRecordWithoutId, insertRecordWithoutId_fn, _insertRecordWithId, insertRecordWithId_fn, _insertRecords, insertRecords_fn, _updateRecordWithID, updateRecordWithID_fn, _updateRecords, updateRecords_fn, _upsertRecordWithID, upsertRecordWithID_fn, _deleteRecord, deleteRecord_fn, _deleteRecords, deleteRecords_fn, _setCacheQuery, setCacheQuery_fn, _getCacheQuery, getCacheQuery_fn, _getSchemaTables$1, getSchemaTables_fn$1;
+var _table, _getFetchProps, _db, _cache, _schemaTables$2, _trace, _insertRecordWithoutId, insertRecordWithoutId_fn, _insertRecordWithId, insertRecordWithId_fn, _insertRecords, insertRecords_fn, _updateRecordWithID, updateRecordWithID_fn, _updateRecords, updateRecords_fn, _upsertRecordWithID, upsertRecordWithID_fn, _deleteRecord, deleteRecord_fn, _deleteRecords, deleteRecords_fn, _setCacheQuery, setCacheQuery_fn, _getCacheQuery, getCacheQuery_fn, _getSchemaTables$1, getSchemaTables_fn$1, _transformObjectToApi, transformObjectToApi_fn;
 const BULK_OPERATION_MAX_SIZE = 1e3;
 class Repository extends Query {
 }
@@ -3116,6 +3365,7 @@ class RestRepository extends Query {
     __privateAdd$4(this, _setCacheQuery);
     __privateAdd$4(this, _getCacheQuery);
     __privateAdd$4(this, _getSchemaTables$1);
+    __privateAdd$4(this, _transformObjectToApi);
     __privateAdd$4(this, _table, void 0);
     __privateAdd$4(this, _getFetchProps, void 0);
     __privateAdd$4(this, _db, void 0);
@@ -3144,24 +3394,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");
@@ -3169,7 +3419,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 [];
@@ -3196,7 +3446,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;
@@ -3238,17 +3494,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) {
@@ -3288,20 +3544,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)) {
@@ -3320,20 +3576,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)) {
@@ -3357,7 +3613,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;
@@ -3391,7 +3647,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}",
@@ -3411,12 +3667,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}",
@@ -3433,7 +3692,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) {
@@ -3476,7 +3738,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);
@@ -3503,27 +3771,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 });
@@ -3531,7 +3810,7 @@ class RestRepository extends Query {
         ...params
       });
     } else {
-      return askTable(params);
+      return askTableSession(params);
     }
   }
 }
@@ -3543,7 +3822,7 @@ _schemaTables$2 = new WeakMap();
 _trace = new WeakMap();
 _insertRecordWithoutId = new WeakSet();
 insertRecordWithoutId_fn = async function(object, columns = ["*"]) {
-  const record = removeLinksFromObject(object);
+  const record = await __privateMethod$2(this, _transformObjectToApi, transformObjectToApi_fn).call(this, object);
   const response = await insertRecord({
     pathParams: {
       workspace: "{workspaceId}",
@@ -3562,7 +3841,7 @@ _insertRecordWithId = new WeakSet();
 insertRecordWithId_fn = async function(recordId, object, columns = ["*"], { createOnly, ifVersion }) {
   if (!recordId)
     return null;
-  const record = removeLinksFromObject(object);
+  const record = await __privateMethod$2(this, _transformObjectToApi, transformObjectToApi_fn).call(this, object);
   const response = await insertRecordWithID({
     pathParams: {
       workspace: "{workspaceId}",
@@ -3580,21 +3859,20 @@ insertRecordWithId_fn = async function(recordId, object, columns = ["*"], { crea
 };
 _insertRecords = new WeakSet();
 insertRecords_fn = async function(objects, { createOnly, ifVersion }) {
-  const chunkedOperations = chunk(
-    objects.map((object) => ({
-      insert: { table: __privateGet$4(this, _table), record: removeLinksFromObject(object), createOnly, ifVersion }
-    })),
-    BULK_OPERATION_MAX_SIZE
-  );
+  const operations = await promiseMap(objects, async (object) => {
+    const record = await __privateMethod$2(this, _transformObjectToApi, transformObjectToApi_fn).call(this, object);
+    return { insert: { table: __privateGet$4(this, _table), record, createOnly, ifVersion } };
+  });
+  const chunkedOperations = chunk(operations, BULK_OPERATION_MAX_SIZE);
   const ids = [];
-  for (const operations of chunkedOperations) {
+  for (const operations2 of chunkedOperations) {
     const { results } = await branchTransaction({
       pathParams: {
         workspace: "{workspaceId}",
         dbBranchName: "{dbBranch}",
         region: "{region}"
       },
-      body: { operations },
+      body: { operations: operations2 },
       ...__privateGet$4(this, _getFetchProps).call(this)
     });
     for (const result of results) {
@@ -3611,7 +3889,7 @@ _updateRecordWithID = new WeakSet();
 updateRecordWithID_fn = async function(recordId, object, columns = ["*"], { ifVersion }) {
   if (!recordId)
     return null;
-  const { id: _id, ...record } = removeLinksFromObject(object);
+  const { id: _id, ...record } = await __privateMethod$2(this, _transformObjectToApi, transformObjectToApi_fn).call(this, object);
   try {
     const response = await updateRecordWithID({
       pathParams: {
@@ -3636,21 +3914,20 @@ updateRecordWithID_fn = async function(recordId, object, columns = ["*"], { ifVe
 };
 _updateRecords = new WeakSet();
 updateRecords_fn = async function(objects, { ifVersion, upsert }) {
-  const chunkedOperations = chunk(
-    objects.map(({ id, ...object }) => ({
-      update: { table: __privateGet$4(this, _table), id, ifVersion, upsert, fields: removeLinksFromObject(object) }
-    })),
-    BULK_OPERATION_MAX_SIZE
-  );
+  const operations = await promiseMap(objects, async ({ id, ...object }) => {
+    const fields = await __privateMethod$2(this, _transformObjectToApi, transformObjectToApi_fn).call(this, object);
+    return { update: { table: __privateGet$4(this, _table), id, ifVersion, upsert, fields } };
+  });
+  const chunkedOperations = chunk(operations, BULK_OPERATION_MAX_SIZE);
   const ids = [];
-  for (const operations of chunkedOperations) {
+  for (const operations2 of chunkedOperations) {
     const { results } = await branchTransaction({
       pathParams: {
         workspace: "{workspaceId}",
         dbBranchName: "{dbBranch}",
         region: "{region}"
       },
-      body: { operations },
+      body: { operations: operations2 },
       ...__privateGet$4(this, _getFetchProps).call(this)
     });
     for (const result of results) {
@@ -3753,12 +4030,40 @@ getSchemaTables_fn$1 = async function() {
   __privateSet$4(this, _schemaTables$2, schema.tables);
   return schema.tables;
 };
-const removeLinksFromObject = (object) => {
-  return Object.entries(object).reduce((acc, [key, value]) => {
+_transformObjectToApi = new WeakSet();
+transformObjectToApi_fn = async function(object) {
+  const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
+  const schema = schemaTables.find((table) => table.name === __privateGet$4(this, _table));
+  if (!schema)
+    throw new Error(`Table ${__privateGet$4(this, _table)} not found in schema`);
+  const result = {};
+  for (const [key, value] of Object.entries(object)) {
     if (key === "xata")
-      return acc;
-    return { ...acc, [key]: isIdentifiable(value) ? value.id : value };
-  }, {});
+      continue;
+    const type = schema.columns.find((column) => column.name === key)?.type;
+    switch (type) {
+      case "link": {
+        result[key] = isIdentifiable(value) ? value.id : value;
+        break;
+      }
+      case "datetime": {
+        result[key] = value instanceof Date ? value.toISOString() : value;
+        break;
+      }
+      case `file`:
+        result[key] = await parseInputFileEntry(value);
+        break;
+      case "file[]":
+        result[key] = await promiseMap(value, (item) => parseInputFileEntry(item));
+        break;
+      case "json":
+        result[key] = stringifyJson(value);
+        break;
+      default:
+        result[key] = value;
+    }
+  }
+  return result;
 };
 const initObject = (db, schemaTables, table, object, selectedColumns) => {
   const data = {};
@@ -3790,18 +4095,33 @@ 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;
         }
         break;
       }
+      case "file":
+        data[column.name] = isDefined(value) ? new XataFile(value) : null;
+        break;
+      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) {
@@ -3811,33 +4131,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 });
@@ -3855,7 +4176,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) {
@@ -3866,12 +4187,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);
@@ -3894,8 +4209,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;
@@ -3940,10 +4253,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 });
@@ -3999,6 +4314,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);
@@ -4034,22 +4423,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 };
       }
     };
   }
@@ -4058,13 +4451,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) {
@@ -4078,6 +4471,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 {
@@ -4093,12 +4558,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);
@@ -4128,9 +4587,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");
       const safeOptions = __privateMethod(this, _parseOptions, parseOptions_fn).call(this, options);
       __privateSet(this, _options, safeOptions);
       const pluginOptions = {
@@ -4141,9 +4597,13 @@ 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)
           continue;
@@ -4240,17 +4700,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;
@@ -4313,34 +4767,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, 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, 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, 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, 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