@robosystems/core 0.11.2 → 0.11.3

package/auth-core/token-storage.js

@@ -7,6 +7,24 @@
 const TOKEN_KEY = 'robosystems_jwt_token';
 const TOKEN_EXPIRY_KEY = 'robosystems_jwt_expiry';
 const TOKEN_THRESHOLD_KEY = 'robosystems_jwt_threshold';
+/**
+ * How long past its local expiry a token is kept. The API renews a
+ * recently-expired token at `/v1/auth/refresh` within this window, so the
+ * token must survive locally until then — for the refresh call only. Data
+ * calls stop sending it at expiry.
+ */
+export const TOKEN_REFRESH_GRACE_MS = 5 * 60 * 1000;
+function readExpiry() {
+    const expiryStr = localStorage.getItem(TOKEN_EXPIRY_KEY);
+    if (!expiryStr)
+        return null;
+    const expiry = parseInt(expiryStr, 10);
+    return Number.isFinite(expiry) ? expiry : null;
+}
+/** Past expiry *and* the refresh grace: nothing can use it any more. */
+function isPastGrace(expiry) {
+    return expiry !== null && Date.now() > expiry + TOKEN_REFRESH_GRACE_MS;
+}
 /**
  * Store JWT token in localStorage
  * @param token - The JWT token from login response
@@ -38,16 +56,15 @@ export function getToken() {
         return null;
     try {
         const token = localStorage.getItem(TOKEN_KEY);
-        const expiryStr = localStorage.getItem(TOKEN_EXPIRY_KEY);
         if (!token)
             return null;
-        // Check if token is expired
-        if (expiryStr) {
-            const expiry = parseInt(expiryStr, 10);
-            if (Date.now() > expiry) {
+        // An expired token is not sent on data calls, but it is kept until the
+        // refresh grace runs out so `getRefreshToken` can still renew it.
+        const expiry = readExpiry();
+        if (expiry !== null && Date.now() > expiry) {
+            if (isPastGrace(expiry))
                 clearToken();
-                return null;
-            }
+            return null;
         }
         return token;
     }
@@ -152,20 +169,29 @@ export function handleAuthResponse(response) {
  * @returns Valid token or null if expired
  */
 export async function getValidToken() {
-    const token = getToken();
-    if (!token)
+    return getToken();
+}
+/**
+ * The token to present to `/v1/auth/refresh`: the stored token while it is
+ * unexpired or within the server's refresh grace after expiry, else null.
+ */
+export function getRefreshToken() {
+    if (typeof window === 'undefined')
         return null;
-    // Just check if token is expired - let backend handle refresh
-    const expiryStr = localStorage.getItem(TOKEN_EXPIRY_KEY);
-    if (expiryStr) {
-        const expiry = parseInt(expiryStr, 10);
-        if (Date.now() > expiry) {
-            // Token is expired, clear it and let AuthProvider handle re-auth
+    try {
+        const token = localStorage.getItem(TOKEN_KEY);
+        if (!token)
+            return null;
+        if (isPastGrace(readExpiry())) {
             clearToken();
             return null;
         }
+        return token;
+    }
+    catch (error) {
+        console.error('[TokenStorage] Failed to retrieve refresh token:', error);
+        return null;
     }
-    return token;
 }
 /**
  * Security event logging (for monitoring)

package/auth-core/types.d.ts

@@ -104,6 +104,7 @@ export interface AuthContextType {
     register: (email: string, password: string, name?: string) => Promise<AuthUser>;
     logout: (reason?: string, options?: {
         redirectTo?: string;
+        skipServerLogout?: boolean;
     }) => Promise<void>;
     refreshUser: () => Promise<AuthUser | null>;
     refreshSession: (force?: boolean) => Promise<void>;

package/components/console/ConsoleContent.js

@@ -61,6 +61,14 @@ export function ConsoleContent({ config }) {
     const { state: graphState } = useGraphContext();
     const graphId = graphState.currentGraphId;
     const streamingQuery = useStreamingQuery();
+    // The graph the console is on right now, readable after an await: an
+    // answer for a graph the user has since left must not be printed (or
+    // recorded as history) under the new graph.
+    const currentGraphRef = useRef(graphId);
+    currentGraphRef.current = graphId;
+    // Moves on at every graph switch, so a switch away and back (A → B → A)
+    // still retires a request made before it.
+    const graphEpochRef = useRef(0);
     // Terminal state
     const [terminalMessages, setTerminalMessages] = useState([]);
     const [commandInput, setCommandInput] = useState('');
@@ -190,16 +198,21 @@ export function ConsoleContent({ config }) {
         if (!graphId)
             return;
         if (previousGraphId.current && previousGraphId.current !== graphId) {
-            if (streamingQuery.isStreaming) {
-                streamingQuery.cancelQuery();
-            }
+            const fromGraphId = previousGraphId.current;
+            previousGraphId.current = graphId;
+            graphEpochRef.current++;
+            // Stop any stream still running for the previous graph; its rows must
+            // not land under the new graph's banner.
+            streamingQuery.cancelQuery();
+            operatorProgressMessageId.current = null;
+            setOperatorProgress({ isRunning: false, message: '' });
             setTerminalMessages([]);
             // A follow-up on the new graph must not resolve against the previous
             // graph's answers.
             conversationRef.current = [];
             setCurrentQueryStartTime(null);
             addSystemMessage(`═══════════════════════════════════════════════════════════════\n` +
-                `${config.welcome.contextLabel} context changed: ${previousGraphId.current} → ${graphId}\n` +
+                `${config.welcome.contextLabel} context changed: ${fromGraphId} → ${graphId}\n` +
                 `═══════════════════════════════════════════════════════════════\n\n` +
                 `Console has been reset for the new ${config.welcome.contextLabel.toLowerCase()} context.\n` +
                 `All queries will now execute against: ${graphId}\n\n` +
@@ -322,6 +335,10 @@ export function ConsoleContent({ config }) {
             return;
         }
         const startTime = Date.now();
+        const requestGraphId = graphId;
+        const requestEpoch = graphEpochRef.current;
+        const isStale = () => currentGraphRef.current !== requestGraphId ||
+            graphEpochRef.current !== requestEpoch;
         try {
             const { clients } = await import('@robosystems/client/clients');
             const history = conversationRef.current;
@@ -331,6 +348,8 @@ export function ConsoleContent({ config }) {
                 mode: 'standard' }, (history.length > 0 ? { history } : {})), {
                 mode: 'auto',
                 onProgress: (message, percentage) => {
+                    if (isStale())
+                        return;
                     setOperatorProgress({
                         isRunning: true,
                         message,
@@ -338,6 +357,11 @@ export function ConsoleContent({ config }) {
                     });
                 },
             });
+            // The user switched graphs while the operator ran: the reset already
+            // cleared the console, so drop the answer rather than print it (and
+            // record it as history) under the new graph.
+            if (isStale())
+                return;
             setOperatorProgress({ isRunning: false, message: '' });
             // The API reports operator failures (credit pre-flight, timeouts,
             // unbilled calls) as HTTP 200 with `error_details` set and an
@@ -419,6 +443,8 @@ export function ConsoleContent({ config }) {
             }
         }
         catch (error) {
+            if (isStale())
+                return;
             setOperatorProgress({ isRunning: false, message: '' });
             const errorMessage = error.message ||
                 ((_a = error.data) === null || _a === void 0 ? void 0 : _a.detail) ||

package/components/console/graphAwareConfig.js

@@ -206,9 +206,11 @@ LIMIT 20`,
 };
 // ── RoboInvestor entity graph (portfolios / positions / securities) ──────────
 // Data: Entity -> Portfolio -> Position -> Security, and Entity -> Security
-// (issuance). Position carries cost_basis and current_value, so gains and
-// allocation are computed in Cypher. Trade/Benchmark/MarketData exist in the
-// schema but aren't populated yet — keep examples off them.
+// (issuance). Position carries cost_basis, current_value and currency, so
+// gains and allocation are computed in Cypher. Sums count only marked
+// positions (current_value set) and never add across currencies.
+// Security.ticker, Trade, Benchmark and MarketData exist in the schema but
+// aren't populated yet — keep examples off them.
 const ROBOINVESTOR_EXAMPLE_SET = {
     subtitle: 'AI analyst for your investment portfolios',
     description: 'Ask in plain English and an AI analyst queries your portfolios, positions, and securities — holdings, cost basis vs. current value, allocation, and unrealized gains. Every answer comes back with the data and the Cypher behind it.',
@@ -221,47 +223,55 @@ const ROBOINVESTOR_EXAMPLE_SET = {
     directQueryExamples: [
         'MATCH (p:Portfolio) RETURN p.name, p.strategy, p.base_currency',
         'MATCH (p:Portfolio)-[:PORTFOLIO_HAS_POSITION]->(pos:Position) RETURN p.name, count(pos) AS positions ORDER BY positions DESC',
-        'MATCH (s:Security) RETURN s.ticker, s.name, s.security_type LIMIT 20',
+        'MATCH (s:Security) RETURN s.name, s.security_type, s.security_subtype LIMIT 20',
     ],
     sampleQueries: [
         {
             name: 'Portfolio value',
             query: `MATCH (p:Portfolio)-[:PORTFOLIO_HAS_POSITION]->(pos:Position)
+WHERE pos.current_value IS NOT NULL
 RETURN p.name,
-       count(pos) AS positions,
-       sum(pos.cost_basis) AS total_cost,
-       sum(pos.current_value) AS total_value
-ORDER BY total_value DESC`,
+       pos.currency AS currency,
+       count(pos) AS marked_positions,
+       sum(pos.current_value) AS total_value,
+       count(pos.cost_basis) AS positions_with_cost,
+       sum(pos.cost_basis) AS total_cost
+ORDER BY p.name, currency`,
         },
         {
             name: 'Holdings by current value',
             query: `MATCH (p:Portfolio)-[:PORTFOLIO_HAS_POSITION]->(pos:Position)-[:POSITION_IN_SECURITY]->(s:Security)
-RETURN p.name, s.ticker, s.name, pos.quantity, pos.cost_basis, pos.current_value
-ORDER BY pos.current_value DESC
+WHERE pos.current_value IS NOT NULL
+RETURN p.name, s.name, pos.quantity, pos.currency AS currency,
+       pos.cost_basis, pos.current_value
+ORDER BY currency, pos.current_value DESC
 LIMIT 25`,
         },
         {
             name: 'Unrealized gains',
             query: `MATCH (p:Portfolio)-[:PORTFOLIO_HAS_POSITION]->(pos:Position)-[:POSITION_IN_SECURITY]->(s:Security)
 WHERE pos.cost_basis IS NOT NULL AND pos.current_value IS NOT NULL
-RETURN s.ticker, s.name, pos.cost_basis, pos.current_value,
+RETURN p.name, s.name, pos.currency AS currency,
+       pos.cost_basis, pos.current_value,
        pos.current_value - pos.cost_basis AS unrealized_gain
-ORDER BY unrealized_gain DESC
+ORDER BY currency, unrealized_gain DESC
 LIMIT 20`,
         },
         {
             name: 'Allocation by security type',
             query: `MATCH (pos:Position)-[:POSITION_IN_SECURITY]->(s:Security)
-RETURN s.security_type,
-       count(pos) AS positions,
+WHERE pos.current_value IS NOT NULL
+RETURN pos.currency AS currency,
+       s.security_type,
+       count(pos) AS marked_positions,
        sum(pos.current_value) AS total_value
-ORDER BY total_value DESC`,
+ORDER BY currency, total_value DESC`,
         },
         {
             name: 'Securities issued by entity',
             query: `MATCH (e:Entity)-[:ENTITY_ISSUES_SECURITY]->(s:Security)
-RETURN e.name, s.ticker, s.name, s.security_type
-ORDER BY s.ticker
+RETURN e.name, s.name, s.security_type, s.security_subtype
+ORDER BY e.name, s.name
 LIMIT 25`,
         },
     ],

package/components/repositories/ActiveSubscriptions.js

@@ -1,5 +1,5 @@
 'use client';
-import { jsx as _jsx, Fragment as _Fragment, jsxs as _jsxs } from "react/jsx-runtime";
+import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
 import * as SDK from '@robosystems/client';
 import { Badge, Button, Card, Spinner } from 'flowbite-react';
 import { useCallback, useEffect, useState } from 'react';
@@ -9,9 +9,13 @@ import { useGraphContext } from '../../contexts/graph-context';
 import { useOrg } from '../../contexts/org-context';
 import { useServiceOfferings } from '../../contexts/service-offerings-context';
 import { useToast } from '../../hooks/use-toast';
+import { unwrapSdk } from '../../lib/sdk-errors';
 export function ActiveSubscriptions({ onOpenConsole, onOpenUsage, onGettingStarted, onBackups, onBilling, onBrowse, onCancel, emptyState, }) {
     const [userSubscriptions, setUserSubscriptions] = useState([]);
     const [loading, setLoading] = useState(true);
+    // A failed subscription read must not render as "no subscriptions": that
+    // offers Subscribe on repositories the org already holds.
+    const [loadError, setLoadError] = useState(null);
     const { showError, ToastContainer } = useToast();
     const { currentOrg } = useOrg();
     const { offerings, isLoading: offeringsLoading } = useServiceOfferings();
@@ -23,11 +27,12 @@ export function ActiveSubscriptions({ onOpenConsole, onOpenUsage, onGettingStart
         }
         try {
             setLoading(true);
-            const subscriptionsResponse = await SDK.listOrgSubscriptions({
+            setLoadError(null);
+            const subscriptions = unwrapSdk(await SDK.listOrgSubscriptions({
                 path: { org_id: currentOrg.id },
-            });
-            if (subscriptionsResponse.data) {
-                const repositorySubscriptions = (subscriptionsResponse.data || []).filter((sub) => sub.resource_type === 'repository');
+            }));
+            if (subscriptions) {
+                const repositorySubscriptions = (subscriptions || []).filter((sub) => sub.resource_type === 'repository');
                 setUserSubscriptions(repositorySubscriptions);
             }
             else {
@@ -36,7 +41,11 @@ export function ActiveSubscriptions({ onOpenConsole, onOpenUsage, onGettingStart
         }
         catch (error) {
             console.error('Failed to load user subscriptions:', error);
-            showError('Failed to load user subscriptions');
+            const message = error instanceof Error && error.message
+                ? `Failed to load subscriptions: ${error.message}`
+                : 'Failed to load subscriptions';
+            setLoadError(message);
+            showError(message);
         }
         finally {
             setLoading(false);
@@ -48,6 +57,9 @@ export function ActiveSubscriptions({ onOpenConsole, onOpenUsage, onGettingStart
     if (loading || offeringsLoading) {
         return (_jsx("div", { className: "flex min-h-[400px] items-center justify-center", children: _jsx(Spinner, { size: "xl" }) }));
     }
+    if (loadError) {
+        return (_jsxs("div", { className: "space-y-6", children: [_jsx(ToastContainer, {}), _jsx(Card, { children: _jsxs("div", { className: "flex flex-col items-center gap-4 py-8 text-center", children: [_jsx("p", { className: "text-zinc-700 dark:text-zinc-300", children: loadError }), _jsx(Button, { color: "gray", onClick: () => loadData(), children: "Try again" })] }) })] }));
+    }
     // Treat a period-end-canceled sub as "still active" until its access
     // window actually closes. Otherwise the row would disappear from this
     // view the moment the user clicks Cancel — even though the modal

package/components/repositories/BrowseRepositories.js

@@ -9,10 +9,14 @@ import { useGraphContext } from '../../contexts/graph-context';
 import { useOrg } from '../../contexts/org-context';
 import { useServiceOfferings } from '../../contexts/service-offerings-context';
 import { useToast } from '../../hooks/use-toast';
+import { unwrapSdk } from '../../lib/sdk-errors';
 import { useRepositorySubscription } from '../../task-monitoring/operationHooks';
 export function BrowseRepositories({ onSubscribed, onViewSubscriptions, showHeader = true, showInfoCard = true, }) {
     const [userSubscriptions, setUserSubscriptions] = useState([]);
     const [loading, setLoading] = useState(true);
+    // A failed subscription read must not render as "no subscriptions": that
+    // offers Subscribe on repositories the org already holds.
+    const [loadError, setLoadError] = useState(null);
     const [subscribing, setSubscribing] = useState(null);
     const [changingPlan, setChangingPlan] = useState(false);
     const [planChangeModal, setPlanChangeModal] = useState(null);
@@ -28,12 +32,13 @@ export function BrowseRepositories({ onSubscribed, onViewSubscriptions, showHead
         }
         try {
             setLoading(true);
+            setLoadError(null);
             // Load org subscriptions and filter for repositories
-            const subscriptionsResponse = await SDK.listOrgSubscriptions({
+            const subscriptions = unwrapSdk(await SDK.listOrgSubscriptions({
                 path: { org_id: currentOrg.id },
-            });
-            if (subscriptionsResponse.data) {
-                const repositorySubscriptions = (subscriptionsResponse.data || []).filter((sub) => sub.resource_type === 'repository');
+            }));
+            if (subscriptions) {
+                const repositorySubscriptions = (subscriptions || []).filter((sub) => sub.resource_type === 'repository');
                 setUserSubscriptions(repositorySubscriptions);
             }
             else {
@@ -42,7 +47,11 @@ export function BrowseRepositories({ onSubscribed, onViewSubscriptions, showHead
         }
         catch (error) {
             console.error('Failed to load user subscriptions:', error);
-            showError('Failed to load user subscriptions');
+            const message = error instanceof Error && error.message
+                ? `Failed to load subscriptions: ${error.message}`
+                : 'Failed to load subscriptions';
+            setLoadError(message);
+            showError(message);
         }
         finally {
             setLoading(false);
@@ -131,6 +140,9 @@ export function BrowseRepositories({ onSubscribed, onViewSubscriptions, showHead
     if (loading || offeringsLoading) {
         return (_jsx("div", { className: "flex min-h-[400px] items-center justify-center", children: _jsx(Spinner, { size: "xl" }) }));
     }
+    if (loadError) {
+        return (_jsxs("div", { className: "space-y-6", children: [_jsx(ToastContainer, {}), _jsx(Card, { children: _jsxs("div", { className: "flex flex-col items-center gap-4 py-8 text-center", children: [_jsx("p", { className: "text-zinc-700 dark:text-zinc-300", children: loadError }), _jsx(Button, { color: "gray", onClick: () => loadData(), children: "Try again" })] }) })] }));
+    }
     // Check if user has any active subscriptions
     const hasActiveSubscriptions = userSubscriptions.some((sub) => sub.status === 'active');
     return (_jsxs("div", { className: "space-y-6", children: [_jsx(ToastContainer, {}), showHeader && (_jsxs("div", { className: "flex flex-col gap-4 sm:flex-row sm:items-start sm:justify-between", children: [_jsxs("div", { className: "flex min-w-0 items-center gap-3 sm:gap-4", children: [_jsx("div", { className: "from-primary-500 to-secondary-600 shrink-0 rounded-lg bg-gradient-to-br p-2.5 sm:p-3", children: _jsx(HiGlobeAlt, { className: "h-6 w-6 text-white sm:h-8 sm:w-8" }) }), _jsxs("div", { className: "min-w-0", children: [_jsx("h1", { className: "font-heading text-2xl font-bold break-words text-zinc-900 sm:text-3xl dark:text-zinc-100", children: "Browse Repositories" }), _jsx("p", { className: "mt-1 text-sm text-zinc-500 dark:text-zinc-400", children: "Subscribe to read-only graph databases containing curated data" })] })] }), hasActiveSubscriptions && onViewSubscriptions && (_jsx(Button, { onClick: onViewSubscriptions, color: "gray", children: "View My Subscriptions" }))] })), showInfoCard && (_jsx(Card, { children: _jsxs("div", { className: "flex items-start gap-4", children: [_jsx("div", { className: "bg-primary-100 dark:bg-primary-900/30 rounded-lg p-3", children: _jsx(HiInformationCircle, { className: "text-primary-600 dark:text-primary-400 h-6 w-6" }) }), _jsxs("div", { className: "flex-1", children: [_jsx("h3", { className: "font-heading mb-3 text-lg font-semibold text-zinc-900 dark:text-white", children: "How Shared Repositories Work" }), _jsxs("div", { className: "grid gap-3 sm:grid-cols-2 lg:grid-cols-4", children: [_jsxs("div", { className: "flex items-start gap-2", children: [_jsx("div", { className: "bg-primary-500 mt-1 h-2 w-2 rounded-full" }), _jsx("span", { className: "text-sm text-zinc-600 dark:text-zinc-400", children: "Subscribe to gain read-only access to curated graph databases" })] }), _jsxs("div", { className: "flex items-start gap-2", children: [_jsx("div", { className: "bg-primary-500 mt-1 h-2 w-2 rounded-full" }), _jsx("span", { className: "text-sm text-zinc-600 dark:text-zinc-400", children: "Credits are used for AI agent calls - queries are unlimited" })] }), _jsxs("div", { className: "flex items-start gap-2", children: [_jsx("div", { className: "bg-primary-500 mt-1 h-2 w-2 rounded-full" }), _jsx("span", { className: "text-sm text-zinc-600 dark:text-zinc-400", children: "Generate API keys to access repositories via MCP tools in AI apps" })] }), _jsxs("div", { className: "flex items-start gap-2", children: [_jsx("div", { className: "bg-primary-500 mt-1 h-2 w-2 rounded-full" }), _jsx("span", { className: "text-sm text-zinc-600 dark:text-zinc-400", children: "Different tiers offer varying credit limits and features" })] })] })] })] }) })), (offerings === null || offerings === void 0 ? void 0 : offerings.repositoryPlans) &&

package/components/search/SearchContent.js

@@ -2,7 +2,7 @@
 import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
 import * as SDK from '@robosystems/client';
 import { Alert, Card, Select, Spinner, TextInput, ToggleSwitch, } from 'flowbite-react';
-import { useCallback, useEffect, useState } from 'react';
+import { useCallback, useEffect, useRef, useState } from 'react';
 import { HiChevronDown, HiChevronUp, HiSearch } from 'react-icons/hi';
 import { useIsRepository } from '../../components/RepositoryGuard';
 import { useGraphContext } from '../../contexts';
@@ -38,6 +38,10 @@ export function SearchContent({ config }) {
     const [expandedDocId, setExpandedDocId] = useState(null);
     const [sectionContent, setSectionContent] = useState(null);
     const [sectionLoading, setSectionLoading] = useState(false);
+    // Request counters: a response is applied only if no newer search (or
+    // section read), and no graph switch, has happened since it was sent.
+    const searchSeq = useRef(0);
+    const sectionSeq = useRef(0);
     // Filters visibility
     const [showFilters, setShowFilters] = useState((_b = config.showFilters) !== null && _b !== void 0 ? _b : false);
     const filters = (_c = config.filters) !== null && _c !== void 0 ? _c : { sourceType: true };
@@ -47,16 +51,24 @@ export function SearchContent({ config }) {
             setDocCount(null);
             return;
         }
+        let cancelled = false;
         SDK.listDocuments({ path: { graph_id: graphId } })
             .then((res) => {
-            if (res.data) {
+            if (!cancelled && res.data) {
                 setDocCount(res.data.total);
             }
         })
             .catch(() => { });
+        return () => {
+            cancelled = true;
+        };
     }, [graphId, isRepository]);
     // Reset when graph changes
     useEffect(() => {
+        searchSeq.current++;
+        sectionSeq.current++;
+        setLoading(false);
+        setSectionLoading(false);
         setResults([]);
         setTotal(0);
         setOffset(0);
@@ -68,6 +80,8 @@ export function SearchContent({ config }) {
     const handleSearch = useCallback(async (newOffset = 0) => {
         if (!graphId || !query.trim())
             return;
+        const seq = ++searchSeq.current;
+        sectionSeq.current++;
         setLoading(true);
         setError(null);
         setExpandedDocId(null);
@@ -95,6 +109,8 @@ export function SearchContent({ config }) {
                 path: { graph_id: graphId },
                 body: body,
             });
+            if (seq !== searchSeq.current)
+                return;
             if (res.data) {
                 const data = res.data;
                 setResults(data.hits);
@@ -107,16 +123,21 @@ export function SearchContent({ config }) {
             }
         }
         catch (_a) {
+            if (seq !== searchSeq.current)
+                return;
             setError('An error occurred while searching.');
         }
         finally {
-            setLoading(false);
+            if (seq === searchSeq.current)
+                setLoading(false);
         }
     }, [graphId, query, sourceType, entity, formType, fiscalYear, semantic]);
     const handleExpand = async (docId) => {
+        const seq = ++sectionSeq.current;
         if (expandedDocId === docId) {
             setExpandedDocId(null);
             setSectionContent(null);
+            setSectionLoading(false);
             return;
         }
         if (!graphId)
@@ -128,15 +149,20 @@ export function SearchContent({ config }) {
             const res = await SDK.getDocumentSection({
                 path: { graph_id: graphId, document_id: docId },
             });
+            if (seq !== sectionSeq.current)
+                return;
             if (res.data) {
                 setSectionContent(res.data);
             }
         }
         catch (_a) {
+            if (seq !== sectionSeq.current)
+                return;
             setSectionContent(null);
         }
         finally {
-            setSectionLoading(false);
+            if (seq === sectionSeq.current)
+                setSectionLoading(false);
         }
     };
     const hasFilters = filters.entity ||

package/contexts/graph-context.js

@@ -1,7 +1,8 @@
 'use client';
 import { jsx as _jsx } from "react/jsx-runtime";
 import * as SDK from '@robosystems/client';
-import { createContext, useCallback, useContext, useEffect, useState, } from 'react';
+import { createContext, useCallback, useContext, useEffect, useRef, useState, } from 'react';
+import { unwrapSdk } from '../lib/sdk-errors';
 // Factory function to create a typed graph context
 export function createGraphContext() {
     return createContext(null);
@@ -15,6 +16,9 @@ export function createGraphProvider(Context) {
             isLoading: false,
             error: null,
         });
+        // Latest graph list, readable synchronously from callbacks
+        const graphsRef = useRef(baseState.graphs);
+        graphsRef.current = baseState.graphs;
         // Transform state if needed
         const state = transformState ? transformState(baseState) : baseState;
         // Load all user graphs
@@ -22,8 +26,9 @@ export function createGraphProvider(Context) {
             setBaseState((prev) => (Object.assign(Object.assign({}, prev), { isLoading: true, error: null })));
             try {
                 const response = await SDK.getGraphs();
-                if (response.data) {
-                    const graphsData = response.data;
+                const data = unwrapSdk(response);
+                if (data) {
+                    const graphsData = data;
                     let graphs = graphsData.graphs || [];
                     // Apply optional filter
                     if (graphFilter) {
@@ -35,6 +40,9 @@ export function createGraphProvider(Context) {
                     const validSelectedId = selectedGraphId && graphIds.includes(selectedGraphId)
                         ? selectedGraphId
                         : null;
+                    // Visible to setCurrentGraph before the next render: a caller that
+                    // awaits refreshGraphs() and then selects the new graph must find it.
+                    graphsRef.current = graphs;
                     setBaseState((prev) => {
                         // Determine which graph should be selected
                         let newCurrentGraphId = validSelectedId || (graphs.length > 0 ? graphs[0].graphId : null);
@@ -51,6 +59,9 @@ export function createGraphProvider(Context) {
                         return Object.assign(Object.assign({}, prev), { graphs, currentGraphId: newCurrentGraphId, isLoading: false });
                     });
                 }
+                else {
+                    throw new Error('Empty response from the graph list endpoint');
+                }
             }
             catch (error) {
                 console.error('Failed to load graphs:', error);
@@ -60,21 +71,20 @@ export function createGraphProvider(Context) {
         // Set current graph with persistence
         const setCurrentGraph = useCallback(async (graphId) => {
             try {
-                // Use functional setState to access current graphs and determine if repository
-                let isUserGraph = false;
-                setBaseState((prev) => {
-                    const graph = prev.graphs.find((g) => g.graphId === graphId);
-                    // Only treat as a user graph if found and NOT a repository
-                    isUserGraph = graph ? !graph.isRepository : false;
-                    return Object.assign(Object.assign({}, prev), { currentGraphId: graphId });
-                });
+                // Decide from the current graph list, not inside a state updater:
+                // React may defer the updater, which left the flag unset and
+                // skipped the API call.
+                const graph = graphsRef.current.find((g) => g.graphId === graphId);
+                // Only treat as a user graph if found and NOT a repository
+                const isUserGraph = graph ? !graph.isRepository : false;
+                setBaseState((prev) => (Object.assign(Object.assign({}, prev), { currentGraphId: graphId })));
                 // Only call selectGraph API for user graphs, not repositories
                 // Repositories can be accessed but not "selected" in the backend
                 // If graph is not found in local state, skip the API call (likely a repository)
                 if (isUserGraph) {
-                    await SDK.selectGraph({
+                    unwrapSdk(await SDK.selectGraph({
                         path: { graph_id: graphId },
-                    });
+                    }));
                 }
             }
             catch (error) {