@robosystems/client 0.2.1 → 0.2.3

package/extensions/GraphClient.d.ts

@@ -21,6 +21,7 @@ export interface GraphInfo {
     status?: string;
 }
 export interface CreateGraphOptions {
+    createEntity?: boolean;
     timeout?: number;
     pollInterval?: number;
     onProgress?: (message: string) => void;
@@ -36,6 +37,18 @@ export declare class GraphClient {
     });
     /**
      * Create a graph and wait for completion
+     *
+     * @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)
+     *   - onProgress: Callback for progress updates
+     * @returns The graph ID when creation completes
      */
     createGraphAndWait(metadata: GraphMetadataInput, initialEntity?: InitialEntityInput, options?: CreateGraphOptions): Promise<string>;
     /**
@@ -50,7 +63,7 @@ export declare class GraphClient {
      * Delete a graph
      * Note: This will be implemented when the deleteGraph endpoint is available in the SDK
      */
-    deleteGraph(graphId: string): Promise<void>;
+    deleteGraph(_graphId: string): Promise<void>;
     /**
      * Clean up resources
      */

package/extensions/GraphClient.js

@@ -15,9 +15,21 @@ class GraphClient {
     }
     /**
      * Create a graph and wait for completion
+     *
+     * @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)
+     *   - onProgress: Callback for progress updates
+     * @returns The graph ID when creation completes
      */
     async createGraphAndWait(metadata, initialEntity, options = {}) {
-        const { timeout = 60000, pollInterval = 2000, onProgress } = options;
+        const { createEntity = true, timeout = 60000, pollInterval = 2000, onProgress } = options;
         if (!this.config.token) {
             throw new Error('No API key provided. Set token in config.');
         }
@@ -47,6 +59,7 @@ class GraphClient {
             body: {
                 metadata: apiMetadata,
                 initial_entity: initialEntityData || null,
+                create_entity: createEntity,
             },
         });
         // Check if we got immediate graph_id
