@datalayer/agent-runtimes 1.0.5 → 1.0.6

package/lib/hooks/useAgentRuntimes.js

@@ -65,6 +65,14 @@ const RUNTIME_STATUS_MAP = {
 function toAgentRuntimeData(raw) {
     const status = typeof raw.status === 'string' ? raw.status.toLowerCase() : '';
     const normalizedStatus = RUNTIME_STATUS_MAP[status] ?? 'running';
+    const rawVolumeUids = Array.isArray(raw.volume_uids)
+        ? raw.volume_uids
+        : raw.volume_uid
+            ? [raw.volume_uid]
+            : [];
+    const volume_uids = rawVolumeUids
+        .map((uid) => String(uid || '').trim())
+        .filter(Boolean);
     return {
         ...raw,
         status: normalizedStatus,
@@ -73,6 +81,8 @@ function toAgentRuntimeData(raw) {
         url: raw.ingress,
         messageCount: 0,
         agent_spec_id: raw.agent_spec_id || undefined,
+        volume_uids,
+        volume_uid: raw.volume_uid || volume_uids[0] || undefined,
     };
 }
 // ═══════════════════════════════════════════════════════════════════════════
@@ -93,7 +103,7 @@ function toAgentRuntimeData(raw) {
  * // Connect mode — attach to an existing runtime
  * const { isReady, endpoint, connectToRuntime } = useAgents({
  *   autoCreateAgent: true,
- *   agentConfig: { model: 'bedrock:us.anthropic.claude-3-5-haiku-20241022-v1:0' },
+ *   agentConfig: { model: 'bedrock:us.anthropic.claude-sonnet-4-5-20250929-v1:0' },
  * });
  *
  * // Lifecycle mode — full lifecycle with agentSpecId
@@ -105,7 +115,8 @@ function toAgentRuntimeData(raw) {
  * ```
  */
 export function useAgentRuntimes(options = {}) {
-    const { agentSpecId, agentConfig, autoCreateAgent = true, autoStart = false, agentSpec, } = options;
+    const { agentSpecId, agentConfig, autoCreateAgent = true, autoStart = false, agentSpec, runtimeCreationTarget = 'backend-services', runtimeCreationBaseUrl, } = options;
+    const { configuration } = useCoreStore();
     // Base store state
     const runtime = useAgentRuntimeConnection();
     const baseStatus = useAgentRuntimeStatus();
@@ -128,6 +139,23 @@ export function useAgentRuntimes(options = {}) {
     agentConfigRef.current = agentConfig;
     // Whether we're managing a full agent lifecycle (agentSpecId provided)
     const hasSpec = !!agentSpecId;
+    const resolvedRuntimeCreationBaseUrl = useMemo(() => {
+        if (runtimeCreationBaseUrl) {
+            return runtimeCreationBaseUrl;
+        }
+        if (runtimeCreationTarget === 'local-agent-runtimes') {
+            return (import.meta.env.VITE_DATALAYER_AGENT_RUNTIMES_URL ||
+                import.meta.env.VITE_BASE_URL ||
+                'http://localhost:8765');
+        }
+        return (configuration?.runtimesRunUrl ||
+            import.meta.env.VITE_DATALAYER_AGENT_RUNTIMES_URL ||
+            'https://r1.datalayer.run');
+    }, [
+        configuration?.runtimesRunUrl,
+        runtimeCreationBaseUrl,
+        runtimeCreationTarget,
+    ]);
     // ─── Auth helpers ─────────────────────────────────────────────────
     const getAuthHeaders = useCallback(async () => {
         try {
@@ -135,30 +163,39 @@ export function useAgentRuntimes(options = {}) {
             const token = iamStore.getState().token || '';
             const config = coreStore.getState().configuration;
             const runUrl = config?.aiagentsRunUrl || '';
-            const runtimesRunUrl = config?.runtimesRunUrl || '';
+            const runtimesRunUrl = resolvedRuntimeCreationBaseUrl;
             return { token, runUrl, runtimesRunUrl };
         }
         catch {
             return { token: '', runUrl: '', runtimesRunUrl: '' };
         }
-    }, []);
+    }, [resolvedRuntimeCreationBaseUrl]);
     // ─── Launch Runtime ─────────────────────────────────────────────────
     const launchRuntime = useCallback(async (runtimeOptions) => {
         if (hasSpec) {
             setLifecycleStatus('launching');
             setLifecycleError(null);
             try {
-                const safeName = `${agentSpecId}`
+                const preferredRuntimeName = (typeof agentConfig?.name === 'string' && agentConfig.name) ||
+                    (typeof agentSpec?.name === 'string' && agentSpec.name) ||
+                    `${agentSpecId}`;
+                const safeName = preferredRuntimeName
                     .replace(/\//g, '-')
                     .replace(/[^a-z0-9-]/g, '-')
                     .replace(/-+/g, '-')
                     .replace(/^-|-$/g, '')
                     .slice(0, 63);
-                const conn = await storeLaunchAgent(runtimeOptions || {
-                    environmentName: 'ai-agents-env',
-                    creditsLimit: 10,
-                    givenName: safeName,
-                });
+                const conn = await storeLaunchAgent(runtimeOptions
+                    ? {
+                        ...runtimeOptions,
+                        runtimesRunUrl: resolvedRuntimeCreationBaseUrl,
+                    }
+                    : {
+                        environmentName: 'ai-agents-env',
+                        creditsLimit: 10,
+                        givenName: safeName,
+                        runtimesRunUrl: resolvedRuntimeCreationBaseUrl,
+                    });
                 setLifecycleStatus('ready');
                 return conn;
             }
@@ -173,9 +210,19 @@ export function useAgentRuntimes(options = {}) {
             if (!runtimeOptions) {
                 throw new Error('Runtime options are required in connect mode');
             }
-            return storeLaunchAgent(runtimeOptions);
+            return storeLaunchAgent({
+                ...runtimeOptions,
+                runtimesRunUrl: resolvedRuntimeCreationBaseUrl,
+            });
         }
-    }, [agentSpecId, hasSpec, storeLaunchAgent]);
+    }, [
+        agentConfig?.name,
+        agentSpec?.name,
+        agentSpecId,
+        hasSpec,
+        resolvedRuntimeCreationBaseUrl,
+        storeLaunchAgent,
+    ]);
     // ─── Create Agent ───────────────────────────────────────────────────
     const createAgent = useCallback(async (config) => {
         if (creatingRef.current) {
@@ -398,6 +445,8 @@ export function useAgentRuntimes(options = {}) {
         // Status
         isReady,
         error,
+        runtimeCreationTarget,
+        runtimeCreationBaseUrl: resolvedRuntimeCreationBaseUrl,
     };
 }
 // ═══════════════════════════════════════════════════════════════════════════
@@ -409,16 +458,36 @@ export function useAgentRuntimes(options = {}) {
  * The backend returns active runtimes from the operator **plus** paused
  * runtimes synthesised from Solr checkpoint records (with ``status="paused"``).
  */
-export function useAgentRuntimesQuery() {
+export function useAgentRuntimesQuery(scope, queryOptions) {
     const { configuration } = useCoreStore();
     const { requestDatalayer } = useDatalayer({ notifyOnError: false });
     const { user } = useIAMStore();
     const queryClient = useQueryClient();
     return useQuery({
-        queryKey: agentQueryKeys.agentRuntimes.lists(),
+        queryKey: [
+            ...agentQueryKeys.agentRuntimes.lists(),
+            scope?.selectedUserUid || '',
+            scope?.selectedOrganizationUid || '',
+            scope?.selectedTeamUid || '',
+            scope?.selectedAgentUid || '',
+        ],
         queryFn: async () => {
+            const params = new URLSearchParams();
+            if (scope?.selectedUserUid) {
+                params.set('selected_user_uid', scope.selectedUserUid);
+            }
+            if (scope?.selectedOrganizationUid) {
+                params.set('selected_organization_uid', scope.selectedOrganizationUid);
+            }
+            if (scope?.selectedTeamUid) {
+                params.set('selected_team_uid', scope.selectedTeamUid);
+            }
+            if (scope?.selectedAgentUid) {
+                params.set('selected_agent_uid', scope.selectedAgentUid);
+            }
+            const query = params.toString();
             const resp = await requestDatalayer({
-                url: `${configuration.runtimesRunUrl}/api/runtimes/v1/runtimes`,
+                url: `${configuration.runtimesRunUrl}/api/runtimes/v1/runtimes${query ? `?${query}` : ''}`,
                 method: 'GET',
             });
             if (resp.success && resp.runtimes) {
@@ -433,8 +502,10 @@ export function useAgentRuntimesQuery() {
             return [];
         },
         ...AGENT_QUERY_OPTIONS,
-        refetchInterval: 10000,
-        enabled: !!user,
+        refetchInterval: queryOptions?.refetchInterval ?? 10000,
+        refetchOnMount: true,
+        refetchOnWindowFocus: true,
+        enabled: (queryOptions?.enabled ?? true) && !!user,
     });
 }
 /**
@@ -474,6 +545,11 @@ export function useCreateAgentRuntime() {
     const queryClient = useQueryClient();
     return useMutation({
         mutationFn: async (data) => {
+            const normalizedVolumeUids = Array.isArray(data.volumeUids)
+                ? data.volumeUids.map(uid => String(uid || '').trim()).filter(Boolean)
+                : data.volumeUid
+                    ? [String(data.volumeUid).trim()]
+                    : [];
             return requestDatalayer({
                 url: `${configuration.runtimesRunUrl}/api/runtimes/v1/runtimes`,
                 method: 'POST',
@@ -486,6 +562,15 @@ export function useCreateAgentRuntime() {
                     enable_codemode: data.enableCodemode ?? false,
                     agent_spec_id: data.agentSpecId || undefined,
                     agent_spec: data.agentSpec || undefined,
+                    user_account_handle: data.userAccountHandle || undefined,
+                    billable_account_uid: data.billableAccountUid || undefined,
+                    billable_account_type: data.billableAccountType || undefined,
+                    billable_account_handle: data.billableAccountHandle || undefined,
+                    billable_source_organization_uid: data.billableSourceOrganizationUid || undefined,
+                    billable_source_organization_handle: data.billableSourceOrganizationHandle || undefined,
+                    mount_home_folder: data.mountHomeFolder ?? false,
+                    volume_uids: normalizedVolumeUids.length > 0 ? normalizedVolumeUids : undefined,
+                    volume_uid: normalizedVolumeUids[0] || data.volumeUid || undefined,
                 },
             });
         },
@@ -515,6 +600,12 @@ export function useDeleteAgentRuntime() {
             });
         },
         onSuccess: (_data, podName) => {
+            queryClient.setQueriesData({ queryKey: agentQueryKeys.agentRuntimes.lists() }, (current) => {
+                if (!Array.isArray(current)) {
+                    return current;
+                }
+                return current.filter(runtime => runtime.pod_name !== podName);
+            });
             queryClient.cancelQueries({
                 queryKey: agentQueryKeys.agentRuntimes.detail(podName),
             });
@@ -524,6 +615,10 @@ export function useDeleteAgentRuntime() {
             queryClient.invalidateQueries({
                 queryKey: agentQueryKeys.agentRuntimes.lists(),
             });
+            queryClient.refetchQueries({
+                queryKey: agentQueryKeys.agentRuntimes.lists(),
+                type: 'active',
+            });
         },
     });
 }
@@ -607,8 +702,8 @@ export const useAgentLifecycleStore = create()(persist((set, get) => ({
 /**
  * Consolidated runtime list and mutations.
  */
-export function useAgentsRuntimes() {
-    const runtimesQuery = useAgentRuntimesQuery();
+export function useAgentsRuntimes(scope, queryOptions) {
+    const runtimesQuery = useAgentRuntimesQuery(scope, queryOptions);
     const createRuntimeMutation = useCreateAgentRuntime();
     const deleteRuntimeMutation = useDeleteAgentRuntime();
     const refreshRuntimes = useRefreshAgentRuntimes();

package/lib/index.d.ts

@@ -6,4 +6,4 @@ export * from './stores';
 export * from './identity';
 export * from './config';
 export * from './specs';
-export type { AgentRuntimeData, AgentSpec, ChatCommonProps, TeamSpec, GuardrailSpec, EvalSpec, TriggerSpec, OutputSpec, NotificationChannelSpec, MCPServer, SkillSpec, SkillInfo, SkillStatus, EnvvarSpec, ToolCallStartContext, ToolCallCompleteContext, } from './types';
+export type { AgentRuntimeData, AgentSpec, BenchmarkSpec, ChatCommonProps, TeamSpec, GuardrailSpec, EvalSpec, TriggerSpec, OutputSpec, NotificationChannelSpec, MCPServer, SkillSpec, SkillInfo, SkillStatus, EnvvarSpec, ToolCallStartContext, ToolCallCompleteContext, } from './types';

package/lib/specs/agents/agents.d.ts

@@ -14,21 +14,28 @@ export declare const CLASSIFY_ROUTE_EMAILS_AGENT_SPEC_0_0_1: AgentSpec;
 export declare const COMPREHENSIVE_SALES_ANALYTICS_AGENT_SPEC_0_0_1: AgentSpec;
 export declare const CRAWLER_AGENT_SPEC_0_0_1: AgentSpec;
 export declare const DATA_ACQUISITION_AGENT_SPEC_0_0_1: AgentSpec;
-export declare const DATALAYER_AGENT_SPEC_0_0_1: AgentSpec;
-export declare const DEMO_FULL_AGENT_SPEC_0_0_1: AgentSpec;
-export declare const DEMO_GUARDRAILS_AGENT_SPEC_0_0_1: AgentSpec;
-export declare const DEMO_HOOKS_AGENT_SPEC_0_0_1: AgentSpec;
-export declare const DEMO_MCP_AGENT_SPEC_0_0_1: AgentSpec;
-export declare const DEMO_MONITORING_AGENT_SPEC_0_0_1: AgentSpec;
-export declare const DEMO_ONE_TRIGGER_APPROVAL_AGENT_SPEC_0_0_1: AgentSpec;
-export declare const DEMO_ONE_TRIGGER_AGENT_SPEC_0_0_1: AgentSpec;
 export declare const DEMO_OUTPUTS_AGENT_SPEC_0_0_1: AgentSpec;
-export declare const DEMO_PARAMETERS_AGENT_SPEC_0_0_1: AgentSpec;
-export declare const DEMO_SIMPLE_AGENT_SPEC_0_0_1: AgentSpec;
-export declare const DEMO_SUBAGENTS_AGENT_SPEC_0_0_1: AgentSpec;
-export declare const DEMO_TAVILY_CODEMODE_AGENT_SPEC_0_0_1: AgentSpec;
-export declare const DEMO_TAVILY_NO_CODEMODE_AGENT_SPEC_0_0_1: AgentSpec;
 export declare const END_OF_MONTH_SALES_PERFORMANCE_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EVAL_EXPERIMENT_RUNNER_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_CODEMODE_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_EVALS_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_FULL_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_GUARDRAILS_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_HOOKS_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_INFERENCE_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_MCP_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_MEMORY_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_MONITORING_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_NO_CODEMODE_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_NOTIFICATIONS_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_ONE_TRIGGER_APPROVAL_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_ONE_TRIGGER_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_OTEL_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_PARAMETERS_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_SIMPLE_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_SKILLS_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_SUBAGENTS_AGENT_SPEC_0_0_1: AgentSpec;
+export declare const EXAMPLE_TOOL_APPROVALS_AGENT_SPEC_0_0_1: AgentSpec;
 export declare const EXTRACT_DATA_FROM_FILES_AGENT_SPEC_0_0_1: AgentSpec;
 export declare const FINANCIAL_VIZ_AGENT_SPEC_0_0_1: AgentSpec;
 export declare const FINANCIAL_AGENT_SPEC_0_0_1: AgentSpec;