@robosystems/client 0.3.0 → 0.3.2

package/artifacts/graphql/client.ts

@@ -0,0 +1,189 @@
+'use client'
+
+/**
+ * GraphQL client factory used internally by the facade clients.
+ *
+ * This is an implementation detail — consumers never interact with it
+ * directly. They call methods on LedgerClient / InvestorClient / etc.,
+ * which use this to execute GraphQL queries against
+ * `/extensions/{graph_id}/graphql`.
+ *
+ * **Graph scoping.** The GraphQL endpoint is mounted per-graph at
+ * `/extensions/{graph_id}/graphql`. Resolvers read the graph from the URL
+ * path parameter — never from GraphQL variables. Every facade method
+ * takes a `graphId` and resolves the correct per-graph client via
+ * `getGraphQLClient(graphId)`, which caches one `GraphQLClient` instance
+ * per graph to avoid rebuilding headers + URLs on every call.
+ *
+ * We pick graphql-request (tiny, no React, no caching) because the facade
+ * doesn't need the hook ergonomics that urql/Apollo provide. Type safety
+ * comes from GraphQL Code Generator, which produces typed DocumentNodes
+ * from the query files in clients/graphql/queries/.
+ */
+
+import { GraphQLClient } from 'graphql-request'
+
+/**
+ * Callback that returns the current auth credential on demand.
+ *
+ * Use this instead of the static `token` field when the credential
+ * can rotate during the lifetime of the client — the primary case
+ * is short-lived JWTs that auto-refresh from browser token storage.
+ * The provider is invoked on **every** GraphQL request, so the
+ * returned value should be cheap to obtain (a localStorage read or
+ * an in-memory lookup, not a network call if avoidable).
+ *
+ * May return `null`/`undefined` to indicate "no credential available
+ * right now" — the request will be sent unauthenticated in that
+ * case, which lets the caller distinguish auth-expired errors from
+ * never-authed errors at the transport layer.
+ */
+export type TokenProvider = () => string | null | undefined | Promise<string | null | undefined>
+
+export interface GraphQLClientConfig {
+  baseUrl: string
+  /**
+   * Static credential captured at construction time. Use this when
+   * the token won't rotate (e.g. CLI/server flows using a long-lived
+   * API key). For JWT flows prefer `tokenProvider` so refreshes are
+   * picked up without rebuilding the client.
+   */
+  token?: string
+  /**
+   * Dynamic credential callback, read on every request. Wins over
+   * `token` when both are set. See `TokenProvider` for semantics.
+   */
+  tokenProvider?: TokenProvider
+  headers?: Record<string, string>
+  credentials?: 'include' | 'same-origin' | 'omit'
+}
+
+/**
+ * Apply a credential to an in-progress request's headers, choosing
+ * the right header based on token shape:
+ *
+ *   - `rfs…` prefix → `X-API-Key` (long-lived API key, validated
+ *     against the database's api_keys table).
+ *   - anything else → `Authorization: Bearer …` (short-lived JWT,
+ *     validated by JWT middleware).
+ *
+ * The two credential types are NOT interchangeable at the backend —
+ * sending a JWT as `X-API-Key` or an API key as Bearer both 401.
+ */
+function applyAuthHeader(headers: Headers, token: string): void {
+  if (token.startsWith('rfs')) {
+    headers.set('X-API-Key', token)
+  } else {
+    headers.set('Authorization', `Bearer ${token}`)
+  }
+}
+
+/**
+ * Build a new GraphQL client for the given graph. Prefer
+ * `GraphQLClientCache.get(graphId)` in facade code — this is the raw
+ * factory and bypasses the cache.
+ */
+export function createGraphQLClient(config: GraphQLClientConfig, graphId: string): GraphQLClient {
+  if (!graphId) {
+    throw new Error('createGraphQLClient requires a non-empty graphId')
+  }
+  const url = `${config.baseUrl.replace(/\/$/, '')}/extensions/${graphId}/graphql`
+  const staticHeaders: Record<string, string> = {
+    ...(config.headers ?? {}),
+  }
+
+  // Dynamic-token path: defer credential injection to a per-request
+  // middleware so JWT refreshes are picked up without rebuilding or
+  // clearing the client. This is the recommended path for browser
+  // flows where the token in localStorage rotates every ~30 minutes.
+  if (config.tokenProvider) {
+    const providerFn = config.tokenProvider
+    return new GraphQLClient(url, {
+      headers: staticHeaders,
+      credentials: config.credentials,
+      requestMiddleware: async (request) => {
+        let token: string | null | undefined
+        try {
+          token = await providerFn()
+        } catch (err) {
+          // A provider failure shouldn't crash the request — fall
+          // through unauthenticated so the backend returns a clean
+          // 401, which is easier to diagnose than a thrown middleware.
+          // We still log a breadcrumb so the failure is visible in
+          // devtools/log aggregators instead of silently disappearing;
+          // silently swallowing provider bugs in production is worse
+          // than the noise.
+
+          console.warn(
+            '[RoboSystems SDK] tokenProvider threw — sending unauthenticated request:',
+            err
+          )
+          token = undefined
+        }
+        if (!token) {
+          return request
+        }
+        // `request.headers` is a HeadersInit (Headers | string[][] |
+        // Record<string, string>), so normalize via Headers before
+        // mutating. Spreading it directly loses keys when it's a
+        // Headers instance.
+        const merged = new Headers(request.headers as HeadersInit | undefined)
+        applyAuthHeader(merged, token)
+        return { ...request, headers: merged }
+      },
+    })
+  }
+
+  // Static-token path: pick the right header at construction time.
+  // Suitable for long-lived API keys that never rotate.
+  if (config.token) {
+    const headers = new Headers(staticHeaders)
+    applyAuthHeader(headers, config.token)
+    return new GraphQLClient(url, {
+      headers,
+      credentials: config.credentials,
+    })
+  }
+
+  // No credentials at all — used by unauthenticated introspection
+  // queries against public dev endpoints.
+  return new GraphQLClient(url, {
+    headers: staticHeaders,
+    credentials: config.credentials,
+  })
+}
+
+/**
+ * Per-graph cache of GraphQL clients. Facade clients own one instance of
+ * this and call `.get(graphId)` on every GraphQL method — cheap after the
+ * first call per graph.
+ */
+export class GraphQLClientCache {
+  private clients = new Map<string, GraphQLClient>()
+
+  constructor(private config: GraphQLClientConfig) {}
+
+  get(graphId: string): GraphQLClient {
+    const existing = this.clients.get(graphId)
+    if (existing !== undefined) {
+      return existing
+    }
+    const client = createGraphQLClient(this.config, graphId)
+    this.clients.set(graphId, client)
+    return client
+  }
+
+  /**
+   * Drop all cached clients. Only needed when swapping **static**
+   * credentials — e.g. replacing the `token` field on a long-lived
+   * facade between tenants, or resetting a CLI session. When the
+   * cache was built from a `tokenProvider`, rotation is handled
+   * per-request inside `requestMiddleware` and `clear()` is a no-op
+   * for auth purposes (the cached `GraphQLClient` instances keep
+   * using the same provider reference and will pick up the next
+   * token automatically).
+   */
+  clear(): void {
+    this.clients.clear()
+  }
+}

