@robosystems/client 0.2.5 → 0.2.7

package/extensions/QueryClient.d.ts

@@ -36,6 +36,7 @@ export declare class QueryClient {
         token?: string;
     });
     executeQuery(graphId: string, request: QueryRequest, options?: QueryOptions): Promise<QueryResult | AsyncIterableIterator<any>>;
+    private parseNDJSONResponse;
     private streamQueryResults;
     private waitForQueryCompletion;
     query(graphId: string, cypher: string, parameters?: Record<string, any>): Promise<QueryResult>;

package/extensions/QueryClient.js

@@ -27,6 +27,15 @@ class QueryClient {
         };
         // Execute the query
         const response = await (0, sdk_gen_1.executeCypherQuery)(data);
+        // Check if this is an NDJSON streaming response
+        // The SDK returns the response object which includes the raw Response
+        if (response.response) {
+            const contentType = response.response.headers.get('content-type') || '';
+            const streamFormat = response.response.headers.get('x-stream-format');
+            if (contentType.includes('application/x-ndjson') || streamFormat === 'ndjson') {
+                return this.parseNDJSONResponse(response.response, graphId);
+            }
+        }
         const responseData = response.data;
         // Check if this is an immediate response
         if (responseData?.data !== undefined && responseData?.columns) {
@@ -59,6 +68,51 @@ class QueryClient {
         // Unexpected response format
         throw new Error('Unexpected response format from query endpoint');
     }
+    async parseNDJSONResponse(response, graphId) {
+        const allData = [];
+        let columns = null;
+        let totalRows = 0;
+        let executionTimeMs = 0;
+        // Parse NDJSON line by line
+        const text = await response.text();
+        const lines = text.trim().split('\n');
+        for (const line of lines) {
+            if (!line.trim())
+                continue;
+            try {
+                const chunk = JSON.parse(line);
+                // Extract columns from first chunk
+                if (columns === null && chunk.columns) {
+                    columns = chunk.columns;
+                }
+                // Aggregate data rows (NDJSON uses "rows", regular JSON uses "data")
+                if (chunk.rows) {
+                    allData.push(...chunk.rows);
+                    totalRows += chunk.rows.length;
+                }
+                else if (chunk.data) {
+                    allData.push(...chunk.data);
+                    totalRows += chunk.data.length;
+                }
+                // Track execution time (use max from all chunks)
+                if (chunk.execution_time_ms) {
+                    executionTimeMs = Math.max(executionTimeMs, chunk.execution_time_ms);
+                }
+            }
+            catch (error) {
+                throw new Error(`Failed to parse NDJSON line: ${error}`);
+            }
+        }
+        // Return aggregated result
+        return {
+            data: allData,
+            columns: columns || [],
+            row_count: totalRows,
+            execution_time_ms: executionTimeMs,
+            graph_id: graphId,
+            timestamp: new Date().toISOString(),
+        };
+    }
     async *streamQueryResults(operationId, options) {
         const buffer = [];
         let completed = false;

package/extensions/QueryClient.ts

@@ -79,6 +79,18 @@ export class QueryClient {
 
     // Execute the query
     const response = await executeCypherQuery(data)
+
+    // Check if this is an NDJSON streaming response
+    // The SDK returns the response object which includes the raw Response
+    if (response.response) {
+      const contentType = response.response.headers.get('content-type') || ''
+      const streamFormat = response.response.headers.get('x-stream-format')
+
+      if (contentType.includes('application/x-ndjson') || streamFormat === 'ndjson') {
+        return this.parseNDJSONResponse(response.response, graphId)
+      }
+    }
+
     const responseData = response.data as any
 
     // Check if this is an immediate response
@@ -117,6 +129,56 @@ export class QueryClient {
     throw new Error('Unexpected response format from query endpoint')
   }
 
+  private async parseNDJSONResponse(response: Response, graphId: string): Promise<QueryResult> {
+    const allData: any[] = []
+    let columns: string[] | null = null
+    let totalRows = 0
+    let executionTimeMs = 0
+
+    // Parse NDJSON line by line
+    const text = await response.text()
+    const lines = text.trim().split('\n')
+
+    for (const line of lines) {
+      if (!line.trim()) continue
+
+      try {
+        const chunk = JSON.parse(line)
+
+        // Extract columns from first chunk
+        if (columns === null && chunk.columns) {
+          columns = chunk.columns
+        }
+
+        // Aggregate data rows (NDJSON uses "rows", regular JSON uses "data")
+        if (chunk.rows) {
+          allData.push(...chunk.rows)
+          totalRows += chunk.rows.length
+        } else if (chunk.data) {
+          allData.push(...chunk.data)
+          totalRows += chunk.data.length
+        }
+
+        // Track execution time (use max from all chunks)
+        if (chunk.execution_time_ms) {
+          executionTimeMs = Math.max(executionTimeMs, chunk.execution_time_ms)
+        }
+      } catch (error) {
+        throw new Error(`Failed to parse NDJSON line: ${error}`)
+      }
+    }
+
+    // Return aggregated result
+    return {
+      data: allData,
+      columns: columns || [],
+      row_count: totalRows,
+      execution_time_ms: executionTimeMs,
+      graph_id: graphId,
+      timestamp: new Date().toISOString(),
+    }
+  }
+
   private async *streamQueryResults(
     operationId: string,
     options: QueryOptions

package/package.json

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