@robosystems/client 0.2.20 → 0.2.22

package/extensions/FileClient.d.ts

@@ -1,6 +1,5 @@
 export interface FileUploadOptions {
     onProgress?: (message: string) => void;
-    fixLocalStackUrl?: boolean;
     fileName?: string;
     ingestToGraph?: boolean;
 }
@@ -30,6 +29,7 @@ export declare class FileClient {
         credentials?: 'include' | 'same-origin' | 'omit';
         headers?: Record<string, string>;
         token?: string;
+        s3EndpointUrl?: string;
     });
     /**
      * Upload a Parquet file to staging

package/extensions/FileClient.js

@@ -48,8 +48,14 @@ class FileClient {
             const uploadData = uploadUrlResponse.data;
             let uploadUrl = uploadData.upload_url;
             const fileId = uploadData.file_id;
-            if (options.fixLocalStackUrl && uploadUrl.includes('localstack:4566')) {
-                uploadUrl = uploadUrl.replace('localstack:4566', 'localhost:4566');
+            // 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);

package/extensions/FileClient.ts

@@ -12,7 +12,6 @@ import type { FileStatusUpdate, FileUploadRequest, FileUploadResponse } from '..
 
 export interface FileUploadOptions {
   onProgress?: (message: string) => void
-  fixLocalStackUrl?: boolean
   fileName?: string
   ingestToGraph?: boolean
 }
@@ -45,6 +44,7 @@ export class FileClient {
     credentials?: 'include' | 'same-origin' | 'omit'
     headers?: Record<string, string>
     token?: string
+    s3EndpointUrl?: string // Override S3 endpoint (e.g., for LocalStack)
   }
 
   constructor(config: {
@@ -52,6 +52,7 @@ export class FileClient {
     credentials?: 'include' | 'same-origin' | 'omit'
     headers?: Record<string, string>
     token?: string
+    s3EndpointUrl?: string
   }) {
     this.config = config
   }
@@ -102,8 +103,14 @@ export class FileClient {
       let uploadUrl = uploadData.upload_url
       const fileId = uploadData.file_id
 
-      if (options.fixLocalStackUrl && uploadUrl.includes('localstack:4566')) {
-        uploadUrl = uploadUrl.replace('localstack:4566', 'localhost:4566')
+      // 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...`)

package/extensions/hooks.d.ts

@@ -121,7 +121,6 @@ export declare function useSDKClients(): {
  * const handleFileUpload = async (file: File) => {
  *   const result = await upload('Entity', file, {
  *     onProgress: (msg) => console.log(msg),
- *     fixLocalStackUrl: true,
  *   })
  *
  *   if (result?.success) {
@@ -129,6 +128,15 @@ export declare function useSDKClients(): {
  *   }
  * }
  * ```
+ *
+ * @remarks
+ * For local development with LocalStack, configure s3EndpointUrl when initializing RoboSystemsExtensions:
+ * ```tsx
+ * const extensions = new RoboSystemsExtensions({
+ *   baseUrl: 'http://localhost:8000',
+ *   s3EndpointUrl: 'http://localhost:4566', // LocalStack endpoint
+ * })
+ * ```
  */
 export declare function useTableUpload(graphId: string): {
     upload: (tableName: string, fileOrBuffer: FileInput, options?: FileUploadOptions) => Promise<FileUploadResult | null>;

package/extensions/hooks.js

@@ -396,7 +396,6 @@ function useSDKClients() {
  * const handleFileUpload = async (file: File) => {
  *   const result = await upload('Entity', file, {
  *     onProgress: (msg) => console.log(msg),
- *     fixLocalStackUrl: true,
  *   })
  *
  *   if (result?.success) {
@@ -404,6 +403,15 @@ function useSDKClients() {
  *   }
  * }
  * ```
+ *
+ * @remarks
+ * For local development with LocalStack, configure s3EndpointUrl when initializing RoboSystemsExtensions:
+ * ```tsx
+ * const extensions = new RoboSystemsExtensions({
+ *   baseUrl: 'http://localhost:8000',
+ *   s3EndpointUrl: 'http://localhost:4566', // LocalStack endpoint
+ * })
+ * ```
  */
 function useTableUpload(graphId) {
     const [loading, setLoading] = (0, react_1.useState)(false);

package/extensions/hooks.ts

@@ -474,7 +474,6 @@ export function useSDKClients() {
  * const handleFileUpload = async (file: File) => {
  *   const result = await upload('Entity', file, {
  *     onProgress: (msg) => console.log(msg),
- *     fixLocalStackUrl: true,
  *   })
  *
  *   if (result?.success) {
@@ -482,6 +481,15 @@ export function useSDKClients() {
  *   }
  * }
  * ```
+ *
+ * @remarks
+ * For local development with LocalStack, configure s3EndpointUrl when initializing RoboSystemsExtensions:
+ * ```tsx
+ * const extensions = new RoboSystemsExtensions({
+ *   baseUrl: 'http://localhost:8000',
+ *   s3EndpointUrl: 'http://localhost:4566', // LocalStack endpoint
+ * })
+ * ```
  */
 export function useTableUpload(graphId: string) {
   const [loading, setLoading] = useState(false)

package/extensions/index.d.ts

@@ -17,6 +17,7 @@ export interface RoboSystemsExtensionConfig {
     headers?: Record<string, string>;
     maxRetries?: number;
     retryDelay?: number;
+    s3EndpointUrl?: string;
 }
 export declare class RoboSystemsExtensions {
     readonly query: QueryClient;

package/extensions/index.js

@@ -49,6 +49,7 @@ class RoboSystemsExtensions {
             headers: config.headers,
             maxRetries: config.maxRetries || 5,
             retryDelay: config.retryDelay || 1000,
+            s3EndpointUrl: config.s3EndpointUrl,
         };
         this.query = new QueryClient_1.QueryClient({
             baseUrl: this.config.baseUrl,
@@ -74,6 +75,7 @@ class RoboSystemsExtensions {
             credentials: this.config.credentials,
             token: this.config.token,
             headers: this.config.headers,
+            s3EndpointUrl: this.config.s3EndpointUrl,
         });
         this.materialization = new MaterializationClient_1.MaterializationClient({
             baseUrl: this.config.baseUrl,

package/extensions/index.ts

@@ -21,6 +21,7 @@ export interface RoboSystemsExtensionConfig {
   headers?: Record<string, string>
   maxRetries?: number
   retryDelay?: number
+  s3EndpointUrl?: string // Override S3 endpoint (e.g., for LocalStack)
 }
 
 // Properly typed configuration interface
@@ -31,6 +32,7 @@ interface ResolvedConfig {
   headers?: Record<string, string>
   maxRetries: number
   retryDelay: number
+  s3EndpointUrl?: string
 }
 
 export class RoboSystemsExtensions {
@@ -57,6 +59,7 @@ export class RoboSystemsExtensions {
       headers: config.headers,
       maxRetries: config.maxRetries || 5,
       retryDelay: config.retryDelay || 1000,
+      s3EndpointUrl: config.s3EndpointUrl,
     }
 
     this.query = new QueryClient({
@@ -86,6 +89,7 @@ export class RoboSystemsExtensions {
       credentials: this.config.credentials,
       token: this.config.token,
       headers: this.config.headers,
+      s3EndpointUrl: this.config.s3EndpointUrl,
     })
 
     this.materialization = new MaterializationClient({

package/package.json

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

package/sdk-extensions/FileClient.d.ts

@@ -1,6 +1,5 @@
 export interface FileUploadOptions {
     onProgress?: (message: string) => void;
-    fixLocalStackUrl?: boolean;
     fileName?: string;
     ingestToGraph?: boolean;
 }
@@ -30,6 +29,7 @@ export declare class FileClient {
         credentials?: 'include' | 'same-origin' | 'omit';
         headers?: Record<string, string>;
         token?: string;
+        s3EndpointUrl?: string;
     });
     /**
      * Upload a Parquet file to staging

package/sdk-extensions/FileClient.js

@@ -48,8 +48,14 @@ class FileClient {
             const uploadData = uploadUrlResponse.data;
             let uploadUrl = uploadData.upload_url;
             const fileId = uploadData.file_id;
-            if (options.fixLocalStackUrl && uploadUrl.includes('localstack:4566')) {
-                uploadUrl = uploadUrl.replace('localstack:4566', 'localhost:4566');
+            // 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);

package/sdk-extensions/FileClient.ts

@@ -12,7 +12,6 @@ import type { FileStatusUpdate, FileUploadRequest, FileUploadResponse } from '..
 
 export interface FileUploadOptions {
   onProgress?: (message: string) => void
-  fixLocalStackUrl?: boolean
   fileName?: string
   ingestToGraph?: boolean
 }
@@ -45,6 +44,7 @@ export class FileClient {
     credentials?: 'include' | 'same-origin' | 'omit'
     headers?: Record<string, string>
     token?: string
+    s3EndpointUrl?: string // Override S3 endpoint (e.g., for LocalStack)
   }
 
   constructor(config: {
@@ -52,6 +52,7 @@ export class FileClient {
     credentials?: 'include' | 'same-origin' | 'omit'
     headers?: Record<string, string>
     token?: string
+    s3EndpointUrl?: string
   }) {
     this.config = config
   }
@@ -102,8 +103,14 @@ export class FileClient {
       let uploadUrl = uploadData.upload_url
       const fileId = uploadData.file_id
 
-      if (options.fixLocalStackUrl && uploadUrl.includes('localstack:4566')) {
-        uploadUrl = uploadUrl.replace('localstack:4566', 'localhost:4566')
+      // 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...`)

package/sdk-extensions/README.md

@@ -114,7 +114,6 @@ const result = await tableClient.uploadParquetFile(
     onProgress: (message) => {
       console.log(`Upload progress: ${message}`)
     },
-    fixLocalStackUrl: true, // Auto-fix for local development
   }
 )
 
@@ -248,7 +247,6 @@ import { readFileSync } from 'fs'
 const fileBuffer = readFileSync('data/entities.parquet')
 const result = await tableClient.uploadParquetFile('graph-id', 'Entity', Buffer.from(fileBuffer), {
   onProgress: (msg) => console.log(msg),
-  fixLocalStackUrl: true, // Auto-fix for local development
 })
 
 if (result.success) {
@@ -673,14 +671,15 @@ NEXT_PUBLIC_PREFER_STREAMING=true
 ### Custom Configuration
 
 ```typescript
-import { createSSEClient } from '@robosystems/client/extensions'
+import { RoboSystemsExtensions } from '@robosystems/client/extensions'
 
-const sseClient = createSSEClient({
+const extensions = new RoboSystemsExtensions({
   // API Configuration
   baseUrl: process.env.NEXT_PUBLIC_ROBOSYSTEMS_API_URL,
 
   // Authentication
   credentials: 'include', // For cookies
+  token: 'your-jwt-token', // Or use JWT token
   headers: {
     'X-API-Key': process.env.ROBOSYSTEMS_API_KEY,
   },
@@ -688,12 +687,29 @@ const sseClient = createSSEClient({
   // Connection Settings
   maxRetries: 5,
   retryDelay: 1000,
-  heartbeatInterval: 30000,
 
-  // Advanced Options
-  eventSourceOptions: {
-    withCredentials: true,
-  },
+  // S3 Configuration (for local development with LocalStack)
+  s3EndpointUrl: process.env.S3_ENDPOINT_URL, // e.g., 'http://localhost:4566'
+})
+```
+
+#### Local Development with LocalStack
+
+For local development using LocalStack, configure the S3 endpoint URL:
+
+```typescript
+import { RoboSystemsExtensions } from '@robosystems/client/extensions'
+
+const extensions = new RoboSystemsExtensions({
+  baseUrl: 'http://localhost:8000',
+  credentials: 'include',
+  // Override S3 endpoint for LocalStack
+  s3EndpointUrl: 'http://localhost:4566',
+})
+
+// File uploads will now use LocalStack instead of AWS S3
+const result = await extensions.files.upload('graph-id', 'TableName', fileBuffer, {
+  onProgress: (msg) => console.log(msg),
 })
 ```
 

package/sdk-extensions/hooks.d.ts

@@ -121,7 +121,6 @@ export declare function useSDKClients(): {
  * const handleFileUpload = async (file: File) => {
  *   const result = await upload('Entity', file, {
  *     onProgress: (msg) => console.log(msg),
- *     fixLocalStackUrl: true,
  *   })
  *
  *   if (result?.success) {
@@ -129,6 +128,15 @@ export declare function useSDKClients(): {
  *   }
  * }
  * ```
+ *
+ * @remarks
+ * For local development with LocalStack, configure s3EndpointUrl when initializing RoboSystemsExtensions:
+ * ```tsx
+ * const extensions = new RoboSystemsExtensions({
+ *   baseUrl: 'http://localhost:8000',
+ *   s3EndpointUrl: 'http://localhost:4566', // LocalStack endpoint
+ * })
+ * ```
  */
 export declare function useTableUpload(graphId: string): {
     upload: (tableName: string, fileOrBuffer: FileInput, options?: FileUploadOptions) => Promise<FileUploadResult | null>;

package/sdk-extensions/hooks.js

@@ -396,7 +396,6 @@ function useSDKClients() {
  * const handleFileUpload = async (file: File) => {
  *   const result = await upload('Entity', file, {
  *     onProgress: (msg) => console.log(msg),
- *     fixLocalStackUrl: true,
  *   })
  *
  *   if (result?.success) {
@@ -404,6 +403,15 @@ function useSDKClients() {
  *   }
  * }
  * ```
+ *
+ * @remarks
+ * For local development with LocalStack, configure s3EndpointUrl when initializing RoboSystemsExtensions:
+ * ```tsx
+ * const extensions = new RoboSystemsExtensions({
+ *   baseUrl: 'http://localhost:8000',
+ *   s3EndpointUrl: 'http://localhost:4566', // LocalStack endpoint
+ * })
+ * ```
  */
 function useTableUpload(graphId) {
     const [loading, setLoading] = (0, react_1.useState)(false);

package/sdk-extensions/hooks.ts

@@ -474,7 +474,6 @@ export function useSDKClients() {
  * const handleFileUpload = async (file: File) => {
  *   const result = await upload('Entity', file, {
  *     onProgress: (msg) => console.log(msg),
- *     fixLocalStackUrl: true,
  *   })
  *
  *   if (result?.success) {
@@ -482,6 +481,15 @@ export function useSDKClients() {
  *   }
  * }
  * ```
+ *
+ * @remarks
+ * For local development with LocalStack, configure s3EndpointUrl when initializing RoboSystemsExtensions:
+ * ```tsx
+ * const extensions = new RoboSystemsExtensions({
+ *   baseUrl: 'http://localhost:8000',
+ *   s3EndpointUrl: 'http://localhost:4566', // LocalStack endpoint
+ * })
+ * ```
  */
 export function useTableUpload(graphId: string) {
   const [loading, setLoading] = useState(false)

package/sdk-extensions/index.d.ts

@@ -17,6 +17,7 @@ export interface RoboSystemsExtensionConfig {
     headers?: Record<string, string>;
     maxRetries?: number;
     retryDelay?: number;
+    s3EndpointUrl?: string;
 }
 export declare class RoboSystemsExtensions {
     readonly query: QueryClient;

package/sdk-extensions/index.js

@@ -49,6 +49,7 @@ class RoboSystemsExtensions {
             headers: config.headers,
             maxRetries: config.maxRetries || 5,
             retryDelay: config.retryDelay || 1000,
+            s3EndpointUrl: config.s3EndpointUrl,
         };
         this.query = new QueryClient_1.QueryClient({
             baseUrl: this.config.baseUrl,
@@ -74,6 +75,7 @@ class RoboSystemsExtensions {
             credentials: this.config.credentials,
             token: this.config.token,
             headers: this.config.headers,
+            s3EndpointUrl: this.config.s3EndpointUrl,
         });
         this.materialization = new MaterializationClient_1.MaterializationClient({
             baseUrl: this.config.baseUrl,

package/sdk-extensions/index.ts

@@ -21,6 +21,7 @@ export interface RoboSystemsExtensionConfig {
   headers?: Record<string, string>
   maxRetries?: number
   retryDelay?: number
+  s3EndpointUrl?: string // Override S3 endpoint (e.g., for LocalStack)
 }
 
 // Properly typed configuration interface
@@ -31,6 +32,7 @@ interface ResolvedConfig {
   headers?: Record<string, string>
   maxRetries: number
   retryDelay: number
+  s3EndpointUrl?: string
 }
 
 export class RoboSystemsExtensions {
@@ -57,6 +59,7 @@ export class RoboSystemsExtensions {
       headers: config.headers,
       maxRetries: config.maxRetries || 5,
       retryDelay: config.retryDelay || 1000,
+      s3EndpointUrl: config.s3EndpointUrl,
     }
 
     this.query = new QueryClient({
@@ -86,6 +89,7 @@ export class RoboSystemsExtensions {
       credentials: this.config.credentials,
       token: this.config.token,
       headers: this.config.headers,
+      s3EndpointUrl: this.config.s3EndpointUrl,
     })
 
     this.materialization = new MaterializationClient({