@xata.io/client 0.0.0-alpha.vecada6d → 0.0.0-alpha.ved155c7

package/dist/index.mjs

@@ -1,3 +1,25 @@
+const defaultTrace = async (_name, fn, _options) => {
+  return await fn({
+    setAttributes: () => {
+      return;
+    }
+  });
+};
+const TraceAttributes = {
+  KIND: "xata.trace.kind",
+  VERSION: "xata.sdk.version",
+  TABLE: "xata.table",
+  HTTP_REQUEST_ID: "http.request_id",
+  HTTP_STATUS_CODE: "http.status_code",
+  HTTP_HOST: "http.host",
+  HTTP_SCHEME: "http.scheme",
+  HTTP_USER_AGENT: "http.user_agent",
+  HTTP_METHOD: "http.method",
+  HTTP_URL: "http.url",
+  HTTP_ROUTE: "http.route",
+  HTTP_TARGET: "http.target"
+};
+
 function notEmpty(value) {
   return value !== null && value !== void 0;
 }
@@ -7,46 +29,101 @@ function compact(arr) {
 function isObject(value) {
   return Boolean(value) && typeof value === "object" && !Array.isArray(value);
 }
+function isDefined(value) {
+  return value !== null && value !== void 0;
+}
 function isString(value) {
-  return value !== void 0 && value !== null && typeof value === "string";
+  return isDefined(value) && typeof value === "string";
+}
+function isStringArray(value) {
+  return isDefined(value) && Array.isArray(value) && value.every(isString);
 }
 function toBase64(value) {
   try {
     return btoa(value);
   } catch (err) {
-    return Buffer.from(value).toString("base64");
+    const buf = Buffer;
+    return buf.from(value).toString("base64");
   }
 }
 
-function getEnvVariable(name) {
+function getEnvironment() {
   try {
-    if (isObject(process) && isString(process?.env?.[name])) {
-      return process.env[name];
+    if (isObject(process) && isObject(process.env)) {
+      return {
+        apiKey: process.env.XATA_API_KEY ?? getGlobalApiKey(),
+        databaseURL: process.env.XATA_DATABASE_URL ?? getGlobalDatabaseURL(),
+        branch: process.env.XATA_BRANCH ?? getGlobalBranch(),
+        envBranch: process.env.VERCEL_GIT_COMMIT_REF ?? process.env.CF_PAGES_BRANCH ?? process.env.BRANCH,
+        fallbackBranch: process.env.XATA_FALLBACK_BRANCH ?? getGlobalFallbackBranch()
+      };
     }
   } catch (err) {
   }
   try {
-    if (isObject(Deno) && isString(Deno?.env?.get(name))) {
-      return Deno.env.get(name);
+    if (isObject(Deno) && isObject(Deno.env)) {
+      return {
+        apiKey: Deno.env.get("XATA_API_KEY") ?? getGlobalApiKey(),
+        databaseURL: Deno.env.get("XATA_DATABASE_URL") ?? getGlobalDatabaseURL(),
+        branch: Deno.env.get("XATA_BRANCH") ?? getGlobalBranch(),
+        envBranch: Deno.env.get("VERCEL_GIT_COMMIT_REF") ?? Deno.env.get("CF_PAGES_BRANCH") ?? Deno.env.get("BRANCH"),
+        fallbackBranch: Deno.env.get("XATA_FALLBACK_BRANCH") ?? getGlobalFallbackBranch()
+      };
     }
   } catch (err) {
   }
+  return {
+    apiKey: getGlobalApiKey(),
+    databaseURL: getGlobalDatabaseURL(),
+    branch: getGlobalBranch(),
+    envBranch: void 0,
+    fallbackBranch: getGlobalFallbackBranch()
+  };
+}
+function getGlobalApiKey() {
+  try {
+    return XATA_API_KEY;
+  } catch (err) {
+    return void 0;
+  }
+}
+function getGlobalDatabaseURL() {
+  try {
+    return XATA_DATABASE_URL;
+  } catch (err) {
+    return void 0;
+  }
+}
+function getGlobalBranch() {
+  try {
+    return XATA_BRANCH;
+  } catch (err) {
+    return void 0;
+  }
+}
+function getGlobalFallbackBranch() {
+  try {
+    return XATA_FALLBACK_BRANCH;
+  } catch (err) {
+    return void 0;
+  }
 }
 async function getGitBranch() {
+  const cmd = ["git", "branch", "--show-current"];
+  const fullCmd = cmd.join(" ");
+  const nodeModule = ["child", "process"].join("_");
+  const execOptions = { encoding: "utf-8", stdio: ["ignore", "pipe", "ignore"] };
   try {
     if (typeof require === "function") {
-      const req = require;
-      return req("child_process").execSync("git branch --show-current", { encoding: "utf-8" }).trim();
+      return require(nodeModule).execSync(fullCmd, execOptions).trim();
     }
+    const { execSync } = await import(nodeModule);
+    return execSync(fullCmd, execOptions).toString().trim();
   } catch (err) {
   }
   try {
     if (isObject(Deno)) {
-      const process2 = Deno.run({
-        cmd: ["git", "branch", "--show-current"],
-        stdout: "piped",
-        stderr: "piped"
-      });
+      const process2 = Deno.run({ cmd, stdout: "piped", stderr: "null" });
       return new TextDecoder().decode(await process2.output()).trim();
     }
   } catch (err) {
@@ -55,7 +132,8 @@ async function getGitBranch() {
 
 function getAPIKey() {
   try {
-    return getEnvVariable("XATA_API_KEY") ?? XATA_API_KEY;
+    const { apiKey } = getEnvironment();
+    return apiKey;
   } catch (err) {
     return void 0;
   }
@@ -65,21 +143,35 @@ function getFetchImplementation(userFetch) {
   const globalFetch = typeof fetch !== "undefined" ? fetch : void 0;
   const fetchImpl = userFetch ?? globalFetch;
   if (!fetchImpl) {
-    throw new Error(`The \`fetch\` option passed to the Xata client is resolving to a falsy value and may not be correctly imported.`);
+    throw new Error(
+      `Couldn't find \`fetch\`. Install a fetch implementation such as \`node-fetch\` and pass it explicitly.`
+    );
   }
   return fetchImpl;
 }
 
-class FetcherError extends Error {
-  constructor(status, data) {
+const VERSION = "0.0.0-alpha.ved155c7";
+
+class ErrorWithCause extends Error {
+  constructor(message, options) {
+    super(message, options);
+  }
+}
+class FetcherError extends ErrorWithCause {
+  constructor(status, data, requestId) {
     super(getMessage(data));
     this.status = status;
     this.errors = isBulkError(data) ? data.errors : void 0;
+    this.requestId = requestId;
     if (data instanceof Error) {
       this.stack = data.stack;
       this.cause = data.cause;
     }
   }
+  toString() {
+    const error = super.toString();
+    return `[${this.status}] (${this.requestId ?? "Unknown"}): ${error}`;
+  }
 }
 function isBulkError(error) {
   return isObject(error) && Array.isArray(error.errors);
@@ -102,9 +194,17 @@ function getMessage(data) {
 }
 
 const resolveUrl = (url, queryParams = {}, pathParams = {}) => {
-  const query = new URLSearchParams(queryParams).toString();
+  const cleanQueryParams = Object.entries(queryParams).reduce((acc, [key, value]) => {
+    if (value === void 0 || value === null)
+      return acc;
+    return { ...acc, [key]: value };
+  }, {});
+  const query = new URLSearchParams(cleanQueryParams).toString();
   const queryString = query.length > 0 ? `?${query}` : "";
-  return url.replace(/\{\w*\}/g, (key) => pathParams[key.slice(1, -1)]) + queryString;
+  const cleanPathParams = Object.entries(pathParams).reduce((acc, [key, value]) => {
+    return { ...acc, [key]: encodeURIComponent(String(value ?? "")).replace("%3A", ":") };
+  }, {});
+  return url.replace(/\{\w*\}/g, (key) => cleanPathParams[key.slice(1, -1)]) + queryString;
 };
 function buildBaseUrl({
   path,
@@ -112,10 +212,10 @@ function buildBaseUrl({
   apiUrl,
   pathParams
 }) {
-  if (!pathParams?.workspace)
+  if (pathParams?.workspace === void 0)
     return `${apiUrl}${path}`;
   const url = typeof workspacesApiUrl === "string" ? `${workspacesApiUrl}${path}` : workspacesApiUrl(path, pathParams);
-  return url.replace("{workspaceId}", pathParams.workspace);
+  return url.replace("{workspaceId}", String(pathParams.workspace));
 }
 function hostHeader(url) {
   const pattern = /.*:\/\/(?<host>[^/]+).*/;
@@ -132,32 +232,61 @@ async function fetch$1({
   fetchImpl,
   apiKey,
   apiUrl,
-  workspacesApiUrl
+  workspacesApiUrl,
+  trace
 }) {
-  const baseUrl = buildBaseUrl({ path, workspacesApiUrl, pathParams, apiUrl });
-  const fullUrl = resolveUrl(baseUrl, queryParams, pathParams);
-  const url = fullUrl.includes("localhost") ? fullUrl.replace(/^[^.]+\./, "http://") : fullUrl;
-  const response = await fetchImpl(url, {
-    method: method.toUpperCase(),
-    body: body ? JSON.stringify(body) : void 0,
-    headers: {
-      "Content-Type": "application/json",
-      ...headers,
-      ...hostHeader(fullUrl),
-      Authorization: `Bearer ${apiKey}`
-    }
-  });
-  if (response.status === 204) {
-    return {};
-  }
+  return trace(
+    `${method.toUpperCase()} ${path}`,
+    async ({ setAttributes }) => {
+      const baseUrl = buildBaseUrl({ path, workspacesApiUrl, pathParams, apiUrl });
+      const fullUrl = resolveUrl(baseUrl, queryParams, pathParams);
+      const url = fullUrl.includes("localhost") ? fullUrl.replace(/^[^.]+\./, "http://") : fullUrl;
+      setAttributes({
+        [TraceAttributes.HTTP_URL]: url,
+        [TraceAttributes.HTTP_TARGET]: resolveUrl(path, queryParams, pathParams)
+      });
+      const response = await fetchImpl(url, {
+        method: method.toUpperCase(),
+        body: body ? JSON.stringify(body) : void 0,
+        headers: {
+          "Content-Type": "application/json",
+          "User-Agent": `Xata client-ts/${VERSION}`,
+          ...headers,
+          ...hostHeader(fullUrl),
+          Authorization: `Bearer ${apiKey}`
+        }
+      });
+      if (response.status === 204) {
+        return {};
+      }
+      const { host, protocol } = parseUrl(response.url);
+      const requestId = response.headers?.get("x-request-id") ?? void 0;
+      setAttributes({
+        [TraceAttributes.KIND]: "http",
+        [TraceAttributes.HTTP_REQUEST_ID]: requestId,
+        [TraceAttributes.HTTP_STATUS_CODE]: response.status,
+        [TraceAttributes.HTTP_HOST]: host,
+        [TraceAttributes.HTTP_SCHEME]: protocol?.replace(":", "")
+      });
+      try {
+        const jsonResponse = await response.json();
+        if (response.ok) {
+          return jsonResponse;
+        }
+        throw new FetcherError(response.status, jsonResponse, requestId);
+      } catch (error) {
+        throw new FetcherError(response.status, error, requestId);
+      }
+    },
+    { [TraceAttributes.HTTP_METHOD]: method.toUpperCase(), [TraceAttributes.HTTP_ROUTE]: path }
+  );
+}
+function parseUrl(url) {
   try {
-    const jsonResponse = await response.json();
-    if (response.ok) {
-      return jsonResponse;
-    }
-    throw new FetcherError(response.status, jsonResponse);
+    const { host, protocol } = new URL(url);
+    return { host, protocol };
   } catch (error) {
-    throw new FetcherError(response.status, error);
+    return {};
   }
 }
 
@@ -216,6 +345,7 @@ const removeWorkspaceMember = (variables) => fetch$1({
   ...variables
 });
 const inviteWorkspaceMember = (variables) => fetch$1({ url: "/workspaces/{workspaceId}/invites", method: "post", ...variables });
+const updateWorkspaceMemberInvite = (variables) => fetch$1({ url: "/workspaces/{workspaceId}/invites/{inviteId}", method: "patch", ...variables });
 const cancelWorkspaceMemberInvite = (variables) => fetch$1({
   url: "/workspaces/{workspaceId}/invites/{inviteId}",
   method: "delete",
@@ -251,6 +381,12 @@ const deleteDatabase = (variables) => fetch$1({
   method: "delete",
   ...variables
 });
+const getDatabaseMetadata = (variables) => fetch$1({
+  url: "/dbs/{dbName}/metadata",
+  method: "get",
+  ...variables
+});
+const patchDatabaseMetadata = (variables) => fetch$1({ url: "/dbs/{dbName}/metadata", method: "patch", ...variables });
 const getGitBranchesMapping = (variables) => fetch$1({ url: "/dbs/{dbName}/gitBranches", method: "get", ...variables });
 const addGitBranchesEntry = (variables) => fetch$1({ url: "/dbs/{dbName}/gitBranches", method: "post", ...variables });
 const removeGitBranchesEntry = (variables) => fetch$1({ url: "/dbs/{dbName}/gitBranches", method: "delete", ...variables });
@@ -259,16 +395,28 @@ const resolveBranch = (variables) => fetch$1({
   method: "get",
   ...variables
 });
-const getBranchDetails = (variables) => fetch$1({
-  url: "/db/{dbBranchName}",
+const listMigrationRequests = (variables) => fetch$1({ url: "/dbs/{dbName}/migrations/list", method: "post", ...variables });
+const createMigrationRequest = (variables) => fetch$1({ url: "/dbs/{dbName}/migrations", method: "post", ...variables });
+const getMigrationRequest = (variables) => fetch$1({
+  url: "/dbs/{dbName}/migrations/{mrNumber}",
   method: "get",
   ...variables
 });
-const createBranch = (variables) => fetch$1({
+const updateMigrationRequest = (variables) => fetch$1({ url: "/dbs/{dbName}/migrations/{mrNumber}", method: "patch", ...variables });
+const listMigrationRequestsCommits = (variables) => fetch$1({ url: "/dbs/{dbName}/migrations/{mrNumber}/commits", method: "post", ...variables });
+const compareMigrationRequest = (variables) => fetch$1({ url: "/dbs/{dbName}/migrations/{mrNumber}/compare", method: "post", ...variables });
+const getMigrationRequestIsMerged = (variables) => fetch$1({ url: "/dbs/{dbName}/migrations/{mrNumber}/merge", method: "get", ...variables });
+const mergeMigrationRequest = (variables) => fetch$1({
+  url: "/dbs/{dbName}/migrations/{mrNumber}/merge",
+  method: "post",
+  ...variables
+});
+const getBranchDetails = (variables) => fetch$1({
   url: "/db/{dbBranchName}",
-  method: "put",
+  method: "get",
   ...variables
 });
+const createBranch = (variables) => fetch$1({ url: "/db/{dbBranchName}", method: "put", ...variables });
 const deleteBranch = (variables) => fetch$1({
   url: "/db/{dbBranchName}",
   method: "delete",
@@ -287,6 +435,16 @@ const getBranchMetadata = (variables) => fetch$1({
 const getBranchMigrationHistory = (variables) => fetch$1({ url: "/db/{dbBranchName}/migrations", method: "get", ...variables });
 const executeBranchMigrationPlan = (variables) => fetch$1({ url: "/db/{dbBranchName}/migrations/execute", method: "post", ...variables });
 const getBranchMigrationPlan = (variables) => fetch$1({ url: "/db/{dbBranchName}/migrations/plan", method: "post", ...variables });
+const compareBranchWithUserSchema = (variables) => fetch$1({ url: "/db/{dbBranchName}/schema/compare", method: "post", ...variables });
+const compareBranchSchemas = (variables) => fetch$1({ url: "/db/{dbBranchName}/schema/compare/{branchName}", method: "post", ...variables });
+const updateBranchSchema = (variables) => fetch$1({
+  url: "/db/{dbBranchName}/schema/update",
+  method: "post",
+  ...variables
+});
+const previewBranchSchemaEdit = (variables) => fetch$1({ url: "/db/{dbBranchName}/schema/preview", method: "post", ...variables });
+const applyBranchSchemaEdit = (variables) => fetch$1({ url: "/db/{dbBranchName}/schema/apply", method: "post", ...variables });
+const getBranchSchemaHistory = (variables) => fetch$1({ url: "/db/{dbBranchName}/schema/history", method: "post", ...variables });
 const getBranchStats = (variables) => fetch$1({
   url: "/db/{dbBranchName}/stats",
   method: "get",
@@ -342,11 +500,7 @@ const updateColumn = (variables) => fetch$1({
   method: "patch",
   ...variables
 });
-const insertRecord = (variables) => fetch$1({
-  url: "/db/{dbBranchName}/tables/{tableName}/data",
-  method: "post",
-  ...variables
-});
+const insertRecord = (variables) => fetch$1({ url: "/db/{dbBranchName}/tables/{tableName}/data", method: "post", ...variables });
 const insertRecordWithID = (variables) => fetch$1({ url: "/db/{dbBranchName}/tables/{tableName}/data/{recordId}", method: "put", ...variables });
 const updateRecordWithID = (variables) => fetch$1({ url: "/db/{dbBranchName}/tables/{tableName}/data/{recordId}", method: "patch", ...variables });
 const upsertRecordWithID = (variables) => fetch$1({ url: "/db/{dbBranchName}/tables/{tableName}/data/{recordId}", method: "post", ...variables });
@@ -388,6 +542,7 @@ const operationsByTag = {
     updateWorkspaceMemberRole,
     removeWorkspaceMember,
     inviteWorkspaceMember,
+    updateWorkspaceMemberInvite,
     cancelWorkspaceMemberInvite,
     resendWorkspaceMemberInvite,
     acceptWorkspaceMemberInvite
@@ -396,6 +551,8 @@ const operationsByTag = {
     getDatabaseList,
     createDatabase,
     deleteDatabase,
+    getDatabaseMetadata,
+    patchDatabaseMetadata,
     getGitBranchesMapping,
     addGitBranchesEntry,
     removeGitBranchesEntry,
@@ -408,10 +565,28 @@ const operationsByTag = {
     deleteBranch,
     updateBranchMetadata,
     getBranchMetadata,
+    getBranchStats
+  },
+  migrationRequests: {
+    listMigrationRequests,
+    createMigrationRequest,
+    getMigrationRequest,
+    updateMigrationRequest,
+    listMigrationRequestsCommits,
+    compareMigrationRequest,
+    getMigrationRequestIsMerged,
+    mergeMigrationRequest
+  },
+  branchSchema: {
     getBranchMigrationHistory,
     executeBranchMigrationPlan,
     getBranchMigrationPlan,
-    getBranchStats
+    compareBranchWithUserSchema,
+    compareBranchSchemas,
+    updateBranchSchema,
+    previewBranchSchemaEdit,
+    applyBranchSchemaEdit,
+    getBranchSchemaHistory
   },
   table: {
     createTable,
@@ -440,9 +615,9 @@ const operationsByTag = {
 };
 
 function getHostUrl(provider, type) {
-  if (isValidAlias(provider)) {
+  if (isHostProviderAlias(provider)) {
     return providers[provider][type];
-  } else if (isValidBuilder(provider)) {
+  } else if (isHostProviderBuilder(provider)) {
     return provider[type];
   }
   throw new Error("Invalid API provider");
@@ -457,10 +632,10 @@ const providers = {
     workspaces: "https://{workspaceId}.staging.xatabase.co"
   }
 };
-function isValidAlias(alias) {
+function isHostProviderAlias(alias) {
   return isString(alias) && Object.keys(providers).includes(alias);
 }
-function isValidBuilder(builder) {
+function isHostProviderBuilder(builder) {
   return isObject(builder) && isString(builder.main) && isString(builder.workspaces);
 }
 
@@ -477,7 +652,7 @@ var __privateAdd$7 = (obj, member, value) => {
     throw TypeError("Cannot add the same private member more than once");
   member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
 };
-var __privateSet$6 = (obj, member, value, setter) => {
+var __privateSet$7 = (obj, member, value, setter) => {
   __accessCheck$7(obj, member, "write to private field");
   setter ? setter.call(obj, value) : member.set(obj, value);
   return value;
@@ -488,15 +663,17 @@ class XataApiClient {
     __privateAdd$7(this, _extraProps, void 0);
     __privateAdd$7(this, _namespaces, {});
     const provider = options.host ?? "production";
-    const apiKey = options?.apiKey ?? getAPIKey();
+    const apiKey = options.apiKey ?? getAPIKey();
+    const trace = options.trace ?? defaultTrace;
     if (!apiKey) {
       throw new Error("Could not resolve a valid apiKey");
     }
-    __privateSet$6(this, _extraProps, {
+    __privateSet$7(this, _extraProps, {
       apiUrl: getHostUrl(provider, "main"),
       workspacesApiUrl: getHostUrl(provider, "workspaces"),
       fetchImpl: getFetchImplementation(options.fetch),
-      apiKey
+      apiKey,
+      trace
     });
   }
   get user() {
@@ -529,6 +706,16 @@ class XataApiClient {
       __privateGet$7(this, _namespaces).records = new RecordsApi(__privateGet$7(this, _extraProps));
     return __privateGet$7(this, _namespaces).records;
   }
+  get migrationRequests() {
+    if (!__privateGet$7(this, _namespaces).migrationRequests)
+      __privateGet$7(this, _namespaces).migrationRequests = new MigrationRequestsApi(__privateGet$7(this, _extraProps));
+    return __privateGet$7(this, _namespaces).migrationRequests;
+  }
+  get branchSchema() {
+    if (!__privateGet$7(this, _namespaces).branchSchema)
+      __privateGet$7(this, _namespaces).branchSchema = new BranchSchemaApi(__privateGet$7(this, _extraProps));
+    return __privateGet$7(this, _namespaces).branchSchema;
+  }
 }
 _extraProps = new WeakMap();
 _namespaces = new WeakMap();
@@ -619,6 +806,13 @@ class WorkspaceApi {
       ...this.extraProps
     });
   }
+  updateWorkspaceMemberInvite(workspaceId, inviteId, role) {
+    return operationsByTag.workspaces.updateWorkspaceMemberInvite({
+      pathParams: { workspaceId, inviteId },
+      body: { role },
+      ...this.extraProps
+    });
+  }
   cancelWorkspaceMemberInvite(workspaceId, inviteId) {
     return operationsByTag.workspaces.cancelWorkspaceMemberInvite({
       pathParams: { workspaceId, inviteId },
@@ -661,6 +855,19 @@ class DatabaseApi {
       ...this.extraProps
     });
   }
+  getDatabaseMetadata(workspace, dbName) {
+    return operationsByTag.database.getDatabaseMetadata({
+      pathParams: { workspace, dbName },
+      ...this.extraProps
+    });
+  }
+  patchDatabaseMetadata(workspace, dbName, options = {}) {
+    return operationsByTag.database.patchDatabaseMetadata({
+      pathParams: { workspace, dbName },
+      body: options,
+      ...this.extraProps
+    });
+  }
   getGitBranchesMapping(workspace, dbName) {
     return operationsByTag.database.getGitBranchesMapping({
       pathParams: { workspace, dbName },
@@ -681,10 +888,10 @@ class DatabaseApi {
       ...this.extraProps
     });
   }
-  resolveBranch(workspace, dbName, gitBranch) {
+  resolveBranch(workspace, dbName, gitBranch, fallbackBranch) {
     return operationsByTag.database.resolveBranch({
       pathParams: { workspace, dbName },
-      queryParams: { gitBranch },
+      queryParams: { gitBranch, fallbackBranch },
       ...this.extraProps
     });
   }
@@ -732,27 +939,6 @@ class BranchApi {
       ...this.extraProps
     });
   }
-  getBranchMigrationHistory(workspace, database, branch, options = {}) {
-    return operationsByTag.branch.getBranchMigrationHistory({
-      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
-      body: options,
-      ...this.extraProps
-    });
-  }
-  executeBranchMigrationPlan(workspace, database, branch, migrationPlan) {
-    return operationsByTag.branch.executeBranchMigrationPlan({
-      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
-      body: migrationPlan,
-      ...this.extraProps
-    });
-  }
-  getBranchMigrationPlan(workspace, database, branch, schema) {
-    return operationsByTag.branch.getBranchMigrationPlan({
-      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
-      body: schema,
-      ...this.extraProps
-    });
-  }
   getBranchStats(workspace, database, branch) {
     return operationsByTag.branch.getBranchStats({
       pathParams: { workspace, dbBranchName: `${database}:${branch}` },
@@ -833,9 +1019,10 @@ class RecordsApi {
   constructor(extraProps) {
     this.extraProps = extraProps;
   }
-  insertRecord(workspace, database, branch, tableName, record) {
+  insertRecord(workspace, database, branch, tableName, record, options = {}) {
     return operationsByTag.records.insertRecord({
       pathParams: { workspace, dbBranchName: `${database}:${branch}`, tableName },
+      queryParams: options,
       body: record,
       ...this.extraProps
     });
@@ -864,21 +1051,24 @@ class RecordsApi {
       ...this.extraProps
     });
   }
-  deleteRecord(workspace, database, branch, tableName, recordId) {
+  deleteRecord(workspace, database, branch, tableName, recordId, options = {}) {
     return operationsByTag.records.deleteRecord({
       pathParams: { workspace, dbBranchName: `${database}:${branch}`, tableName, recordId },
+      queryParams: options,
       ...this.extraProps
     });
   }
   getRecord(workspace, database, branch, tableName, recordId, options = {}) {
     return operationsByTag.records.getRecord({
       pathParams: { workspace, dbBranchName: `${database}:${branch}`, tableName, recordId },
+      queryParams: options,
       ...this.extraProps
     });
   }
-  bulkInsertTableRecords(workspace, database, branch, tableName, records) {
+  bulkInsertTableRecords(workspace, database, branch, tableName, records, options = {}) {
     return operationsByTag.records.bulkInsertTableRecords({
       pathParams: { workspace, dbBranchName: `${database}:${branch}`, tableName },
+      queryParams: options,
       body: { records },
       ...this.extraProps
     });
@@ -905,6 +1095,131 @@ class RecordsApi {
     });
   }
 }
+class MigrationRequestsApi {
+  constructor(extraProps) {
+    this.extraProps = extraProps;
+  }
+  listMigrationRequests(workspace, database, options = {}) {
+    return operationsByTag.migrationRequests.listMigrationRequests({
+      pathParams: { workspace, dbName: database },
+      body: options,
+      ...this.extraProps
+    });
+  }
+  createMigrationRequest(workspace, database, options) {
+    return operationsByTag.migrationRequests.createMigrationRequest({
+      pathParams: { workspace, dbName: database },
+      body: options,
+      ...this.extraProps
+    });
+  }
+  getMigrationRequest(workspace, database, migrationRequest) {
+    return operationsByTag.migrationRequests.getMigrationRequest({
+      pathParams: { workspace, dbName: database, mrNumber: migrationRequest },
+      ...this.extraProps
+    });
+  }
+  updateMigrationRequest(workspace, database, migrationRequest, options) {
+    return operationsByTag.migrationRequests.updateMigrationRequest({
+      pathParams: { workspace, dbName: database, mrNumber: migrationRequest },
+      body: options,
+      ...this.extraProps
+    });
+  }
+  listMigrationRequestsCommits(workspace, database, migrationRequest, options = {}) {
+    return operationsByTag.migrationRequests.listMigrationRequestsCommits({
+      pathParams: { workspace, dbName: database, mrNumber: migrationRequest },
+      body: options,
+      ...this.extraProps
+    });
+  }
+  compareMigrationRequest(workspace, database, migrationRequest) {
+    return operationsByTag.migrationRequests.compareMigrationRequest({
+      pathParams: { workspace, dbName: database, mrNumber: migrationRequest },
+      ...this.extraProps
+    });
+  }
+  getMigrationRequestIsMerged(workspace, database, migrationRequest) {
+    return operationsByTag.migrationRequests.getMigrationRequestIsMerged({
+      pathParams: { workspace, dbName: database, mrNumber: migrationRequest },
+      ...this.extraProps
+    });
+  }
+  mergeMigrationRequest(workspace, database, migrationRequest) {
+    return operationsByTag.migrationRequests.mergeMigrationRequest({
+      pathParams: { workspace, dbName: database, mrNumber: migrationRequest },
+      ...this.extraProps
+    });
+  }
+}
+class BranchSchemaApi {
+  constructor(extraProps) {
+    this.extraProps = extraProps;
+  }
+  getBranchMigrationHistory(workspace, database, branch, options = {}) {
+    return operationsByTag.branchSchema.getBranchMigrationHistory({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
+      body: options,
+      ...this.extraProps
+    });
+  }
+  executeBranchMigrationPlan(workspace, database, branch, migrationPlan) {
+    return operationsByTag.branchSchema.executeBranchMigrationPlan({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
+      body: migrationPlan,
+      ...this.extraProps
+    });
+  }
+  getBranchMigrationPlan(workspace, database, branch, schema) {
+    return operationsByTag.branchSchema.getBranchMigrationPlan({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
+      body: schema,
+      ...this.extraProps
+    });
+  }
+  compareBranchWithUserSchema(workspace, database, branch, schema) {
+    return operationsByTag.branchSchema.compareBranchWithUserSchema({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
+      body: { schema },
+      ...this.extraProps
+    });
+  }
+  compareBranchSchemas(workspace, database, branch, branchName, schema) {
+    return operationsByTag.branchSchema.compareBranchSchemas({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}`, branchName },
+      body: { schema },
+      ...this.extraProps
+    });
+  }
+  updateBranchSchema(workspace, database, branch, migration) {
+    return operationsByTag.branchSchema.updateBranchSchema({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
+      body: migration,
+      ...this.extraProps
+    });
+  }
+  previewBranchSchemaEdit(workspace, database, branch, migration) {
+    return operationsByTag.branchSchema.previewBranchSchemaEdit({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
+      body: migration,
+      ...this.extraProps
+    });
+  }
+  applyBranchSchemaEdit(workspace, database, branch, edits) {
+    return operationsByTag.branchSchema.applyBranchSchemaEdit({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
+      body: { edits },
+      ...this.extraProps
+    });
+  }
+  getBranchSchemaHistory(workspace, database, branch, options = {}) {
+    return operationsByTag.branchSchema.getBranchSchemaHistory({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
+      body: options,
+      ...this.extraProps
+    });
+  }
+}
 
 class XataApiPlugin {
   async build(options) {
@@ -929,18 +1244,18 @@ var __privateAdd$6 = (obj, member, value) => {
     throw TypeError("Cannot add the same private member more than once");
   member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
 };
-var __privateSet$5 = (obj, member, value, setter) => {
+var __privateSet$6 = (obj, member, value, setter) => {
   __accessCheck$6(obj, member, "write to private field");
   setter ? setter.call(obj, value) : member.set(obj, value);
   return value;
 };
-var _query;
+var _query, _page;
 class Page {
   constructor(query, meta, records = []) {
     __privateAdd$6(this, _query, void 0);
-    __privateSet$5(this, _query, query);
+    __privateSet$6(this, _query, query);
     this.meta = meta;
-    this.records = records;
+    this.records = new RecordArray(this, records);
   }
   async nextPage(size, offset) {
     return __privateGet$6(this, _query).getPaginated({ pagination: { size, offset, after: this.meta.page.cursor } });
@@ -960,9 +1275,56 @@ class Page {
 }
 _query = new WeakMap();
 const PAGINATION_MAX_SIZE = 200;
-const PAGINATION_DEFAULT_SIZE = 200;
+const PAGINATION_DEFAULT_SIZE = 20;
 const PAGINATION_MAX_OFFSET = 800;
 const PAGINATION_DEFAULT_OFFSET = 0;
+function isCursorPaginationOptions(options) {
+  return isDefined(options) && (isDefined(options.first) || isDefined(options.last) || isDefined(options.after) || isDefined(options.before));
+}
+const _RecordArray = class extends Array {
+  constructor(...args) {
+    super(..._RecordArray.parseConstructorParams(...args));
+    __privateAdd$6(this, _page, void 0);
+    __privateSet$6(this, _page, isObject(args[0]?.meta) ? args[0] : { meta: { page: { cursor: "", more: false } }, records: [] });
+  }
+  static parseConstructorParams(...args) {
+    if (args.length === 1 && typeof args[0] === "number") {
+      return new Array(args[0]);
+    }
+    if (args.length <= 2 && isObject(args[0]?.meta) && Array.isArray(args[1] ?? [])) {
+      const result = args[1] ?? args[0].records ?? [];
+      return new Array(...result);
+    }
+    return new Array(...args);
+  }
+  toArray() {
+    return new Array(...this);
+  }
+  map(callbackfn, thisArg) {
+    return this.toArray().map(callbackfn, thisArg);
+  }
+  async nextPage(size, offset) {
+    const newPage = await __privateGet$6(this, _page).nextPage(size, offset);
+    return new _RecordArray(newPage);
+  }
+  async previousPage(size, offset) {
+    const newPage = await __privateGet$6(this, _page).previousPage(size, offset);
+    return new _RecordArray(newPage);
+  }
+  async firstPage(size, offset) {
+    const newPage = await __privateGet$6(this, _page).firstPage(size, offset);
+    return new _RecordArray(newPage);
+  }
+  async lastPage(size, offset) {
+    const newPage = await __privateGet$6(this, _page).lastPage(size, offset);
+    return new _RecordArray(newPage);
+  }
+  hasNextPage() {
+    return __privateGet$6(this, _page).meta.page.more;
+  }
+};
+let RecordArray = _RecordArray;
+_page = new WeakMap();
 
 var __accessCheck$5 = (obj, member, msg) => {
   if (!member.has(obj))
@@ -977,25 +1339,26 @@ var __privateAdd$5 = (obj, member, value) => {
     throw TypeError("Cannot add the same private member more than once");
   member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
 };
-var __privateSet$4 = (obj, member, value, setter) => {
+var __privateSet$5 = (obj, member, value, setter) => {
   __accessCheck$5(obj, member, "write to private field");
   setter ? setter.call(obj, value) : member.set(obj, value);
   return value;
 };
 var _table$1, _repository, _data;
 const _Query = class {
-  constructor(repository, table, data, parent) {
+  constructor(repository, table, data, rawParent) {
     __privateAdd$5(this, _table$1, void 0);
     __privateAdd$5(this, _repository, void 0);
     __privateAdd$5(this, _data, { filter: {} });
     this.meta = { page: { cursor: "start", more: true } };
-    this.records = [];
-    __privateSet$4(this, _table$1, table);
+    this.records = new RecordArray(this, []);
+    __privateSet$5(this, _table$1, table);
     if (repository) {
-      __privateSet$4(this, _repository, repository);
+      __privateSet$5(this, _repository, repository);
     } else {
-      __privateSet$4(this, _repository, this);
+      __privateSet$5(this, _repository, this);
     }
+    const parent = cleanParent(data, rawParent);
     __privateGet$5(this, _data).filter = data.filter ?? parent?.filter ?? {};
     __privateGet$5(this, _data).filter.$any = data.filter?.$any ?? parent?.filter?.$any;
     __privateGet$5(this, _data).filter.$all = data.filter?.$all ?? parent?.filter?.$all;
@@ -1040,21 +1403,34 @@ const _Query = class {
   }
   filter(a, b) {
     if (arguments.length === 1) {
-      const constraints = Object.entries(a).map(([column, constraint]) => ({ [column]: constraint }));
+      const constraints = Object.entries(a ?? {}).map(([column, constraint]) => ({ [column]: constraint }));
       const $all = compact([__privateGet$5(this, _data).filter?.$all].flat().concat(constraints));
       return new _Query(__privateGet$5(this, _repository), __privateGet$5(this, _table$1), { filter: { $all } }, __privateGet$5(this, _data));
     } else {
-      const $all = compact([__privateGet$5(this, _data).filter?.$all].flat().concat([{ [a]: b }]));
+      const constraints = isDefined(a) && isDefined(b) ? [{ [a]: this.defaultFilter(a, b) }] : void 0;
+      const $all = compact([__privateGet$5(this, _data).filter?.$all].flat().concat(constraints));
       return new _Query(__privateGet$5(this, _repository), __privateGet$5(this, _table$1), { filter: { $all } }, __privateGet$5(this, _data));
     }
   }
-  sort(column, direction) {
+  defaultFilter(column, value) {
+    const columnType = __privateGet$5(this, _table$1).schema?.columns.find(({ name }) => name === column)?.type;
+    if (columnType === "multiple" && (isString(value) || isStringArray(value))) {
+      return { $includes: value };
+    }
+    return value;
+  }
+  sort(column, direction = "asc") {
     const originalSort = [__privateGet$5(this, _data).sort ?? []].flat();
     const sort = [...originalSort, { column, direction }];
     return new _Query(__privateGet$5(this, _repository), __privateGet$5(this, _table$1), { sort }, __privateGet$5(this, _data));
   }
   select(columns) {
-    return new _Query(__privateGet$5(this, _repository), __privateGet$5(this, _table$1), { columns }, __privateGet$5(this, _data));
+    return new _Query(
+      __privateGet$5(this, _repository),
+      __privateGet$5(this, _table$1),
+      { columns },
+      __privateGet$5(this, _data)
+    );
   }
   getPaginated(options = {}) {
     const query = new _Query(__privateGet$5(this, _repository), __privateGet$5(this, _table$1), options, __privateGet$5(this, _data));
@@ -1067,18 +1443,21 @@ const _Query = class {
   }
   async *getIterator(options = {}) {
     const { batchSize = 1 } = options;
-    let offset = 0;
-    let end = false;
-    while (!end) {
-      const { records, meta } = await this.getPaginated({ ...options, pagination: { size: batchSize, offset } });
-      yield records;
-      offset += batchSize;
-      end = !meta.page.more;
+    let page = await this.getPaginated({ ...options, pagination: { size: batchSize, offset: 0 } });
+    let more = page.hasNextPage();
+    yield page.records;
+    while (more) {
+      page = await page.nextPage();
+      more = page.hasNextPage();
+      yield page.records;
     }
   }
   async getMany(options = {}) {
-    const { records } = await this.getPaginated(options);
-    return records;
+    const page = await this.getPaginated(options);
+    if (page.hasNextPage() && options.pagination?.size === void 0) {
+      console.trace("Calling getMany does not return all results. Paginate to get all results or call getAll.");
+    }
+    return page.records;
   }
   async getAll(options = {}) {
     const { batchSize = PAGINATION_MAX_SIZE, ...rest } = options;
@@ -1115,12 +1494,20 @@ let Query = _Query;
 _table$1 = new WeakMap();
 _repository = new WeakMap();
 _data = new WeakMap();
+function cleanParent(data, parent) {
+  if (isCursorPaginationOptions(data.pagination)) {
+    return { ...parent, sorting: void 0, filter: void 0 };
+  }
+  return parent;
+}
 
 function isIdentifiable(x) {
   return isObject(x) && isString(x?.id);
 }
 function isXataRecord(x) {
-  return isIdentifiable(x) && typeof x?.xata === "object" && typeof x?.xata?.version === "number";
+  const record = x;
+  const metadata = record?.getMetadata();
+  return isIdentifiable(x) && isObject(metadata) && typeof metadata.version === "number";
 }
 
 function isSortFilterString(value) {
@@ -1159,7 +1546,7 @@ var __privateAdd$4 = (obj, member, value) => {
     throw TypeError("Cannot add the same private member more than once");
   member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
 };
-var __privateSet$3 = (obj, member, value, setter) => {
+var __privateSet$4 = (obj, member, value, setter) => {
   __accessCheck$4(obj, member, "write to private field");
   setter ? setter.call(obj, value) : member.set(obj, value);
   return value;
@@ -1168,184 +1555,228 @@ var __privateMethod$2 = (obj, member, method) => {
   __accessCheck$4(obj, member, "access private method");
   return method;
 };
-var _table, _getFetchProps, _cache, _schema$1, _insertRecordWithoutId, insertRecordWithoutId_fn, _insertRecordWithId, insertRecordWithId_fn, _bulkInsertTableRecords, bulkInsertTableRecords_fn, _updateRecordWithID, updateRecordWithID_fn, _upsertRecordWithID, upsertRecordWithID_fn, _deleteRecord, deleteRecord_fn, _invalidateCache, invalidateCache_fn, _setCacheRecord, setCacheRecord_fn, _getCacheRecord, getCacheRecord_fn, _setCacheQuery, setCacheQuery_fn, _getCacheQuery, getCacheQuery_fn, _getSchema$1, getSchema_fn$1;
+var _table, _getFetchProps, _db, _cache, _schemaTables$2, _trace, _insertRecordWithoutId, insertRecordWithoutId_fn, _insertRecordWithId, insertRecordWithId_fn, _bulkInsertTableRecords, bulkInsertTableRecords_fn, _updateRecordWithID, updateRecordWithID_fn, _upsertRecordWithID, upsertRecordWithID_fn, _deleteRecord, deleteRecord_fn, _setCacheQuery, setCacheQuery_fn, _getCacheQuery, getCacheQuery_fn, _getSchemaTables$1, getSchemaTables_fn$1;
 class Repository extends Query {
 }
 class RestRepository extends Query {
   constructor(options) {
-    super(null, options.table, {});
+    super(
+      null,
+      { name: options.table, schema: options.schemaTables?.find((table) => table.name === options.table) },
+      {}
+    );
     __privateAdd$4(this, _insertRecordWithoutId);
     __privateAdd$4(this, _insertRecordWithId);
     __privateAdd$4(this, _bulkInsertTableRecords);
     __privateAdd$4(this, _updateRecordWithID);
     __privateAdd$4(this, _upsertRecordWithID);
     __privateAdd$4(this, _deleteRecord);
-    __privateAdd$4(this, _invalidateCache);
-    __privateAdd$4(this, _setCacheRecord);
-    __privateAdd$4(this, _getCacheRecord);
     __privateAdd$4(this, _setCacheQuery);
     __privateAdd$4(this, _getCacheQuery);
-    __privateAdd$4(this, _getSchema$1);
+    __privateAdd$4(this, _getSchemaTables$1);
     __privateAdd$4(this, _table, void 0);
     __privateAdd$4(this, _getFetchProps, void 0);
+    __privateAdd$4(this, _db, void 0);
     __privateAdd$4(this, _cache, void 0);
-    __privateAdd$4(this, _schema$1, void 0);
-    __privateSet$3(this, _table, options.table);
-    __privateSet$3(this, _getFetchProps, options.pluginOptions.getFetchProps);
-    this.db = options.db;
-    __privateSet$3(this, _cache, options.pluginOptions.cache);
-  }
-  async create(a, b) {
-    if (Array.isArray(a)) {
-      const records = await __privateMethod$2(this, _bulkInsertTableRecords, bulkInsertTableRecords_fn).call(this, a);
-      await Promise.all(records.map((record) => __privateMethod$2(this, _setCacheRecord, setCacheRecord_fn).call(this, record)));
-      return records;
-    }
-    if (isString(a) && isObject(b)) {
-      if (a === "")
-        throw new Error("The id can't be empty");
-      const record = await __privateMethod$2(this, _insertRecordWithId, insertRecordWithId_fn).call(this, a, b);
-      await __privateMethod$2(this, _setCacheRecord, setCacheRecord_fn).call(this, record);
-      return record;
-    }
-    if (isObject(a) && isString(a.id)) {
-      if (a.id === "")
-        throw new Error("The id can't be empty");
-      const record = await __privateMethod$2(this, _insertRecordWithId, insertRecordWithId_fn).call(this, a.id, { ...a, id: void 0 });
-      await __privateMethod$2(this, _setCacheRecord, setCacheRecord_fn).call(this, record);
-      return record;
-    }
-    if (isObject(a)) {
-      const record = await __privateMethod$2(this, _insertRecordWithoutId, insertRecordWithoutId_fn).call(this, a);
-      await __privateMethod$2(this, _setCacheRecord, setCacheRecord_fn).call(this, record);
-      return record;
-    }
-    throw new Error("Invalid arguments for create method");
-  }
-  async read(recordId) {
-    const cacheRecord = await __privateMethod$2(this, _getCacheRecord, getCacheRecord_fn).call(this, recordId);
-    if (cacheRecord)
-      return cacheRecord;
-    const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
-    try {
-      const response = await getRecord({
-        pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", tableName: __privateGet$4(this, _table), recordId },
-        ...fetchProps
+    __privateAdd$4(this, _schemaTables$2, void 0);
+    __privateAdd$4(this, _trace, void 0);
+    __privateSet$4(this, _table, options.table);
+    __privateSet$4(this, _getFetchProps, options.pluginOptions.getFetchProps);
+    __privateSet$4(this, _db, options.db);
+    __privateSet$4(this, _cache, options.pluginOptions.cache);
+    __privateSet$4(this, _schemaTables$2, options.schemaTables);
+    const trace = options.pluginOptions.trace ?? defaultTrace;
+    __privateSet$4(this, _trace, async (name, fn, options2 = {}) => {
+      return trace(name, fn, {
+        ...options2,
+        [TraceAttributes.TABLE]: __privateGet$4(this, _table),
+        [TraceAttributes.KIND]: "sdk-operation",
+        [TraceAttributes.VERSION]: VERSION
       });
-      const schema = await __privateMethod$2(this, _getSchema$1, getSchema_fn$1).call(this);
-      return initObject(this.db, schema, __privateGet$4(this, _table), response);
-    } catch (e) {
-      if (isObject(e) && e.status === 404) {
-        return null;
+    });
+  }
+  async create(a, b, c) {
+    return __privateGet$4(this, _trace).call(this, "create", async () => {
+      if (Array.isArray(a)) {
+        if (a.length === 0)
+          return [];
+        const columns = isStringArray(b) ? b : void 0;
+        return __privateMethod$2(this, _bulkInsertTableRecords, bulkInsertTableRecords_fn).call(this, a, columns);
       }
-      throw e;
-    }
+      if (isString(a) && isObject(b)) {
+        if (a === "")
+          throw new Error("The id can't be empty");
+        const columns = isStringArray(c) ? c : void 0;
+        return __privateMethod$2(this, _insertRecordWithId, insertRecordWithId_fn).call(this, a, b, columns);
+      }
+      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;
+        return __privateMethod$2(this, _insertRecordWithId, insertRecordWithId_fn).call(this, a.id, { ...a, id: void 0 }, columns);
+      }
+      if (isObject(a)) {
+        const columns = isStringArray(b) ? b : void 0;
+        return __privateMethod$2(this, _insertRecordWithoutId, insertRecordWithoutId_fn).call(this, a, columns);
+      }
+      throw new Error("Invalid arguments for create method");
+    });
   }
-  async update(a, b) {
-    if (Array.isArray(a)) {
-      if (a.length > 100) {
-        console.warn("Bulk update operation is not optimized in the Xata API yet, this request might be slow");
+  async read(a, b) {
+    return __privateGet$4(this, _trace).call(this, "read", async () => {
+      const columns = isStringArray(b) ? b : ["*"];
+      if (Array.isArray(a)) {
+        if (a.length === 0)
+          return [];
+        const ids = a.map((item) => extractId(item));
+        const finalObjects = await this.getAll({ filter: { id: { $any: compact(ids) } }, columns });
+        const dictionary = finalObjects.reduce((acc, object) => {
+          acc[object.id] = object;
+          return acc;
+        }, {});
+        return ids.map((id2) => dictionary[id2 ?? ""] ?? null);
       }
-      return Promise.all(a.map((object) => this.update(object)));
-    }
-    if (isString(a) && isObject(b)) {
-      await __privateMethod$2(this, _invalidateCache, invalidateCache_fn).call(this, a);
-      const record = await __privateMethod$2(this, _updateRecordWithID, updateRecordWithID_fn).call(this, a, b);
-      await __privateMethod$2(this, _setCacheRecord, setCacheRecord_fn).call(this, record);
-      return record;
-    }
-    if (isObject(a) && isString(a.id)) {
-      await __privateMethod$2(this, _invalidateCache, invalidateCache_fn).call(this, a.id);
-      const record = await __privateMethod$2(this, _updateRecordWithID, updateRecordWithID_fn).call(this, a.id, { ...a, id: void 0 });
-      await __privateMethod$2(this, _setCacheRecord, setCacheRecord_fn).call(this, record);
-      return record;
-    }
-    throw new Error("Invalid arguments for update method");
+      const id = extractId(a);
+      if (id) {
+        const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
+        try {
+          const response = await getRecord({
+            pathParams: {
+              workspace: "{workspaceId}",
+              dbBranchName: "{dbBranch}",
+              tableName: __privateGet$4(this, _table),
+              recordId: id
+            },
+            queryParams: { columns },
+            ...fetchProps
+          });
+          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);
+        } catch (e) {
+          if (isObject(e) && e.status === 404) {
+            return null;
+          }
+          throw e;
+        }
+      }
+      return null;
+    });
   }
-  async createOrUpdate(a, b) {
-    if (Array.isArray(a)) {
-      if (a.length > 100) {
-        console.warn("Bulk update operation is not optimized in the Xata API yet, this request might be slow");
+  async update(a, b, c) {
+    return __privateGet$4(this, _trace).call(this, "update", async () => {
+      if (Array.isArray(a)) {
+        if (a.length === 0)
+          return [];
+        if (a.length > 100) {
+          console.warn("Bulk update operation is not optimized in the Xata API yet, this request might be slow");
+        }
+        const columns = isStringArray(b) ? b : ["*"];
+        return Promise.all(a.map((object) => this.update(object, columns)));
       }
-      return Promise.all(a.map((object) => this.createOrUpdate(object)));
-    }
-    if (isString(a) && isObject(b)) {
-      await __privateMethod$2(this, _invalidateCache, invalidateCache_fn).call(this, a);
-      const record = await __privateMethod$2(this, _upsertRecordWithID, upsertRecordWithID_fn).call(this, a, b);
-      await __privateMethod$2(this, _setCacheRecord, setCacheRecord_fn).call(this, record);
-      return record;
-    }
-    if (isObject(a) && isString(a.id)) {
-      await __privateMethod$2(this, _invalidateCache, invalidateCache_fn).call(this, a.id);
-      const record = await __privateMethod$2(this, _upsertRecordWithID, upsertRecordWithID_fn).call(this, a.id, { ...a, id: void 0 });
-      await __privateMethod$2(this, _setCacheRecord, setCacheRecord_fn).call(this, record);
-      return record;
-    }
-    throw new Error("Invalid arguments for createOrUpdate method");
+      if (isString(a) && isObject(b)) {
+        const columns = isStringArray(c) ? c : void 0;
+        return __privateMethod$2(this, _updateRecordWithID, updateRecordWithID_fn).call(this, a, b, columns);
+      }
+      if (isObject(a) && isString(a.id)) {
+        const columns = isStringArray(b) ? b : void 0;
+        return __privateMethod$2(this, _updateRecordWithID, updateRecordWithID_fn).call(this, a.id, { ...a, id: void 0 }, columns);
+      }
+      throw new Error("Invalid arguments for update method");
+    });
   }
-  async delete(a) {
-    if (Array.isArray(a)) {
-      if (a.length > 100) {
-        console.warn("Bulk delete operation is not optimized in the Xata API yet, this request might be slow");
+  async createOrUpdate(a, b, c) {
+    return __privateGet$4(this, _trace).call(this, "createOrUpdate", async () => {
+      if (Array.isArray(a)) {
+        if (a.length === 0)
+          return [];
+        if (a.length > 100) {
+          console.warn("Bulk update operation is not optimized in the Xata API yet, this request might be slow");
+        }
+        const columns = isStringArray(b) ? b : ["*"];
+        return Promise.all(a.map((object) => this.createOrUpdate(object, columns)));
       }
-      await Promise.all(a.map((id) => this.delete(id)));
-      return;
-    }
-    if (isString(a)) {
-      await __privateMethod$2(this, _deleteRecord, deleteRecord_fn).call(this, a);
-      await __privateMethod$2(this, _invalidateCache, invalidateCache_fn).call(this, a);
-      return;
-    }
-    if (isObject(a) && isString(a.id)) {
-      await __privateMethod$2(this, _deleteRecord, deleteRecord_fn).call(this, a.id);
-      await __privateMethod$2(this, _invalidateCache, invalidateCache_fn).call(this, a.id);
-      return;
-    }
-    throw new Error("Invalid arguments for delete method");
+      if (isString(a) && isObject(b)) {
+        const columns = isStringArray(c) ? c : void 0;
+        return __privateMethod$2(this, _upsertRecordWithID, upsertRecordWithID_fn).call(this, a, b, columns);
+      }
+      if (isObject(a) && isString(a.id)) {
+        const columns = isStringArray(c) ? c : void 0;
+        return __privateMethod$2(this, _upsertRecordWithID, upsertRecordWithID_fn).call(this, a.id, { ...a, id: void 0 }, columns);
+      }
+      throw new Error("Invalid arguments for createOrUpdate method");
+    });
+  }
+  async delete(a, b) {
+    return __privateGet$4(this, _trace).call(this, "delete", async () => {
+      if (Array.isArray(a)) {
+        if (a.length === 0)
+          return [];
+        if (a.length > 100) {
+          console.warn("Bulk delete operation is not optimized in the Xata API yet, this request might be slow");
+        }
+        return Promise.all(a.map((id) => this.delete(id, b)));
+      }
+      if (isString(a)) {
+        return __privateMethod$2(this, _deleteRecord, deleteRecord_fn).call(this, a, b);
+      }
+      if (isObject(a) && isString(a.id)) {
+        return __privateMethod$2(this, _deleteRecord, deleteRecord_fn).call(this, a.id, b);
+      }
+      throw new Error("Invalid arguments for delete method");
+    });
   }
   async search(query, options = {}) {
-    const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
-    const { records } = await searchTable({
-      pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", tableName: __privateGet$4(this, _table) },
-      body: {
-        query,
-        fuzziness: options.fuzziness,
-        filter: options.filter
-      },
-      ...fetchProps
+    return __privateGet$4(this, _trace).call(this, "search", async () => {
+      const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
+      const { records } = await searchTable({
+        pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", tableName: __privateGet$4(this, _table) },
+        body: {
+          query,
+          fuzziness: options.fuzziness,
+          prefix: options.prefix,
+          highlight: options.highlight,
+          filter: options.filter,
+          boosters: options.boosters
+        },
+        ...fetchProps
+      });
+      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));
     });
-    const schema = await __privateMethod$2(this, _getSchema$1, getSchema_fn$1).call(this);
-    return records.map((item) => initObject(this.db, schema, __privateGet$4(this, _table), item));
   }
   async query(query) {
-    const cacheQuery = await __privateMethod$2(this, _getCacheQuery, getCacheQuery_fn).call(this, query);
-    if (cacheQuery)
-      return new Page(query, cacheQuery.meta, cacheQuery.records);
-    const data = query.getQueryOptions();
-    const body = {
-      filter: Object.values(data.filter ?? {}).some(Boolean) ? data.filter : void 0,
-      sort: data.sort !== void 0 ? buildSortFilter(data.sort) : void 0,
-      page: data.pagination,
-      columns: data.columns
-    };
-    const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
-    const { meta, records: objects } = await queryTable({
-      pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", tableName: __privateGet$4(this, _table) },
-      body,
-      ...fetchProps
+    return __privateGet$4(this, _trace).call(this, "query", async () => {
+      const cacheQuery = await __privateMethod$2(this, _getCacheQuery, getCacheQuery_fn).call(this, query);
+      if (cacheQuery)
+        return new Page(query, cacheQuery.meta, cacheQuery.records);
+      const data = query.getQueryOptions();
+      const body = {
+        filter: cleanFilter(data.filter),
+        sort: data.sort !== void 0 ? buildSortFilter(data.sort) : void 0,
+        page: data.pagination,
+        columns: data.columns
+      };
+      const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
+      const { meta, records: objects } = await queryTable({
+        pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", tableName: __privateGet$4(this, _table) },
+        body,
+        ...fetchProps
+      });
+      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));
+      await __privateMethod$2(this, _setCacheQuery, setCacheQuery_fn).call(this, query, meta, records);
+      return new Page(query, meta, records);
     });
-    const schema = await __privateMethod$2(this, _getSchema$1, getSchema_fn$1).call(this);
-    const records = objects.map((record) => initObject(this.db, schema, __privateGet$4(this, _table), record));
-    await __privateMethod$2(this, _setCacheQuery, setCacheQuery_fn).call(this, query, meta, records);
-    return new Page(query, meta, records);
   }
 }
 _table = new WeakMap();
 _getFetchProps = new WeakMap();
+_db = new WeakMap();
 _cache = new WeakMap();
-_schema$1 = new WeakMap();
+_schemaTables$2 = new WeakMap();
+_trace = new WeakMap();
 _insertRecordWithoutId = new WeakSet();
-insertRecordWithoutId_fn = async function(object) {
+insertRecordWithoutId_fn = async function(object, columns = ["*"]) {
   const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
   const record = transformObjectLinks(object);
   const response = await insertRecord({
@@ -1354,17 +1785,15 @@ insertRecordWithoutId_fn = async function(object) {
       dbBranchName: "{dbBranch}",
       tableName: __privateGet$4(this, _table)
     },
+    queryParams: { columns },
     body: record,
     ...fetchProps
   });
-  const finalObject = await this.read(response.id);
-  if (!finalObject) {
-    throw new Error("The server failed to save the record");
-  }
-  return finalObject;
+  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);
 };
 _insertRecordWithId = new WeakSet();
-insertRecordWithId_fn = async function(recordId, object) {
+insertRecordWithId_fn = async function(recordId, object, columns = ["*"]) {
   const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
   const record = transformObjectLinks(object);
   const response = await insertRecordWithID({
@@ -1375,88 +1804,78 @@ insertRecordWithId_fn = async function(recordId, object) {
       recordId
     },
     body: record,
-    queryParams: { createOnly: true },
+    queryParams: { createOnly: true, columns },
     ...fetchProps
   });
-  const finalObject = await this.read(response.id);
-  if (!finalObject) {
-    throw new Error("The server failed to save the record");
-  }
-  return finalObject;
+  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);
 };
 _bulkInsertTableRecords = new WeakSet();
-bulkInsertTableRecords_fn = async function(objects) {
+bulkInsertTableRecords_fn = async function(objects, columns = ["*"]) {
   const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
   const records = objects.map((object) => transformObjectLinks(object));
   const response = await bulkInsertTableRecords({
     pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", tableName: __privateGet$4(this, _table) },
+    queryParams: { columns },
     body: { records },
     ...fetchProps
   });
-  const finalObjects = await this.any(...response.recordIDs.map((id) => this.filter("id", id))).getAll();
-  if (finalObjects.length !== objects.length) {
-    throw new Error("The server failed to save some records");
+  if (!isResponseWithRecords(response)) {
+    throw new Error("Request included columns but server didn't include them");
   }
-  return finalObjects;
+  const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
+  return response.records?.map((item) => initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), item));
 };
 _updateRecordWithID = new WeakSet();
-updateRecordWithID_fn = async function(recordId, object) {
+updateRecordWithID_fn = async function(recordId, object, columns = ["*"]) {
   const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
   const record = transformObjectLinks(object);
-  const response = await updateRecordWithID({
-    pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", tableName: __privateGet$4(this, _table), recordId },
-    body: record,
-    ...fetchProps
-  });
-  const item = await this.read(response.id);
-  if (!item)
-    throw new Error("The server failed to save the record");
-  return item;
+  try {
+    const response = await updateRecordWithID({
+      pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", tableName: __privateGet$4(this, _table), recordId },
+      queryParams: { columns },
+      body: record,
+      ...fetchProps
+    });
+    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);
+  } catch (e) {
+    if (isObject(e) && e.status === 404) {
+      return null;
+    }
+    throw e;
+  }
 };
 _upsertRecordWithID = new WeakSet();
-upsertRecordWithID_fn = async function(recordId, object) {
+upsertRecordWithID_fn = async function(recordId, object, columns = ["*"]) {
   const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
   const response = await upsertRecordWithID({
     pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", tableName: __privateGet$4(this, _table), recordId },
+    queryParams: { columns },
     body: object,
     ...fetchProps
   });
-  const item = await this.read(response.id);
-  if (!item)
-    throw new Error("The server failed to save the record");
-  return item;
+  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);
 };
 _deleteRecord = new WeakSet();
-deleteRecord_fn = async function(recordId) {
+deleteRecord_fn = async function(recordId, columns = ["*"]) {
   const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
-  await deleteRecord({
-    pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", tableName: __privateGet$4(this, _table), recordId },
-    ...fetchProps
-  });
-};
-_invalidateCache = new WeakSet();
-invalidateCache_fn = async function(recordId) {
-  await __privateGet$4(this, _cache).delete(`rec_${__privateGet$4(this, _table)}:${recordId}`);
-  const cacheItems = await __privateGet$4(this, _cache).getAll();
-  const queries = Object.entries(cacheItems).filter(([key]) => key.startsWith("query_"));
-  for (const [key, value] of queries) {
-    const ids = getIds(value);
-    if (ids.includes(recordId))
-      await __privateGet$4(this, _cache).delete(key);
+  try {
+    const response = await deleteRecord({
+      pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", tableName: __privateGet$4(this, _table), recordId },
+      queryParams: { columns },
+      ...fetchProps
+    });
+    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);
+  } catch (e) {
+    if (isObject(e) && e.status === 404) {
+      return null;
+    }
+    throw e;
   }
 };
-_setCacheRecord = new WeakSet();
-setCacheRecord_fn = async function(record) {
-  if (!__privateGet$4(this, _cache).cacheRecords)
-    return;
-  await __privateGet$4(this, _cache).set(`rec_${__privateGet$4(this, _table)}:${record.id}`, record);
-};
-_getCacheRecord = new WeakSet();
-getCacheRecord_fn = async function(recordId) {
-  if (!__privateGet$4(this, _cache).cacheRecords)
-    return null;
-  return __privateGet$4(this, _cache).get(`rec_${__privateGet$4(this, _table)}:${recordId}`);
-};
 _setCacheQuery = new WeakSet();
 setCacheQuery_fn = async function(query, meta, records) {
   await __privateGet$4(this, _cache).set(`query_${__privateGet$4(this, _table)}:${query.key()}`, { date: new Date(), meta, records });
@@ -1473,17 +1892,17 @@ getCacheQuery_fn = async function(query) {
   const hasExpired = result.date.getTime() + ttl < Date.now();
   return hasExpired ? null : result;
 };
-_getSchema$1 = new WeakSet();
-getSchema_fn$1 = async function() {
-  if (__privateGet$4(this, _schema$1))
-    return __privateGet$4(this, _schema$1);
+_getSchemaTables$1 = new WeakSet();
+getSchemaTables_fn$1 = async function() {
+  if (__privateGet$4(this, _schemaTables$2))
+    return __privateGet$4(this, _schemaTables$2);
   const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
   const { schema } = await getBranchDetails({
     pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}" },
     ...fetchProps
   });
-  __privateSet$3(this, _schema$1, schema);
-  return schema;
+  __privateSet$4(this, _schemaTables$2, schema.tables);
+  return schema.tables;
 };
 const transformObjectLinks = (object) => {
   return Object.entries(object).reduce((acc, [key, value]) => {
@@ -1492,20 +1911,21 @@ const transformObjectLinks = (object) => {
     return { ...acc, [key]: isIdentifiable(value) ? value.id : value };
   }, {});
 };
-const initObject = (db, schema, table, object) => {
+const initObject = (db, schemaTables, table, object) => {
   const result = {};
-  Object.assign(result, object);
-  const { columns } = schema.tables.find(({ name }) => name === table) ?? {};
+  const { xata, ...rest } = object ?? {};
+  Object.assign(result, rest);
+  const { columns } = schemaTables.find(({ name }) => name === table) ?? {};
   if (!columns)
     console.error(`Table ${table} not found in schema`);
   for (const column of columns ?? []) {
     const value = result[column.name];
     switch (column.type) {
       case "datetime": {
-        const date = new Date(value);
-        if (isNaN(date.getTime())) {
+        const date = value !== void 0 ? new Date(value) : void 0;
+        if (date && isNaN(date.getTime())) {
           console.error(`Failed to parse date ${value} for field ${column.name}`);
-        } else {
+        } else if (date) {
           result[column.name] = date;
         }
         break;
@@ -1515,35 +1935,45 @@ const initObject = (db, schema, table, object) => {
         if (!linkTable) {
           console.error(`Failed to parse link for field ${column.name}`);
         } else if (isObject(value)) {
-          result[column.name] = initObject(db, schema, linkTable, value);
+          result[column.name] = initObject(db, schemaTables, linkTable, value);
         }
         break;
       }
     }
   }
-  result.read = function() {
-    return db[table].read(result["id"]);
+  result.read = function(columns2) {
+    return db[table].read(result["id"], columns2);
   };
-  result.update = function(data) {
-    return db[table].update(result["id"], data);
+  result.update = function(data, columns2) {
+    return db[table].update(result["id"], data, columns2);
   };
   result.delete = function() {
     return db[table].delete(result["id"]);
   };
-  for (const prop of ["read", "update", "delete"]) {
+  result.getMetadata = function() {
+    return xata;
+  };
+  for (const prop of ["read", "update", "delete", "getMetadata"]) {
     Object.defineProperty(result, prop, { enumerable: false });
   }
   Object.freeze(result);
   return result;
 };
-function getIds(value) {
-  if (Array.isArray(value)) {
-    return value.map((item) => getIds(item)).flat();
-  }
-  if (!isObject(value))
-    return [];
-  const nestedIds = Object.values(value).map((item) => getIds(item)).flat();
-  return isString(value.id) ? [value.id, ...nestedIds] : nestedIds;
+function isResponseWithRecords(value) {
+  return isObject(value) && Array.isArray(value.records);
+}
+function extractId(value) {
+  if (isString(value))
+    return value;
+  if (isObject(value) && isString(value.id))
+    return value.id;
+  return void 0;
+}
+function cleanFilter(filter) {
+  if (!filter)
+    return void 0;
+  const values = Object.values(filter).filter(Boolean).filter((value) => Array.isArray(value) ? value.length > 0 : true);
+  return values.length > 0 ? filter : void 0;
 }
 
 var __accessCheck$3 = (obj, member, msg) => {
@@ -1559,7 +1989,7 @@ var __privateAdd$3 = (obj, member, value) => {
     throw TypeError("Cannot add the same private member more than once");
   member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
 };
-var __privateSet$2 = (obj, member, value, setter) => {
+var __privateSet$3 = (obj, member, value, setter) => {
   __accessCheck$3(obj, member, "write to private field");
   setter ? setter.call(obj, value) : member.set(obj, value);
   return value;
@@ -1568,9 +1998,8 @@ var _map;
 class SimpleCache {
   constructor(options = {}) {
     __privateAdd$3(this, _map, void 0);
-    __privateSet$2(this, _map, /* @__PURE__ */ new Map());
+    __privateSet$3(this, _map, /* @__PURE__ */ new Map());
     this.capacity = options.max ?? 500;
-    this.cacheRecords = options.cacheRecords ?? true;
     this.defaultQueryTTL = options.defaultQueryTTL ?? 60 * 1e3;
   }
   async getAll() {
@@ -1596,18 +2025,25 @@ class SimpleCache {
 }
 _map = new WeakMap();
 
-const gt = (value) => ({ $gt: value });
-const ge = (value) => ({ $ge: value });
-const gte = (value) => ({ $ge: value });
-const lt = (value) => ({ $lt: value });
-const lte = (value) => ({ $le: value });
-const le = (value) => ({ $le: value });
+const greaterThan = (value) => ({ $gt: value });
+const gt = greaterThan;
+const greaterThanEquals = (value) => ({ $ge: value });
+const greaterEquals = greaterThanEquals;
+const gte = greaterThanEquals;
+const ge = greaterThanEquals;
+const lessThan = (value) => ({ $lt: value });
+const lt = lessThan;
+const lessThanEquals = (value) => ({ $le: value });
+const lessEquals = lessThanEquals;
+const lte = lessThanEquals;
+const le = lessThanEquals;
 const exists = (column) => ({ $exists: column });
 const notExists = (column) => ({ $notExists: column });
 const startsWith = (value) => ({ $startsWith: value });
 const endsWith = (value) => ({ $endsWith: value });
 const pattern = (value) => ({ $pattern: value });
 const is = (value) => ({ $is: value });
+const equals = is;
 const isNot = (value) => ({ $isNot: value });
 const contains = (value) => ({ $contains: value });
 const includes = (value) => ({ $includes: value });
@@ -1628,31 +2064,42 @@ var __privateAdd$2 = (obj, member, value) => {
     throw TypeError("Cannot add the same private member more than once");
   member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
 };
-var _tables;
+var __privateSet$2 = (obj, member, value, setter) => {
+  __accessCheck$2(obj, member, "write to private field");
+  setter ? setter.call(obj, value) : member.set(obj, value);
+  return value;
+};
+var _tables, _schemaTables$1;
 class SchemaPlugin extends XataPlugin {
-  constructor(tableNames) {
+  constructor(schemaTables) {
     super();
-    this.tableNames = tableNames;
     __privateAdd$2(this, _tables, {});
+    __privateAdd$2(this, _schemaTables$1, void 0);
+    __privateSet$2(this, _schemaTables$1, schemaTables);
   }
   build(pluginOptions) {
-    const db = new Proxy({}, {
-      get: (_target, table) => {
-        if (!isString(table))
-          throw new Error("Invalid table name");
-        if (__privateGet$2(this, _tables)[table] === void 0) {
-          __privateGet$2(this, _tables)[table] = new RestRepository({ db, pluginOptions, table });
+    const db = new Proxy(
+      {},
+      {
+        get: (_target, table) => {
+          if (!isString(table))
+            throw new Error("Invalid table name");
+          if (__privateGet$2(this, _tables)[table] === void 0) {
+            __privateGet$2(this, _tables)[table] = new RestRepository({ db, pluginOptions, table, schemaTables: __privateGet$2(this, _schemaTables$1) });
+          }
+          return __privateGet$2(this, _tables)[table];
         }
-        return __privateGet$2(this, _tables)[table];
       }
-    });
-    for (const table of this.tableNames ?? []) {
-      db[table] = new RestRepository({ db, pluginOptions, table });
+    );
+    const tableNames = __privateGet$2(this, _schemaTables$1)?.map(({ name }) => name) ?? [];
+    for (const table of tableNames) {
+      db[table] = new RestRepository({ db, pluginOptions, table, schemaTables: __privateGet$2(this, _schemaTables$1) });
     }
     return db;
   }
 }
 _tables = new WeakMap();
+_schemaTables$1 = new WeakMap();
 
 var __accessCheck$1 = (obj, member, msg) => {
   if (!member.has(obj))
@@ -1676,82 +2123,77 @@ var __privateMethod$1 = (obj, member, method) => {
   __accessCheck$1(obj, member, "access private method");
   return method;
 };
-var _schema, _search, search_fn, _getSchema, getSchema_fn;
+var _schemaTables, _search, search_fn, _getSchemaTables, getSchemaTables_fn;
 class SearchPlugin extends XataPlugin {
-  constructor(db) {
+  constructor(db, schemaTables) {
     super();
     this.db = db;
     __privateAdd$1(this, _search);
-    __privateAdd$1(this, _getSchema);
-    __privateAdd$1(this, _schema, void 0);
+    __privateAdd$1(this, _getSchemaTables);
+    __privateAdd$1(this, _schemaTables, void 0);
+    __privateSet$1(this, _schemaTables, schemaTables);
   }
   build({ getFetchProps }) {
     return {
       all: async (query, options = {}) => {
         const records = await __privateMethod$1(this, _search, search_fn).call(this, query, options, getFetchProps);
-        const schema = await __privateMethod$1(this, _getSchema, getSchema_fn).call(this, getFetchProps);
+        const schemaTables = await __privateMethod$1(this, _getSchemaTables, getSchemaTables_fn).call(this, getFetchProps);
         return records.map((record) => {
           const { table = "orphan" } = record.xata;
-          return { table, record: initObject(this.db, schema, table, record) };
+          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, getFetchProps);
-        const schema = await __privateMethod$1(this, _getSchema, getSchema_fn).call(this, getFetchProps);
+        const schemaTables = await __privateMethod$1(this, _getSchemaTables, getSchemaTables_fn).call(this, getFetchProps);
         return records.reduce((acc, record) => {
           const { table = "orphan" } = record.xata;
           const items = acc[table] ?? [];
-          const item = initObject(this.db, schema, table, record);
+          const item = initObject(this.db, schemaTables, table, record);
           return { ...acc, [table]: [...items, item] };
         }, {});
       }
     };
   }
 }
-_schema = new WeakMap();
+_schemaTables = new WeakMap();
 _search = new WeakSet();
 search_fn = async function(query, options, getFetchProps) {
   const fetchProps = await getFetchProps();
-  const { tables, fuzziness } = options ?? {};
+  const { tables, fuzziness, highlight, prefix } = options ?? {};
   const { records } = await searchBranch({
     pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}" },
-    body: { tables, query, fuzziness },
+    body: { tables, query, fuzziness, prefix, highlight },
     ...fetchProps
   });
   return records;
 };
-_getSchema = new WeakSet();
-getSchema_fn = async function(getFetchProps) {
-  if (__privateGet$1(this, _schema))
-    return __privateGet$1(this, _schema);
+_getSchemaTables = new WeakSet();
+getSchemaTables_fn = async function(getFetchProps) {
+  if (__privateGet$1(this, _schemaTables))
+    return __privateGet$1(this, _schemaTables);
   const fetchProps = await getFetchProps();
   const { schema } = await getBranchDetails({
     pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}" },
     ...fetchProps
   });
-  __privateSet$1(this, _schema, schema);
-  return schema;
+  __privateSet$1(this, _schemaTables, schema.tables);
+  return schema.tables;
 };
 
 const isBranchStrategyBuilder = (strategy) => {
   return typeof strategy === "function";
 };
 
-const envBranchNames = [
-  "XATA_BRANCH",
-  "VERCEL_GIT_COMMIT_REF",
-  "CF_PAGES_BRANCH",
-  "BRANCH"
-];
 async function getCurrentBranchName(options) {
-  const env = getBranchByEnvVariable();
-  if (env) {
-    const details = await getDatabaseBranch(env, options);
+  const { branch, envBranch } = getEnvironment();
+  if (branch) {
+    const details = await getDatabaseBranch(branch, options);
     if (details)
-      return env;
-    console.warn(`Branch ${env} not found in Xata. Ignoring...`);
+      return branch;
+    console.warn(`Branch ${branch} not found in Xata. Ignoring...`);
   }
-  const gitBranch = await getGitBranch();
+  const gitBranch = envBranch || await getGitBranch();
   return resolveXataBranch(gitBranch, options);
 }
 async function getCurrentBranchDetails(options) {
@@ -1762,18 +2204,24 @@ async function resolveXataBranch(gitBranch, options) {
   const databaseURL = options?.databaseURL || getDatabaseURL();
   const apiKey = options?.apiKey || getAPIKey();
   if (!databaseURL)
-    throw new Error("A databaseURL was not defined. Either set the XATA_DATABASE_URL env variable or pass the argument explicitely");
+    throw new Error(
+      "A databaseURL was not defined. Either set the XATA_DATABASE_URL env variable or pass the argument explicitely"
+    );
   if (!apiKey)
-    throw new Error("An API key was not defined. Either set the XATA_API_KEY env variable or pass the argument explicitely");
+    throw new Error(
+      "An API key was not defined. Either set the XATA_API_KEY env variable or pass the argument explicitely"
+    );
   const [protocol, , host, , dbName] = databaseURL.split("/");
   const [workspace] = host.split(".");
+  const { fallbackBranch } = getEnvironment();
   const { branch } = await resolveBranch({
     apiKey,
     apiUrl: databaseURL,
     fetchImpl: getFetchImplementation(options?.fetchImpl),
     workspacesApiUrl: `${protocol}//${host}`,
     pathParams: { dbName, workspace },
-    queryParams: { gitBranch, fallbackBranch: getEnvVariable("XATA_FALLBACK_BRANCH") }
+    queryParams: { gitBranch, fallbackBranch },
+    trace: defaultTrace
   });
   return branch;
 }
@@ -1781,9 +2229,13 @@ async function getDatabaseBranch(branch, options) {
   const databaseURL = options?.databaseURL || getDatabaseURL();
   const apiKey = options?.apiKey || getAPIKey();
   if (!databaseURL)
-    throw new Error("A databaseURL was not defined. Either set the XATA_DATABASE_URL env variable or pass the argument explicitely");
+    throw new Error(
+      "A databaseURL was not defined. Either set the XATA_DATABASE_URL env variable or pass the argument explicitely"
+    );
   if (!apiKey)
-    throw new Error("An API key was not defined. Either set the XATA_API_KEY env variable or pass the argument explicitely");
+    throw new Error(
+      "An API key was not defined. Either set the XATA_API_KEY env variable or pass the argument explicitely"
+    );
   const [protocol, , host, , database] = databaseURL.split("/");
   const [workspace] = host.split(".");
   const dbBranchName = `${database}:${branch}`;
@@ -1793,10 +2245,8 @@ async function getDatabaseBranch(branch, options) {
       apiUrl: databaseURL,
       fetchImpl: getFetchImplementation(options?.fetchImpl),
       workspacesApiUrl: `${protocol}//${host}`,
-      pathParams: {
-        dbBranchName,
-        workspace
-      }
+      pathParams: { dbBranchName, workspace },
+      trace: defaultTrace
     });
   } catch (err) {
     if (isObject(err) && err.status === 404)
@@ -1804,21 +2254,10 @@ async function getDatabaseBranch(branch, options) {
     throw err;
   }
 }
-function getBranchByEnvVariable() {
-  for (const name of envBranchNames) {
-    const value = getEnvVariable(name);
-    if (value) {
-      return value;
-    }
-  }
-  try {
-    return XATA_BRANCH;
-  } catch (err) {
-  }
-}
 function getDatabaseURL() {
   try {
-    return getEnvVariable("XATA_DATABASE_URL") ?? XATA_DATABASE_URL;
+    const { databaseURL } = getEnvironment();
+    return databaseURL;
   } catch (err) {
     return void 0;
   }
@@ -1847,20 +2286,23 @@ var __privateMethod = (obj, member, method) => {
   return method;
 };
 const buildClient = (plugins) => {
-  var _branch, _parseOptions, parseOptions_fn, _getFetchProps, getFetchProps_fn, _evaluateBranch, evaluateBranch_fn, _a;
+  var _branch, _options, _parseOptions, parseOptions_fn, _getFetchProps, getFetchProps_fn, _evaluateBranch, evaluateBranch_fn, _a;
   return _a = class {
-    constructor(options = {}, tables) {
+    constructor(options = {}, schemaTables) {
       __privateAdd(this, _parseOptions);
       __privateAdd(this, _getFetchProps);
       __privateAdd(this, _evaluateBranch);
       __privateAdd(this, _branch, void 0);
+      __privateAdd(this, _options, void 0);
       const safeOptions = __privateMethod(this, _parseOptions, parseOptions_fn).call(this, options);
+      __privateSet(this, _options, safeOptions);
       const pluginOptions = {
         getFetchProps: () => __privateMethod(this, _getFetchProps, getFetchProps_fn).call(this, safeOptions),
-        cache: safeOptions.cache
+        cache: safeOptions.cache,
+        trace: safeOptions.trace
       };
-      const db = new SchemaPlugin(tables).build(pluginOptions);
-      const search = new SearchPlugin(db).build(pluginOptions);
+      const db = new SchemaPlugin(schemaTables).build(pluginOptions);
+      const search = new SearchPlugin(db, schemaTables).build(pluginOptions);
       this.db = db;
       this.search = search;
       for (const [key, namespace] of Object.entries(plugins ?? {})) {
@@ -1876,22 +2318,26 @@ const buildClient = (plugins) => {
         }
       }
     }
-  }, _branch = new WeakMap(), _parseOptions = new WeakSet(), parseOptions_fn = function(options) {
+    async getConfig() {
+      const databaseURL = __privateGet(this, _options).databaseURL;
+      const branch = await __privateGet(this, _options).branch();
+      return { databaseURL, branch };
+    }
+  }, _branch = new WeakMap(), _options = new WeakMap(), _parseOptions = new WeakSet(), parseOptions_fn = function(options) {
     const fetch = getFetchImplementation(options?.fetch);
     const databaseURL = options?.databaseURL || getDatabaseURL();
     const apiKey = options?.apiKey || getAPIKey();
-    const cache = options?.cache ?? new SimpleCache({ cacheRecords: false, defaultQueryTTL: 0 });
+    const cache = options?.cache ?? new SimpleCache({ defaultQueryTTL: 0 });
+    const trace = options?.trace ?? defaultTrace;
     const branch = async () => options?.branch !== void 0 ? await __privateMethod(this, _evaluateBranch, evaluateBranch_fn).call(this, options.branch) : await getCurrentBranchName({ apiKey, databaseURL, fetchImpl: options?.fetch });
-    if (!databaseURL || !apiKey) {
-      throw new Error("Options databaseURL and apiKey are required");
+    if (!apiKey) {
+      throw new Error("Option apiKey is required");
     }
-    return { fetch, databaseURL, apiKey, branch, cache };
-  }, _getFetchProps = new WeakSet(), getFetchProps_fn = async function({
-    fetch,
-    apiKey,
-    databaseURL,
-    branch
-  }) {
+    if (!databaseURL) {
+      throw new Error("Option databaseURL is required");
+    }
+    return { fetch, databaseURL, apiKey, branch, cache, trace };
+  }, _getFetchProps = new WeakSet(), getFetchProps_fn = async function({ fetch, apiKey, databaseURL, branch, trace }) {
     const branchValue = await __privateMethod(this, _evaluateBranch, evaluateBranch_fn).call(this, branch);
     if (!branchValue)
       throw new Error("Unable to resolve branch value");
@@ -1901,9 +2347,10 @@ const buildClient = (plugins) => {
       apiUrl: "",
       workspacesApiUrl: (path, params) => {
         const hasBranch = params.dbBranchName ?? params.branch;
-        const newPath = path.replace(/^\/db\/[^/]+/, hasBranch ? `:${branchValue}` : "");
+        const newPath = path.replace(/^\/db\/[^/]+/, hasBranch !== void 0 ? `:${branchValue}` : "");
         return databaseURL + newPath;
-      }
+      },
+      trace
     };
   }, _evaluateBranch = new WeakSet(), evaluateBranch_fn = async function(param) {
     if (__privateGet(this, _branch))
@@ -1926,6 +2373,88 @@ const buildClient = (plugins) => {
 class BaseClient extends buildClient() {
 }
 
+const META = "__";
+const VALUE = "___";
+class Serializer {
+  constructor() {
+    this.classes = {};
+  }
+  add(clazz) {
+    this.classes[clazz.name] = clazz;
+  }
+  toJSON(data) {
+    function visit(obj) {
+      if (Array.isArray(obj))
+        return obj.map(visit);
+      const type = typeof obj;
+      if (type === "undefined")
+        return { [META]: "undefined" };
+      if (type === "bigint")
+        return { [META]: "bigint", [VALUE]: obj.toString() };
+      if (obj === null || type !== "object")
+        return obj;
+      const constructor = obj.constructor;
+      const o = { [META]: constructor.name };
+      for (const [key, value] of Object.entries(obj)) {
+        o[key] = visit(value);
+      }
+      if (constructor === Date)
+        o[VALUE] = obj.toISOString();
+      if (constructor === Map)
+        o[VALUE] = Object.fromEntries(obj);
+      if (constructor === Set)
+        o[VALUE] = [...obj];
+      return o;
+    }
+    return JSON.stringify(visit(data));
+  }
+  fromJSON(json) {
+    return JSON.parse(json, (key, value) => {
+      if (value && typeof value === "object" && !Array.isArray(value)) {
+        const { [META]: clazz, [VALUE]: val, ...rest } = value;
+        const constructor = this.classes[clazz];
+        if (constructor) {
+          return Object.assign(Object.create(constructor.prototype), rest);
+        }
+        if (clazz === "Date")
+          return new Date(val);
+        if (clazz === "Set")
+          return new Set(val);
+        if (clazz === "Map")
+          return new Map(Object.entries(val));
+        if (clazz === "bigint")
+          return BigInt(val);
+        if (clazz === "undefined")
+          return void 0;
+        return rest;
+      }
+      return value;
+    });
+  }
+}
+const defaultSerializer = new Serializer();
+const serialize = (data) => {
+  return defaultSerializer.toJSON(data);
+};
+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);
+    };
+  };
+}
+
 class XataError extends Error {
   constructor(message, status) {
     super(message);
@@ -1933,5 +2462,5 @@ class XataError extends Error {
   }
 }
 
-export { BaseClient, operationsByTag as Operations, PAGINATION_DEFAULT_OFFSET, PAGINATION_DEFAULT_SIZE, PAGINATION_MAX_OFFSET, PAGINATION_MAX_SIZE, Page, Query, Repository, RestRepository, SchemaPlugin, SearchPlugin, SimpleCache, XataApiClient, XataApiPlugin, XataError, XataPlugin, acceptWorkspaceMemberInvite, addGitBranchesEntry, addTableColumn, buildClient, bulkInsertTableRecords, cancelWorkspaceMemberInvite, contains, createBranch, createDatabase, createTable, createUserAPIKey, createWorkspace, deleteBranch, deleteColumn, deleteDatabase, deleteRecord, deleteTable, deleteUser, deleteUserAPIKey, deleteWorkspace, endsWith, executeBranchMigrationPlan, exists, ge, getAPIKey, getBranchDetails, getBranchList, getBranchMetadata, getBranchMigrationHistory, getBranchMigrationPlan, getBranchStats, getColumn, getCurrentBranchDetails, getCurrentBranchName, getDatabaseList, getDatabaseURL, getGitBranchesMapping, getRecord, getTableColumns, getTableSchema, getUser, getUserAPIKeys, getWorkspace, getWorkspaceMembersList, getWorkspacesList, gt, gte, includes, includesAll, includesAny, includesNone, insertRecord, insertRecordWithID, inviteWorkspaceMember, is, isIdentifiable, isNot, isXataRecord, le, lt, lte, notExists, operationsByTag, pattern, queryTable, removeGitBranchesEntry, removeWorkspaceMember, resendWorkspaceMemberInvite, resolveBranch, searchBranch, searchTable, setTableSchema, startsWith, updateBranchMetadata, updateColumn, updateRecordWithID, updateTable, updateUser, updateWorkspace, updateWorkspaceMemberRole, upsertRecordWithID };
+export { BaseClient, operationsByTag as Operations, PAGINATION_DEFAULT_OFFSET, PAGINATION_DEFAULT_SIZE, PAGINATION_MAX_OFFSET, PAGINATION_MAX_SIZE, Page, Query, RecordArray, Repository, RestRepository, SchemaPlugin, SearchPlugin, Serializer, SimpleCache, XataApiClient, XataApiPlugin, XataError, XataPlugin, acceptWorkspaceMemberInvite, addGitBranchesEntry, addTableColumn, applyBranchSchemaEdit, buildClient, buildWorkerRunner, bulkInsertTableRecords, cancelWorkspaceMemberInvite, compareBranchSchemas, compareBranchWithUserSchema, compareMigrationRequest, contains, createBranch, createDatabase, createMigrationRequest, createTable, createUserAPIKey, createWorkspace, deleteBranch, deleteColumn, deleteDatabase, deleteRecord, deleteTable, deleteUser, deleteUserAPIKey, deleteWorkspace, deserialize, endsWith, equals, executeBranchMigrationPlan, exists, ge, getAPIKey, getBranchDetails, getBranchList, getBranchMetadata, getBranchMigrationHistory, getBranchMigrationPlan, getBranchSchemaHistory, getBranchStats, getColumn, getCurrentBranchDetails, getCurrentBranchName, getDatabaseList, getDatabaseMetadata, getDatabaseURL, getGitBranchesMapping, getMigrationRequest, getMigrationRequestIsMerged, getRecord, getTableColumns, getTableSchema, getUser, getUserAPIKeys, getWorkspace, getWorkspaceMembersList, getWorkspacesList, greaterEquals, greaterThan, greaterThanEquals, gt, gte, includes, includesAll, includesAny, includesNone, insertRecord, insertRecordWithID, inviteWorkspaceMember, is, isCursorPaginationOptions, isIdentifiable, isNot, isXataRecord, le, lessEquals, lessThan, lessThanEquals, listMigrationRequests, listMigrationRequestsCommits, lt, lte, mergeMigrationRequest, notExists, operationsByTag, patchDatabaseMetadata, pattern, previewBranchSchemaEdit, queryTable, removeGitBranchesEntry, removeWorkspaceMember, resendWorkspaceMemberInvite, resolveBranch, searchBranch, searchTable, serialize, setTableSchema, startsWith, updateBranchMetadata, updateBranchSchema, updateColumn, updateMigrationRequest, updateRecordWithID, updateTable, updateUser, updateWorkspace, updateWorkspaceMemberInvite, updateWorkspaceMemberRole, upsertRecordWithID };
 //# sourceMappingURL=index.mjs.map