@datalayer/agent-runtimes 1.5.17 → 1.5.19

package/lib/hooks/useAgentRuntimes.js

@@ -24,6 +24,7 @@ import { parseAgentStreamMessage, } from '../types/stream';
 import { DEFAULT_AGENT_CONFIG } from '../types/config';
 import { ServiceManager } from '@jupyterlab/services/lib/manager';
 import { disposeSandboxServiceManagers } from '../services/sandboxServiceManagers';
+import { agentDeletionUrl, launchesOwnRoutesOnly, runtimeStatusUrl, } from './ownRoutes';
 import { runtimeUrl, runtimesUrl } from '../runtimes/lifecycle';
 import { legacyTargetOf, locationOf, runsInBrowser, toAgentRuntimeVariant, } from '../runtimes/variants';
 // ═══════════════════════════════════════════════════════════════════════════
@@ -232,26 +233,22 @@ export async function createServiceManagerFromAgentSandbox(info, authToken) {
         .replace(/\/$/, '');
     const agentId = String(info.agentId || '').trim();
     const bearer = String(authToken || '').trim();
-    if (agentBaseUrl && agentId && bearer) {
-        const statusCandidates = [
-            `${agentBaseUrl}/api/v1/runtime/status?agent_id=${encodeURIComponent(agentId)}`,
-        ];
-        for (const statusUrl of statusCandidates) {
-            try {
-                const response = await fetch(statusUrl, {
-                    headers: {
-                        Authorization: `Bearer ${bearer}`,
-                    },
-                });
-                if (!response.ok) {
-                    continue;
-                }
-                const payload = (await response
+    const statusUrl = agentBaseUrl && agentId && bearer
+        ? runtimeStatusUrl(agentBaseUrl, agentId, Boolean(info.ownRoutesOnly))
+        : undefined;
+    if (statusUrl) {
+        try {
+            const response = await fetch(statusUrl, {
+                headers: {
+                    Authorization: `Bearer ${bearer}`,
+                },
+            });
+            const payload = response.ok
+                ? (await response
                     .json()
-                    .catch(() => null));
-                if (!payload) {
-                    continue;
-                }
+                    .catch(() => null))
+                : null;
+            if (payload) {
                 payloadFromStatus = payload;
                 const ingressFromStatus = resolveSandboxIngress(payload);
                 if (ingressFromStatus) {
@@ -268,12 +265,11 @@ export async function createServiceManagerFromAgentSandbox(info, authToken) {
                     }
                 }
                 tokenFromStatus = resolveSandboxToken(payload);
-                break;
-            }
-            catch {
-                // Keep fallback ingress/token when status endpoint is unavailable.
             }
         }
+        catch {
+            // Keep fallback ingress/token when status endpoint is unavailable.
+        }
     }
     let ingress = resolvedIngress.replace(/\/$/, '');
     let tokenFromIngress = '';
@@ -366,6 +362,9 @@ export function useAgentRuntimes(options = {}) {
     // Both vocabularies funnel into one value here, so no branch below has to
     // remember that the old names exist.
     const variant = toAgentRuntimeVariant(variantOption ?? runtimeCreationTarget);
+    // Launched for a deployment, its runtime answers only the application's
+    // own routes (OwnRoutesOnly): nothing else is asked of it.
+    const ownRoutesOnly = launchesOwnRoutesOnly(appLaunch);
     const { configuration } = useCoreStore();
     // Base store state
     const runtime = useAgentRuntimeConnection();
@@ -575,10 +574,14 @@ export function useAgentRuntimes(options = {}) {
     const teardown = useCallback(async () => {
         const agentId = runtime?.agentId;
         const agentBaseUrl = runtime?.agentBaseUrl;
-        if (agentId && agentBaseUrl) {
+        // A deployment's runtime keeps its agent: nobody outside deletes it.
+        const deletion = agentId && agentBaseUrl
+            ? agentDeletionUrl(agentBaseUrl, agentId, ownRoutesOnly)
+            : undefined;
+        if (deletion) {
             try {
                 const { token } = await getAuthHeaders();
-                await fetch(`${agentBaseUrl}/api/v1/agents/${encodeURIComponent(agentId)}`, {
+                await fetch(deletion, {
                     method: 'DELETE',
                     headers: token ? { Authorization: `Bearer ${token}` } : {},
                 });
@@ -593,7 +596,7 @@ export function useAgentRuntimes(options = {}) {
         setLifecycleStatus('idle');
         setLifecycleError(null);
         agentRuntimeStore.getState().reset();
-    }, [runtime?.agentId, runtime?.agentBaseUrl, getAuthHeaders]);
+    }, [runtime?.agentId, runtime?.agentBaseUrl, getAuthHeaders, ownRoutesOnly]);
     // ─── Auto-create agent when runtime is ready (connect mode) ───────
     useEffect(() => {
         if (!hasSpec &&

package/lib/hooks/useBrowserAgent.d.ts

@@ -14,7 +14,7 @@ export type UseBrowserAgentOptions = CreateBrowserAgentOptions & {
  * ```tsx
  * const frontendTools = useNotebookTools(notebookId);
  * const { messages, sendMessage, status } = useBrowserAgent({
- *   spec: JUPYTER_NOTEBOOK_COMPACTOR_AGENTSPEC_0_0_1,
+ *   spec: AGENTSPECS['jupyter-notebook-compactor'],
  *   frontendTools,
  *   inference: { inferenceUrl, token },
  * });

package/lib/hooks/useBrowserAgent.js

@@ -35,7 +35,7 @@ import { createBrowserAgent, } from '../runtimes/browser';
  * ```tsx
  * const frontendTools = useNotebookTools(notebookId);
  * const { messages, sendMessage, status } = useBrowserAgent({
- *   spec: JUPYTER_NOTEBOOK_COMPACTOR_AGENTSPEC_0_0_1,
+ *   spec: AGENTSPECS['jupyter-notebook-compactor'],
  *   frontendTools,
  *   inference: { inferenceUrl, token },
  * });

package/lib/protocols/AGUIAdapter.d.ts

@@ -15,6 +15,11 @@ export declare const RUN_CUT_SHORT = "The run stopped before it finished, withou
 export interface AGUIAdapterConfig extends ProtocolAdapterConfig {
     /** Runtime URL (e.g., /api/copilotkit) */
     runtimeUrl?: string;
+    /**
+     * An application's sessions (`…/api/v1/apps/sessions`): Stop is the
+     * thread's session's (`ProtocolConfig.appSessions`).
+     */
+    appSessions?: string;
 }
 /**
  * AG-UI protocol adapter
@@ -63,6 +68,12 @@ export declare class AGUIAdapter extends BaseProtocolAdapter {
      * Terminate a running agent thread on the backend
      */
     terminateAgent(threadId?: string): Promise<void>;
+    /**
+     * Stop an application's session (`POST {sessions}/{uid}/stop`): what runs
+     * is cancelled and it waits to be resumed. Its thread is the session; with
+     * none, nothing was sent and there is nothing to stop.
+     */
+    private stopSession;
     /**
      * Send a message through AG-UI protocol
      * @param message - The new message to send

package/lib/protocols/AGUIAdapter.js

@@ -86,8 +86,10 @@ export class AGUIAdapter extends BaseProtocolAdapter {
     disconnect() {
         // Abort every ongoing fetch request, not merely the newest.
         this.stopGeneration();
-        // Send terminate request to backend if we have a thread ID
-        if (this.currentThreadId) {
+        // Send terminate request to backend if we have a thread ID. Never an
+        // application's session: a page going away, or the chat rebuilt with a
+        // fresh token, is not the person's Stop — it waits for them to come back.
+        if (this.currentThreadId && !this.aguiConfig.appSessions) {
             this.terminateAgent(this.currentThreadId).catch(err => {
                 console.warn('[AGUIAdapter] Failed to terminate agent:', err);
             });
@@ -98,6 +100,11 @@ export class AGUIAdapter extends BaseProtocolAdapter {
      * Terminate a running agent thread on the backend
      */
     async terminateAgent(threadId) {
+        const sessions = this.aguiConfig.appSessions;
+        if (sessions) {
+            await this.stopSession(sessions, threadId || this.currentThreadId);
+            return;
+        }
         try {
             // Derive the terminate endpoint from the base URL
             // e.g., http://localhost:8765/api/v1/ag-ui/default -> http://localhost:8765/api/v1/ag-ui/terminate
@@ -121,6 +128,24 @@ export class AGUIAdapter extends BaseProtocolAdapter {
             // Ignore errors - agent may have already completed
         }
     }
+    /**
+     * Stop an application's session (`POST {sessions}/{uid}/stop`): what runs
+     * is cancelled and it waits to be resumed. Its thread is the session; with
+     * none, nothing was sent and there is nothing to stop.
+     */
+    async stopSession(sessions, threadId) {
+        if (!threadId) {
+            return;
+        }
+        const url = `${sessions.replace(/\/+$/, '')}/${encodeURIComponent(threadId)}/stop`;
+        const response = await this.fetchOf()(url, {
+            method: 'POST',
+            headers: this.buildHeaders(),
+        });
+        if (!response.ok) {
+            throw new Error(`The session ${threadId} could not be stopped (${response.status}).`);
+        }
+    }
     /**
      * Send a message through AG-UI protocol
      * @param message - The new message to send

package/lib/specs/apps.d.ts

@@ -8,38 +8,38 @@
  * DO NOT EDIT MANUALLY - run 'make specs' to regenerate.
  */
 import type { AppBuilt, AppKind, AppSpec } from '../types/agentspecs';
-export declare const ACCOUNTING_APP_0_0_1: AppSpec;
-export declare const BOOKKEEPING_DESK_APP_0_0_1: AppSpec;
-export declare const CHANGE_DETECTION_APP_0_0_1: AppSpec;
-export declare const CROP_MONITORING_APP_0_0_1: AppSpec;
-export declare const CUSTOMER_INTERVIEW_APP_0_0_2: AppSpec;
-export declare const DATA_QUALITY_APP_0_0_2: AppSpec;
-export declare const DECIDE_APP_0_0_1: AppSpec;
-export declare const DISASTER_ASSESSMENT_APP_0_0_1: AppSpec;
-export declare const ERRAND_RUNNER_APP_0_0_1: AppSpec;
-export declare const EVENT_RESPONSE_APP_0_0_2: AppSpec;
-export declare const EXCHANGE_DESK_APP_0_0_2: AppSpec;
-export declare const GUEST_DESK_APP_0_0_1: AppSpec;
-export declare const INBOX_TRIAGE_APP_0_0_1: AppSpec;
-export declare const LIVE_SHOP_CHAT_APP_0_0_1: AppSpec;
-export declare const MODEL_CHOICE_APP_0_0_2: AppSpec;
-export declare const MONTH_END_CLOSE_APP_0_0_1: AppSpec;
-export declare const ORDER_DESK_APP_0_0_1: AppSpec;
-export declare const ORDER_TRACKER_APP_0_0_1: AppSpec;
-export declare const PIPELINE_REPORT_APP_0_0_1: AppSpec;
-export declare const QUOTE_CALCULATOR_APP_0_0_5: AppSpec;
-export declare const REPORT_FROM_A_FILE_APP_0_0_4: AppSpec;
-export declare const RETURNS_DESK_APP_0_0_1: AppSpec;
-export declare const SALES_APP_0_0_2: AppSpec;
-export declare const SHIP_OR_FIX_APP_0_0_2: AppSpec;
-export declare const SHOP_CONCIERGE_APP_0_0_1: AppSpec;
-export declare const SHOPPING_DESK_APP_0_0_1: AppSpec;
-export declare const SIGN_IN_DESK_APP_0_0_1: AppSpec;
-export declare const SIGNED_READER_APP_0_0_1: AppSpec;
-export declare const SUPPLIER_COMPARISON_APP_0_0_2: AppSpec;
-export declare const SUPPORT_DESK_APP_0_0_4: AppSpec;
-export declare const TRAVEL_DESK_APP_0_0_1: AppSpec;
-export declare const WEB_RESEARCH_APP_0_0_1: AppSpec;
+export declare const ACCOUNTING_APP_0_0_2: AppSpec;
+export declare const BOOKKEEPING_DESK_APP_0_0_2: AppSpec;
+export declare const CHANGE_DETECTION_APP_0_0_2: AppSpec;
+export declare const CROP_MONITORING_APP_0_0_2: AppSpec;
+export declare const CUSTOMER_INTERVIEW_APP_0_0_3: AppSpec;
+export declare const DATA_QUALITY_APP_0_0_3: AppSpec;
+export declare const DECIDE_APP_0_0_2: AppSpec;
+export declare const DISASTER_ASSESSMENT_APP_0_0_2: AppSpec;
+export declare const ERRAND_RUNNER_APP_0_0_2: AppSpec;
+export declare const EVENT_RESPONSE_APP_0_0_3: AppSpec;
+export declare const EXCHANGE_DESK_APP_0_0_3: AppSpec;
+export declare const GUEST_DESK_APP_0_0_2: AppSpec;
+export declare const INBOX_TRIAGE_APP_0_0_2: AppSpec;
+export declare const LIVE_SHOP_CHAT_APP_0_0_2: AppSpec;
+export declare const MODEL_CHOICE_APP_0_0_3: AppSpec;
+export declare const MONTH_END_CLOSE_APP_0_0_2: AppSpec;
+export declare const ORDER_DESK_APP_0_0_2: AppSpec;
+export declare const ORDER_TRACKER_APP_0_0_2: AppSpec;
+export declare const PIPELINE_REPORT_APP_0_0_2: AppSpec;
+export declare const QUOTE_CALCULATOR_APP_0_0_6: AppSpec;
+export declare const REPORT_FROM_A_FILE_APP_0_0_5: AppSpec;
+export declare const RETURNS_DESK_APP_0_0_2: AppSpec;
+export declare const SALES_APP_0_0_3: AppSpec;
+export declare const SHIP_OR_FIX_APP_0_0_3: AppSpec;
+export declare const SHOP_CONCIERGE_APP_0_0_2: AppSpec;
+export declare const SHOPPING_DESK_APP_0_0_2: AppSpec;
+export declare const SIGN_IN_DESK_APP_0_0_2: AppSpec;
+export declare const SIGNED_READER_APP_0_0_2: AppSpec;
+export declare const SUPPLIER_COMPARISON_APP_0_0_3: AppSpec;
+export declare const SUPPORT_DESK_APP_0_0_5: AppSpec;
+export declare const TRAVEL_DESK_APP_0_0_2: AppSpec;
+export declare const WEB_RESEARCH_APP_0_0_2: AppSpec;
 export declare const APP_CATALOGUE: Record<string, AppSpec>;
 /** An application, by `id` or `id:version`, or undefined. */
 export declare function getApp(ref: string): AppSpec | undefined;