@robosystems/client 0.2.11 → 0.2.13

package/sdk.gen.js

@@ -1,9 +1,9 @@
 "use strict";
 // This file is auto-generated by @hey-api/openapi-ts
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getAgentMetadata = exports.listAgents = exports.batchProcessQueries = exports.executeSpecificAgent = exports.autoSelectAgent = exports.syncConnection = exports.getConnection = exports.deleteConnection = exports.oauthCallback = exports.initOAuth = exports.createLinkToken = exports.exchangeLinkToken = exports.getConnectionOptions = exports.createConnection = exports.listConnections = exports.getOrgUsage = exports.getOrgLimits = exports.updateOrgMemberRole = exports.removeOrgMember = exports.inviteOrgMember = exports.listOrgMembers = exports.listOrgGraphs = exports.updateOrg = exports.getOrg = exports.createOrg = exports.listUserOrgs = exports.updateUserApiKey = exports.revokeUserApiKey = exports.createUserApiKey = exports.listUserApiKeys = exports.updateUserPassword = exports.updateUser = exports.getCurrentUser = exports.getServiceStatus = exports.getCaptchaConfig = exports.completeSsoAuth = exports.ssoTokenExchange = exports.generateSsoToken = exports.resetPassword = exports.validateResetToken = exports.forgotPassword = exports.checkPasswordStrength = exports.getPasswordPolicy = exports.verifyEmail = exports.resendVerificationEmail = exports.refreshAuthSession = exports.getCurrentAuthUser = exports.logoutUser = exports.loginUser = exports.registerUser = void 0;
-exports.updateOrgPaymentMethod = exports.getOrgBillingCustomer = exports.cancelOperation = exports.getOperationStatus = exports.streamOperationEvents = exports.getServiceOfferings = exports.selectGraph = exports.getAvailableGraphTiers = exports.getAvailableExtensions = exports.createGraph = exports.getGraphs = exports.queryTables = exports.ingestTables = exports.updateFileStatus = exports.getFileInfo = exports.deleteFile = exports.getUploadUrl = exports.listTableFiles = exports.listTables = exports.upgradeSubscription = exports.createRepositorySubscription = exports.getGraphSubscription = exports.cancelSubscription = exports.getSubgraphQuota = exports.getSubgraphInfo = exports.deleteSubgraph = exports.createSubgraph = exports.listSubgraphs = exports.getGraphLimits = exports.getDatabaseInfo = exports.getDatabaseHealth = exports.checkStorageLimits = exports.getStorageUsage = exports.checkCreditBalance = exports.listCreditTransactions = exports.getCreditSummary = exports.validateSchema = exports.exportGraphSchema = exports.getGraphSchema = exports.executeCypherQuery = exports.getGraphUsageAnalytics = exports.getGraphMetrics = exports.getBackupStats = exports.restoreBackup = exports.getBackupDownloadUrl = exports.createBackup = exports.listBackups = exports.callMcpTool = exports.listMcpTools = exports.recommendAgent = void 0;
-exports.getCheckoutStatus = exports.createCheckoutSession = exports.getOrgUpcomingInvoice = exports.listOrgInvoices = exports.cancelOrgSubscription = exports.getOrgSubscription = exports.listOrgSubscriptions = void 0;
+exports.batchProcessQueries = exports.executeSpecificAgent = exports.getAgentMetadata = exports.autoSelectAgent = exports.listAgents = exports.syncConnection = exports.getConnection = exports.deleteConnection = exports.oauthCallback = exports.initOAuth = exports.createLinkToken = exports.exchangeLinkToken = exports.getConnectionOptions = exports.createConnection = exports.listConnections = exports.getOrgUsage = exports.getOrgLimits = exports.updateOrgMemberRole = exports.removeOrgMember = exports.inviteOrgMember = exports.listOrgMembers = exports.listOrgGraphs = exports.updateOrg = exports.getOrg = exports.createOrg = exports.listUserOrgs = exports.updateUserApiKey = exports.revokeUserApiKey = exports.createUserApiKey = exports.listUserApiKeys = exports.updateUserPassword = exports.updateUser = exports.getCurrentUser = exports.getServiceStatus = exports.getCaptchaConfig = exports.completeSsoAuth = exports.ssoTokenExchange = exports.generateSsoToken = exports.resetPassword = exports.validateResetToken = exports.forgotPassword = exports.checkPasswordStrength = exports.getPasswordPolicy = exports.verifyEmail = exports.resendVerificationEmail = exports.refreshAuthSession = exports.getCurrentAuthUser = exports.logoutUser = exports.loginUser = exports.registerUser = void 0;
+exports.listOrgSubscriptions = exports.createPortalSession = exports.getOrgBillingCustomer = exports.cancelOperation = exports.getOperationStatus = exports.streamOperationEvents = exports.getServiceOfferings = exports.selectGraph = exports.getAvailableGraphTiers = exports.getAvailableExtensions = exports.createGraph = exports.getGraphs = exports.queryTables = exports.ingestTables = exports.updateFileStatus = exports.getFileInfo = exports.deleteFile = exports.getUploadUrl = exports.listTableFiles = exports.listTables = exports.upgradeSubscription = exports.createRepositorySubscription = exports.getGraphSubscription = exports.getSubgraphQuota = exports.getSubgraphInfo = exports.deleteSubgraph = exports.createSubgraph = exports.listSubgraphs = exports.getGraphLimits = exports.getDatabaseInfo = exports.getDatabaseHealth = exports.checkStorageLimits = exports.getStorageUsage = exports.checkCreditBalance = exports.listCreditTransactions = exports.getCreditSummary = exports.validateSchema = exports.exportGraphSchema = exports.getGraphSchema = exports.executeCypherQuery = exports.getGraphUsageAnalytics = exports.getGraphMetrics = exports.getBackupStats = exports.restoreBackup = exports.getBackupDownloadUrl = exports.createBackup = exports.listBackups = exports.callMcpTool = exports.listMcpTools = exports.recommendAgent = void 0;
+exports.getCheckoutStatus = exports.createCheckoutSession = exports.getOrgUpcomingInvoice = exports.listOrgInvoices = exports.cancelOrgSubscription = exports.getOrgSubscription = void 0;
 const client_gen_1 = require("./client.gen");
 /**
  * Register New User
@@ -1022,9 +1022,38 @@ const syncConnection = (options) => {
     });
 };
 exports.syncConnection = syncConnection;
+/**
+ * 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.
+ */
+const listAgents = (options) => {
+    return (options.client ?? client_gen_1.client).get({
+        security: [
+            {
+                name: 'X-API-Key',
+                type: 'apiKey'
+            },
+            {
+                scheme: 'bearer',
+                type: 'http'
+            }
+        ],
+        url: '/v1/graphs/{graph_id}/agent',
+        ...options
+    });
+};
+exports.listAgents = listAgents;
 /**
  * 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:**
  *
@@ -1033,7 +1062,8 @@ exports.syncConnection = syncConnection;
  * 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
@@ -1046,6 +1076,14 @@ exports.syncConnection = syncConnection;
  * - `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
@@ -1088,15 +1126,54 @@ const autoSelectAgent = (options) => {
     });
 };
 exports.autoSelectAgent = autoSelectAgent;
+/**
+ * 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.
+ */
+const getAgentMetadata = (options) => {
+    return (options.client ?? client_gen_1.client).get({
+        security: [
+            {
+                name: 'X-API-Key',
+                type: 'apiKey'
+            },
+            {
+                scheme: 'bearer',
+                type: 'http'
+            }
+        ],
+        url: '/v1/graphs/{graph_id}/agent/{agent_type}',
+        ...options
+    });
+};
+exports.getAgentMetadata = getAgentMetadata;
 /**
  * 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.
  */
 const executeSpecificAgent = (options) => {
@@ -1158,66 +1235,6 @@ const batchProcessQueries = (options) => {
     });
 };
 exports.batchProcessQueries = batchProcessQueries;
