@robosystems/client 0.2.43 → 0.2.45

package/extensions/FileClient.js

@@ -1,246 +0,0 @@
-'use client';
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.FileClient = void 0;
-/**
- * File Client for RoboSystems API
- *
- * Manages file upload operations including presigned URLs, S3 uploads,
- * and file status tracking.
- */
-const sdk_gen_1 = require("../sdk.gen");
-class FileClient {
-    constructor(config) {
-        this.config = config;
-    }
-    /**
-     * 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)
-     */
-    async upload(graphId, tableName, fileOrBuffer, options = {}) {
-        const fileName = this.getFileName(fileOrBuffer, options.fileName);
-        try {
-            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.createFileUpload)({
-                path: { graph_id: graphId },
-                body: uploadRequest,
-            });
-            if (uploadUrlResponse.error || !uploadUrlResponse.data) {
-                return {
-                    fileId: '',
-                    fileSize: 0,
-                    rowCount: 0,
-                    tableName,
-                    fileName,
-                    success: false,
-                    error: `Failed to get upload URL: ${uploadUrlResponse.error}`,
-                };
-            }
-            const uploadData = uploadUrlResponse.data;
-            let uploadUrl = uploadData.upload_url;
-            const fileId = uploadData.file_id;
-            // Override S3 endpoint if configured (e.g., for LocalStack)
-            if (this.config.s3EndpointUrl) {
-                const originalUrl = new URL(uploadUrl);
-                const overrideUrl = new URL(this.config.s3EndpointUrl);
-                // Replace scheme, host, and port with the override endpoint
-                originalUrl.protocol = overrideUrl.protocol;
-                originalUrl.host = overrideUrl.host;
-                uploadUrl = originalUrl.toString();
-            }
-            options.onProgress?.(`Uploading ${fileName} to S3...`);
-            const fileContent = await this.getFileContent(fileOrBuffer);
-            const fileSize = fileContent.byteLength;
-            const s3Response = await fetch(uploadUrl, {
-                method: 'PUT',
-                body: fileContent,
-                headers: {
-                    'Content-Type': 'application/x-parquet',
-                },
-            });
-            if (!s3Response.ok) {
-                return {
-                    fileId,
-                    fileSize,
-                    rowCount: 0,
-                    tableName,
-                    fileName,
-                    success: false,
-                    error: `S3 upload failed: ${s3Response.status} ${s3Response.statusText}`,
-                };
-            }
-            options.onProgress?.(`Marking ${fileName} as uploaded...`);
-            const statusUpdate = {
-                status: 'uploaded',
-            };
-            const updateResponse = await (0, sdk_gen_1.updateFile)({
-                path: { graph_id: graphId, file_id: fileId },
-                body: statusUpdate,
-            });
-            if (updateResponse.error || !updateResponse.data) {
-                return {
-                    fileId,
-                    fileSize,
-                    rowCount: 0,
-                    tableName,
-                    fileName,
-                    success: false,
-                    error: 'Failed to complete file upload',
-                };
-            }
-            const responseData = updateResponse.data;
-            const actualFileSize = responseData.file_size_bytes || 0;
-            const actualRowCount = responseData.row_count || 0;
-            options.onProgress?.(`✅ Uploaded ${fileName} (${actualFileSize.toLocaleString()} bytes, ${actualRowCount.toLocaleString()} rows)`);
-            return {
-                fileId,
-                fileSize: actualFileSize,
-                rowCount: actualRowCount,
-                tableName,
-                fileName,
-                success: true,
-            };
-        }
-        catch (error) {
-            return {
-                fileId: '',
-                fileSize: 0,
-                rowCount: 0,
-                tableName,
-                fileName,
-                success: false,
-                error: error instanceof Error ? error.message : String(error),
-            };
-        }
-    }
-    /**
-     * List files in a graph
-     */
-    async list(graphId, tableName, status) {
-        try {
-            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 files:', response.error);
-                return [];
-            }
-            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 files:', error);
-            return [];
-        }
-    }
-    /**
-     * Get file information
-     */
-    async get(graphId, fileId) {
-        try {
-            const response = await (0, sdk_gen_1.getFile)({
-                path: { graph_id: graphId, file_id: fileId },
-            });
-            if (response.error || !response.data) {
-                console.error('Failed to get file:', response.error);
-                return null;
-            }
-            const file = response.data;
-            return {
-                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 get file:', error);
-            return null;
-        }
-    }
-    /**
-     * Delete a file
-     */
-    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;
-        }
-    }
-    getFileName(fileOrBuffer, override) {
-        if (override)
-            return override;
-        if ('name' in fileOrBuffer && typeof fileOrBuffer.name === 'string') {
-            return fileOrBuffer.name;
-        }
-        return 'data.parquet';
-    }
-    async getFileContent(fileOrBuffer) {
-        if (fileOrBuffer instanceof Blob || fileOrBuffer instanceof File) {
-            return fileOrBuffer.arrayBuffer();
-        }
-        if (Buffer.isBuffer(fileOrBuffer)) {
-            const buffer = fileOrBuffer.buffer.slice(fileOrBuffer.byteOffset, fileOrBuffer.byteOffset + fileOrBuffer.byteLength);
-            // Convert SharedArrayBuffer to ArrayBuffer if needed
-            if (buffer instanceof ArrayBuffer) {
-                return buffer;
-            }
-            // Handle SharedArrayBuffer by copying to ArrayBuffer
-            const arrayBuffer = new ArrayBuffer(buffer.byteLength);
-            new Uint8Array(arrayBuffer).set(new Uint8Array(buffer));
-            return arrayBuffer;
-        }
-        if ('getReader' in fileOrBuffer) {
-            const reader = fileOrBuffer.getReader();
-            const chunks = [];
-            while (true) {
-                const { done, value } = await reader.read();
-                if (done)
-                    break;
-                if (value)
-                    chunks.push(value);
-            }
-            const totalLength = chunks.reduce((sum, chunk) => sum + chunk.length, 0);
-            const result = new Uint8Array(totalLength);
-            let offset = 0;
-            for (const chunk of chunks) {
-                result.set(chunk, offset);
-                offset += chunk.length;
-            }
-            return result.buffer;
-        }
-        throw new Error('Unsupported file input type');
-    }
-}
-exports.FileClient = FileClient;