@@ -145,10 +158,10 @@ class GraphClient {
      * Delete a graph
      * Note: This will be implemented when the deleteGraph endpoint is available in the SDK
      */
-    async deleteGraph(graphId) {
+    async deleteGraph(_graphId) {
         throw new Error('deleteGraph is not yet implemented - waiting for SDK endpoint to be generated');
         // TODO: Implement when deleteGraph endpoint is available
-        // const response = await deleteGraph({ path: { graph_id: graphId } })
+        // const response = await deleteGraph({ path: { graph_id: _graphId } })
         // if (response.status !== 200 && response.status !== 204) {
         //   throw new Error(`Failed to delete graph: ${response.status}`)
         // }

package/extensions/GraphClient.ts

@@ -66,6 +66,7 @@ export interface GraphInfo {
 }
 
 export interface CreateGraphOptions {
+  createEntity?: boolean
   timeout?: number
   pollInterval?: number
   onProgress?: (message: string) => void
@@ -92,13 +93,25 @@ export class GraphClient {
 
   /**
    * Create a graph and wait for completion
+   *
+   * @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)
+   *   - onProgress: Callback for progress updates
+   * @returns The graph ID when creation completes
    */
   async createGraphAndWait(
     metadata: GraphMetadataInput,
     initialEntity?: InitialEntityInput,
     options: CreateGraphOptions = {}
   ): Promise<string> {
-    const { timeout = 60000, pollInterval = 2000, onProgress } = options
+    const { createEntity = true, timeout = 60000, pollInterval = 2000, onProgress } = options
 
     if (!this.config.token) {
       throw new Error('No API key provided. Set token in config.')
@@ -133,6 +146,7 @@ export class GraphClient {
       body: {
         metadata: apiMetadata,
         initial_entity: initialEntityData || null,
+        create_entity: createEntity,
       },
     })
 
@@ -249,10 +263,10 @@ export class GraphClient {
    * Delete a graph
    * Note: This will be implemented when the deleteGraph endpoint is available in the SDK
    */
-  async deleteGraph(graphId: string): Promise<void> {
+  async deleteGraph(_graphId: string): Promise<void> {
     throw new Error('deleteGraph is not yet implemented - waiting for SDK endpoint to be generated')
     // TODO: Implement when deleteGraph endpoint is available
-    // const response = await deleteGraph({ path: { graph_id: graphId } })
+    // const response = await deleteGraph({ path: { graph_id: _graphId } })
     // if (response.status !== 200 && response.status !== 204) {
     //   throw new Error(`Failed to delete graph: ${response.status}`)
     // }

package/extensions/QueryClient.js

@@ -23,7 +23,6 @@ class QueryClient {
             query: {
                 mode: options.mode,
                 test_mode: options.testMode,
-                token: this.config.token, // Pass JWT token for SSE authentication
             },
         };
         // Execute the query

package/extensions/QueryClient.ts

@@ -74,7 +74,6 @@ export class QueryClient {
       query: {
         mode: options.mode,
         test_mode: options.testMode,
-        token: this.config.token, // Pass JWT token for SSE authentication
       },
     }
 

package/extensions/TableIngestClient.d.ts

@@ -44,7 +44,7 @@ export declare class TableIngestClient {
      * This method handles the complete 3-step upload process:
      * 1. Get presigned upload URL
      * 2. Upload file to S3
-     * 3. Update file metadata
+     * 3. Mark file as 'uploaded' (backend validates, calculates size/row count)
      *
      * Supports File (browser), Blob (browser), Buffer (Node.js), and ReadableStream.
      */

package/extensions/TableIngestClient.js

@@ -19,7 +19,7 @@ class TableIngestClient {
      * This method handles the complete 3-step upload process:
      * 1. Get presigned upload URL
      * 2. Upload file to S3
-     * 3. Update file metadata
+     * 3. Mark file as 'uploaded' (backend validates, calculates size/row count)
      *
      * Supports File (browser), Blob (browser), Buffer (Node.js), and ReadableStream.
      */
@@ -32,10 +32,9 @@ class TableIngestClient {
                 file_name: fileName,
                 content_type: 'application/x-parquet',
             };
-            const uploadUrlResponse = await (0, sdk_gen_1.getUploadUrlV1GraphsGraphIdTablesTableNameFilesPost)({
+            const uploadUrlResponse = await (0, sdk_gen_1.getUploadUrl)({
                 path: { graph_id: graphId, table_name: tableName },
                 body: uploadRequest,
-                query: this.config.token ? { token: this.config.token } : undefined,
             });
             if (uploadUrlResponse.error || !uploadUrlResponse.data) {
                 return {
@@ -77,36 +76,35 @@ class TableIngestClient {
                     error: `S3 upload failed: ${s3Response.status} ${s3Response.statusText}`,
                 };
             }
-            // Step 3: Update file metadata
-            options.onProgress?.(`Updating file metadata for ${fileName}...`);
-            // Estimate row count (TypeScript can't read Parquet metadata without library)
-            // Rough estimate: ~100 bytes per row for typical data
-            const estimatedRowCount = Math.floor(fileSize / 100);
-            const metadataUpdate = {
-                file_size_bytes: fileSize,
-                row_count: estimatedRowCount,
+            // 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.updateFileV1GraphsGraphIdTablesFilesFileIdPatch)({
+            const updateResponse = await (0, sdk_gen_1.updateFileStatus)({
                 path: { graph_id: graphId, file_id: fileId },
-                body: metadataUpdate,
-                query: this.config.token ? { token: this.config.token } : undefined,
+                body: statusUpdate,
             });
-            if (updateResponse.error) {
+            if (updateResponse.error || !updateResponse.data) {
                 return {
                     fileId,
                     fileSize,
-                    rowCount: estimatedRowCount,
+                    rowCount: 0,
                     tableName,
                     fileName,
                     success: false,
-                    error: 'Failed to update file metadata',
+                    error: 'Failed to complete file upload',
                 };
             }
-            options.onProgress?.(`✅ Uploaded ${fileName} (${fileSize.toLocaleString()} bytes, ~${estimatedRowCount.toLocaleString()} rows)`);
+            // 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;
+            options.onProgress?.(`✅ Uploaded ${fileName} (${actualFileSize.toLocaleString()} bytes, ${actualRowCount.toLocaleString()} rows)`);
             return {
                 fileId,
-                fileSize,
-                rowCount: estimatedRowCount,
+                fileSize: actualFileSize,
+                rowCount: actualRowCount,
                 tableName,
                 fileName,
                 success: true,
@@ -129,9 +127,8 @@ class TableIngestClient {
      */
     async listStagingTables(graphId) {
         try {
-            const response = await (0, sdk_gen_1.listTablesV1GraphsGraphIdTablesGet)({
+            const response = await (0, sdk_gen_1.listTables)({
                 path: { graph_id: graphId },
-                query: this.config.token ? { token: this.config.token } : undefined,
             });
             if (response.error || !response.data) {
                 console.error('Failed to list tables:', response.error);
@@ -160,10 +157,9 @@ class TableIngestClient {
                 ignore_errors: options.ignoreErrors ?? true,
                 rebuild: options.rebuild ?? false,
             };
-            const response = await (0, sdk_gen_1.ingestTablesV1GraphsGraphIdTablesIngestPost)({
+            const response = await (0, sdk_gen_1.ingestTables)({
                 path: { graph_id: graphId },
                 body: ingestRequest,
-                query: this.config.token ? { token: this.config.token } : undefined,
             });
             if (response.error || !response.data) {
                 return {

package/extensions/TableIngestClient.test.ts

@@ -153,7 +153,7 @@ describe('TableIngestClient', () => {
       const result = await tableClient.uploadParquetFile('graph_123', 'Entity', buffer)
 
       expect(result.success).toBe(false)
-      expect(result.error).toContain('Failed to update file metadata')
+      expect(result.error).toContain('Failed to complete file upload')
     })
   })
 

package/extensions/TableIngestClient.ts

@@ -7,15 +7,10 @@
  * Supports File (browser), Blob (browser), Buffer (Node.js), and ReadableStream.
  */
 
-import {
-  getUploadUrlV1GraphsGraphIdTablesTableNameFilesPost,
-  ingestTablesV1GraphsGraphIdTablesIngestPost,
-  listTablesV1GraphsGraphIdTablesGet,
-  updateFileV1GraphsGraphIdTablesFilesFileIdPatch,
-} from '../sdk.gen'
+import { getUploadUrl, ingestTables, listTables, updateFileStatus } from '../sdk.gen'
 import type {
   BulkIngestRequest,
-  FileUpdateRequest,
+  FileStatusUpdate,
   FileUploadRequest,
   FileUploadResponse,
   TableListResponse,
@@ -83,7 +78,7 @@ export class TableIngestClient {
    * This method handles the complete 3-step upload process:
    * 1. Get presigned upload URL
    * 2. Upload file to S3
-   * 3. Update file metadata
+   * 3. Mark file as 'uploaded' (backend validates, calculates size/row count)
    *
    * Supports File (browser), Blob (browser), Buffer (Node.js), and ReadableStream.
    */
@@ -104,10 +99,9 @@ export class TableIngestClient {
         content_type: 'application/x-parquet',
       }
 
-      const uploadUrlResponse = await getUploadUrlV1GraphsGraphIdTablesTableNameFilesPost({
+      const uploadUrlResponse = await getUploadUrl({
         path: { graph_id: graphId, table_name: tableName },
         body: uploadRequest,
-        query: this.config.token ? { token: this.config.token } : undefined,
       })
 
       if (uploadUrlResponse.error || !uploadUrlResponse.data) {
@@ -157,44 +151,43 @@ export class TableIngestClient {
         }
       }
 
-      // Step 3: Update file metadata
-      options.onProgress?.(`Updating file metadata for ${fileName}...`)
+      // Step 3: Mark file as uploaded (backend validates and calculates size/row count)
+      options.onProgress?.(`Marking ${fileName} as uploaded...`)
 
-      // Estimate row count (TypeScript can't read Parquet metadata without library)
-      // Rough estimate: ~100 bytes per row for typical data
-      const estimatedRowCount = Math.floor(fileSize / 100)
-
-      const metadataUpdate: FileUpdateRequest = {
-        file_size_bytes: fileSize,
-        row_count: estimatedRowCount,
+      const statusUpdate: FileStatusUpdate = {
+        status: 'uploaded',
       }
 
-      const updateResponse = await updateFileV1GraphsGraphIdTablesFilesFileIdPatch({
+      const updateResponse = await updateFileStatus({
         path: { graph_id: graphId, file_id: fileId },
-        body: metadataUpdate,
-        query: this.config.token ? { token: this.config.token } : undefined,
+        body: statusUpdate,
       })
 
-      if (updateResponse.error) {
+      if (updateResponse.error || !updateResponse.data) {
         return {
           fileId,
           fileSize,
-          rowCount: estimatedRowCount,
+          rowCount: 0,
           tableName,
           fileName,
           success: false,
-          error: 'Failed to update file metadata',
+          error: 'Failed to complete file upload',
         }
       }
 
+      // 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
+
       options.onProgress?.(
-        `✅ Uploaded ${fileName} (${fileSize.toLocaleString()} bytes, ~${estimatedRowCount.toLocaleString()} rows)`
+        `✅ Uploaded ${fileName} (${actualFileSize.toLocaleString()} bytes, ${actualRowCount.toLocaleString()} rows)`
       )
 
       return {
         fileId,
-        fileSize,
-        rowCount: estimatedRowCount,
+        fileSize: actualFileSize,
+        rowCount: actualRowCount,
         tableName,
         fileName,
         success: true,
@@ -217,9 +210,8 @@ export class TableIngestClient {
    */
   async listStagingTables(graphId: string): Promise<TableInfo[]> {
     try {
-      const response = await listTablesV1GraphsGraphIdTablesGet({
+      const response = await listTables({
         path: { graph_id: graphId },
-        query: this.config.token ? { token: this.config.token } : undefined,
       })
 
       if (response.error || !response.data) {
@@ -255,10 +247,9 @@ export class TableIngestClient {
         rebuild: options.rebuild ?? false,
       }
 
-      const response = await ingestTablesV1GraphsGraphIdTablesIngestPost({
+      const response = await ingestTables({
         path: { graph_id: graphId },
         body: ingestRequest,
-        query: this.config.token ? { token: this.config.token } : undefined,
       })
 
       if (response.error || !response.data) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@robosystems/client",
-  "version": "0.2.1",
+  "version": "0.2.3",
   "description": "TypeScript client library for RoboSystems Financial Knowledge Graph API",
   "main": "index.js",
   "types": "index.d.ts",