-/**
- * 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.
- */
-const listAgents = (options) => {
-    return (options.client ?? client_gen_1.client).get({
-        security: [
-            {
-                name: 'X-API-Key',
-                type: 'apiKey'
-            },
-            {
-                scheme: 'bearer',
-                type: 'http'
-            }
-        ],
-        url: '/v1/graphs/{graph_id}/agent/list',
-        ...options
-    });
-};
-exports.listAgents = listAgents;
-/**
- * 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.
- */
-const getAgentMetadata = (options) => {
-    return (options.client ?? client_gen_1.client).get({
-        security: [
-            {
-                name: 'X-API-Key',
-                type: 'apiKey'
-            },
-            {
-                scheme: 'bearer',
-                type: 'http'
-            }
-        ],
-        url: '/v1/graphs/{graph_id}/agent/{agent_type}/metadata',
-        ...options
-    });
-};
-exports.getAgentMetadata = getAgentMetadata;
 /**
  * Get agent recommendations
  * Get intelligent agent recommendations for a specific query.
@@ -2226,7 +2243,7 @@ exports.listSubgraphs = listSubgraphs;
  * - 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
  *
@@ -2272,8 +2289,8 @@ exports.createSubgraph = createSubgraph;
  * 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`
  */
 const deleteSubgraph = (options) => {
     return (options.client ?? client_gen_1.client).delete({
@@ -2372,32 +2389,6 @@ const getSubgraphQuota = (options) => {
     });
 };
 exports.getSubgraphQuota = getSubgraphQuota;
-/**
- * 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.
- */
-const cancelSubscription = (options) => {
-    return (options.client ?? client_gen_1.client).delete({
-        security: [
-            {
-                name: 'X-API-Key',
-                type: 'apiKey'
-            },
-            {
-                scheme: 'bearer',
-                type: 'http'
-            }
-        ],
-        url: '/v1/graphs/{graph_id}/subscriptions',
-        ...options
-    });
-};
-exports.cancelSubscription = cancelSubscription;
 /**
  * Get Subscription
  * Get subscription details for a graph or shared repository.
@@ -3430,15 +3421,22 @@ const getOrgBillingCustomer = (options) => {
 };
 exports.getOrgBillingCustomer = getOrgBillingCustomer;
 /**
- * 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)
  */
-const updateOrgPaymentMethod = (options) => {
+const createPortalSession = (options) => {
     return (options.client ?? client_gen_1.client).post({
         security: [
             {
@@ -3450,15 +3448,11 @@ const updateOrgPaymentMethod = (options) => {
                 type: 'http'
             }
         ],
-        url: '/v1/billing/customer/{org_id}/payment-method',
-        ...options,
-        headers: {
-            'Content-Type': 'application/json',
-            ...options.headers
-        }
+        url: '/v1/billing/customer/{org_id}/portal',
+        ...options
     });
 };
-exports.updateOrgPaymentMethod = updateOrgPaymentMethod;
+exports.createPortalSession = createPortalSession;
 /**
  * List Organization Subscriptions
  * List all active and past subscriptions for an organization.

package/sdk.gen.ts

@@ -1,7 +1,7 @@
 // This file is auto-generated by @hey-api/openapi-ts
 
 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';
 import { client as _heyApiClient } from './client.gen';
 
 export type Options<TData extends TDataShape = TDataShape, ThrowOnError extends boolean = boolean> = ClientOptions<TData, ThrowOnError> & {
@@ -1035,9 +1035,38 @@ export const syncConnection = <ThrowOnError extends boolean = false>(options: Op
     });
 };
 
+/**
+ * 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 const listAgents = <ThrowOnError extends boolean = false>(options: Options<ListAgentsData, ThrowOnError>) => {
+    return (options.client ?? _heyApiClient).get<ListAgentsResponses, ListAgentsErrors, ThrowOnError>({
+        security: [
+            {
+                name: 'X-API-Key',
+                type: 'apiKey'
+            },
+            {
+                scheme: 'bearer',
+                type: 'http'
+            }
+        ],
+        url: '/v1/graphs/{graph_id}/agent',
+        ...options
+    });
+};
+
 /**
  * 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:**
  *
@@ -1046,7 +1075,8 @@ export const syncConnection = <ThrowOnError extends boolean = false>(options: Op
  * 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
@@ -1059,6 +1089,14 @@ export const syncConnection = <ThrowOnError extends boolean = false>(options: Op
  * - `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
@@ -1101,15 +1139,54 @@ export const autoSelectAgent = <ThrowOnError extends boolean = false>(options: O
     });
 };
 
+/**
+ * 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 const getAgentMetadata = <ThrowOnError extends boolean = false>(options: Options<GetAgentMetadataData, ThrowOnError>) => {
+    return (options.client ?? _heyApiClient).get<GetAgentMetadataResponses, GetAgentMetadataErrors, ThrowOnError>({
+        security: [
+            {
+                name: 'X-API-Key',
+                type: 'apiKey'
+            },
+            {
+                scheme: 'bearer',
+                type: 'http'
+            }
+        ],
+        url: '/v1/graphs/{graph_id}/agent/{agent_type}',
+        ...options
+    });
+};
+
 /**
  * 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 const executeSpecificAgent = <ThrowOnError extends boolean = false>(options: Options<ExecuteSpecificAgentData, ThrowOnError>) => {
@@ -1171,66 +1248,6 @@ export const batchProcessQueries = <ThrowOnError extends boolean = false>(option
     });
 };
 
-/**
- * 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 const listAgents = <ThrowOnError extends boolean = false>(options: Options<ListAgentsData, ThrowOnError>) => {
-    return (options.client ?? _heyApiClient).get<ListAgentsResponses, ListAgentsErrors, ThrowOnError>({
-        security: [
-            {
-                name: 'X-API-Key',
-                type: 'apiKey'
-            },
-            {
-                scheme: 'bearer',
-                type: 'http'
-            }
-        ],
-        url: '/v1/graphs/{graph_id}/agent/list',
-        ...options
-    });
-};
-
-/**
- * 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 const getAgentMetadata = <ThrowOnError extends boolean = false>(options: Options<GetAgentMetadataData, ThrowOnError>) => {
-    return (options.client ?? _heyApiClient).get<GetAgentMetadataResponses, GetAgentMetadataErrors, ThrowOnError>({
-        security: [
-            {
-                name: 'X-API-Key',
-                type: 'apiKey'
-            },
-            {
-                scheme: 'bearer',
-                type: 'http'
-            }
-        ],
-        url: '/v1/graphs/{graph_id}/agent/{agent_type}/metadata',
-        ...options
-    });
-};
-
 /**
  * Get agent recommendations
  * Get intelligent agent recommendations for a specific query.
@@ -2239,7 +2256,7 @@ export const listSubgraphs = <ThrowOnError extends boolean = false>(options: Opt
  * - 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
  *
@@ -2285,8 +2302,8 @@ export const createSubgraph = <ThrowOnError extends boolean = false>(options: Op
  * 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 const deleteSubgraph = <ThrowOnError extends boolean = false>(options: Options<DeleteSubgraphData, ThrowOnError>) => {
     return (options.client ?? _heyApiClient).delete<DeleteSubgraphResponses, DeleteSubgraphErrors, ThrowOnError>({
@@ -2385,32 +2402,6 @@ export const getSubgraphQuota = <ThrowOnError extends boolean = false>(options:
     });
 };
 
-/**
- * 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 const cancelSubscription = <ThrowOnError extends boolean = false>(options: Options<CancelSubscriptionData, ThrowOnError>) => {
-    return (options.client ?? _heyApiClient).delete<CancelSubscriptionResponses, CancelSubscriptionErrors, ThrowOnError>({
-        security: [
-            {
-                name: 'X-API-Key',
-                type: 'apiKey'
-            },
-            {
-                scheme: 'bearer',
-                type: 'http'
-            }
-        ],
-        url: '/v1/graphs/{graph_id}/subscriptions',
-        ...options
-    });
-};
-
 /**
  * Get Subscription
  * Get subscription details for a graph or shared repository.
@@ -3443,16 +3434,23 @@ export const getOrgBillingCustomer = <ThrowOnError extends boolean = false>(opti
 };
 
 /**
- * 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 const updateOrgPaymentMethod = <ThrowOnError extends boolean = false>(options: Options<UpdateOrgPaymentMethodData, ThrowOnError>) => {
-    return (options.client ?? _heyApiClient).post<UpdateOrgPaymentMethodResponses, UpdateOrgPaymentMethodErrors, ThrowOnError>({
+export const createPortalSession = <ThrowOnError extends boolean = false>(options: Options<CreatePortalSessionData, ThrowOnError>) => {
+    return (options.client ?? _heyApiClient).post<CreatePortalSessionResponses, CreatePortalSessionErrors, ThrowOnError>({
         security: [
             {
                 name: 'X-API-Key',
@@ -3463,12 +3461,8 @@ export const updateOrgPaymentMethod = <ThrowOnError extends boolean = false>(opt
                 type: 'http'
             }
         ],
-        url: '/v1/billing/customer/{org_id}/payment-method',
-        ...options,
-        headers: {
-            'Content-Type': 'application/json',
-            ...options.headers
-        }
+        url: '/v1/billing/customer/{org_id}/portal',
+        ...options
     });
 };