@robosystems/client 0.2.14 → 0.2.16

package/extensions/FileClient.d.ts

@@ -0,0 +1,57 @@
+export interface FileUploadOptions {
+    onProgress?: (message: string) => void;
+    fixLocalStackUrl?: boolean;
+    fileName?: string;
+    ingestToGraph?: boolean;
+}
+export interface FileUploadResult {
+    fileId: string;
+    fileSize: number;
+    rowCount: number;
+    tableName: string;
+    fileName: string;
+    success: boolean;
+    error?: string;
+}
+export interface FileInfo {
+    fileId: string;
+    fileName: string;
+    tableName: string;
+    status: string;
+    fileSize: number;
+    rowCount: number;
+    createdAt: string;
+}
+export type FileInput = File | Blob | Buffer | ReadableStream<Uint8Array>;
+export declare class FileClient {
+    private config;
+    constructor(config: {
+        baseUrl: string;
+        credentials?: 'include' | 'same-origin' | 'omit';
+        headers?: Record<string, string>;
+        token?: string;
+    });
+    /**
+     * Upload a Parquet file to staging
+     *
+     * Handles the complete 3-step upload process:
+     * 1. Get presigned upload URL
+     * 2. Upload file to S3
+     * 3. Mark file as 'uploaded' (backend validates, calculates size/row count)
+     */
+    upload(graphId: string, tableName: string, fileOrBuffer: FileInput, options?: FileUploadOptions): Promise<FileUploadResult>;
+    /**
+     * List files in a graph
+     */
+    list(graphId: string, tableName?: string, status?: string): Promise<FileInfo[]>;
+    /**
+     * Get file information
+     */
+    get(graphId: string, fileId: string): Promise<FileInfo | null>;
+    /**
+     * Delete a file
+     */
+    delete(graphId: string, fileId: string, cascade?: boolean): Promise<boolean>;
+    private getFileName;
+    private getFileContent;
+}

package/extensions/{TableIngestClient.js → FileClient.js}

@@ -1,39 +1,37 @@
 'use client';
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.TableIngestClient = void 0;
+exports.FileClient = void 0;
 /**
- * Table Ingest Client for RoboSystems API
+ * File Client for RoboSystems API
  *
- * Simplifies uploading Parquet files to staging tables and ingesting them into graphs.
- * Supports File (browser), Blob (browser), Buffer (Node.js), and ReadableStream.
+ * Manages file upload operations including presigned URLs, S3 uploads,
+ * and file status tracking.
  */
 const sdk_gen_1 = require("../sdk.gen");