package/extensions/FileClient.ts

@@ -1,330 +0,0 @@
-'use client'
-
-/**
- * File Client for RoboSystems API
- *
- * Manages file upload operations including presigned URLs, S3 uploads,
- * and file status tracking.
- */
-
-import { createFileUpload, deleteFile, getFile, listFiles, updateFile } from '../sdk.gen'
-import type { FileStatusUpdate, FileUploadRequest, FileUploadResponse } from '../types.gen'
-
-export interface FileUploadOptions {
-  onProgress?: (message: string) => void
-  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 class FileClient {
-  private config: {
-    baseUrl: string
-    credentials?: 'include' | 'same-origin' | 'omit'
-    headers?: Record<string, string>
-    token?: string
-    s3EndpointUrl?: string // Override S3 endpoint (e.g., for LocalStack)
-  }
-
-  constructor(config: {
-    baseUrl: string
-    credentials?: 'include' | 'same-origin' | 'omit'
-    headers?: Record<string, string>
-    token?: string
-    s3EndpointUrl?: string
-  }) {
-    this.config = config
-  }
-
-  /**
-   * 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)
-   */
-  async upload(
-    graphId: string,
-    tableName: string,
-    fileOrBuffer: FileInput,
-    options: FileUploadOptions = {}
-  ): Promise<FileUploadResult> {
-    const fileName = this.getFileName(fileOrBuffer, options.fileName)
-
-    try {
-      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 createFileUpload({
-        path: { graph_id: graphId },
-        body: uploadRequest,
-      })
-
-      if (uploadUrlResponse.error || !uploadUrlResponse.data) {
-        return {
-          fileId: '',
-          fileSize: 0,
-          rowCount: 0,
-          tableName,
-          fileName,
-          success: false,
-          error: `Failed to get upload URL: ${uploadUrlResponse.error}`,
-        }
-      }
-
-      const uploadData = uploadUrlResponse.data as FileUploadResponse
-      let uploadUrl = uploadData.upload_url
-      const fileId = uploadData.file_id
-
-      // Override S3 endpoint if configured (e.g., for LocalStack)
-      if (this.config.s3EndpointUrl) {
-        const originalUrl = new URL(uploadUrl)
-        const overrideUrl = new URL(this.config.s3EndpointUrl)
-        // Replace scheme, host, and port with the override endpoint
-        originalUrl.protocol = overrideUrl.protocol
-        originalUrl.host = overrideUrl.host
-        uploadUrl = originalUrl.toString()
-      }
-
-      options.onProgress?.(`Uploading ${fileName} to S3...`)
-
-      const fileContent = await this.getFileContent(fileOrBuffer)
-      const fileSize = fileContent.byteLength
-
-      const s3Response = await fetch(uploadUrl, {
-        method: 'PUT',
-        body: fileContent,
-        headers: {
-          'Content-Type': 'application/x-parquet',
-        },
-      })
-
-      if (!s3Response.ok) {
-        return {
-          fileId,
-          fileSize,
-          rowCount: 0,
-          tableName,
-          fileName,
-          success: false,
-          error: `S3 upload failed: ${s3Response.status} ${s3Response.statusText}`,
-        }
-      }
-
-      options.onProgress?.(`Marking ${fileName} as uploaded...`)
-
-      const statusUpdate: FileStatusUpdate = {
-        status: 'uploaded',
-      }
-
-      const updateResponse = await updateFile({
-        path: { graph_id: graphId, file_id: fileId },
-        body: statusUpdate,
-      })
-
-      if (updateResponse.error || !updateResponse.data) {
-        return {
-          fileId,
-          fileSize,
-          rowCount: 0,
-          tableName,
-          fileName,
-          success: false,
-          error: 'Failed to complete file upload',
-        }
-      }
-
-      const responseData = updateResponse.data as any
-      const actualFileSize = responseData.file_size_bytes || 0
-      const actualRowCount = responseData.row_count || 0
-
-      options.onProgress?.(
-        `✅ Uploaded ${fileName} (${actualFileSize.toLocaleString()} bytes, ${actualRowCount.toLocaleString()} rows)`
-      )
-
-      return {
-        fileId,
-        fileSize: actualFileSize,
-        rowCount: actualRowCount,
-        tableName,
-        fileName,
-        success: true,
-      }
-    } catch (error) {
-      return {
-        fileId: '',
-        fileSize: 0,
-        rowCount: 0,
-        tableName,
-        fileName,
-        success: false,
-        error: error instanceof Error ? error.message : String(error),
-      }
-    }
-  }
-
-  /**
-   * List files in a graph
-   */
-  async list(graphId: string, tableName?: string, status?: string): Promise<FileInfo[]> {
-    try {
-      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 files:', response.error)
-        return []
-      }
-
-      const fileData = response.data as any
-
-      return (
-        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 files:', error)
-      return []
-    }
-  }
-
-  /**
-   * Get file information
-   */
-  async get(graphId: string, fileId: string): Promise<FileInfo | null> {
-    try {
-      const response = await getFile({
-        path: { graph_id: graphId, file_id: fileId },
-      })
-
-      if (response.error || !response.data) {
-        console.error('Failed to get file:', response.error)
-        return null
-      }
-
-      const file = response.data as any
-
-      return {
-        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 get file:', error)
-      return null
-    }
-  }
-
-  /**
-   * Delete a file
-   */
-  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 !response.error
-    } catch (error) {
-      console.error('Failed to delete file:', error)
-      return false
-    }
-  }
-
-  private getFileName(fileOrBuffer: FileInput, override?: string): string {
-    if (override) return override
-
-    if ('name' in fileOrBuffer && typeof fileOrBuffer.name === 'string') {
-      return fileOrBuffer.name
-    }
-
-    return 'data.parquet'
-  }
-
-  private async getFileContent(fileOrBuffer: FileInput): Promise<ArrayBuffer> {
-    if (fileOrBuffer instanceof Blob || fileOrBuffer instanceof File) {
-      return fileOrBuffer.arrayBuffer()
-    }
-
-    if (Buffer.isBuffer(fileOrBuffer)) {
-      const buffer = fileOrBuffer.buffer.slice(
-        fileOrBuffer.byteOffset,
-        fileOrBuffer.byteOffset + fileOrBuffer.byteLength
-      )
-      // Convert SharedArrayBuffer to ArrayBuffer if needed
-      if (buffer instanceof ArrayBuffer) {
-        return buffer
-      }
-      // Handle SharedArrayBuffer by copying to ArrayBuffer
-      const arrayBuffer = new ArrayBuffer(buffer.byteLength)
-      new Uint8Array(arrayBuffer).set(new Uint8Array(buffer))
-      return arrayBuffer
-    }
-
-    if ('getReader' in fileOrBuffer) {
-      const reader = fileOrBuffer.getReader()
-      const chunks: Uint8Array[] = []
-
-      while (true) {
-        const { done, value } = await reader.read()
-        if (done) break
-        if (value) chunks.push(value)
-      }
-
-      const totalLength = chunks.reduce((sum, chunk) => sum + chunk.length, 0)
-      const result = new Uint8Array(totalLength)
-      let offset = 0
-      for (const chunk of chunks) {
-        result.set(chunk, offset)
-        offset += chunk.length
-      }
-      return result.buffer
-    }
-
-    throw new Error('Unsupported file input type')
-  }
-}

