@robosystems/client 0.2.11 → 0.2.13

package/sdk-extensions/QueryClient.ts

@@ -75,19 +75,31 @@ export class QueryClient {
         mode: options.mode,
         test_mode: options.testMode,
       },
+      // For streaming mode, don't parse response - get raw Response object
+      ...(options.mode === 'stream' ? { parseAs: 'stream' as const } : {}),
     }
 
-    // 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) {
+    // Check if this is a raw stream response (when parseAs: 'stream')
+    if (options.mode === 'stream' && 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') {
+      if (contentType.includes('application/x-ndjson')) {
         return this.parseNDJSONResponse(response.response, graphId)
+      } else if (contentType.includes('application/json')) {
+        // Fallback: parse JSON manually
+        const data = await response.response.json()
+        if (data.data !== undefined && data.columns) {
+          return {
+            data: data.data,
+            columns: data.columns,
+            row_count: data.row_count || data.data.length,
+            execution_time_ms: data.execution_time_ms || 0,
+            graph_id: graphId,
+            timestamp: data.timestamp || new Date().toISOString(),
+          }
+        }
       }
     }
 
@@ -118,11 +130,7 @@ export class QueryClient {
       }
 
       // Use SSE to monitor the operation
-      if (options.mode === 'stream') {
-        return this.streamQueryResults(queuedResponse.operation_id, options)
-      } else {
-        return this.waitForQueryCompletion(queuedResponse.operation_id, options)
-      }
+      return this.waitForQueryCompletion(queuedResponse.operation_id, options)
     }
 
     // Unexpected response format
