@meshagent/meshagent 0.42.0 → 0.43.0

package/dist/esm/datasets-client.js

@@ -1,14 +1,10 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.DatasetsClient = exports.DatasetUuid = exports.DatasetJson = exports.DatasetStruct = exports.DatasetDate = exports.DatasetExpression = exports.DatasetValueEncoder = void 0;
-const apache_arrow_1 = require("apache-arrow");
-const room_server_client_1 = require("./room-server-client");
-const response_1 = require("./response");
+import { Schema, Table, tableFromIPC, tableToIPC } from "apache-arrow";
+import { RoomServerException } from "./room-server-client.js";
+import { BinaryContent, ControlContent, EmptyContent, ErrorContent, JsonContent } from "./response.js";
 const ARROW_IPC_STREAM_MIME_TYPE = "application/vnd.apache.arrow.stream";
-class DatasetValueEncoder {
+export class DatasetValueEncoder {
 }
-exports.DatasetValueEncoder = DatasetValueEncoder;
-class DatasetExpression extends DatasetValueEncoder {
+export class DatasetExpression extends DatasetValueEncoder {
     constructor(expression) {
         super();
         const normalized = expression.trim();
@@ -26,8 +22,7 @@ class DatasetExpression extends DatasetValueEncoder {
         return this.expression;
     }
 }
-exports.DatasetExpression = DatasetExpression;
-class DatasetDate extends DatasetValueEncoder {
+export class DatasetDate extends DatasetValueEncoder {
     constructor(value) {
         super();
         const normalized = value.trim();
@@ -46,8 +41,7 @@ class DatasetDate extends DatasetValueEncoder {
         return this.value;
     }
 }
-exports.DatasetDate = DatasetDate;
-class DatasetStruct extends DatasetValueEncoder {
+export class DatasetStruct extends DatasetValueEncoder {
     constructor(fields) {
         super();
         this.fields = Object.fromEntries(Object.entries(fields).map(([key, value]) => {
@@ -66,8 +60,7 @@ class DatasetStruct extends DatasetValueEncoder {
         };
     }
 }
-exports.DatasetStruct = DatasetStruct;
-class DatasetJson extends DatasetValueEncoder {
+export class DatasetJson extends DatasetValueEncoder {
     constructor(value) {
         super();
         this.value = normalizeDatasetJsonValue(value);
@@ -81,14 +74,13 @@ class DatasetJson extends DatasetValueEncoder {
         };
     }
 }
-exports.DatasetJson = DatasetJson;
 const globalScope = globalThis;
 const UUID_HEX_REGEX = /^[0-9a-f]{32}$/;
 const ISO_DATE_REGEX = /^\d{4}-\d{2}-\d{2}$/;
 function normalizeUuidHex(value) {
     const normalized = value.trim().toLowerCase().replace(/-/g, "");
     if (!UUID_HEX_REGEX.test(normalized)) {
-        throw new room_server_client_1.RoomServerException("invalid uuid format");
+        throw new RoomServerException("invalid uuid format");
     }
     return normalized;
 }
@@ -99,7 +91,7 @@ function formatUuidHex(value) {
         `${value.substring(16, 20)}-` +
         `${value.substring(20)}`);
 }
-class DatasetUuid {
+export class DatasetUuid {
     constructor(value) {
         this.value = formatUuidHex(normalizeUuidHex(value));
     }
@@ -107,7 +99,6 @@ class DatasetUuid {
         return this.value;
     }
 }
-exports.DatasetUuid = DatasetUuid;
 function bytesToBase64(bytes) {
     if (globalScope.Buffer) {
         return globalScope.Buffer.from(bytes).toString("base64");
@@ -191,7 +182,7 @@ function encodeRecordValue(value) {
         };
     }
     if (isRecord(value)) {
-        throw new room_server_client_1.RoomServerException("dataset object values must use DatasetStruct or DatasetJson");
+        throw new RoomServerException("dataset object values must use DatasetStruct or DatasetJson");
     }
     return value;
 }
@@ -216,62 +207,62 @@ function datasetSqlLiteral(value) {
 }
 function decodeRecordValue(value) {
     if (Array.isArray(value)) {
-        throw new room_server_client_1.RoomServerException("dataset list values must use a {'list': [...]} wrapper");
+        throw new RoomServerException("dataset list values must use a {'list': [...]} wrapper");
     }
     if (!isRecord(value)) {
         return value;
     }
     const entries = Object.entries(value);
     if (entries.length !== 1) {
-        throw new room_server_client_1.RoomServerException("dataset object values must use a single-key type wrapper");
+        throw new RoomServerException("dataset object values must use a single-key type wrapper");
     }
     const [wrapper, payload] = entries[0];
     switch (wrapper) {
         case "binary":
             if (typeof payload !== "string") {
-                throw new room_server_client_1.RoomServerException("dataset binary values must be base64 strings");
+                throw new RoomServerException("dataset binary values must be base64 strings");
             }
             return base64ToBytes(payload);
         case "uuid":
             if (typeof payload !== "string") {
-                throw new room_server_client_1.RoomServerException("dataset uuid values must be strings");
+                throw new RoomServerException("dataset uuid values must be strings");
             }
             return new DatasetUuid(payload);
         case "expression":
             if (typeof payload !== "string") {
-                throw new room_server_client_1.RoomServerException("dataset expression values must be strings");
+                throw new RoomServerException("dataset expression values must be strings");
             }
             return new DatasetExpression(payload);
         case "date":
             if (typeof payload !== "string") {
-                throw new room_server_client_1.RoomServerException("dataset date values must be strings");
+                throw new RoomServerException("dataset date values must be strings");
             }
             return new DatasetDate(payload);
         case "timestamp":
             if (typeof payload !== "string") {
-                throw new room_server_client_1.RoomServerException("dataset timestamp values must be strings");
+                throw new RoomServerException("dataset timestamp values must be strings");
             }
             {
                 const parsed = new Date(payload);
                 if (Number.isNaN(parsed.getTime())) {
-                    throw new room_server_client_1.RoomServerException("dataset timestamp value is not valid");
+                    throw new RoomServerException("dataset timestamp value is not valid");
                 }
                 return parsed;
             }
         case "list":
             if (!Array.isArray(payload)) {
-                throw new room_server_client_1.RoomServerException("dataset list values must be arrays");
+                throw new RoomServerException("dataset list values must be arrays");
             }
             return payload.map((item) => decodeRecordValue(item));
         case "struct":
             if (!isRecord(payload)) {
-                throw new room_server_client_1.RoomServerException("dataset struct values must be objects");
+                throw new RoomServerException("dataset struct values must be objects");
             }
             return new DatasetStruct(Object.fromEntries(Object.entries(payload).map(([key, item]) => [key, decodeRecordValue(item)])));
         case "json":
             return new DatasetJson(normalizeDatasetJsonValue(payload));
         default:
-            throw new room_server_client_1.RoomServerException(`unsupported dataset value wrapper '${wrapper}'`);
+            throw new RoomServerException(`unsupported dataset value wrapper '${wrapper}'`);
     }
 }
 function encodeDatasetRecord(record) {
@@ -290,28 +281,28 @@ function rowsChunk(records) {
 }
 function recordsFromRowsChunk(payload, operation) {
     if (!isRecord(payload) || payload.kind !== "rows" || !Array.isArray(payload.rows)) {
-        throw new room_server_client_1.RoomServerException(`unexpected return type from datasets.${operation}`);
+        throw new RoomServerException(`unexpected return type from datasets.${operation}`);
     }
     return payload.rows.map((row) => {
         if (!isRecord(row) || !Array.isArray(row.columns)) {
-            throw new room_server_client_1.RoomServerException(`unexpected return type from datasets.${operation}`);
+            throw new RoomServerException(`unexpected return type from datasets.${operation}`);
         }
         return Object.fromEntries(row.columns.map((column) => {
             if (!isRecord(column) || typeof column.name !== "string") {
-                throw new room_server_client_1.RoomServerException(`unexpected return type from datasets.${operation}`);
+                throw new RoomServerException(`unexpected return type from datasets.${operation}`);
             }
             try {
                 return [column.name, decodeRecordValue(column.value)];
             }
             catch {
-                throw new room_server_client_1.RoomServerException(`unexpected return type from datasets.${operation}`);
+                throw new RoomServerException(`unexpected return type from datasets.${operation}`);
             }
         }));
     });
 }
 function rowChunkList(records, rowsPerChunk = 128) {
     if (rowsPerChunk <= 0) {
-        throw new room_server_client_1.RoomServerException("rowsPerChunk must be greater than zero");
+        throw new RoomServerException("rowsPerChunk must be greater than zero");
     }
     const chunks = [];
     for (let index = 0; index < records.length; index += rowsPerChunk) {
@@ -346,14 +337,14 @@ function normalizeTableRefs(tables) {
 }
 function tableIndexFromJson(value) {
     if (!isRecord(value) || typeof value.name !== "string" || typeof value.type !== "string" || !Array.isArray(value.columns)) {
-        throw new room_server_client_1.RoomServerException("unexpected return type from datasets.list_indexes");
+        throw new RoomServerException("unexpected return type from datasets.list_indexes");
     }
     return {
         name: value.name,
         type: value.type,
         columns: value.columns.map((column) => {
             if (typeof column !== "string") {
-                throw new room_server_client_1.RoomServerException("unexpected return type from datasets.list_indexes");
+                throw new RoomServerException("unexpected return type from datasets.list_indexes");
             }
             return column;
         }),
@@ -368,7 +359,7 @@ function tableIndexFromJson(value) {
 }
 function optimizeResultFromJson(value) {
     if (!isRecord(value)) {
-        throw new room_server_client_1.RoomServerException("unexpected return type from datasets.optimize");
+        throw new RoomServerException("unexpected return type from datasets.optimize");
     }
     return {
         compaction: typeof value.compaction_json === "string" ? JSON.parse(value.compaction_json) : isRecord(value.compaction) ? value.compaction : null,
@@ -378,32 +369,32 @@ function optimizeResultFromJson(value) {
 }
 function tableStatsFromJson(value) {
     if (!isRecord(value)) {
-        throw new room_server_client_1.RoomServerException("unexpected return type from datasets.stats");
+        throw new RoomServerException("unexpected return type from datasets.stats");
     }
     const dataset = typeof value.dataset_json === "string" ? JSON.parse(value.dataset_json) : value.dataset;
     const data = typeof value.data_json === "string" ? JSON.parse(value.data_json) : value.data;
     if (!isRecord(dataset) || !isRecord(data)) {
-        throw new room_server_client_1.RoomServerException("unexpected return type from datasets.stats");
+        throw new RoomServerException("unexpected return type from datasets.stats");
     }
     return { dataset, data };
 }
 function tableVersionFromJson(value) {
     if (!isRecord(value) || typeof value.metadata_json !== "string" || typeof value.timestamp !== "string" || typeof value.version !== "number") {
-        throw new room_server_client_1.RoomServerException("unexpected return type from datasets.list_versions");
+        throw new RoomServerException("unexpected return type from datasets.list_versions");
     }
     const timestamp = new Date(value.timestamp);
     if (Number.isNaN(timestamp.getTime())) {
-        throw new room_server_client_1.RoomServerException("unexpected return type from datasets.list_versions");
+        throw new RoomServerException("unexpected return type from datasets.list_versions");
     }
     let metadata;
     try {
         metadata = JSON.parse(value.metadata_json);
     }
     catch (_) {
-        throw new room_server_client_1.RoomServerException("unexpected return type from datasets.list_versions");
+        throw new RoomServerException("unexpected return type from datasets.list_versions");
     }
     if (!isRecord(metadata)) {
-        throw new room_server_client_1.RoomServerException("unexpected return type from datasets.list_versions");
+        throw new RoomServerException("unexpected return type from datasets.list_versions");
     }
     return {
         version: value.version,
@@ -413,27 +404,27 @@ function tableVersionFromJson(value) {
 }
 function tableBranchFromJson(value) {
     if (!isRecord(value) || typeof value.name !== "string") {
-        throw new room_server_client_1.RoomServerException("unexpected return type from datasets.list_branches");
+        throw new RoomServerException("unexpected return type from datasets.list_branches");
     }
     if (value.parent_branch != null && typeof value.parent_branch !== "string") {
-        throw new room_server_client_1.RoomServerException("unexpected return type from datasets.list_branches");
+        throw new RoomServerException("unexpected return type from datasets.list_branches");
     }
     if (value.parent_version != null
         && (typeof value.parent_version !== "number" || !Number.isInteger(value.parent_version))) {
-        throw new room_server_client_1.RoomServerException("unexpected return type from datasets.list_branches");
+        throw new RoomServerException("unexpected return type from datasets.list_branches");
     }
     if (value.manifest_size != null
         && (typeof value.manifest_size !== "number" || !Number.isInteger(value.manifest_size))) {
-        throw new room_server_client_1.RoomServerException("unexpected return type from datasets.list_branches");
+        throw new RoomServerException("unexpected return type from datasets.list_branches");
     }
     let createdAt = null;
     if (value.created_at != null) {
         if (typeof value.created_at !== "string") {
-            throw new room_server_client_1.RoomServerException("unexpected return type from datasets.list_branches");
+            throw new RoomServerException("unexpected return type from datasets.list_branches");
         }
         createdAt = new Date(value.created_at);
         if (Number.isNaN(createdAt.getTime())) {
-            throw new room_server_client_1.RoomServerException("unexpected return type from datasets.list_branches");
+            throw new RoomServerException("unexpected return type from datasets.list_branches");
         }
     }
     return {
@@ -499,7 +490,7 @@ class DatasetWriteInputStream {
         });
     }
     async *stream() {
-        yield new response_1.JsonContent({ json: this.start });
+        yield new JsonContent({ json: this.start });
         while (!this.closed) {
             await this.waitForPull();
             if (this.closed) {
@@ -512,7 +503,7 @@ class DatasetWriteInputStream {
             if (nextChunk.value.length === 0) {
                 continue;
             }
-            yield new response_1.JsonContent({ json: rowsChunk(nextChunk.value) });
+            yield new JsonContent({ json: rowsChunk(nextChunk.value) });
         }
     }
 }
@@ -528,18 +519,18 @@ async function* toAsyncArrowIterable(chunks) {
     }
 }
 function schemaToIPC(schema) {
-    return (0, apache_arrow_1.tableToIPC)(new apache_arrow_1.Table(schema, []), "stream");
+    return tableToIPC(new Table(schema, []), "stream");
 }
 function tableFromIPCBytes(data) {
-    const table = (0, apache_arrow_1.tableFromIPC)(data);
+    const table = tableFromIPC(data);
     if (table instanceof Promise) {
-        throw new room_server_client_1.RoomServerException("unexpected async Arrow IPC result");
+        throw new RoomServerException("unexpected async Arrow IPC result");
     }
     return table;
 }
 function tableFromChunks(chunks) {
     if (chunks.length === 0) {
-        return new apache_arrow_1.Table(new apache_arrow_1.Schema([]), []);
+        return new Table(new Schema([]), []);
     }
     const [first, ...rest] = chunks;
     return first.concat(...rest);
@@ -591,7 +582,7 @@ class DatasetArrowWriteInputStream {
         });
     }
     async *stream() {
-        yield new response_1.BinaryContent({
+        yield new BinaryContent({
             data: this.schema == null ? new Uint8Array() : schemaToIPC(this.schema),
             headers: this.start,
         });
@@ -607,8 +598,8 @@ class DatasetArrowWriteInputStream {
             if (nextChunk.value.numRows === 0) {
                 continue;
             }
-            yield new response_1.BinaryContent({
-                data: (0, apache_arrow_1.tableToIPC)(nextChunk.value, "stream"),
+            yield new BinaryContent({
+                data: tableToIPC(nextChunk.value, "stream"),
                 headers: { kind: "data", content_type: ARROW_IPC_STREAM_MIME_TYPE },
             });
         }
@@ -655,13 +646,13 @@ class DatasetReadInputStream {
         });
     }
     async *stream() {
-        yield new response_1.JsonContent({ json: this.start });
+        yield new JsonContent({ json: this.start });
         while (!this.closed) {
             await this.waitForPull();
             if (this.closed) {
                 return;
             }
-            yield new response_1.JsonContent({ json: { kind: "pull" } });
+            yield new JsonContent({ json: { kind: "pull" } });
         }
     }
 }
@@ -706,26 +697,26 @@ class DatasetArrowReadInputStream {
         });
     }
     async *stream() {
-        yield new response_1.BinaryContent({ data: new Uint8Array(), headers: this.start });
+        yield new BinaryContent({ data: new Uint8Array(), headers: this.start });
         while (!this.closed) {
             await this.waitForPull();
             if (this.closed) {
                 return;
             }
-            yield new response_1.BinaryContent({ data: new Uint8Array(), headers: { kind: "pull" } });
+            yield new BinaryContent({ data: new Uint8Array(), headers: { kind: "pull" } });
         }
     }
 }
-class DatasetsClient {
+export class DatasetsClient {
     constructor({ room }) {
         this.room = room;
     }
     _unexpectedResponseError(operation) {
-        return new room_server_client_1.RoomServerException(`unexpected return type from datasets.${operation}`);
+        return new RoomServerException(`unexpected return type from datasets.${operation}`);
     }
     async invoke(operation, input) {
         const response = await this.room.invoke({ toolkit: "dataset", tool: operation, input });
-        if (response instanceof response_1.JsonContent) {
+        if (response instanceof JsonContent) {
             return response;
         }
         if (response == null) {
@@ -743,23 +734,23 @@ class DatasetsClient {
         const response = await this.invokeStream(operation, input.stream());
         try {
             for await (const chunk of response) {
-                if (chunk instanceof response_1.ErrorContent) {
-                    throw new room_server_client_1.RoomServerException(chunk.text, chunk.code);
+                if (chunk instanceof ErrorContent) {
+                    throw new RoomServerException(chunk.text, chunk.code);
                 }
-                if (chunk instanceof response_1.ControlContent) {
+                if (chunk instanceof ControlContent) {
                     if (chunk.method === "close") {
                         return;
                     }
                     throw this._unexpectedResponseError(operation);
                 }
-                if (chunk instanceof response_1.BinaryContent) {
+                if (chunk instanceof BinaryContent) {
                     if (chunk.headers.kind !== "pull") {
                         throw this._unexpectedResponseError(operation);
                     }
                     input.requestNext();
                     continue;
                 }
-                if (!(chunk instanceof response_1.JsonContent) || chunk.json.kind !== "pull") {
+                if (!(chunk instanceof JsonContent) || chunk.json.kind !== "pull") {
                     throw this._unexpectedResponseError(operation);
                 }
                 input.requestNext();
@@ -775,16 +766,16 @@ class DatasetsClient {
         input.requestNext();
         try {
             for await (const chunk of response) {
-                if (chunk instanceof response_1.ErrorContent) {
-                    throw new room_server_client_1.RoomServerException(chunk.text, chunk.code);
+                if (chunk instanceof ErrorContent) {
+                    throw new RoomServerException(chunk.text, chunk.code);
                 }
-                if (chunk instanceof response_1.ControlContent) {
+                if (chunk instanceof ControlContent) {
                     if (chunk.method === "close") {
                         return;
                     }
                     throw this._unexpectedResponseError(operation);
                 }
-                if (!(chunk instanceof response_1.JsonContent)) {
+                if (!(chunk instanceof JsonContent)) {
                     throw this._unexpectedResponseError(operation);
                 }
                 yield recordsFromRowsChunk(chunk.json, operation);
@@ -801,16 +792,16 @@ class DatasetsClient {
         input.requestNext();
         try {
             for await (const chunk of response) {
-                if (chunk instanceof response_1.ErrorContent) {
-                    throw new room_server_client_1.RoomServerException(chunk.text, chunk.code);
+                if (chunk instanceof ErrorContent) {
+                    throw new RoomServerException(chunk.text, chunk.code);
                 }
-                if (chunk instanceof response_1.ControlContent) {
+                if (chunk instanceof ControlContent) {
                     if (chunk.method === "close") {
                         return;
                     }
                     throw this._unexpectedResponseError(operation);
                 }
-                if (!(chunk instanceof response_1.BinaryContent) || chunk.headers.kind !== "data") {
+                if (!(chunk instanceof BinaryContent) || chunk.headers.kind !== "data") {
                     throw this._unexpectedResponseError(operation);
                 }
                 yield tableFromIPCBytes(chunk.data);
@@ -826,7 +817,7 @@ class DatasetsClient {
             namespace: namespace ?? null,
             branch: branch ?? null,
         });
-        if (!(response instanceof response_1.JsonContent)) {
+        if (!(response instanceof JsonContent)) {
             throw this._unexpectedResponseError("list_tables");
         }
         return Array.isArray(response.json.tables) ? response.json.tables : [];
@@ -846,7 +837,7 @@ class DatasetsClient {
         return this.createTable({
             name,
             schema,
-            data: data == null ? undefined : data instanceof apache_arrow_1.Table ? [data] : data,
+            data: data == null ? undefined : data instanceof Table ? [data] : data,
             mode,
             namespace,
             branch,
@@ -888,7 +879,7 @@ class DatasetsClient {
     async createTableFromData({ name, data, mode = "create", namespace, branch, metadata }) {
         return this.createTable({
             name,
-            data: data == null ? undefined : data instanceof apache_arrow_1.Table ? [data] : data,
+            data: data == null ? undefined : data instanceof Table ? [data] : data,
             mode,
             namespace,
             branch,
@@ -972,8 +963,8 @@ class DatasetsClient {
         });
     }
     async addColumns({ table, newColumns, namespace, branch }) {
-        if (newColumns instanceof apache_arrow_1.Schema) {
-            await this.invokeContent("add_columns", new response_1.BinaryContent({
+        if (newColumns instanceof Schema) {
+            await this.invokeContent("add_columns", new BinaryContent({
                 data: schemaToIPC(newColumns),
                 headers: {
                     table,
@@ -1064,8 +1055,8 @@ class DatasetsClient {
         return tableFromChunks(await this.sql({ query, tables, params, namespace, branch }));
     }
     async openSqlQuery({ query, tables, params, namespace, branch }) {
-        const response = await this.invokeContent("open_sql_query", new response_1.BinaryContent({
-            data: params == null ? new Uint8Array() : (0, apache_arrow_1.tableToIPC)(params, "stream"),
+        const response = await this.invokeContent("open_sql_query", new BinaryContent({
+            data: params == null ? new Uint8Array() : tableToIPC(params, "stream"),
             headers: {
                 query,
                 tables: normalizeTableRefs(tables ?? []),
@@ -1073,7 +1064,7 @@ class DatasetsClient {
                 branch: branch ?? null,
             },
         }));
-        if (!(response instanceof response_1.BinaryContent)) {
+        if (!(response instanceof BinaryContent)) {
             throw this._unexpectedResponseError("open_sql_query");
         }
         const queryId = response.headers.query_id;
@@ -1087,8 +1078,8 @@ class DatasetsClient {
         };
     }
     async executeSql({ query, tables, params, namespace, branch }) {
-        const response = await this.invokeContent("execute_sql", new response_1.BinaryContent({
-            data: params == null ? new Uint8Array() : (0, apache_arrow_1.tableToIPC)(params, "stream"),
+        const response = await this.invokeContent("execute_sql", new BinaryContent({
+            data: params == null ? new Uint8Array() : tableToIPC(params, "stream"),
             headers: {
                 query,
                 tables: normalizeTableRefs(tables ?? []),
@@ -1096,7 +1087,7 @@ class DatasetsClient {
                 branch: branch ?? null,
             },
         }));
-        if (response instanceof response_1.BinaryContent) {
+        if (response instanceof BinaryContent) {
             if (response.headers.kind !== "query") {
                 throw this._unexpectedResponseError("execute_sql");
             }
@@ -1110,7 +1101,7 @@ class DatasetsClient {
                 queryId,
             };
         }
-        if (response instanceof response_1.JsonContent) {
+        if (response instanceof JsonContent) {
             if (response.json.kind !== "statement"
                 || typeof response.json.rows_affected !== "number"
                 || !Number.isInteger(response.json.rows_affected)) {
@@ -1126,7 +1117,7 @@ class DatasetsClient {
     async *sqlStream({ query, tables, params, namespace, branch }) {
         const result = await this.executeSql({ query, tables, params, namespace, branch });
         if (result.kind === "statement") {
-            throw new room_server_client_1.RoomServerException(`SQL statement did not return rows; rows_affected=${result.rowsAffected}`);
+            throw new RoomServerException(`SQL statement did not return rows; rows_affected=${result.rowsAffected}`);
         }
         const opened = result;
         try {
@@ -1144,21 +1135,21 @@ class DatasetsClient {
     }
     async closeSqlQuery({ queryId }) {
         const response = await this.room.invoke({ toolkit: "dataset", tool: "close_sql_query", input: { query_id: queryId } });
-        if (!(response instanceof response_1.EmptyContent)) {
+        if (!(response instanceof EmptyContent)) {
             throw this._unexpectedResponseError("close_sql_query");
         }
     }
     async cancelSqlQuery({ queryId }) {
         const response = await this.room.invoke({ toolkit: "dataset", tool: "cancel_sql_query", input: { query_id: queryId } });
-        if (!(response instanceof response_1.JsonContent)
+        if (!(response instanceof JsonContent)
             || !["cancelled", "cancelling", "not_cancellable"].includes(response.json.status)) {
             throw this._unexpectedResponseError("cancel_sql_query");
         }
         return { status: response.json.status };
     }
     async executeSqlStatement({ query, tables, params, namespace, branch }) {
-        const response = await this.invokeContent("execute_sql_statement", new response_1.BinaryContent({
-            data: params == null ? new Uint8Array() : (0, apache_arrow_1.tableToIPC)(params, "stream"),
+        const response = await this.invokeContent("execute_sql_statement", new BinaryContent({
+            data: params == null ? new Uint8Array() : tableToIPC(params, "stream"),
             headers: {
                 query,
                 tables: normalizeTableRefs(tables ?? []),
@@ -1166,7 +1157,7 @@ class DatasetsClient {
                 branch: branch ?? null,
             },
         }));
-        if (!(response instanceof response_1.JsonContent)
+        if (!(response instanceof JsonContent)
             || typeof response.json.rows_affected !== "number"
             || !Number.isInteger(response.json.rows_affected)) {
             throw this._unexpectedResponseError("execute_sql_statement");
@@ -1233,16 +1224,16 @@ class DatasetsClient {
         input.requestNext();
         try {
             for await (const chunk of response) {
-                if (chunk instanceof response_1.ErrorContent) {
-                    throw new room_server_client_1.RoomServerException(chunk.text, chunk.code);
+                if (chunk instanceof ErrorContent) {
+                    throw new RoomServerException(chunk.text, chunk.code);
                 }
-                if (chunk instanceof response_1.ControlContent) {
+                if (chunk instanceof ControlContent) {
                     if (chunk.method === "close") {
                         return;
                     }
                     throw this._unexpectedResponseError("watch_table");
                 }
-                if (chunk instanceof response_1.BinaryContent) {
+                if (chunk instanceof BinaryContent) {
                     if (chunk.headers.kind !== "data") {
                         throw this._unexpectedResponseError("watch_table");
                     }
@@ -1260,7 +1251,7 @@ class DatasetsClient {
                     input.requestNext();
                     continue;
                 }
-                if (chunk instanceof response_1.JsonContent) {
+                if (chunk instanceof JsonContent) {
                     if (!isRecord(chunk.json)) {
                         throw this._unexpectedResponseError("watch_table");
                     }
@@ -1299,7 +1290,7 @@ class DatasetsClient {
             branch: branch ?? null,
             version: version ?? null,
         });
-        if (!(response instanceof response_1.JsonContent) || typeof response.json.count !== "number" || !Number.isInteger(response.json.count)) {
+        if (!(response instanceof JsonContent) || typeof response.json.count !== "number" || !Number.isInteger(response.json.count)) {
             throw this._unexpectedResponseError("count");
         }
         return response.json.count;
@@ -1315,7 +1306,7 @@ class DatasetsClient {
                 version: version ?? null,
             },
         });
-        if (!(response instanceof response_1.BinaryContent)) {
+        if (!(response instanceof BinaryContent)) {
             throw this._unexpectedResponseError("inspect");
         }
         return tableFromIPCBytes(response.data).schema;
@@ -1331,7 +1322,7 @@ class DatasetsClient {
             branch: branch ?? null,
             config: config ?? null,
         });
-        if (!(response instanceof response_1.JsonContent)) {
+        if (!(response instanceof JsonContent)) {
             throw this._unexpectedResponseError("optimize");
         }
         return optimizeResultFromJson(response.json);
@@ -1344,7 +1335,7 @@ class DatasetsClient {
             version: version ?? null,
             max_rows_per_group: maxRowsPerGroup ?? null,
         });
-        if (!(response instanceof response_1.JsonContent)) {
+        if (!(response instanceof JsonContent)) {
             throw this._unexpectedResponseError("stats");
         }
         return tableStatsFromJson(response.json);
@@ -1362,7 +1353,7 @@ class DatasetsClient {
             namespace: namespace ?? null,
             branch: branch ?? null,
         });
-        if (!(response instanceof response_1.JsonContent) || !Array.isArray(response.json.versions)) {
+        if (!(response instanceof JsonContent) || !Array.isArray(response.json.versions)) {
             throw this._unexpectedResponseError("list_versions");
         }
         return response.json.versions.map((version) => tableVersionFromJson(version));
@@ -1381,7 +1372,7 @@ class DatasetsClient {
             branch: branch ?? null,
             version: version ?? null,
         });
-        if (!(response instanceof response_1.JsonContent) || !Array.isArray(response.json.indexes)) {
+        if (!(response instanceof JsonContent) || !Array.isArray(response.json.indexes)) {
             throw this._unexpectedResponseError("list_indexes");
         }
         return response.json.indexes.map((index) => tableIndexFromJson(index));
@@ -1390,7 +1381,7 @@ class DatasetsClient {
         const response = await this.invoke("list_branches", {
             namespace: namespace ?? null,
         });
-        if (!(response instanceof response_1.JsonContent) || !Array.isArray(response.json.branches)) {
+        if (!(response instanceof JsonContent) || !Array.isArray(response.json.branches)) {
             throw this._unexpectedResponseError("list_branches");
         }
         return response.json.branches.map((branch) => tableBranchFromJson(branch));
@@ -1414,4 +1405,3 @@ class DatasetsClient {
         });
     }
 }
-exports.DatasetsClient = DatasetsClient;