@datalayer/agent-runtimes 1.0.8 → 1.0.9

package/lib/chat/Chat.js

@@ -18,7 +18,7 @@ import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
  *
  * @module chat/Chat
  */
-import { useEffect, useMemo, useState, useRef } from 'react';
+import { useEffect, useMemo, useState, useRef, useCallback } from 'react';
 import { QueryClient, QueryClientProvider } from '@tanstack/react-query';
 import { Text, Button, Spinner } from '@primer/react';
 import { AlertIcon, SyncIcon } from '@primer/octicons-react';
@@ -145,9 +145,7 @@ export function Chat({ protocol: transport, extensions: _extensions, baseUrl = '
     const [showDetails, setShowDetails] = useState(false);
     const [messageCount, setMessageCount] = useState(0);
     const [focusTrigger, setFocusTrigger] = useState(0);
-    const renderCountRef = useRef(0);
-    renderCountRef.current += 1;
-    console.log(`[AgentRuntimes][Chat] render count: ${renderCountRef.current}`);
+    const lastMessageCountRef = useRef(null);
     // Get connected identities to pass to backend for skill execution
     const connectedIdentities = useConnectedIdentities();
     // Map identities to the format expected by ChatBase
@@ -272,6 +270,26 @@ export function Chat({ protocol: transport, extensions: _extensions, baseUrl = '
     const handleNewChat = () => {
         onDisconnect?.();
     };
+    const handleInformationClick = useCallback(() => {
+        setShowDetails(true);
+    }, []);
+    const handleBackFromDetails = useCallback(() => {
+        setShowDetails(false);
+    }, []);
+    const handleMessagesChange = useCallback((messages) => {
+        const nextCount = messages.length;
+        if (lastMessageCountRef.current === nextCount) {
+            return;
+        }
+        lastMessageCountRef.current = nextCount;
+        setMessageCount(nextCount);
+    }, []);
+    const headerButtons = useMemo(() => ({
+        showNewChat: showNewChatButton,
+        showClear: showClearButton,
+        onNewChat: handleNewChat,
+    }), [showNewChatButton, showClearButton]);
+    const avatarConfig = useMemo(() => ({ showAvatars: true }), []);
     // Render error state
     if (error) {
         return (_jsx(QueryClientProvider, { client: queryClient, children: _jsxs(Box, { className: className, sx: {
@@ -306,7 +324,7 @@ export function Chat({ protocol: transport, extensions: _extensions, baseUrl = '
                         display: showDetails ? 'flex' : 'none',
                         flexDirection: 'column',
                         height: '100%',
-                    }, children: _jsx(AgentDetails, { name: title || 'AI Agent', icon: brandIcon, protocol: transport, url: protocolConfig?.endpoint || baseUrl, messageCount: messageCount, agentId: agentId, apiBase: baseUrl, identityProviders: identityProviders, onIdentityConnect: onIdentityConnect, onIdentityDisconnect: onIdentityDisconnect, onBack: () => setShowDetails(false), mcpStatusData: mcpStatusData, codemodeStatusData: codemodeStatusData }) }), _jsxs(Box, { sx: {
+                    }, children: _jsx(AgentDetails, { name: title || 'AI Agent', icon: brandIcon, protocol: transport, url: protocolConfig?.endpoint || baseUrl, messageCount: messageCount, agentId: agentId, apiBase: baseUrl, identityProviders: identityProviders, onIdentityConnect: onIdentityConnect, onIdentityDisconnect: onIdentityDisconnect, onBack: handleBackFromDetails, mcpStatusData: mcpStatusData, codemodeStatusData: codemodeStatusData }) }), _jsxs(Box, { sx: {
                         display: showDetails ? 'none' : 'flex',
                         flexDirection: 'column',
                         height: '100%',
@@ -331,12 +349,6 @@ export function Chat({ protocol: transport, extensions: _extensions, baseUrl = '
                                             ? 'attention.fg'
                                             : 'danger.fg',
                                         flex: 1,
-                                    }, children: errorBanner.message })] })), _jsx(ChatBase, { title: title, subtitle: subtitle, brandIcon: brandIcon, showHeader: showHeader, protocol: protocolConfig, placeholder: placeholder, description: description, suggestions: suggestions, submitOnSuggestionClick: submitOnSuggestionClick, autoFocus: autoFocus, runtimeId: runtimeId, historyEndpoint: historyEndpoint, pendingPrompt: pendingPrompt, showInformation: showInformation, onInformationClick: () => setShowDetails(true), headerContent: headerContent, headerActions: headerActions, showModelSelector: showModelSelector, showToolsMenu: showToolsMenu, showInput: showInput, disableInputPrompt: disableInputPrompt, showSkillsMenu: showSkillsMenu, showTokenUsage: showTokenUsage, codemodeEnabled: codemodeEnabled, onToggleCodemode: onToggleCodemode, initialModel: initialModel, availableModels: availableModels, mcpServers: mcpServers, initialSkills: initialSkills, connectedIdentities: identitiesForChat, onNewChat: handleNewChat, onMessagesChange: messages => setMessageCount(messages.length), headerButtons: {
-                                showNewChat: showNewChatButton,
-                                showClear: showClearButton,
-                                onNewChat: handleNewChat,
-                            }, avatarConfig: {
-                                showAvatars: true,
-                            }, backgroundColor: "canvas.default", focusTrigger: focusTrigger, chatViewMode: chatViewMode, onChatViewModeChange: onChatViewModeChange, frontendTools: frontendTools, onToolCallStart: onToolCallStart, onToolCallComplete: onToolCallComplete, renderToolResult: renderToolResult, hideMessagesAfterToolUI: hideMessagesAfterToolUI, contextSnapshot: contextSnapshot, mcpStatusData: mcpStatusData, sandboxStatusData: sandboxStatusData, showToolApprovalBanner: showToolApprovalBanner, pendingApprovals: pendingApprovals, onApproveApproval: onApproveApproval, onRejectApproval: onRejectApproval })] })] }) }));
+                                    }, children: errorBanner.message })] })), _jsx(ChatBase, { title: title, subtitle: subtitle, brandIcon: brandIcon, showHeader: showHeader, protocol: protocolConfig, placeholder: placeholder, description: description, suggestions: suggestions, submitOnSuggestionClick: submitOnSuggestionClick, autoFocus: autoFocus, runtimeId: runtimeId, historyEndpoint: historyEndpoint, pendingPrompt: pendingPrompt, showInformation: showInformation, onInformationClick: handleInformationClick, headerContent: headerContent, headerActions: headerActions, showModelSelector: showModelSelector, showToolsMenu: showToolsMenu, showInput: showInput, disableInputPrompt: disableInputPrompt, showSkillsMenu: showSkillsMenu, showTokenUsage: showTokenUsage, codemodeEnabled: codemodeEnabled, onToggleCodemode: onToggleCodemode, initialModel: initialModel, availableModels: availableModels, mcpServers: mcpServers, initialSkills: initialSkills, connectedIdentities: identitiesForChat, onNewChat: handleNewChat, onMessagesChange: handleMessagesChange, headerButtons: headerButtons, avatarConfig: avatarConfig, backgroundColor: "canvas.default", focusTrigger: focusTrigger, chatViewMode: chatViewMode, onChatViewModeChange: onChatViewModeChange, frontendTools: frontendTools, onToolCallStart: onToolCallStart, onToolCallComplete: onToolCallComplete, renderToolResult: renderToolResult, hideMessagesAfterToolUI: hideMessagesAfterToolUI, contextSnapshot: contextSnapshot, mcpStatusData: mcpStatusData, sandboxStatusData: sandboxStatusData, showToolApprovalBanner: showToolApprovalBanner, pendingApprovals: pendingApprovals, onApproveApproval: onApproveApproval, onRejectApproval: onRejectApproval })] })] }) }));
 }
 export default Chat;