@@ -135,37 +143,77 @@ export class QueryClient {
     let totalRows = 0
     let executionTimeMs = 0
 
-    // Parse NDJSON line by line
-    const text = await response.text()
-    const lines = text.trim().split('\n')
+    // Use streaming reader to avoid "body already read" error
+    const reader = response.body?.getReader()
+    if (!reader) {
+      throw new Error('Response body is not readable')
+    }
+
+    const decoder = new TextDecoder()
+    let buffer = ''
 
-    for (const line of lines) {
-      if (!line.trim()) continue
+    try {
+      while (true) {
+        const { done, value } = await reader.read()
+        if (done) break
 
-      try {
-        const chunk = JSON.parse(line)
+        buffer += decoder.decode(value, { stream: true })
+        const lines = buffer.split('\n')
+        buffer = lines.pop() || ''
 
-        // Extract columns from first chunk
-        if (columns === null && chunk.columns) {
-          columns = chunk.columns
-        }
+        for (const line of lines) {
+          if (!line.trim()) continue
+
+          try {
+            const chunk = JSON.parse(line)
 
-        // 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
+            // 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) {
+            console.error('Failed to parse NDJSON line:', error)
+          }
         }
+      }
 
-        // Track execution time (use max from all chunks)
-        if (chunk.execution_time_ms) {
-          executionTimeMs = Math.max(executionTimeMs, chunk.execution_time_ms)
+      // Parse any remaining buffer
+      if (buffer.trim()) {
+        try {
+          const chunk = JSON.parse(buffer)
+          if (columns === null && chunk.columns) {
+            columns = chunk.columns
+          }
+          if (chunk.rows) {
+            allData.push(...chunk.rows)
+            totalRows += chunk.rows.length
+          } else if (chunk.data) {
+            allData.push(...chunk.data)
+            totalRows += chunk.data.length
+          }
+          if (chunk.execution_time_ms) {
+            executionTimeMs = Math.max(executionTimeMs, chunk.execution_time_ms)
+          }
+        } catch (error) {
+          console.error('Failed to parse final NDJSON line:', error)
         }
-      } catch (error) {
-        throw new Error(`Failed to parse NDJSON line: ${error}`)
       }
+    } catch (error) {
+      throw new Error(`NDJSON stream reading error: ${error}`)
     }
 
     // Return aggregated result

package/sdk-extensions/index.d.ts

@@ -4,6 +4,7 @@
  */
 import { OperationClient } from './OperationClient';
 import { QueryClient } from './QueryClient';
+import { AgentClient } from './AgentClient';
 import { SSEClient } from './SSEClient';
 import { TableIngestClient } from './TableIngestClient';
 import { GraphClient } from './GraphClient';
@@ -17,6 +18,7 @@ export interface RoboSystemsExtensionConfig {
 }
 export declare class RoboSystemsExtensions {
     readonly query: QueryClient;
+    readonly agent: AgentClient;
     readonly operations: OperationClient;
     readonly tables: TableIngestClient;
     readonly graphs: GraphClient;
@@ -37,13 +39,16 @@ export declare class RoboSystemsExtensions {
 }
 export * from './OperationClient';
 export * from './QueryClient';
+export * from './AgentClient';
 export * from './SSEClient';
 export * from './TableIngestClient';
 export * from './GraphClient';
-export { OperationClient, QueryClient, SSEClient, TableIngestClient, GraphClient };
+export * from './config';
+export { OperationClient, QueryClient, AgentClient, SSEClient, TableIngestClient, GraphClient };
 export { useMultipleOperations, useOperation, useQuery, useSDKClients, useStreamingQuery, useTableUpload, } from './hooks';
 export declare const extensions: {
     readonly query: QueryClient;
+    readonly agent: AgentClient;
     readonly operations: OperationClient;
     readonly graphs: GraphClient;
     monitorOperation: (operationId: string, onProgress?: (progress: any) => void) => Promise<any>;
@@ -53,3 +58,5 @@ export declare const extensions: {
 export declare const monitorOperation: (operationId: string, onProgress?: (progress: any) => void) => Promise<any>;
 export declare const executeQuery: (graphId: string, query: string, parameters?: Record<string, any>) => Promise<import("./QueryClient").QueryResult>;
 export declare const streamQuery: (graphId: string, query: string, parameters?: Record<string, any>, chunkSize?: number) => AsyncIterableIterator<any>;
+export declare const agentQuery: (graphId: string, message: string, context?: Record<string, any>) => Promise<import("./AgentClient").AgentResult>;
+export declare const analyzeFinancials: (graphId: string, message: string) => Promise<import("./AgentClient").AgentResult>;

package/sdk-extensions/index.js

@@ -18,13 +18,15 @@ 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.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.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.TableIngestClient = exports.SSEClient = exports.AgentClient = exports.QueryClient = exports.OperationClient = exports.RoboSystemsExtensions = void 0;
 const client_gen_1 = require("../sdk/client.gen");
 const config_1 = require("./config");
 const OperationClient_1 = require("./OperationClient");
 Object.defineProperty(exports, "OperationClient", { enumerable: true, get: function () { return OperationClient_1.OperationClient; } });
 const QueryClient_1 = require("./QueryClient");
 Object.defineProperty(exports, "QueryClient", { enumerable: true, get: function () { return QueryClient_1.QueryClient; } });
+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");
@@ -51,6 +53,12 @@ class RoboSystemsExtensions {
             token: this.config.token,
             headers: this.config.headers,
         });
+        this.agent = new AgentClient_1.AgentClient({
+            baseUrl: this.config.baseUrl,
+            credentials: this.config.credentials,
+            token: this.config.token,
+            headers: this.config.headers,
+        });
         this.operations = new OperationClient_1.OperationClient({
             baseUrl: this.config.baseUrl,
             credentials: this.config.credentials,
@@ -95,6 +103,7 @@ class RoboSystemsExtensions {
      */
     close() {
         this.query.close();
+        this.agent.close();
         this.operations.closeAll();
         this.graphs.close();
     }
@@ -103,9 +112,11 @@ exports.RoboSystemsExtensions = RoboSystemsExtensions;
 // Export all types and classes
 __exportStar(require("./OperationClient"), exports);
 __exportStar(require("./QueryClient"), exports);
+__exportStar(require("./AgentClient"), exports);
 __exportStar(require("./SSEClient"), exports);
 __exportStar(require("./TableIngestClient"), exports);
 __exportStar(require("./GraphClient"), exports);
+__exportStar(require("./config"), exports);
 // Export React hooks
 var hooks_1 = require("./hooks");
 Object.defineProperty(exports, "useMultipleOperations", { enumerable: true, get: function () { return hooks_1.useMultipleOperations; } });
@@ -126,6 +137,9 @@ exports.extensions = {
     get query() {
         return getExtensions().query;
     },
+    get agent() {
+        return getExtensions().agent;
+    },
     get operations() {
         return getExtensions().operations;
     },
@@ -143,3 +157,7 @@ const executeQuery = (graphId, query, parameters) => getExtensions().query.query
 exports.executeQuery = executeQuery;
 const streamQuery = (graphId, query, parameters, chunkSize) => getExtensions().query.streamQuery(graphId, query, parameters, chunkSize);
 exports.streamQuery = streamQuery;
+const agentQuery = (graphId, message, context) => getExtensions().agent.query(graphId, message, context);
+exports.agentQuery = agentQuery;
+const analyzeFinancials = (graphId, message) => getExtensions().agent.analyzeFinancials(graphId, message);
+exports.analyzeFinancials = analyzeFinancials;

package/sdk-extensions/index.ts

@@ -7,6 +7,7 @@ import { client } from '../sdk/client.gen'
 import { extractTokenFromSDKClient } from './config'
 import { OperationClient } from './OperationClient'
 import { QueryClient } from './QueryClient'
+import { AgentClient } from './AgentClient'
 import { SSEClient } from './SSEClient'
 import { TableIngestClient } from './TableIngestClient'
 import { GraphClient } from './GraphClient'
@@ -32,6 +33,7 @@ interface ResolvedConfig {
 
 export class RoboSystemsExtensions {
   public readonly query: QueryClient
+  public readonly agent: AgentClient
   public readonly operations: OperationClient
   public readonly tables: TableIngestClient
   public readonly graphs: GraphClient
@@ -60,6 +62,13 @@ export class RoboSystemsExtensions {
       headers: this.config.headers,
     })
 
+    this.agent = new AgentClient({
+      baseUrl: this.config.baseUrl,
+      credentials: this.config.credentials,
+      token: this.config.token,
+      headers: this.config.headers,
+    })
+
     this.operations = new OperationClient({
       baseUrl: this.config.baseUrl,
       credentials: this.config.credentials,
@@ -109,6 +118,7 @@ export class RoboSystemsExtensions {
    */
   close(): void {
     this.query.close()
+    this.agent.close()
     this.operations.closeAll()
     this.graphs.close()
   }
@@ -117,10 +127,12 @@ export class RoboSystemsExtensions {
 // Export all types and classes
 export * from './OperationClient'
 export * from './QueryClient'
+export * from './AgentClient'
 export * from './SSEClient'
 export * from './TableIngestClient'
 export * from './GraphClient'
-export { OperationClient, QueryClient, SSEClient, TableIngestClient, GraphClient }
+export * from './config'
+export { OperationClient, QueryClient, AgentClient, SSEClient, TableIngestClient, GraphClient }
 
 // Export React hooks
 export {
@@ -146,6 +158,9 @@ export const extensions = {
   get query() {
     return getExtensions().query
   },
+  get agent() {
+    return getExtensions().agent
+  },
   get operations() {
     return getExtensions().operations
   },
@@ -171,3 +186,9 @@ export const streamQuery = (
   parameters?: Record<string, any>,
   chunkSize?: number
 ) => getExtensions().query.streamQuery(graphId, query, parameters, chunkSize)
+
+export const agentQuery = (graphId: string, message: string, context?: Record<string, any>) =>
+  getExtensions().agent.query(graphId, message, context)
+
+export const analyzeFinancials = (graphId: string, message: string) =>
+  getExtensions().agent.analyzeFinancials(graphId, message)

package/sdk.gen.d.ts

@@ -1,5 +1,5 @@
 import type { Options as ClientOptions, TDataShape, Client } from './client';
-import type { RegisterUserData, RegisterUserResponses, RegisterUserErrors, LoginUserData, LoginUserResponses, LoginUserErrors, LogoutUserData, LogoutUserResponses, GetCurrentAuthUserData, GetCurrentAuthUserResponses, GetCurrentAuthUserErrors, RefreshAuthSessionData, RefreshAuthSessionResponses, RefreshAuthSessionErrors, ResendVerificationEmailData, ResendVerificationEmailResponses, ResendVerificationEmailErrors, VerifyEmailData, VerifyEmailResponses, VerifyEmailErrors, GetPasswordPolicyData, GetPasswordPolicyResponses, CheckPasswordStrengthData, CheckPasswordStrengthResponses, CheckPasswordStrengthErrors, ForgotPasswordData, ForgotPasswordResponses, ForgotPasswordErrors, ValidateResetTokenData, ValidateResetTokenResponses, ValidateResetTokenErrors, ResetPasswordData, ResetPasswordResponses, ResetPasswordErrors, GenerateSsoTokenData, GenerateSsoTokenResponses, GenerateSsoTokenErrors, SsoTokenExchangeData, SsoTokenExchangeResponses, SsoTokenExchangeErrors, CompleteSsoAuthData, CompleteSsoAuthResponses, CompleteSsoAuthErrors, GetCaptchaConfigData, GetCaptchaConfigResponses, GetServiceStatusData, GetServiceStatusResponses, GetCurrentUserData, GetCurrentUserResponses, UpdateUserData, UpdateUserResponses, UpdateUserErrors, UpdateUserPasswordData, UpdateUserPasswordResponses, UpdateUserPasswordErrors, ListUserApiKeysData, ListUserApiKeysResponses, CreateUserApiKeyData, CreateUserApiKeyResponses, CreateUserApiKeyErrors, RevokeUserApiKeyData, RevokeUserApiKeyResponses, RevokeUserApiKeyErrors, UpdateUserApiKeyData, UpdateUserApiKeyResponses, UpdateUserApiKeyErrors, ListUserOrgsData, ListUserOrgsResponses, CreateOrgData, CreateOrgResponses, CreateOrgErrors, GetOrgData, GetOrgResponses, GetOrgErrors, UpdateOrgData, UpdateOrgResponses, UpdateOrgErrors, ListOrgGraphsData, ListOrgGraphsResponses, ListOrgGraphsErrors, ListOrgMembersData, ListOrgMembersResponses, ListOrgMembersErrors, InviteOrgMemberData, InviteOrgMemberResponses, InviteOrgMemberErrors, RemoveOrgMemberData, RemoveOrgMemberResponses, RemoveOrgMemberErrors, UpdateOrgMemberRoleData, UpdateOrgMemberRoleResponses, UpdateOrgMemberRoleErrors, GetOrgLimitsData, GetOrgLimitsResponses, GetOrgLimitsErrors, GetOrgUsageData, GetOrgUsageResponses, GetOrgUsageErrors, ListConnectionsData, ListConnectionsResponses, ListConnectionsErrors, CreateConnectionData, CreateConnectionResponses, CreateConnectionErrors, GetConnectionOptionsData, GetConnectionOptionsResponses, GetConnectionOptionsErrors, ExchangeLinkTokenData, ExchangeLinkTokenResponses, ExchangeLinkTokenErrors, CreateLinkTokenData, CreateLinkTokenResponses, CreateLinkTokenErrors, InitOAuthData, InitOAuthResponses, InitOAuthErrors, OauthCallbackData, OauthCallbackResponses, OauthCallbackErrors, DeleteConnectionData, DeleteConnectionResponses, DeleteConnectionErrors, GetConnectionData, GetConnectionResponses, GetConnectionErrors, SyncConnectionData, SyncConnectionResponses, SyncConnectionErrors, AutoSelectAgentData, AutoSelectAgentResponses, AutoSelectAgentErrors, ExecuteSpecificAgentData, ExecuteSpecificAgentResponses, ExecuteSpecificAgentErrors, BatchProcessQueriesData, BatchProcessQueriesResponses, BatchProcessQueriesErrors, ListAgentsData, ListAgentsResponses, ListAgentsErrors, GetAgentMetadataData, GetAgentMetadataResponses, GetAgentMetadataErrors, RecommendAgentData, RecommendAgentResponses, RecommendAgentErrors, ListMcpToolsData, ListMcpToolsResponses, ListMcpToolsErrors, CallMcpToolData, CallMcpToolResponses, CallMcpToolErrors, ListBackupsData, ListBackupsResponses, ListBackupsErrors, CreateBackupData, CreateBackupResponses, CreateBackupErrors, GetBackupDownloadUrlData, GetBackupDownloadUrlResponses, GetBackupDownloadUrlErrors, RestoreBackupData, RestoreBackupResponses, RestoreBackupErrors, GetBackupStatsData, GetBackupStatsResponses, GetBackupStatsErrors, GetGraphMetricsData, GetGraphMetricsResponses, GetGraphMetricsErrors, GetGraphUsageAnalyticsData, GetGraphUsageAnalyticsResponses, GetGraphUsageAnalyticsErrors, ExecuteCypherQueryData, ExecuteCypherQueryResponses, ExecuteCypherQueryErrors, GetGraphSchemaData, GetGraphSchemaResponses, GetGraphSchemaErrors, ExportGraphSchemaData, ExportGraphSchemaResponses, ExportGraphSchemaErrors, ValidateSchemaData, ValidateSchemaResponses, ValidateSchemaErrors, GetCreditSummaryData, GetCreditSummaryResponses, GetCreditSummaryErrors, ListCreditTransactionsData, ListCreditTransactionsResponses, ListCreditTransactionsErrors, CheckCreditBalanceData, CheckCreditBalanceResponses, CheckCreditBalanceErrors, GetStorageUsageData, GetStorageUsageResponses, GetStorageUsageErrors, CheckStorageLimitsData, CheckStorageLimitsResponses, CheckStorageLimitsErrors, GetDatabaseHealthData, GetDatabaseHealthResponses, GetDatabaseHealthErrors, GetDatabaseInfoData, GetDatabaseInfoResponses, GetDatabaseInfoErrors, GetGraphLimitsData, GetGraphLimitsResponses, GetGraphLimitsErrors, ListSubgraphsData, ListSubgraphsResponses, ListSubgraphsErrors, CreateSubgraphData, CreateSubgraphResponses, CreateSubgraphErrors, DeleteSubgraphData, DeleteSubgraphResponses, DeleteSubgraphErrors, GetSubgraphInfoData, GetSubgraphInfoResponses, GetSubgraphInfoErrors, GetSubgraphQuotaData, GetSubgraphQuotaResponses, GetSubgraphQuotaErrors, CancelSubscriptionData, CancelSubscriptionResponses, CancelSubscriptionErrors, GetGraphSubscriptionData, GetGraphSubscriptionResponses, GetGraphSubscriptionErrors, CreateRepositorySubscriptionData, CreateRepositorySubscriptionResponses, CreateRepositorySubscriptionErrors, UpgradeSubscriptionData, UpgradeSubscriptionResponses, UpgradeSubscriptionErrors, ListTablesData, ListTablesResponses, ListTablesErrors, ListTableFilesData, ListTableFilesResponses, ListTableFilesErrors, GetUploadUrlData, GetUploadUrlResponses, GetUploadUrlErrors, DeleteFileData, DeleteFileResponses, DeleteFileErrors, GetFileInfoData, GetFileInfoResponses, GetFileInfoErrors, UpdateFileStatusData, UpdateFileStatusResponses, UpdateFileStatusErrors, IngestTablesData, IngestTablesResponses, IngestTablesErrors, QueryTablesData, QueryTablesResponses, QueryTablesErrors, GetGraphsData, GetGraphsResponses, GetGraphsErrors, CreateGraphData, CreateGraphResponses, CreateGraphErrors, GetAvailableExtensionsData, GetAvailableExtensionsResponses, GetAvailableExtensionsErrors, GetAvailableGraphTiersData, GetAvailableGraphTiersResponses, GetAvailableGraphTiersErrors, SelectGraphData, SelectGraphResponses, SelectGraphErrors, GetServiceOfferingsData, GetServiceOfferingsResponses, GetServiceOfferingsErrors, StreamOperationEventsData, StreamOperationEventsResponses, StreamOperationEventsErrors, GetOperationStatusData, GetOperationStatusResponses, GetOperationStatusErrors, CancelOperationData, CancelOperationResponses, CancelOperationErrors, GetOrgBillingCustomerData, GetOrgBillingCustomerResponses, GetOrgBillingCustomerErrors, UpdateOrgPaymentMethodData, UpdateOrgPaymentMethodResponses, UpdateOrgPaymentMethodErrors, ListOrgSubscriptionsData, ListOrgSubscriptionsResponses, ListOrgSubscriptionsErrors, GetOrgSubscriptionData, GetOrgSubscriptionResponses, GetOrgSubscriptionErrors, CancelOrgSubscriptionData, CancelOrgSubscriptionResponses, CancelOrgSubscriptionErrors, ListOrgInvoicesData, ListOrgInvoicesResponses, ListOrgInvoicesErrors, GetOrgUpcomingInvoiceData, GetOrgUpcomingInvoiceResponses, GetOrgUpcomingInvoiceErrors, CreateCheckoutSessionData, CreateCheckoutSessionResponses, CreateCheckoutSessionErrors, GetCheckoutStatusData, GetCheckoutStatusResponses, GetCheckoutStatusErrors } from './types.gen';
+import type { RegisterUserData, RegisterUserResponses, RegisterUserErrors, LoginUserData, LoginUserResponses, LoginUserErrors, LogoutUserData, LogoutUserResponses, GetCurrentAuthUserData, GetCurrentAuthUserResponses, GetCurrentAuthUserErrors, RefreshAuthSessionData, RefreshAuthSessionResponses, RefreshAuthSessionErrors, ResendVerificationEmailData, ResendVerificationEmailResponses, ResendVerificationEmailErrors, VerifyEmailData, VerifyEmailResponses, VerifyEmailErrors, GetPasswordPolicyData, GetPasswordPolicyResponses, CheckPasswordStrengthData, CheckPasswordStrengthResponses, CheckPasswordStrengthErrors, ForgotPasswordData, ForgotPasswordResponses, ForgotPasswordErrors, ValidateResetTokenData, ValidateResetTokenResponses, ValidateResetTokenErrors, ResetPasswordData, ResetPasswordResponses, ResetPasswordErrors, GenerateSsoTokenData, GenerateSsoTokenResponses, GenerateSsoTokenErrors, SsoTokenExchangeData, SsoTokenExchangeResponses, SsoTokenExchangeErrors, CompleteSsoAuthData, CompleteSsoAuthResponses, CompleteSsoAuthErrors, GetCaptchaConfigData, GetCaptchaConfigResponses, GetServiceStatusData, GetServiceStatusResponses, GetCurrentUserData, GetCurrentUserResponses, UpdateUserData, UpdateUserResponses, UpdateUserErrors, UpdateUserPasswordData, UpdateUserPasswordResponses, UpdateUserPasswordErrors, ListUserApiKeysData, ListUserApiKeysResponses, CreateUserApiKeyData, CreateUserApiKeyResponses, CreateUserApiKeyErrors, RevokeUserApiKeyData, RevokeUserApiKeyResponses, RevokeUserApiKeyErrors, UpdateUserApiKeyData, UpdateUserApiKeyResponses, UpdateUserApiKeyErrors, ListUserOrgsData, ListUserOrgsResponses, CreateOrgData, CreateOrgResponses, CreateOrgErrors, GetOrgData, GetOrgResponses, GetOrgErrors, UpdateOrgData, UpdateOrgResponses, UpdateOrgErrors, ListOrgGraphsData, ListOrgGraphsResponses, ListOrgGraphsErrors, ListOrgMembersData, ListOrgMembersResponses, ListOrgMembersErrors, InviteOrgMemberData, InviteOrgMemberResponses, InviteOrgMemberErrors, RemoveOrgMemberData, RemoveOrgMemberResponses, RemoveOrgMemberErrors, UpdateOrgMemberRoleData, UpdateOrgMemberRoleResponses, UpdateOrgMemberRoleErrors, GetOrgLimitsData, GetOrgLimitsResponses, GetOrgLimitsErrors, GetOrgUsageData, GetOrgUsageResponses, GetOrgUsageErrors, ListConnectionsData, ListConnectionsResponses, ListConnectionsErrors, CreateConnectionData, CreateConnectionResponses, CreateConnectionErrors, GetConnectionOptionsData, GetConnectionOptionsResponses, GetConnectionOptionsErrors, ExchangeLinkTokenData, ExchangeLinkTokenResponses, ExchangeLinkTokenErrors, CreateLinkTokenData, CreateLinkTokenResponses, CreateLinkTokenErrors, InitOAuthData, InitOAuthResponses, InitOAuthErrors, OauthCallbackData, OauthCallbackResponses, OauthCallbackErrors, DeleteConnectionData, DeleteConnectionResponses, DeleteConnectionErrors, GetConnectionData, GetConnectionResponses, GetConnectionErrors, SyncConnectionData, SyncConnectionResponses, SyncConnectionErrors, ListAgentsData, ListAgentsResponses, ListAgentsErrors, AutoSelectAgentData, AutoSelectAgentResponses, AutoSelectAgentErrors, GetAgentMetadataData, GetAgentMetadataResponses, GetAgentMetadataErrors, ExecuteSpecificAgentData, ExecuteSpecificAgentResponses, ExecuteSpecificAgentErrors, BatchProcessQueriesData, BatchProcessQueriesResponses, BatchProcessQueriesErrors, RecommendAgentData, RecommendAgentResponses, RecommendAgentErrors, ListMcpToolsData, ListMcpToolsResponses, ListMcpToolsErrors, CallMcpToolData, CallMcpToolResponses, CallMcpToolErrors, ListBackupsData, ListBackupsResponses, ListBackupsErrors, CreateBackupData, CreateBackupResponses, CreateBackupErrors, GetBackupDownloadUrlData, GetBackupDownloadUrlResponses, GetBackupDownloadUrlErrors, RestoreBackupData, RestoreBackupResponses, RestoreBackupErrors, GetBackupStatsData, GetBackupStatsResponses, GetBackupStatsErrors, GetGraphMetricsData, GetGraphMetricsResponses, GetGraphMetricsErrors, GetGraphUsageAnalyticsData, GetGraphUsageAnalyticsResponses, GetGraphUsageAnalyticsErrors, ExecuteCypherQueryData, ExecuteCypherQueryResponses, ExecuteCypherQueryErrors, GetGraphSchemaData, GetGraphSchemaResponses, GetGraphSchemaErrors, ExportGraphSchemaData, ExportGraphSchemaResponses, ExportGraphSchemaErrors, ValidateSchemaData, ValidateSchemaResponses, ValidateSchemaErrors, GetCreditSummaryData, GetCreditSummaryResponses, GetCreditSummaryErrors, ListCreditTransactionsData, ListCreditTransactionsResponses, ListCreditTransactionsErrors, CheckCreditBalanceData, CheckCreditBalanceResponses, CheckCreditBalanceErrors, GetStorageUsageData, GetStorageUsageResponses, GetStorageUsageErrors, CheckStorageLimitsData, CheckStorageLimitsResponses, CheckStorageLimitsErrors, GetDatabaseHealthData, GetDatabaseHealthResponses, GetDatabaseHealthErrors, GetDatabaseInfoData, GetDatabaseInfoResponses, GetDatabaseInfoErrors, GetGraphLimitsData, GetGraphLimitsResponses, GetGraphLimitsErrors, ListSubgraphsData, ListSubgraphsResponses, ListSubgraphsErrors, CreateSubgraphData, CreateSubgraphResponses, CreateSubgraphErrors, DeleteSubgraphData, DeleteSubgraphResponses, DeleteSubgraphErrors, GetSubgraphInfoData, GetSubgraphInfoResponses, GetSubgraphInfoErrors, GetSubgraphQuotaData, GetSubgraphQuotaResponses, GetSubgraphQuotaErrors, GetGraphSubscriptionData, GetGraphSubscriptionResponses, GetGraphSubscriptionErrors, CreateRepositorySubscriptionData, CreateRepositorySubscriptionResponses, CreateRepositorySubscriptionErrors, UpgradeSubscriptionData, UpgradeSubscriptionResponses, UpgradeSubscriptionErrors, ListTablesData, ListTablesResponses, ListTablesErrors, ListTableFilesData, ListTableFilesResponses, ListTableFilesErrors, GetUploadUrlData, GetUploadUrlResponses, GetUploadUrlErrors, DeleteFileData, DeleteFileResponses, DeleteFileErrors, GetFileInfoData, GetFileInfoResponses, GetFileInfoErrors, UpdateFileStatusData, UpdateFileStatusResponses, UpdateFileStatusErrors, IngestTablesData, IngestTablesResponses, IngestTablesErrors, QueryTablesData, QueryTablesResponses, QueryTablesErrors, GetGraphsData, GetGraphsResponses, GetGraphsErrors, CreateGraphData, CreateGraphResponses, CreateGraphErrors, GetAvailableExtensionsData, GetAvailableExtensionsResponses, GetAvailableExtensionsErrors, GetAvailableGraphTiersData, GetAvailableGraphTiersResponses, GetAvailableGraphTiersErrors, SelectGraphData, SelectGraphResponses, SelectGraphErrors, GetServiceOfferingsData, GetServiceOfferingsResponses, GetServiceOfferingsErrors, StreamOperationEventsData, StreamOperationEventsResponses, StreamOperationEventsErrors, GetOperationStatusData, GetOperationStatusResponses, GetOperationStatusErrors, CancelOperationData, CancelOperationResponses, CancelOperationErrors, GetOrgBillingCustomerData, GetOrgBillingCustomerResponses, GetOrgBillingCustomerErrors, CreatePortalSessionData, CreatePortalSessionResponses, CreatePortalSessionErrors, ListOrgSubscriptionsData, ListOrgSubscriptionsResponses, ListOrgSubscriptionsErrors, GetOrgSubscriptionData, GetOrgSubscriptionResponses, GetOrgSubscriptionErrors, CancelOrgSubscriptionData, CancelOrgSubscriptionResponses, CancelOrgSubscriptionErrors, ListOrgInvoicesData, ListOrgInvoicesResponses, ListOrgInvoicesErrors, GetOrgUpcomingInvoiceData, GetOrgUpcomingInvoiceResponses, GetOrgUpcomingInvoiceErrors, CreateCheckoutSessionData, CreateCheckoutSessionResponses, CreateCheckoutSessionErrors, GetCheckoutStatusData, GetCheckoutStatusResponses, GetCheckoutStatusErrors } from './types.gen';
 export type Options<TData extends TDataShape = TDataShape, ThrowOnError extends boolean = boolean> = ClientOptions<TData, ThrowOnError> & {
     /**
      * You can provide a client instance returned by `createClient()` instead of
@@ -392,9 +392,22 @@ export declare const getConnection: <ThrowOnError extends boolean = false>(optio
  * Returns a task ID for monitoring sync progress.
  */
 export declare const syncConnection: <ThrowOnError extends boolean = false>(options: Options<SyncConnectionData, ThrowOnError>) => import("./client").RequestResult<SyncConnectionResponses, SyncConnectionErrors, ThrowOnError, "fields">;
+/**
+ * List available agents
+ * Get a comprehensive list of all available agents with their metadata.
+ *
+ * **Returns:**
+ * - Agent types and names
+ * - Capabilities and supported modes
+ * - Version information
+ * - Credit requirements
+ *
+ * Use the optional `capability` filter to find agents with specific capabilities.
+ */
+export declare const listAgents: <ThrowOnError extends boolean = false>(options: Options<ListAgentsData, ThrowOnError>) => import("./client").RequestResult<ListAgentsResponses, ListAgentsErrors, ThrowOnError, "fields">;
 /**
  * Auto-select agent for query
- * Automatically select the best agent for your query.
+ * Automatically select the best agent for your query with intelligent execution strategy.
  *
  * **Agent Selection Process:**
  *
@@ -403,7 +416,8 @@ export declare const syncConnection: <ThrowOnError extends boolean = false>(opti
  * 2. Enriching context with RAG if enabled
  * 3. Evaluating all available agents against selection criteria
  * 4. Selecting the best match based on confidence scores
- * 5. Executing the query with the selected agent
+ * 5. Choosing execution strategy (sync/SSE/async) based on expected time
+ * 6. Executing the query with the selected agent
  *
  * **Available Agent Types:**
  * - `financial`: Financial analysis, SEC filings, company metrics
@@ -416,6 +430,14 @@ export declare const syncConnection: <ThrowOnError extends boolean = false>(opti
  * - `extended`: Comprehensive analysis (~15-60s), deep research
  * - `streaming`: Real-time response streaming
  *
+ * **Execution Strategies (automatic):**
+ * - Fast operations (<5s): Immediate synchronous response
+ * - Medium operations (5-30s): SSE streaming with progress updates
+ * - Long operations (>30s): Async Celery worker with operation tracking
+ *
+ * **Response Mode Override:**
+ * Use query parameter `?mode=sync|async` to override automatic strategy selection.
+ *
  * **Confidence Score Interpretation:**
  * - `0.9-1.0`: High confidence, agent is ideal match
  * - `0.7-0.9`: Good confidence, agent is suitable
@@ -438,15 +460,38 @@ export declare const syncConnection: <ThrowOnError extends boolean = false>(opti
  * See request/response examples in the "Examples" dropdown below.
  */
 export declare const autoSelectAgent: <ThrowOnError extends boolean = false>(options: Options<AutoSelectAgentData, ThrowOnError>) => import("./client").RequestResult<AutoSelectAgentResponses, AutoSelectAgentErrors, ThrowOnError, "fields">;
+/**
+ * Get agent metadata
+ * Get comprehensive metadata for a specific agent type.
+ *
+ * **Returns:**
+ * - Agent name and description
+ * - Version information
+ * - Supported capabilities and modes
+ * - Credit requirements
+ * - Author and tags
+ * - Configuration options
+ *
+ * Use this to understand agent capabilities before execution.
+ */
+export declare const getAgentMetadata: <ThrowOnError extends boolean = false>(options: Options<GetAgentMetadataData, ThrowOnError>) => import("./client").RequestResult<GetAgentMetadataResponses, GetAgentMetadataErrors, ThrowOnError, "fields">;
 /**
  * Execute specific agent
- * Execute a specific agent type directly.
+ * Execute a specific agent type directly with intelligent execution strategy.
  *
  * Available agents:
  * - **financial**: Financial analysis, SEC filings, accounting data
  * - **research**: Deep research and comprehensive analysis
  * - **rag**: Fast retrieval without AI (no credits required)
  *
+ * **Execution Strategies (automatic):**
+ * - Fast operations (<5s): Immediate synchronous response
+ * - Medium operations (5-30s): SSE streaming with progress updates
+ * - Long operations (>30s): Async Celery worker with operation tracking
+ *
+ * **Response Mode Override:**
+ * Use query parameter `?mode=sync|async` to override automatic strategy selection.
+ *
  * Use this endpoint when you know which agent you want to use.
  */
 export declare const executeSpecificAgent: <ThrowOnError extends boolean = false>(options: Options<ExecuteSpecificAgentData, ThrowOnError>) => import("./client").RequestResult<ExecuteSpecificAgentResponses, ExecuteSpecificAgentErrors, ThrowOnError, "fields">;
@@ -468,34 +513,6 @@ export declare const executeSpecificAgent: <ThrowOnError extends boolean = false
  * Returns individual results for each query with execution metrics.
  */
 export declare const batchProcessQueries: <ThrowOnError extends boolean = false>(options: Options<BatchProcessQueriesData, ThrowOnError>) => import("./client").RequestResult<BatchProcessQueriesResponses, BatchProcessQueriesErrors, ThrowOnError, "fields">;
-/**
- * List available agents
- * Get a comprehensive list of all available agents with their metadata.
- *
- * **Returns:**
- * - Agent types and names
- * - Capabilities and supported modes
- * - Version information
- * - Credit requirements
- *
- * Use the optional `capability` filter to find agents with specific capabilities.
- */
-export declare const listAgents: <ThrowOnError extends boolean = false>(options: Options<ListAgentsData, ThrowOnError>) => import("./client").RequestResult<ListAgentsResponses, ListAgentsErrors, ThrowOnError, "fields">;
-/**
- * Get agent metadata
- * Get comprehensive metadata for a specific agent type.
- *
- * **Returns:**
- * - Agent name and description
- * - Version information
- * - Supported capabilities and modes
- * - Credit requirements
- * - Author and tags
- * - Configuration options
- *
- * Use this to understand agent capabilities before execution.
- */
-export declare const getAgentMetadata: <ThrowOnError extends boolean = false>(options: Options<GetAgentMetadataData, ThrowOnError>) => import("./client").RequestResult<GetAgentMetadataResponses, GetAgentMetadataErrors, ThrowOnError, "fields">;
 /**
  * Get agent recommendations
  * Get intelligent agent recommendations for a specific query.
@@ -1112,7 +1129,7 @@ export declare const listSubgraphs: <ThrowOnError extends boolean = false>(optio
  * - Valid authentication
  * - Parent graph must exist and be accessible to the user
  * - User must have 'admin' permission on the parent graph
- * - Parent graph tier must support subgraphs (Enterprise or Premium only)
+ * - Parent graph tier must support subgraphs (Kuzu Large/XLarge or Neo4j Enterprise XLarge)
  * - Must be within subgraph quota limits
  * - Subgraph name must be unique within the parent graph
  *
@@ -1138,8 +1155,8 @@ export declare const createSubgraph: <ThrowOnError extends boolean = false>(opti
  * All data in the subgraph will be lost.
  *
  * **Backup Location:**
- * If backup requested, stored in S3 at:
- * `s3://robosystems-backups/{instance_id}/{database_name}_{timestamp}.backup`
+ * If backup requested, stored in S3 Kuzu database bucket at:
+ * `s3://{kuzu_s3_bucket}/{instance_id}/{database_name}_{timestamp}.backup`
  */
 export declare const deleteSubgraph: <ThrowOnError extends boolean = false>(options: Options<DeleteSubgraphData, ThrowOnError>) => import("./client").RequestResult<DeleteSubgraphResponses, DeleteSubgraphErrors, ThrowOnError, "fields">;
 /**
@@ -1186,16 +1203,6 @@ export declare const getSubgraphInfo: <ThrowOnError extends boolean = false>(opt
  * Individual subgraph sizes shown in list endpoint.
  */
 export declare const getSubgraphQuota: <ThrowOnError extends boolean = false>(options: Options<GetSubgraphQuotaData, ThrowOnError>) => import("./client").RequestResult<GetSubgraphQuotaResponses, GetSubgraphQuotaErrors, ThrowOnError, "fields">;
-/**
- * Cancel Subscription
- * Cancel a subscription.
- *
- * For shared repositories: Cancels the user's personal subscription
- * For user graphs: Not allowed - delete the graph instead
- *
- * The subscription will be marked as canceled and will end at the current period end date.
- */
-export declare const cancelSubscription: <ThrowOnError extends boolean = false>(options: Options<CancelSubscriptionData, ThrowOnError>) => import("./client").RequestResult<CancelSubscriptionResponses, CancelSubscriptionErrors, ThrowOnError, "fields">;
 /**
  * Get Subscription
  * Get subscription details for a graph or shared repository.
@@ -1874,15 +1881,22 @@ export declare const cancelOperation: <ThrowOnError extends boolean = false>(opt
  */
 export declare const getOrgBillingCustomer: <ThrowOnError extends boolean = false>(options: Options<GetOrgBillingCustomerData, ThrowOnError>) => import("./client").RequestResult<GetOrgBillingCustomerResponses, GetOrgBillingCustomerErrors, ThrowOnError, "fields">;
 /**
- * Update Organization Default Payment Method
- * Update the default payment method for the organization.
+ * Create Customer Portal Session
+ * Create a Stripe Customer Portal session for managing payment methods.
+ *
+ * The portal allows users to:
+ * - Add new payment methods
+ * - Remove existing payment methods
+ * - Update default payment method
+ * - View billing history
  *
- * This changes which payment method will be used for future subscription charges.
+ * The user will be redirected to Stripe's hosted portal page and returned to the billing page when done.
  *
  * **Requirements:**
  * - User must be an OWNER of the organization
+ * - Organization must have a Stripe customer ID (i.e., has gone through checkout at least once)
  */
-export declare const updateOrgPaymentMethod: <ThrowOnError extends boolean = false>(options: Options<UpdateOrgPaymentMethodData, ThrowOnError>) => import("./client").RequestResult<UpdateOrgPaymentMethodResponses, UpdateOrgPaymentMethodErrors, ThrowOnError, "fields">;
+export declare const createPortalSession: <ThrowOnError extends boolean = false>(options: Options<CreatePortalSessionData, ThrowOnError>) => import("./client").RequestResult<CreatePortalSessionResponses, CreatePortalSessionErrors, ThrowOnError, "fields">;
 /**
  * List Organization Subscriptions
  * List all active and past subscriptions for an organization.