package/{extensions → artifacts}/hooks.js

@@ -7,7 +7,7 @@ exports.useOperation = useOperation;
 exports.useMultipleOperations = useMultipleOperations;
 exports.useSDKClients = useSDKClients;
 /**
- * React hooks for SDK extensions
+ * React hooks for SDK clients
  * Provides easy-to-use hooks for Next.js/React applications
  */
 const react_1 = require("react");
@@ -36,7 +36,7 @@ function useQuery(graphId) {
     const clientRef = (0, react_1.useRef)(null);
     // Initialize client
     (0, react_1.useEffect)(() => {
-        const sdkConfig = (0, config_1.getSDKExtensionsConfig)();
+        const sdkConfig = (0, config_1.getSDKClientConfig)();
         const clientConfig = client_gen_1.client.getConfig();
         // Extract JWT token (uses centralized logic)
         const token = (0, config_1.extractTokenFromSDKClient)();
@@ -115,7 +115,7 @@ function useStreamingQuery(graphId) {
     const [rowsReceived, setRowsReceived] = (0, react_1.useState)(0);
     const clientRef = (0, react_1.useRef)(null);
     (0, react_1.useEffect)(() => {
-        const sdkConfig = (0, config_1.getSDKExtensionsConfig)();
+        const sdkConfig = (0, config_1.getSDKClientConfig)();
         const clientConfig = client_gen_1.client.getConfig();
         clientRef.current = new QueryClient_1.QueryClient({
             baseUrl: sdkConfig.baseUrl || clientConfig.baseUrl || 'http://localhost:8000',
@@ -193,7 +193,7 @@ function useOperation(operationId) {
     const [result, setResult] = (0, react_1.useState)(null);
     const clientRef = (0, react_1.useRef)(null);
     (0, react_1.useEffect)(() => {
-        const sdkConfig = (0, config_1.getSDKExtensionsConfig)();
+        const sdkConfig = (0, config_1.getSDKClientConfig)();
         const clientConfig = client_gen_1.client.getConfig();
         // Extract JWT token (uses centralized logic)
         const token = (0, config_1.extractTokenFromSDKClient)();
@@ -292,7 +292,7 @@ function useMultipleOperations() {
     const [errors, setErrors] = (0, react_1.useState)(new Map());
     const clientRef = (0, react_1.useRef)(null);
     (0, react_1.useEffect)(() => {
-        const sdkConfig = (0, config_1.getSDKExtensionsConfig)();
+        const sdkConfig = (0, config_1.getSDKClientConfig)();
         const clientConfig = client_gen_1.client.getConfig();
         // Extract JWT token (uses centralized logic)
         const token = (0, config_1.extractTokenFromSDKClient)();
@@ -359,7 +359,7 @@ function useSDKClients() {
         operations: null,
     });
     (0, react_1.useEffect)(() => {
-        const sdkConfig = (0, config_1.getSDKExtensionsConfig)();
+        const sdkConfig = (0, config_1.getSDKClientConfig)();
         const clientConfig = client_gen_1.client.getConfig();
         // Extract JWT token (uses centralized logic)
         const token = (0, config_1.extractTokenFromSDKClient)();

package/{extensions → artifacts}/hooks.ts

@@ -1,13 +1,13 @@
 'use client'
 
 /**
- * React hooks for SDK extensions
+ * React hooks for SDK clients
  * Provides easy-to-use hooks for Next.js/React applications
  */
 
 import { useCallback, useEffect, useRef, useState } from 'react'
 import { client } from '../client.gen'
-import { extractTokenFromSDKClient, getSDKExtensionsConfig } from './config'
+import { extractTokenFromSDKClient, getSDKClientConfig } from './config'
 import type { OperationProgress, OperationResult } from './OperationClient'
 import { OperationClient } from './OperationClient'
 import type { QueryOptions, QueryResult } from './QueryClient'
@@ -35,7 +35,7 @@ export function useQuery(graphId: string) {
 
   // Initialize client
   useEffect(() => {
-    const sdkConfig = getSDKExtensionsConfig()
+    const sdkConfig = getSDKClientConfig()
     const clientConfig = client.getConfig()
 
     // Extract JWT token (uses centralized logic)
@@ -136,7 +136,7 @@ export function useStreamingQuery(graphId: string) {
   const clientRef = useRef<QueryClient>(null)
 
   useEffect(() => {
-    const sdkConfig = getSDKExtensionsConfig()
+    const sdkConfig = getSDKClientConfig()
     const clientConfig = client.getConfig()
     clientRef.current = new QueryClient({
       baseUrl: sdkConfig.baseUrl || clientConfig.baseUrl || 'http://localhost:8000',
@@ -230,7 +230,7 @@ export function useOperation<T = any>(operationId?: string) {
   const clientRef = useRef<OperationClient>(null)
 
   useEffect(() => {
-    const sdkConfig = getSDKExtensionsConfig()
+    const sdkConfig = getSDKClientConfig()
     const clientConfig = client.getConfig()
 
     // Extract JWT token (uses centralized logic)
@@ -343,7 +343,7 @@ export function useMultipleOperations<T = any>() {
   const clientRef = useRef<OperationClient>(null)
 
   useEffect(() => {
-    const sdkConfig = getSDKExtensionsConfig()
+    const sdkConfig = getSDKClientConfig()
     const clientConfig = client.getConfig()
 
     // Extract JWT token (uses centralized logic)
@@ -428,7 +428,7 @@ export function useSDKClients() {
   })
 
   useEffect(() => {
-    const sdkConfig = getSDKExtensionsConfig()
+    const sdkConfig = getSDKClientConfig()
     const clientConfig = client.getConfig()
 
     // Extract JWT token (uses centralized logic)

package/{extensions → artifacts}/index.d.ts

@@ -1,31 +1,47 @@
 /**
- * RoboSystems SDK Extensions
- * Enhanced clients with SSE support for the RoboSystems API
+ * RoboSystems SDK Clients
+ * High-level clients with SSE support for the RoboSystems API
  */
-import { OperationClient } from './OperationClient';
-import { QueryClient } from './QueryClient';
 import { AgentClient } from './AgentClient';
-import { SSEClient } from './SSEClient';
+import type { TokenProvider } from './graphql/client';
 import { InvestorClient } from './InvestorClient';
 import { LedgerClient } from './LedgerClient';
-import { ReportClient } from './ReportClient';
-export interface RoboSystemsExtensionConfig {
+import { OperationClient } from './OperationClient';
+import { QueryClient } from './QueryClient';
+import { SSEClient } from './SSEClient';
+export type { TokenProvider } from './graphql/client';
+export interface RoboSystemsClientConfig {
     baseUrl?: string;
     credentials?: 'include' | 'same-origin' | 'omit';
+    /**
+     * Static credential captured at construction time. Fine for
+     * long-lived API keys; use `tokenProvider` instead when the JWT
+     * can rotate (browser login flows).
+     */
     token?: string;
+    /**
+     * Dynamic credential callback invoked on every GraphQL request.
+     * When set, JWT refreshes are picked up automatically — no need
+     * to rebuild or clear cached clients after a refresh.
+     */
+    tokenProvider?: TokenProvider;
     headers?: Record<string, string>;
     maxRetries?: number;
     retryDelay?: number;
 }
-export declare class RoboSystemsExtensions {
+export declare class RoboSystemsClients {
     readonly query: QueryClient;
     readonly agent: AgentClient;
     readonly operations: OperationClient;
     readonly ledger: LedgerClient;
     readonly investor: InvestorClient;
-    readonly reports: ReportClient;
+    /**
+     * @deprecated Use `ledger` instead — reports and publish lists are
+     * now part of the LedgerClient.
+     */
+    readonly reports: LedgerClient;
     private config;
-    constructor(config?: RoboSystemsExtensionConfig);
+    constructor(config?: RoboSystemsClientConfig);
     /**
      * Convenience method to monitor any operation
      */
@@ -39,23 +55,23 @@ export declare class RoboSystemsExtensions {
      */
     close(): void;
 }
+export * from './AgentClient';
+export * from './config';
+export * from './InvestorClient';
+export * from './LedgerClient';
 export * from './OperationClient';
 export * from './QueryClient';
-export * from './AgentClient';
 export * from './SSEClient';
-export * from './LedgerClient';
-export * from './InvestorClient';
-export * from './ReportClient';
-export * from './config';
-export { OperationClient, QueryClient, AgentClient, SSEClient, LedgerClient, InvestorClient, ReportClient, };
+export { AgentClient, InvestorClient, LedgerClient, OperationClient, QueryClient, SSEClient };
 export { useMultipleOperations, useOperation, useQuery, useSDKClients, useStreamingQuery, } from './hooks';
-export declare const extensions: {
+export declare const clients: {
     readonly query: QueryClient;
     readonly agent: AgentClient;
     readonly operations: OperationClient;
     readonly ledger: LedgerClient;
     readonly investor: InvestorClient;
-    readonly reports: ReportClient;
+    /** @deprecated Use `ledger` instead */
+    readonly reports: LedgerClient;
     monitorOperation: (operationId: string, onProgress?: (progress: any) => void) => Promise<any>;
     createSSEClient: () => SSEClient;
     close: () => void;

package/{extensions → artifacts}/index.js

@@ -1,7 +1,7 @@
 "use strict";
 /**
- * RoboSystems SDK Extensions
- * Enhanced clients with SSE support for the RoboSystems API
+ * RoboSystems SDK Clients
+ * High-level clients with SSE support for the RoboSystems API
  */
 var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
     if (k2 === undefined) k2 = k;
@@ -18,33 +18,36 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.analyzeFinancials = exports.agentQuery = exports.streamQuery = exports.executeQuery = exports.monitorOperation = exports.extensions = exports.useStreamingQuery = exports.useSDKClients = exports.useQuery = exports.useOperation = exports.useMultipleOperations = exports.ReportClient = exports.InvestorClient = exports.LedgerClient = exports.SSEClient = exports.AgentClient = exports.QueryClient = exports.OperationClient = exports.RoboSystemsExtensions = void 0;
+exports.analyzeFinancials = exports.agentQuery = exports.streamQuery = exports.executeQuery = exports.monitorOperation = exports.clients = exports.useStreamingQuery = exports.useSDKClients = exports.useQuery = exports.useOperation = exports.useMultipleOperations = exports.SSEClient = exports.QueryClient = exports.OperationClient = exports.LedgerClient = exports.InvestorClient = exports.AgentClient = exports.RoboSystemsClients = void 0;
 const client_gen_1 = require("../client.gen");
+const AgentClient_1 = require("./AgentClient");
+Object.defineProperty(exports, "AgentClient", { enumerable: true, get: function () { return AgentClient_1.AgentClient; } });
 const config_1 = require("./config");
+const InvestorClient_1 = require("./InvestorClient");
+Object.defineProperty(exports, "InvestorClient", { enumerable: true, get: function () { return InvestorClient_1.InvestorClient; } });
+const LedgerClient_1 = require("./LedgerClient");
+Object.defineProperty(exports, "LedgerClient", { enumerable: true, get: function () { return LedgerClient_1.LedgerClient; } });
 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 InvestorClient_1 = require("./InvestorClient");
-Object.defineProperty(exports, "InvestorClient", { enumerable: true, get: function () { return InvestorClient_1.InvestorClient; } });
-const LedgerClient_1 = require("./LedgerClient");
-Object.defineProperty(exports, "LedgerClient", { enumerable: true, get: function () { return LedgerClient_1.LedgerClient; } });
-const ReportClient_1 = require("./ReportClient");
-Object.defineProperty(exports, "ReportClient", { enumerable: true, get: function () { return ReportClient_1.ReportClient; } });
-class RoboSystemsExtensions {
+class RoboSystemsClients {
     constructor(config = {}) {
         // Get base URL from SDK client config or use provided/default
         const sdkConfig = client_gen_1.client.getConfig();
         // Extract JWT token using centralized logic
         const token = config.token || (0, config_1.extractTokenFromSDKClient)();
+        // `tokenProvider` falls back to the global SDK clients config so
+        // apps can wire refresh once via `setSDKClientConfig({ tokenProvider })`
+        // and have the lazy default singleton pick it up automatically.
+        const tokenProvider = config.tokenProvider || (0, config_1.getSDKClientConfig)().tokenProvider;
         this.config = {
             baseUrl: config.baseUrl || sdkConfig.baseUrl || 'http://localhost:8000',
             credentials: config.credentials || 'include',
             token,
+            tokenProvider,
             headers: config.headers,
             maxRetries: config.maxRetries || 5,
             retryDelay: config.retryDelay || 1000,
@@ -68,24 +71,24 @@ class RoboSystemsExtensions {
             maxRetries: this.config.maxRetries,
             retryDelay: this.config.retryDelay,
         });
+        // LedgerClient / InvestorClient use GraphQL internally, so they
+        // get the tokenProvider — REST-only clients do not.
         this.ledger = new LedgerClient_1.LedgerClient({
             baseUrl: this.config.baseUrl,
             credentials: this.config.credentials,
             token: this.config.token,
+            tokenProvider: this.config.tokenProvider,
             headers: this.config.headers,
         });
         this.investor = new InvestorClient_1.InvestorClient({
             baseUrl: this.config.baseUrl,
             credentials: this.config.credentials,
             token: this.config.token,
+            tokenProvider: this.config.tokenProvider,
             headers: this.config.headers,
         });
-        this.reports = new ReportClient_1.ReportClient({
-            baseUrl: this.config.baseUrl,
-            credentials: this.config.credentials,
-            token: this.config.token,
-            headers: this.config.headers,
-        });
+        // Reports consolidated into LedgerClient — alias for backward compat
+        this.reports = this.ledger;
     }
     /**
      * Convenience method to monitor any operation
@@ -115,16 +118,15 @@ class RoboSystemsExtensions {
         this.operations.closeAll();
     }
 }
-exports.RoboSystemsExtensions = RoboSystemsExtensions;
+exports.RoboSystemsClients = RoboSystemsClients;
 // Export all types and classes
+__exportStar(require("./AgentClient"), exports);
+__exportStar(require("./config"), exports);
+__exportStar(require("./InvestorClient"), exports);
+__exportStar(require("./LedgerClient"), exports);
 __exportStar(require("./OperationClient"), exports);
 __exportStar(require("./QueryClient"), exports);
-__exportStar(require("./AgentClient"), exports);
 __exportStar(require("./SSEClient"), exports);
-__exportStar(require("./LedgerClient"), exports);
-__exportStar(require("./InvestorClient"), exports);
-__exportStar(require("./ReportClient"), 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; } });
@@ -133,44 +135,45 @@ Object.defineProperty(exports, "useQuery", { enumerable: true, get: function ()
 Object.defineProperty(exports, "useSDKClients", { enumerable: true, get: function () { return hooks_1.useSDKClients; } });
 Object.defineProperty(exports, "useStreamingQuery", { enumerable: true, get: function () { return hooks_1.useStreamingQuery; } });
 // Lazy initialization of default instance
-let _extensions = null;
-function getExtensions() {
-    if (!_extensions) {
-        _extensions = new RoboSystemsExtensions();
+let _clients = null;
+function getClients() {
+    if (!_clients) {
+        _clients = new RoboSystemsClients();
     }
-    return _extensions;
+    return _clients;
 }
-exports.extensions = {
+exports.clients = {
     get query() {
-        return getExtensions().query;
+        return getClients().query;
     },
     get agent() {
-        return getExtensions().agent;
+        return getClients().agent;
     },
     get operations() {
-        return getExtensions().operations;
+        return getClients().operations;
     },
     get ledger() {
-        return getExtensions().ledger;
+        return getClients().ledger;
     },
     get investor() {
-        return getExtensions().investor;
+        return getClients().investor;
     },
+    /** @deprecated Use `ledger` instead */
     get reports() {
-        return getExtensions().reports;
+        return getClients().ledger;
     },
-    monitorOperation: (operationId, onProgress) => getExtensions().monitorOperation(operationId, onProgress),
-    createSSEClient: () => getExtensions().createSSEClient(),
-    close: () => getExtensions().close(),
+    monitorOperation: (operationId, onProgress) => getClients().monitorOperation(operationId, onProgress),
+    createSSEClient: () => getClients().createSSEClient(),
+    close: () => getClients().close(),
 };
 // Export convenience functions that use the default instance
-const monitorOperation = (operationId, onProgress) => getExtensions().monitorOperation(operationId, onProgress);
+const monitorOperation = (operationId, onProgress) => getClients().monitorOperation(operationId, onProgress);
 exports.monitorOperation = monitorOperation;
-const executeQuery = (graphId, query, parameters) => getExtensions().query.query(graphId, query, parameters);
+const executeQuery = (graphId, query, parameters) => getClients().query.query(graphId, query, parameters);
 exports.executeQuery = executeQuery;
-const streamQuery = (graphId, query, parameters, chunkSize) => getExtensions().query.streamQuery(graphId, query, parameters, chunkSize);
+const streamQuery = (graphId, query, parameters, chunkSize) => getClients().query.streamQuery(graphId, query, parameters, chunkSize);
 exports.streamQuery = streamQuery;
-const agentQuery = (graphId, message, context) => getExtensions().agent.query(graphId, message, context);
+const agentQuery = (graphId, message, context) => getClients().agent.query(graphId, message, context);
 exports.agentQuery = agentQuery;
-const analyzeFinancials = (graphId, message) => getExtensions().agent.analyzeFinancials(graphId, message);
+const analyzeFinancials = (graphId, message) => getClients().agent.analyzeFinancials(graphId, message);
 exports.analyzeFinancials = analyzeFinancials;