@meshagent/meshagent 0.41.4 → 0.41.6

package/dist/esm/datasets-client.d.ts

@@ -206,6 +206,31 @@ export declare class DatasetsClient {
         branch?: string;
         metadata?: Record<string, unknown>;
     }): Promise<void>;
+    createTableWithArrowSchema({ name, schema, batches, mode, namespace, branch, metadata }: {
+        name: string;
+        schema: Schema;
+        batches?: ArrowTableChunks;
+        mode?: CreateMode;
+        namespace?: string[];
+        branch?: string;
+        metadata?: Record<string, unknown>;
+    }): Promise<void>;
+    createTableFromArrowBatches({ name, batches, mode, namespace, branch, metadata }: {
+        name: string;
+        batches: ArrowTableChunks;
+        mode?: CreateMode;
+        namespace?: string[];
+        branch?: string;
+        metadata?: Record<string, unknown>;
+    }): Promise<void>;
+    createTableFromArrowTable({ name, table, mode, namespace, branch, metadata }: {
+        name: string;
+        table: Table;
+        mode?: CreateMode;
+        namespace?: string[];
+        branch?: string;
+        metadata?: Record<string, unknown>;
+    }): Promise<void>;
     createTableFromData({ name, data, mode, namespace, branch, metadata }: {
         name: string;
         data?: Iterable<Table> | Table;
@@ -237,6 +262,12 @@ export declare class DatasetsClient {
         namespace?: string[];
         branch?: string;
     }): Promise<void>;
+    renameTable({ name, newName, namespace, branch }: {
+        name: string;
+        newName: string;
+        namespace?: string[];
+        branch?: string;
+    }): Promise<void>;
     importFromStorage({ table, path, mode, format, on, sheet, batchSize, namespace, branch }: {
         table: string;
         path: string;
@@ -256,6 +287,13 @@ export declare class DatasetsClient {
         branch?: string;
         version?: number;
     }): Promise<void>;
+    updateColumnMetadata({ table, column, metadata, namespace, branch }: {
+        table: string;
+        column: string;
+        metadata: Record<string, string>;
+        namespace?: string[];
+        branch?: string;
+    }): Promise<void>;
     dropIndex({ table, name, namespace, branch }: {
         table: string;
         name: string;
@@ -280,6 +318,12 @@ export declare class DatasetsClient {
         namespace?: string[];
         branch?: string;
     }): Promise<void>;
+    insertTable({ table, records, namespace, branch }: {
+        table: string;
+        records: Table;
+        namespace?: string[];
+        branch?: string;
+    }): Promise<void>;
     insertStream({ table, chunks, namespace, branch }: {
         table: string;
         chunks: ArrowTableChunks;
@@ -306,6 +350,13 @@ export declare class DatasetsClient {
         namespace?: string[];
         branch?: string;
     }): Promise<void>;
+    mergeTable({ table, on, records, namespace, branch }: {
+        table: string;
+        on: string;
+        records: Table;
+        namespace?: string[];
+        branch?: string;
+    }): Promise<void>;
     mergeStream({ table, on, chunks, namespace, branch }: {
         table: string;
         on: string;
@@ -320,6 +371,13 @@ export declare class DatasetsClient {
         namespace?: string[];
         branch?: string;
     }): Promise<Table[]>;
+    sqlTable({ query, tables, params, namespace, branch }: {
+        query: string;
+        tables?: Array<TableRef | string>;
+        params?: Table;
+        namespace?: string[];
+        branch?: string;
+    }): Promise<Table>;
     openSqlQuery({ query, tables, params, namespace, branch }: {
         query: string;
         tables?: Array<TableRef | string>;
@@ -369,6 +427,18 @@ export declare class DatasetsClient {
         branch?: string;
         version?: number;
     }): Promise<Table[]>;
+    searchTable({ table, text, vector, where, offset, limit, select, namespace, branch, version }: {
+        table: string;
+        text?: string;
+        vector?: number[];
+        where?: DatasetWhere;
+        offset?: number;
+        limit?: number;
+        select?: string[];
+        namespace?: string[];
+        branch?: string;
+        version?: number;
+    }): Promise<Table>;
     searchStream({ table, text, vector, where, offset, limit, select, namespace, branch, version }: {
         table: string;
         text?: string;

package/dist/esm/datasets-client.js

@@ -537,6 +537,13 @@ function tableFromIPCBytes(data) {
     }
     return table;
 }
+function tableFromChunks(chunks) {
+    if (chunks.length === 0) {
+        return new apache_arrow_1.Table(new apache_arrow_1.Schema([]), []);
+    }
+    const [first, ...rest] = chunks;
+    return first.concat(...rest);
+}
 function schemaFromIPCBytes(data) {
     return tableFromIPCBytes(data).schema;
 }
@@ -846,6 +853,38 @@ class DatasetsClient {
             metadata,
         });
     }
+    async createTableWithArrowSchema({ name, schema, batches, mode = "create", namespace, branch, metadata }) {
+        return this.createTable({
+            name,
+            schema,
+            data: batches,
+            mode,
+            namespace,
+            branch,
+            metadata,
+        });
+    }
+    async createTableFromArrowBatches({ name, batches, mode = "create", namespace, branch, metadata }) {
+        return this.createTable({
+            name,
+            data: batches,
+            mode,
+            namespace,
+            branch,
+            metadata,
+        });
+    }
+    async createTableFromArrowTable({ name, table, mode = "create", namespace, branch, metadata }) {
+        return this.createTableWithArrowSchema({
+            name,
+            schema: table.schema,
+            batches: [table],
+            mode,
+            namespace,
+            branch,
+            metadata,
+        });
+    }
     async createTableFromData({ name, data, mode = "create", namespace, branch, metadata }) {
         return this.createTable({
             name,
@@ -882,6 +921,14 @@ class DatasetsClient {
             },
         });
     }
+    async renameTable({ name, newName, namespace, branch }) {
+        await this.invoke("rename_table", {
+            name,
+            new_name: newName,
+            namespace: namespace ?? null,
+            branch: branch ?? null,
+        });
+    }
     async importFromStorage({ table, path, mode = "create", format = "auto", on, sheet, batchSize, namespace, branch }) {
         const input = {
             table,
@@ -908,6 +955,15 @@ class DatasetsClient {
             version: version ?? null,
         });
     }
+    async updateColumnMetadata({ table, column, metadata, namespace, branch }) {
+        await this.invoke("update_column_metadata", {
+            table,
+            column,
+            metadata: metadataEntries(metadata),
+            namespace: namespace ?? null,
+            branch: branch ?? null,
+        });
+    }
     async dropIndex({ table, name, namespace, branch }) {
         await this.room.invoke({
             toolkit: "dataset",
@@ -949,6 +1005,9 @@ class DatasetsClient {
     async insert({ table, records, namespace, branch }) {
         await this.insertStream({ table, chunks: [records], namespace, branch });
     }
+    async insertTable({ table, records, namespace, branch }) {
+        await this.insertStream({ table, chunks: [records], namespace, branch });
+    }
     async insertStream({ table, chunks, namespace, branch }) {
         const input = new DatasetArrowWriteInputStream({
             kind: "start",
@@ -981,6 +1040,9 @@ class DatasetsClient {
     async merge({ table, on, records, namespace, branch }) {
         await this.mergeStream({ table, on, chunks: [records], namespace, branch });
     }
+    async mergeTable({ table, on, records, namespace, branch }) {
+        await this.mergeStream({ table, on, chunks: [records], namespace, branch });
+    }
     async mergeStream({ table, on, chunks, namespace, branch }) {
         const input = new DatasetArrowWriteInputStream({
             kind: "start",
@@ -998,6 +1060,9 @@ class DatasetsClient {
         }
         return results;
     }
+    async sqlTable({ query, tables, params, namespace, branch }) {
+        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"),
@@ -1126,6 +1191,20 @@ class DatasetsClient {
         }
         return results;
     }
+    async searchTable({ table, text, vector, where, offset, limit, select, namespace, branch, version }) {
+        return tableFromChunks(await this.search({
+            table,
+            text,
+            vector,
+            where,
+            offset,
+            limit,
+            select,
+            namespace,
+            branch,
+            version,
+        }));
+    }
     async *searchStream({ table, text, vector, where, offset, limit, select, namespace, branch, version }) {
         yield* this.streamArrow("search", {
             kind: "start",