package/extensions/GraphClient.d.ts

@@ -1,91 +0,0 @@
-/**
- * Error thrown when a graph operation fails (as reported by the server)
- * This is distinct from connection/SSE errors
- */
-export declare class GraphOperationError extends Error {
-    constructor(message: string);
-}
-export interface GraphMetadataInput {
-    graphName: string;
-    description?: string;
-    schemaExtensions?: string[];
-    tags?: string[];
-}
-export interface InitialEntityInput {
-    name: string;
-    uri: string;
-    category?: string;
-    sic?: string;
-    sicDescription?: string;
-}
-export interface GraphInfo {
-    graphId: string;
-    graphName: string;
-    description?: string;
-    schemaExtensions?: string[];
-    tags?: string[];
-    createdAt?: string;
-    status?: string;
-}
-export interface CreateGraphOptions {
-    createEntity?: boolean;
-    timeout?: number;
-    pollInterval?: number;
-    onProgress?: (message: string) => void;
-    useSSE?: boolean;
-}
-export declare class GraphClient {
-    private operationClient;
-    private config;
-    constructor(config: {
-        baseUrl: string;
-        credentials?: 'include' | 'same-origin' | 'omit';
-        headers?: Record<string, string>;
-        token?: string;
-    });
-    /**
-     * Create a graph and wait for completion
-     *
-     * Uses SSE (Server-Sent Events) for real-time progress updates with
-     * automatic fallback to polling if SSE connection fails.
-     *
-     * @param metadata - Graph metadata (name, description, etc.)
-     * @param initialEntity - Optional initial entity to create
-     * @param options - Additional options including:
-     *   - createEntity: Whether to create the entity node and upload initial data.
-     *                   Only applies when initialEntity is provided. Set to false to
-     *                   create graph without populating entity data (useful for file-based ingestion).
-     *                   Defaults to true.
-     *   - timeout: Maximum time to wait in milliseconds (default: 60000)
-     *   - pollInterval: Time between status checks in milliseconds (default: 2000, for polling fallback)
-     *   - onProgress: Callback for progress updates
-     *   - useSSE: Whether to try SSE first (default: true). Falls back to polling on failure.
-     * @returns The graph ID when creation completes
-     */
-    createGraphAndWait(metadata: GraphMetadataInput, initialEntity?: InitialEntityInput, options?: CreateGraphOptions): Promise<string>;
-    /**
-     * Wait for operation completion using SSE stream
-     */
-    private waitWithSSE;
-    /**
-     * Wait for operation completion using polling
-     */
-    private waitWithPolling;
-    /**
-     * Get information about a graph
-     */
-    getGraphInfo(graphId: string): Promise<GraphInfo>;
-    /**
-     * List all graphs
-     */
-    listGraphs(): Promise<GraphInfo[]>;
-    /**
-     * Delete a graph
-     * Note: This will be implemented when the deleteGraph endpoint is available in the SDK
-     */
-    deleteGraph(_graphId: string): Promise<void>;
-    /**
-     * Clean up resources
-     */
-    close(): void;
-}