@robosystems/client 0.2.14 → 0.2.16

package/sdk-extensions/hooks.js

@@ -14,9 +14,11 @@ exports.useTableUpload = useTableUpload;
 const react_1 = require("react");
 const client_gen_1 = require("../sdk/client.gen");
 const config_1 = require("./config");
+const FileClient_1 = require("./FileClient");
+const MaterializationClient_1 = require("./MaterializationClient");
 const OperationClient_1 = require("./OperationClient");
 const QueryClient_1 = require("./QueryClient");
-const TableIngestClient_1 = require("./TableIngestClient");
+const TableClient_1 = require("./TableClient");
 /**
  * Hook for executing Cypher queries with loading states and error handling
  *
@@ -385,11 +387,11 @@ function useSDKClients() {
     return clients;
 }
 /**
- * Hook for uploading Parquet files to staging tables
+ * Hook for uploading Parquet files to staging and materializing graphs
  *
  * @example
  * ```tsx
- * const { upload, uploadAndIngest, loading, error, progress } = useTableUpload('graph_123')
+ * const { upload, listTables, materialize, loading, error, progress } = useTableUpload('graph_123')
  *
  * const handleFileUpload = async (file: File) => {
  *   const result = await upload('Entity', file, {
@@ -408,28 +410,32 @@ function useTableUpload(graphId) {
     const [error, setError] = (0, react_1.useState)(null);
     const [progress, setProgress] = (0, react_1.useState)(null);
     const [uploadResult, setUploadResult] = (0, react_1.useState)(null);
-    const clientRef = (0, react_1.useRef)(null);
+    const fileClientRef = (0, react_1.useRef)(null);
+    const materializationClientRef = (0, react_1.useRef)(null);
+    const tableClientRef = (0, react_1.useRef)(null);
     (0, react_1.useEffect)(() => {
         const sdkConfig = (0, config_1.getSDKExtensionsConfig)();
         const clientConfig = client_gen_1.client.getConfig();
-        // Extract JWT token (uses centralized logic)
         const token = (0, config_1.extractTokenFromSDKClient)();
-        clientRef.current = new TableIngestClient_1.TableIngestClient({
+        const baseConfig = {
             baseUrl: sdkConfig.baseUrl || clientConfig.baseUrl || 'http://localhost:8000',
             credentials: sdkConfig.credentials,
             token,
             headers: sdkConfig.headers,
-        });
+        };
+        fileClientRef.current = new FileClient_1.FileClient(baseConfig);
+        materializationClientRef.current = new MaterializationClient_1.MaterializationClient(baseConfig);
+        tableClientRef.current = new TableClient_1.TableClient(baseConfig);
     }, []);
     const upload = (0, react_1.useCallback)(async (tableName, fileOrBuffer, options) => {
-        if (!clientRef.current)
+        if (!fileClientRef.current)
             return null;
         setLoading(true);
         setError(null);
         setUploadResult(null);
         setProgress(null);
         try {
-            const result = await clientRef.current.uploadParquetFile(graphId, tableName, fileOrBuffer, {
+            const result = await fileClientRef.current.upload(graphId, tableName, fileOrBuffer, {
                 ...options,
                 onProgress: (msg) => {
                     setProgress(msg);
@@ -452,65 +458,25 @@ function useTableUpload(graphId) {
             setProgress(null);
         }
     }, [graphId]);
-    const uploadAndIngest = (0, react_1.useCallback)(async (tableName, fileOrBuffer, uploadOptions, ingestOptions) => {
-        if (!clientRef.current)
-            return null;
-        setLoading(true);
-        setError(null);
-        setUploadResult(null);
-        setProgress(null);
-        try {
-            const result = await clientRef.current.uploadAndIngest(graphId, tableName, fileOrBuffer, {
-                ...uploadOptions,
-                onProgress: (msg) => {
-                    setProgress(msg);
-                    uploadOptions?.onProgress?.(msg);
-                },
-            }, {
-                ...ingestOptions,
-                onProgress: (msg) => {
-                    setProgress(msg);
-                    ingestOptions?.onProgress?.(msg);
-                },
-            });
-            setUploadResult(result.upload);
-            if (!result.upload.success && result.upload.error) {
-                setError(new Error(result.upload.error));
-            }
-            else if (result.ingest && !result.ingest.success && result.ingest.error) {
-                setError(new Error(result.ingest.error));
-            }
-            return result;
-        }
-        catch (err) {
-            const error = err;
-            setError(error);
-            return null;
-        }
-        finally {
-            setLoading(false);
-            setProgress(null);
-        }
-    }, [graphId]);
     const listTables = (0, react_1.useCallback)(async () => {
-        if (!clientRef.current)
+        if (!tableClientRef.current)
             return [];
         try {
-            return await clientRef.current.listStagingTables(graphId);
+            return await tableClientRef.current.list(graphId);
         }
         catch (err) {
             setError(err);
             return [];
         }
     }, [graphId]);
-    const ingestAllTables = (0, react_1.useCallback)(async (options) => {
-        if (!clientRef.current)
+    const materialize = (0, react_1.useCallback)(async (options) => {
+        if (!materializationClientRef.current)
             return null;
         setLoading(true);
         setError(null);
         setProgress(null);
         try {
-            const result = await clientRef.current.ingestAllTables(graphId, {
+            const result = await materializationClientRef.current.materialize(graphId, {
                 ...options,
                 onProgress: (msg) => {
                     setProgress(msg);
@@ -534,9 +500,8 @@ function useTableUpload(graphId) {
     }, [graphId]);
     return {
         upload,
-        uploadAndIngest,
         listTables,
-        ingestAllTables,
+        materialize,
         loading,
         error,
         progress,

package/sdk-extensions/hooks.ts

@@ -8,12 +8,16 @@
 import { useCallback, useEffect, useRef, useState } from 'react'
 import { client } from '../sdk/client.gen'
 import { extractTokenFromSDKClient, getSDKExtensionsConfig } from './config'
+import type { FileInput, FileUploadOptions, FileUploadResult } from './FileClient'
+import { FileClient } from './FileClient'
+import type { MaterializationOptions } from './MaterializationClient'
+import { MaterializationClient } from './MaterializationClient'
 import type { OperationProgress, OperationResult } from './OperationClient'
 import { OperationClient } from './OperationClient'
 import type { QueryOptions, QueryResult } from './QueryClient'
 import { QueryClient } from './QueryClient'
-import type { FileInput, IngestOptions, UploadOptions, UploadResult } from './TableIngestClient'
-import { TableIngestClient } from './TableIngestClient'
+import type { TableInfo } from './TableClient'
+import { TableClient } from './TableClient'
 
 /**
  * Hook for executing Cypher queries with loading states and error handling
@@ -461,11 +465,11 @@ export function useSDKClients() {
 }
 
 /**
- * Hook for uploading Parquet files to staging tables
+ * Hook for uploading Parquet files to staging and materializing graphs
  *
  * @example
  * ```tsx
- * const { upload, uploadAndIngest, loading, error, progress } = useTableUpload('graph_123')
+ * const { upload, listTables, materialize, loading, error, progress } = useTableUpload('graph_123')
  *
  * const handleFileUpload = async (file: File) => {
  *   const result = await upload('Entity', file, {
@@ -483,31 +487,35 @@ export function useTableUpload(graphId: string) {
   const [loading, setLoading] = useState(false)
   const [error, setError] = useState<Error | null>(null)
   const [progress, setProgress] = useState<string | null>(null)
-  const [uploadResult, setUploadResult] = useState<UploadResult | null>(null)
-  const clientRef = useRef<TableIngestClient>(null)
+  const [uploadResult, setUploadResult] = useState<FileUploadResult | null>(null)
+  const fileClientRef = useRef<FileClient | null>(null)
+  const materializationClientRef = useRef<MaterializationClient | null>(null)
+  const tableClientRef = useRef<TableClient | null>(null)
 
   useEffect(() => {
     const sdkConfig = getSDKExtensionsConfig()
     const clientConfig = client.getConfig()
-
-    // Extract JWT token (uses centralized logic)
     const token = extractTokenFromSDKClient()
 
-    clientRef.current = new TableIngestClient({
+    const baseConfig = {
       baseUrl: sdkConfig.baseUrl || clientConfig.baseUrl || 'http://localhost:8000',
       credentials: sdkConfig.credentials,
       token,
       headers: sdkConfig.headers,
-    })
+    }
+
+    fileClientRef.current = new FileClient(baseConfig)
+    materializationClientRef.current = new MaterializationClient(baseConfig)
+    tableClientRef.current = new TableClient(baseConfig)
   }, [])
 
   const upload = useCallback(
     async (
       tableName: string,
       fileOrBuffer: FileInput,
-      options?: UploadOptions
-    ): Promise<UploadResult | null> => {
-      if (!clientRef.current) return null
+      options?: FileUploadOptions
+    ): Promise<FileUploadResult | null> => {
+      if (!fileClientRef.current) return null
 
       setLoading(true)
       setError(null)
@@ -515,7 +523,7 @@ export function useTableUpload(graphId: string) {
       setProgress(null)
 
       try {
-        const result = await clientRef.current.uploadParquetFile(graphId, tableName, fileOrBuffer, {
+        const result = await fileClientRef.current.upload(graphId, tableName, fileOrBuffer, {
           ...options,
           onProgress: (msg) => {
             setProgress(msg)
@@ -542,83 +550,27 @@ export function useTableUpload(graphId: string) {
     [graphId]
   )
 
-  const uploadAndIngest = useCallback(
-    async (
-      tableName: string,
-      fileOrBuffer: FileInput,
-      uploadOptions?: UploadOptions,
-      ingestOptions?: IngestOptions
-    ): Promise<{ upload: UploadResult; ingest: any } | null> => {
-      if (!clientRef.current) return null
-
-      setLoading(true)
-      setError(null)
-      setUploadResult(null)
-      setProgress(null)
-
-      try {
-        const result = await clientRef.current.uploadAndIngest(
-          graphId,
-          tableName,
-          fileOrBuffer,
-          {
-            ...uploadOptions,
-            onProgress: (msg) => {
-              setProgress(msg)
-              uploadOptions?.onProgress?.(msg)
-            },
-          },
-          {
-            ...ingestOptions,
-            onProgress: (msg) => {
-              setProgress(msg)
-              ingestOptions?.onProgress?.(msg)
-            },
-          }
-        )
-
-        setUploadResult(result.upload)
-
-        if (!result.upload.success && result.upload.error) {
-          setError(new Error(result.upload.error))
-        } else if (result.ingest && !result.ingest.success && result.ingest.error) {
-          setError(new Error(result.ingest.error))
-        }
-
-        return result
-      } catch (err) {
-        const error = err as Error
-        setError(error)
-        return null
-      } finally {
-        setLoading(false)
-        setProgress(null)
-      }
-    },
-    [graphId]
-  )
-
-  const listTables = useCallback(async () => {
-    if (!clientRef.current) return []
+  const listTables = useCallback(async (): Promise<TableInfo[]> => {
+    if (!tableClientRef.current) return []
 
     try {
-      return await clientRef.current.listStagingTables(graphId)
+      return await tableClientRef.current.list(graphId)
     } catch (err) {
       setError(err as Error)
       return []
     }
   }, [graphId])
 
-  const ingestAllTables = useCallback(
-    async (options?: IngestOptions) => {
-      if (!clientRef.current) return null
+  const materialize = useCallback(
+    async (options?: MaterializationOptions) => {
+      if (!materializationClientRef.current) return null
 
       setLoading(true)
       setError(null)
       setProgress(null)
 
       try {
-        const result = await clientRef.current.ingestAllTables(graphId, {
+        const result = await materializationClientRef.current.materialize(graphId, {
           ...options,
           onProgress: (msg) => {
             setProgress(msg)
@@ -645,9 +597,8 @@ export function useTableUpload(graphId: string) {
 
   return {
     upload,
-    uploadAndIngest,
     listTables,
-    ingestAllTables,
+    materialize,
     loading,
     error,
     progress,

package/sdk-extensions/index.d.ts

@@ -6,7 +6,9 @@ import { OperationClient } from './OperationClient';
 import { QueryClient } from './QueryClient';
 import { AgentClient } from './AgentClient';
 import { SSEClient } from './SSEClient';
-import { TableIngestClient } from './TableIngestClient';
+import { FileClient } from './FileClient';
+import { MaterializationClient } from './MaterializationClient';
+import { TableClient } from './TableClient';
 import { GraphClient } from './GraphClient';
 export interface RoboSystemsExtensionConfig {
     baseUrl?: string;
@@ -20,7 +22,9 @@ export declare class RoboSystemsExtensions {
     readonly query: QueryClient;
     readonly agent: AgentClient;
     readonly operations: OperationClient;
-    readonly tables: TableIngestClient;
+    readonly files: FileClient;
+    readonly materialization: MaterializationClient;
+    readonly tables: TableClient;
     readonly graphs: GraphClient;
     private config;
     constructor(config?: RoboSystemsExtensionConfig);
@@ -41,15 +45,21 @@ export * from './OperationClient';
 export * from './QueryClient';
 export * from './AgentClient';
 export * from './SSEClient';
-export * from './TableIngestClient';
+export * from './FileClient';
+export * from './MaterializationClient';
 export * from './GraphClient';
 export * from './config';
-export { OperationClient, QueryClient, AgentClient, SSEClient, TableIngestClient, GraphClient };
+export type { TableInfo, TableQueryResult } from './TableClient';
+export { TableClient } from './TableClient';
+export { OperationClient, QueryClient, AgentClient, SSEClient, FileClient, MaterializationClient, GraphClient, };
 export { useMultipleOperations, useOperation, useQuery, useSDKClients, useStreamingQuery, useTableUpload, } from './hooks';
 export declare const extensions: {
     readonly query: QueryClient;
     readonly agent: AgentClient;
     readonly operations: OperationClient;
+    readonly files: FileClient;
+    readonly materialization: MaterializationClient;
+    readonly tables: TableClient;
     readonly graphs: GraphClient;
     monitorOperation: (operationId: string, onProgress?: (progress: any) => void) => Promise<any>;
     createSSEClient: () => SSEClient;

package/sdk-extensions/index.js

@@ -18,7 +18,7 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.analyzeFinancials = exports.agentQuery = exports.streamQuery = exports.executeQuery = exports.monitorOperation = exports.extensions = exports.useTableUpload = exports.useStreamingQuery = exports.useSDKClients = exports.useQuery = exports.useOperation = exports.useMultipleOperations = exports.GraphClient = exports.TableIngestClient = exports.SSEClient = exports.AgentClient = exports.QueryClient = exports.OperationClient = exports.RoboSystemsExtensions = void 0;
+exports.analyzeFinancials = exports.agentQuery = exports.streamQuery = exports.executeQuery = exports.monitorOperation = exports.extensions = exports.useTableUpload = exports.useStreamingQuery = exports.useSDKClients = exports.useQuery = exports.useOperation = exports.useMultipleOperations = exports.GraphClient = exports.MaterializationClient = exports.FileClient = exports.SSEClient = exports.AgentClient = exports.QueryClient = exports.OperationClient = exports.TableClient = exports.RoboSystemsExtensions = void 0;
 const client_gen_1 = require("../sdk/client.gen");
 const config_1 = require("./config");
 const OperationClient_1 = require("./OperationClient");
@@ -29,8 +29,11 @@ const AgentClient_1 = require("./AgentClient");
 Object.defineProperty(exports, "AgentClient", { enumerable: true, get: function () { return AgentClient_1.AgentClient; } });
 const SSEClient_1 = require("./SSEClient");
 Object.defineProperty(exports, "SSEClient", { enumerable: true, get: function () { return SSEClient_1.SSEClient; } });
-const TableIngestClient_1 = require("./TableIngestClient");
-Object.defineProperty(exports, "TableIngestClient", { enumerable: true, get: function () { return TableIngestClient_1.TableIngestClient; } });
+const FileClient_1 = require("./FileClient");
+Object.defineProperty(exports, "FileClient", { enumerable: true, get: function () { return FileClient_1.FileClient; } });
+const MaterializationClient_1 = require("./MaterializationClient");
+Object.defineProperty(exports, "MaterializationClient", { enumerable: true, get: function () { return MaterializationClient_1.MaterializationClient; } });
+const TableClient_1 = require("./TableClient");
 const GraphClient_1 = require("./GraphClient");
 Object.defineProperty(exports, "GraphClient", { enumerable: true, get: function () { return GraphClient_1.GraphClient; } });
 class RoboSystemsExtensions {
@@ -66,7 +69,19 @@ class RoboSystemsExtensions {
             maxRetries: this.config.maxRetries,
             retryDelay: this.config.retryDelay,
         });
-        this.tables = new TableIngestClient_1.TableIngestClient({
+        this.files = new FileClient_1.FileClient({
+            baseUrl: this.config.baseUrl,
+            credentials: this.config.credentials,
+            token: this.config.token,
+            headers: this.config.headers,
+        });
+        this.materialization = new MaterializationClient_1.MaterializationClient({
+            baseUrl: this.config.baseUrl,
+            credentials: this.config.credentials,
+            token: this.config.token,
+            headers: this.config.headers,
+        });
+        this.tables = new TableClient_1.TableClient({
             baseUrl: this.config.baseUrl,
             credentials: this.config.credentials,
             token: this.config.token,
@@ -114,9 +129,12 @@ __exportStar(require("./OperationClient"), exports);
 __exportStar(require("./QueryClient"), exports);
 __exportStar(require("./AgentClient"), exports);
 __exportStar(require("./SSEClient"), exports);
-__exportStar(require("./TableIngestClient"), exports);
+__exportStar(require("./FileClient"), exports);
+__exportStar(require("./MaterializationClient"), exports);
 __exportStar(require("./GraphClient"), exports);
 __exportStar(require("./config"), exports);
+var TableClient_2 = require("./TableClient");
+Object.defineProperty(exports, "TableClient", { enumerable: true, get: function () { return TableClient_2.TableClient; } });
 // Export React hooks
 var hooks_1 = require("./hooks");
 Object.defineProperty(exports, "useMultipleOperations", { enumerable: true, get: function () { return hooks_1.useMultipleOperations; } });
@@ -143,6 +161,15 @@ exports.extensions = {
     get operations() {
         return getExtensions().operations;
     },
+    get files() {
+        return getExtensions().files;
+    },
+    get materialization() {
+        return getExtensions().materialization;
+    },
+    get tables() {
+        return getExtensions().tables;
+    },
     get graphs() {
         return getExtensions().graphs;
     },

package/sdk-extensions/index.test.ts

@@ -23,12 +23,16 @@ describe('RoboSystemsExtensions', () => {
       expect(extensions).toBeDefined()
       expect(extensions.query).toBeDefined()
       expect(extensions.operations).toBeDefined()
+      expect(extensions.files).toBeDefined()
+      expect(extensions.materialization).toBeDefined()
       expect(extensions.tables).toBeDefined()
       expect(extensions.graphs).toBeDefined()
       // Verify clients have expected methods
       expect(typeof extensions.query.executeQuery).toBe('function')
       expect(typeof extensions.operations.monitorOperation).toBe('function')
-      expect(typeof extensions.tables.uploadParquetFile).toBe('function')
+      expect(typeof extensions.files.upload).toBe('function')
+      expect(typeof extensions.materialization.materialize).toBe('function')
+      expect(typeof extensions.tables.list).toBe('function')
       expect(typeof extensions.graphs.createGraphAndWait).toBe('function')
     })
 
@@ -193,11 +197,15 @@ describe('RoboSystemsExtensions', () => {
 
       expect(extensions.query).toBeDefined()
       expect(extensions.operations).toBeDefined()
+      expect(extensions.files).toBeDefined()
+      expect(extensions.materialization).toBeDefined()
       expect(extensions.tables).toBeDefined()
       expect(extensions.graphs).toBeDefined()
       expect(typeof extensions.query.executeQuery).toBe('function')
       expect(typeof extensions.operations.monitorOperation).toBe('function')
-      expect(typeof extensions.tables.uploadParquetFile).toBe('function')
+      expect(typeof extensions.files.upload).toBe('function')
+      expect(typeof extensions.materialization.materialize).toBe('function')
+      expect(typeof extensions.tables.list).toBe('function')
       expect(typeof extensions.graphs.createGraphAndWait).toBe('function')
     })
 

package/sdk-extensions/index.ts

@@ -9,7 +9,9 @@ import { OperationClient } from './OperationClient'
 import { QueryClient } from './QueryClient'
 import { AgentClient } from './AgentClient'
 import { SSEClient } from './SSEClient'
-import { TableIngestClient } from './TableIngestClient'
+import { FileClient } from './FileClient'
+import { MaterializationClient } from './MaterializationClient'
+import { TableClient } from './TableClient'
 import { GraphClient } from './GraphClient'
 
 export interface RoboSystemsExtensionConfig {
@@ -35,7 +37,9 @@ export class RoboSystemsExtensions {
   public readonly query: QueryClient
   public readonly agent: AgentClient
   public readonly operations: OperationClient
-  public readonly tables: TableIngestClient
+  public readonly files: FileClient
+  public readonly materialization: MaterializationClient
+  public readonly tables: TableClient
   public readonly graphs: GraphClient
   private config: ResolvedConfig
 
@@ -77,7 +81,21 @@ export class RoboSystemsExtensions {
       retryDelay: this.config.retryDelay,
     })
 
-    this.tables = new TableIngestClient({
+    this.files = new FileClient({
+      baseUrl: this.config.baseUrl,
+      credentials: this.config.credentials,
+      token: this.config.token,
+      headers: this.config.headers,
+    })
+
+    this.materialization = new MaterializationClient({
+      baseUrl: this.config.baseUrl,
+      credentials: this.config.credentials,
+      token: this.config.token,
+      headers: this.config.headers,
+    })
+
+    this.tables = new TableClient({
       baseUrl: this.config.baseUrl,
       credentials: this.config.credentials,
       token: this.config.token,
@@ -129,10 +147,24 @@ export * from './OperationClient'
 export * from './QueryClient'
 export * from './AgentClient'
 export * from './SSEClient'
-export * from './TableIngestClient'
+export * from './FileClient'
+export * from './MaterializationClient'
 export * from './GraphClient'
 export * from './config'
-export { OperationClient, QueryClient, AgentClient, SSEClient, TableIngestClient, GraphClient }
+
+// Export TableClient types individually to avoid conflicts with QueryClient's QueryResult
+export type { TableInfo, TableQueryResult } from './TableClient'
+export { TableClient } from './TableClient'
+
+export {
+  OperationClient,
+  QueryClient,
+  AgentClient,
+  SSEClient,
+  FileClient,
+  MaterializationClient,
+  GraphClient,
+}
 
 // Export React hooks
 export {
@@ -164,6 +196,15 @@ export const extensions = {
   get operations() {
     return getExtensions().operations
   },
+  get files() {
+    return getExtensions().files
+  },
+  get materialization() {
+    return getExtensions().materialization
+  },
+  get tables() {
+    return getExtensions().tables
+  },
   get graphs() {
     return getExtensions().graphs
   },