package/lib/identity/identityStore.js

@@ -11,7 +11,7 @@
  */
 import { create } from 'zustand';
 import { devtools, subscribeWithSelector, persist } from 'zustand/middleware';
-import { shallow } from 'zustand/shallow';
+import { useShallow } from 'zustand/react/shallow';
 import { GITHUB_PROVIDER, GOOGLE_PROVIDER, KAGGLE_PROVIDER } from './types';
 import { generatePKCEPair, generateState } from './pkce';
 /**
@@ -460,15 +460,15 @@ export const useIdentityStore = create()(devtools(subscribeWithSelector(persist(
  * Get all connected identities
  */
 export function useConnectedIdentities() {
-    return useIdentityStore(state => Array.from(state.identities.values()).filter(i => i.isConnected), shallow);
+    return useIdentityStore(useShallow(state => Array.from(state.identities.values()).filter(i => i.isConnected)));
 }
 /**
  * Get connected provider names
  */
 export function useConnectedProviders() {
-    return useIdentityStore(state => Array.from(state.identities.entries())
+    return useIdentityStore(useShallow(state => Array.from(state.identities.entries())
         .filter(([_, identity]) => identity.isConnected)
-        .map(([provider]) => provider), shallow);
+        .map(([provider]) => provider)));
 }
 /**
  * Get identity for a specific provider

package/lib/tools/adapters/agent-runtimes/AgentRuntimesToolAdapter.js

@@ -24,7 +24,6 @@ function isRecentDuplicate(toolName, params) {
     if (lastExecution) {
         const timeSinceExecution = Date.now() - lastExecution;
         if (timeSinceExecution < CACHE_TTL_MS) {
-            console.log(`[agent-runtimes] 🚫 DUPLICATE DETECTED - Skipping execution (${timeSinceExecution}ms since last call)`);
             return true;
         }
     }
@@ -53,7 +52,6 @@ function processParameters(params) {
             (value.startsWith('{') || value.startsWith('['))) {
             try {
                 processed[key] = JSON.parse(value);
-                console.log(`[agent-runtimes] 📝 Parsed JSON string for parameter '${key}'`);
             }
             catch {
                 // If parsing fails, keep the original string
@@ -82,13 +80,8 @@ export function createAgentRuntimesTool(definition, operation, context) {
         description: definition.description,
         parameters: definition.parameters,
         handler: async (params) => {
-            console.log(`[agent-runtimes] ========== HANDLER CALLED ==========`);
-            console.log(`[agent-runtimes] Tool: ${definition.name}`);
-            console.log(`[agent-runtimes] Operation: ${definition.operation}`);
-            console.log(`[agent-runtimes] Params:`, params);
             // Check for duplicate execution
             if (isRecentDuplicate(definition.name, params)) {
-                console.log(`[agent-runtimes] 🚫 DUPLICATE DETECTED - Returning early`);
                 return {
                     success: false,
                     message: 'Operation already executed recently (duplicate detected)',
@@ -97,13 +90,11 @@ export function createAgentRuntimesTool(definition, operation, context) {
             try {
                 // Process parameters to handle JSON strings
                 const processedParams = processParameters(params);
-                console.log(`[agent-runtimes] Processed params:`, processedParams);
                 // Use OperationRunner to execute operation with TOON format
                 const result = await runner.execute(operation, processedParams, {
                     ...context,
                     format: 'toon', // Return human/LLM-readable string
                 });
-                console.log(`[agent-runtimes] Operation result:`, result);
                 return result;
             }
             catch (error) {
@@ -136,6 +127,5 @@ export function createAllAgentRuntimesTools(definitions, operations, context) {
         const tool = createAgentRuntimesTool(definition, operation, context);
         tools.push(tool);
     }
-    console.log(`[agent-runtimes Tools] Created ${tools.length} frontend tools`);
     return tools;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@datalayer/agent-runtimes",
-  "version": "1.0.8",
+  "version": "1.0.9",
   "type": "module",
   "workspaces": [
     ".",