-class TableIngestClient {
+class FileClient {
     constructor(config) {
         this.config = config;
     }
     /**
-     * Upload a Parquet file to a staging table
+     * Upload a Parquet file to staging
      *
-     * This method handles the complete 3-step upload process:
+     * Handles the complete 3-step upload process:
      * 1. Get presigned upload URL
      * 2. Upload file to S3
      * 3. Mark file as 'uploaded' (backend validates, calculates size/row count)
-     *
-     * Supports File (browser), Blob (browser), Buffer (Node.js), and ReadableStream.
      */
-    async uploadParquetFile(graphId, tableName, fileOrBuffer, options = {}) {
+    async upload(graphId, tableName, fileOrBuffer, options = {}) {
         const fileName = this.getFileName(fileOrBuffer, options.fileName);
         try {
-            // Step 1: Get presigned upload URL
             options.onProgress?.(`Getting upload URL for ${fileName} -> table '${tableName}'...`);
             const uploadRequest = {
                 file_name: fileName,
                 content_type: 'application/x-parquet',
+                table_name: tableName,
             };
-            const uploadUrlResponse = await (0, sdk_gen_1.getUploadUrl)({
-                path: { graph_id: graphId, table_name: tableName },
+            const uploadUrlResponse = await (0, sdk_gen_1.createFileUpload)({
+                path: { graph_id: graphId },
                 body: uploadRequest,
             });
             if (uploadUrlResponse.error || !uploadUrlResponse.data) {
@@ -50,11 +48,9 @@ class TableIngestClient {
             const uploadData = uploadUrlResponse.data;
             let uploadUrl = uploadData.upload_url;
             const fileId = uploadData.file_id;
-            // Fix LocalStack URL if needed
             if (options.fixLocalStackUrl && uploadUrl.includes('localstack:4566')) {
                 uploadUrl = uploadUrl.replace('localstack:4566', 'localhost:4566');
             }
-            // Step 2: Upload file to S3
             options.onProgress?.(`Uploading ${fileName} to S3...`);
             const fileContent = await this.getFileContent(fileOrBuffer);
             const fileSize = fileContent.byteLength;
@@ -76,12 +72,11 @@ class TableIngestClient {
                     error: `S3 upload failed: ${s3Response.status} ${s3Response.statusText}`,
                 };
             }
-            // Step 3: Mark file as uploaded (backend validates and calculates size/row count)
             options.onProgress?.(`Marking ${fileName} as uploaded...`);
             const statusUpdate = {
                 status: 'uploaded',
             };
-            const updateResponse = await (0, sdk_gen_1.updateFileStatus)({
+            const updateResponse = await (0, sdk_gen_1.updateFile)({
                 path: { graph_id: graphId, file_id: fileId },
                 body: statusUpdate,
             });
@@ -96,7 +91,6 @@ class TableIngestClient {
                     error: 'Failed to complete file upload',
                 };
             }
-            // Extract size and row count from response (calculated by backend)
             const responseData = updateResponse.data;
             const actualFileSize = responseData.file_size_bytes || 0;
             const actualRowCount = responseData.row_count || 0;
@@ -123,110 +117,96 @@ class TableIngestClient {
         }
     }
     /**
-     * List all staging tables in a graph
+     * List files in a graph
      */
-    async listStagingTables(graphId) {
+    async list(graphId, tableName, status) {
         try {
-            const response = await (0, sdk_gen_1.listTables)({
+            const response = await (0, sdk_gen_1.listFiles)({
                 path: { graph_id: graphId },
+                query: {
+                    table_name: tableName,
+                    status: status,
+                },
             });
             if (response.error || !response.data) {
-                console.error('Failed to list tables:', response.error);
+                console.error('Failed to list files:', response.error);
                 return [];
             }
-            const tableData = response.data;
-            return (tableData.tables?.map((table) => ({
-                tableName: table.table_name,
-                rowCount: table.row_count,
-                fileCount: table.file_count || 0,
-                totalSizeBytes: table.total_size_bytes || 0,
+            const fileData = response.data;
+            return (fileData.files?.map((file) => ({
+                fileId: file.file_id,
+                fileName: file.file_name,
+                tableName: file.table_name,
+                status: file.status,
+                fileSize: file.file_size_bytes || 0,
+                rowCount: file.row_count || 0,
+                createdAt: file.created_at,
             })) || []);
         }
         catch (error) {
-            console.error('Failed to list tables:', error);
+            console.error('Failed to list files:', error);
             return [];
         }
     }
     /**
-     * Ingest all staging tables into the graph
+     * Get file information
      */
-    async ingestAllTables(graphId, options = {}) {
+    async get(graphId, fileId) {
         try {
-            options.onProgress?.('Starting table ingestion...');
-            const ingestRequest = {
-                ignore_errors: options.ignoreErrors ?? true,
-                rebuild: options.rebuild ?? false,
-            };
-            const response = await (0, sdk_gen_1.ingestTables)({
-                path: { graph_id: graphId },
-                body: ingestRequest,
+            const response = await (0, sdk_gen_1.getFile)({
+                path: { graph_id: graphId, file_id: fileId },
             });
             if (response.error || !response.data) {
-                return {
-                    success: false,
-                    error: `Failed to ingest tables: ${response.error}`,
-                };
+                console.error('Failed to get file:', response.error);
+                return null;
             }
-            const result = response.data;
-            options.onProgress?.('✅ Table ingestion completed');
+            const file = response.data;
             return {
-                success: true,
-                operationId: result.operation_id,
-                message: result.message || 'Ingestion started',
+                fileId: file.file_id,
+                fileName: file.file_name,
+                tableName: file.table_name,
+                status: file.status,
+                fileSize: file.file_size_bytes || 0,
+                rowCount: file.row_count || 0,
+                createdAt: file.created_at,
             };
         }
         catch (error) {
-            return {
-                success: false,
-                error: error instanceof Error ? error.message : String(error),
-            };
+            console.error('Failed to get file:', error);
+            return null;
         }
     }
     /**
-     * Convenience method to upload a file and immediately ingest it
+     * Delete a file
      */
-    async uploadAndIngest(graphId, tableName, fileOrBuffer, uploadOptions = {}, ingestOptions = {}) {
-        // Upload the file
-        const uploadResult = await this.uploadParquetFile(graphId, tableName, fileOrBuffer, uploadOptions);
-        if (!uploadResult.success) {
-            return {
-                upload: uploadResult,
-                ingest: null,
-            };
+    async delete(graphId, fileId, cascade = false) {
+        try {
+            const response = await (0, sdk_gen_1.deleteFile)({
+                path: { graph_id: graphId, file_id: fileId },
+                query: { cascade },
+            });
+            return !response.error;
+        }
+        catch (error) {
+            console.error('Failed to delete file:', error);
+            return false;
         }
-        // Ingest the table
-        const ingestResult = await this.ingestAllTables(graphId, ingestOptions);
-        return {
-            upload: uploadResult,
-            ingest: ingestResult,
-        };
     }
-    /**
-     * Get file name from input or use provided override
-     */
     getFileName(fileOrBuffer, override) {
         if (override)
             return override;
-        // File object (browser)
         if ('name' in fileOrBuffer && typeof fileOrBuffer.name === 'string') {
             return fileOrBuffer.name;
         }
-        // Default name for buffers/blobs/streams
         return 'data.parquet';
     }
-    /**
-     * Convert various file inputs to ArrayBuffer for upload
-     */
     async getFileContent(fileOrBuffer) {
-        // File or Blob (browser)
         if (fileOrBuffer instanceof Blob || fileOrBuffer instanceof File) {
             return fileOrBuffer.arrayBuffer();
         }
-        // Buffer (Node.js)
         if (Buffer.isBuffer(fileOrBuffer)) {
             return fileOrBuffer.buffer.slice(fileOrBuffer.byteOffset, fileOrBuffer.byteOffset + fileOrBuffer.byteLength);
         }
-        // ReadableStream
         if ('getReader' in fileOrBuffer) {
             const reader = fileOrBuffer.getReader();
             const chunks = [];
@@ -237,7 +217,6 @@ class TableIngestClient {
                 if (value)
                     chunks.push(value);
             }
-            // Concatenate chunks
             const totalLength = chunks.reduce((sum, chunk) => sum + chunk.length, 0);
             const result = new Uint8Array(totalLength);
             let offset = 0;
@@ -250,4 +229,4 @@ class TableIngestClient {
         throw new Error('Unsupported file input type');
     }
 }
-exports.TableIngestClient = TableIngestClient;
+exports.FileClient = FileClient;

package/extensions/{TableIngestClient.ts → FileClient.ts}

@@ -1,34 +1,23 @@
 'use client'
 
 /**
- * Table Ingest Client for RoboSystems API
+ * File Client for RoboSystems API
  *
- * Simplifies uploading Parquet files to staging tables and ingesting them into graphs.
- * Supports File (browser), Blob (browser), Buffer (Node.js), and ReadableStream.
+ * Manages file upload operations including presigned URLs, S3 uploads,
+ * and file status tracking.
  */
 
-import { getUploadUrl, ingestTables, listTables, updateFileStatus } from '../sdk.gen'
-import type {
-  BulkIngestRequest,
-  FileStatusUpdate,
-  FileUploadRequest,
-  FileUploadResponse,
-  TableListResponse,
-} from '../types.gen'
+import { createFileUpload, deleteFile, getFile, listFiles, updateFile } from '../sdk.gen'
+import type { FileStatusUpdate, FileUploadRequest, FileUploadResponse } from '../types.gen'
 
-export interface UploadOptions {
+export interface FileUploadOptions {
   onProgress?: (message: string) => void
-  fixLocalStackUrl?: boolean // Auto-fix LocalStack URLs for localhost
-  fileName?: string // Override file name (useful for buffer/blob uploads)
+  fixLocalStackUrl?: boolean
+  fileName?: string
+  ingestToGraph?: boolean
 }
 
-export interface IngestOptions {
-  ignoreErrors?: boolean
-  rebuild?: boolean
-  onProgress?: (message: string) => void
-}
-
-export interface UploadResult {
+export interface FileUploadResult {
   fileId: string
   fileSize: number
   rowCount: number
@@ -38,24 +27,19 @@ export interface UploadResult {
   error?: string
 }
 
-export interface TableInfo {
+export interface FileInfo {
+  fileId: string
+  fileName: string
   tableName: string
+  status: string
+  fileSize: number
   rowCount: number
-  fileCount: number
-  totalSizeBytes: number
+  createdAt: string
 }
 
-export interface IngestResult {
-  success: boolean
-  operationId?: string
-  message?: string
-  error?: string
-}
-
-// Union type for all supported file inputs
 export type FileInput = File | Blob | Buffer | ReadableStream<Uint8Array>
 
-export class TableIngestClient {
+export class FileClient {
   private config: {
     baseUrl: string
     credentials?: 'include' | 'same-origin' | 'omit'
@@ -73,34 +57,32 @@ export class TableIngestClient {
   }
 
   /**
-   * Upload a Parquet file to a staging table
+   * Upload a Parquet file to staging
    *
-   * This method handles the complete 3-step upload process:
+   * Handles the complete 3-step upload process:
    * 1. Get presigned upload URL
    * 2. Upload file to S3
    * 3. Mark file as 'uploaded' (backend validates, calculates size/row count)
-   *
-   * Supports File (browser), Blob (browser), Buffer (Node.js), and ReadableStream.
    */
-  async uploadParquetFile(
+  async upload(
     graphId: string,
     tableName: string,
     fileOrBuffer: FileInput,
-    options: UploadOptions = {}
-  ): Promise<UploadResult> {
+    options: FileUploadOptions = {}
+  ): Promise<FileUploadResult> {
     const fileName = this.getFileName(fileOrBuffer, options.fileName)
 
     try {
-      // Step 1: Get presigned upload URL
       options.onProgress?.(`Getting upload URL for ${fileName} -> table '${tableName}'...`)
 
       const uploadRequest: FileUploadRequest = {
         file_name: fileName,
         content_type: 'application/x-parquet',
+        table_name: tableName,
       }
 
-      const uploadUrlResponse = await getUploadUrl({
-        path: { graph_id: graphId, table_name: tableName },
+      const uploadUrlResponse = await createFileUpload({
+        path: { graph_id: graphId },
         body: uploadRequest,
       })
 
@@ -120,12 +102,10 @@ export class TableIngestClient {
       let uploadUrl = uploadData.upload_url
       const fileId = uploadData.file_id
 
-      // Fix LocalStack URL if needed
       if (options.fixLocalStackUrl && uploadUrl.includes('localstack:4566')) {
         uploadUrl = uploadUrl.replace('localstack:4566', 'localhost:4566')
       }
 
-      // Step 2: Upload file to S3
       options.onProgress?.(`Uploading ${fileName} to S3...`)
 
       const fileContent = await this.getFileContent(fileOrBuffer)
@@ -151,14 +131,13 @@ export class TableIngestClient {
         }
       }
 
-      // Step 3: Mark file as uploaded (backend validates and calculates size/row count)
       options.onProgress?.(`Marking ${fileName} as uploaded...`)
 
       const statusUpdate: FileStatusUpdate = {
         status: 'uploaded',
       }
 
-      const updateResponse = await updateFileStatus({
+      const updateResponse = await updateFile({
         path: { graph_id: graphId, file_id: fileId },
         body: statusUpdate,
       })
@@ -175,7 +154,6 @@ export class TableIngestClient {
         }
       }
 
-      // Extract size and row count from response (calculated by backend)
       const responseData = updateResponse.data as any
       const actualFileSize = responseData.file_size_bytes || 0
       const actualRowCount = responseData.row_count || 0
@@ -206,135 +184,105 @@ export class TableIngestClient {
   }
 
   /**
-   * List all staging tables in a graph
+   * List files in a graph
    */
-  async listStagingTables(graphId: string): Promise<TableInfo[]> {
+  async list(graphId: string, tableName?: string, status?: string): Promise<FileInfo[]> {
     try {
-      const response = await listTables({
+      const response = await listFiles({
         path: { graph_id: graphId },
+        query: {
+          table_name: tableName,
+          status: status as any,
+        },
       })
 
       if (response.error || !response.data) {
-        console.error('Failed to list tables:', response.error)
+        console.error('Failed to list files:', response.error)
         return []
       }
 
-      const tableData = response.data as TableListResponse
+      const fileData = response.data as any
 
       return (
-        tableData.tables?.map((table) => ({
-          tableName: table.table_name,
-          rowCount: table.row_count,
-          fileCount: table.file_count || 0,
-          totalSizeBytes: table.total_size_bytes || 0,
+        fileData.files?.map((file: any) => ({
+          fileId: file.file_id,
+          fileName: file.file_name,
+          tableName: file.table_name,
+          status: file.status,
+          fileSize: file.file_size_bytes || 0,
+          rowCount: file.row_count || 0,
+          createdAt: file.created_at,
         })) || []
       )
     } catch (error) {
-      console.error('Failed to list tables:', error)
+      console.error('Failed to list files:', error)
       return []
     }
   }
 
   /**
-   * Ingest all staging tables into the graph
+   * Get file information
    */
-  async ingestAllTables(graphId: string, options: IngestOptions = {}): Promise<IngestResult> {
+  async get(graphId: string, fileId: string): Promise<FileInfo | null> {
     try {
-      options.onProgress?.('Starting table ingestion...')
-
-      const ingestRequest: BulkIngestRequest = {
-        ignore_errors: options.ignoreErrors ?? true,
-        rebuild: options.rebuild ?? false,
-      }
-
-      const response = await ingestTables({
-        path: { graph_id: graphId },
-        body: ingestRequest,
+      const response = await getFile({
+        path: { graph_id: graphId, file_id: fileId },
       })
 
       if (response.error || !response.data) {
-        return {
-          success: false,
-          error: `Failed to ingest tables: ${response.error}`,
-        }
+        console.error('Failed to get file:', response.error)
+        return null
       }
 
-      const result = response.data as any
-
-      options.onProgress?.('✅ Table ingestion completed')
+      const file = response.data as any
 
       return {
-        success: true,
-        operationId: result.operation_id,
-        message: result.message || 'Ingestion started',
+        fileId: file.file_id,
+        fileName: file.file_name,
+        tableName: file.table_name,
+        status: file.status,
+        fileSize: file.file_size_bytes || 0,
+        rowCount: file.row_count || 0,
+        createdAt: file.created_at,
       }
     } catch (error) {
-      return {
-        success: false,
-        error: error instanceof Error ? error.message : String(error),
-      }
+      console.error('Failed to get file:', error)
+      return null
     }
   }
 
   /**
-   * Convenience method to upload a file and immediately ingest it
+   * Delete a file
    */
-  async uploadAndIngest(
-    graphId: string,
-    tableName: string,
-    fileOrBuffer: FileInput,
-    uploadOptions: UploadOptions = {},
-    ingestOptions: IngestOptions = {}
-  ): Promise<{ upload: UploadResult; ingest: IngestResult | null }> {
-    // Upload the file
-    const uploadResult = await this.uploadParquetFile(
-      graphId,
-      tableName,
-      fileOrBuffer,
-      uploadOptions
-    )
-
-    if (!uploadResult.success) {
-      return {
-        upload: uploadResult,
-        ingest: null,
-      }
-    }
-
-    // Ingest the table
-    const ingestResult = await this.ingestAllTables(graphId, ingestOptions)
+  async delete(graphId: string, fileId: string, cascade: boolean = false): Promise<boolean> {
+    try {
+      const response = await deleteFile({
+        path: { graph_id: graphId, file_id: fileId },
+        query: { cascade },
+      })
 
-    return {
-      upload: uploadResult,
-      ingest: ingestResult,
+      return !response.error
+    } catch (error) {
+      console.error('Failed to delete file:', error)
+      return false
     }
   }
 
-  /**
-   * Get file name from input or use provided override
-   */
   private getFileName(fileOrBuffer: FileInput, override?: string): string {
     if (override) return override
 
-    // File object (browser)
     if ('name' in fileOrBuffer && typeof fileOrBuffer.name === 'string') {
       return fileOrBuffer.name
     }
 
-    // Default name for buffers/blobs/streams
     return 'data.parquet'
   }
 
-  /**
-   * Convert various file inputs to ArrayBuffer for upload
-   */
   private async getFileContent(fileOrBuffer: FileInput): Promise<ArrayBuffer> {
-    // File or Blob (browser)
     if (fileOrBuffer instanceof Blob || fileOrBuffer instanceof File) {
       return fileOrBuffer.arrayBuffer()
     }
 
-    // Buffer (Node.js)
     if (Buffer.isBuffer(fileOrBuffer)) {
       return fileOrBuffer.buffer.slice(
         fileOrBuffer.byteOffset,
@@ -342,7 +290,6 @@ export class TableIngestClient {
       )
     }
 
-    // ReadableStream
     if ('getReader' in fileOrBuffer) {
       const reader = fileOrBuffer.getReader()
       const chunks: Uint8Array[] = []
@@ -353,7 +300,6 @@ export class TableIngestClient {
         if (value) chunks.push(value)
       }
 
-      // Concatenate chunks
       const totalLength = chunks.reduce((sum, chunk) => sum + chunk.length, 0)
       const result = new Uint8Array(totalLength)
       let offset = 0