brave search tool + fetch weather tool custom render; bug fixes

Author: code-forge-temple

client/src/components/nodes/ToolNode/tools/BraveSearchTool/components/BraveSearchToolConfigFields.tsx

@@ -0,0 +1,46 @@
+/************************************************************************
+ *    Copyright (C) 2025 Code Forge Temple                              *
+ *    This file is part of agentic-signal project                       *
+ *    See the LICENSE file in the project root for license details.     *
+ ************************************************************************/
+
+import {useGlobalConfig} from "../../../../../../stores/globalConfig";
+import {globalToolProp} from "../../utils/utils";
+import {DebouncedTextField} from "../../../../../DebouncedTextField";
+import {UserConfigFields} from "../../../UserConfigFields";
+import {excludeKeysFromObject} from "../../../../../../utils";
+
+type BraveSearchToolConfigFieldsProps = {
+    userConfig: Record<string, any>;
+    onConfigChange: (key: string, value: string | number | boolean) => void;
+};
+
+export function BraveSearchToolConfigFields ({userConfig, onConfigChange}: BraveSearchToolConfigFieldsProps) {
+    const {globalData, setGlobalData} = useGlobalConfig();
+    const globalApiKeyKey = globalToolProp("brave-search", "apiKey");
+    const apiKeyValue = globalData[globalApiKeyKey] || userConfig?.apiKey || "";
+
+    return (
+        <>
+            <DebouncedTextField
+                label="Brave Search API Key"
+                value={apiKeyValue}
+                onChange={(value) => {
+                    onConfigChange("apiKey", value);
+                    setGlobalData(globalApiKeyKey, value, true);
+                }}
+                fullWidth
+                sx={{mb: 1}}
+                type="password"
+                helperText="Your Brave Search API key is retained globally for this tool."
+            />
+            <UserConfigFields
+                userConfigSchema={excludeKeysFromObject({
+                    apiKey: {type: "string", description: "Brave Search API Key", required: true},
+                }, ["apiKey"])}
+                userConfig={userConfig}
+                onConfigChange={onConfigChange}
+            />
+        </>
+    );
+}

client/src/components/nodes/ToolNode/tools/BraveSearchTool/descriptor.tsx

@@ -9,7 +9,7 @@ import {GraphQLService} from "./services/graphqlService";
 import {ToolDefinition} from "../types";
 import Brave from "./assets/brave.svg";
 import {extendSystemUserConfigSchema} from "../../../../../types/ollama.types";
-
+import {BraveSearchToolConfigFields} from "./components/BraveSearchToolConfigFields";
 
 export const BraveSearchToolDescriptor:ToolDefinition = {
     toolSubtype: "brave-search",
@@ -37,6 +37,14 @@ export const BraveSearchToolDescriptor:ToolDefinition = {
         }
     }),
     toSanitize: ["userConfig.apiKey"],
+    renderConfig: function ({userConfig, onConfigChange}) {
+        return (
+            <BraveSearchToolConfigFields
+                userConfig={userConfig}
+                onConfigChange={onConfigChange}
+            />
+        );
+    },
     handlerFactory: (userConfig: { apiKey?: string, maxResults?: number }) => async ({query}: { query: string }) => {
         if (!userConfig.apiKey) {
             return {error: "API key must be specified. Please set apiKey in the configuration."};

client/src/components/nodes/ToolNode/tools/FetchWeatherDataTool/components/FetchWeatherDataToolConfigFields.tsx

@@ -0,0 +1,35 @@
+/************************************************************************
+ *    Copyright (C) 2025 Code Forge Temple                              *
+ *    This file is part of agentic-signal project                       *
+ *    See the LICENSE file in the project root for license details.     *
+ ************************************************************************/
+
+import {useGlobalConfig} from "../../../../../../stores/globalConfig";
+import {globalToolProp} from "../../utils/utils";
+import {DebouncedTextField} from "../../../../../DebouncedTextField";
+
+type FetchWeatherDataToolConfigFieldsProps = {
+    userConfig: Record<string, any>;
+    onConfigChange: (key: string, value: string | number | boolean) => void;
+};
+
+export function FetchWeatherDataToolConfigFields ({userConfig, onConfigChange}: FetchWeatherDataToolConfigFieldsProps) {
+    const {globalData, setGlobalData} = useGlobalConfig();
+    const globalApiKeyKey = globalToolProp("fetch-weather-data", "apiKey");
+    const apiKeyValue = globalData[globalApiKeyKey] || userConfig?.apiKey || "";
+
+    return (
+        <DebouncedTextField
+            label="WeatherAPI.com API Key"
+            value={apiKeyValue}
+            onChange={(value) => {
+                onConfigChange("apiKey", value);
+                setGlobalData(globalApiKeyKey, value, true);
+            }}
+            fullWidth
+            sx={{mb: 1}}
+            type="password"
+            helperText="Your Weather API key is retained globally for this tool."
+        />
+    );
+}

client/src/components/nodes/ToolNode/tools/FetchWeatherDataTool/descriptor.tsx

@@ -8,6 +8,7 @@ import {Thunderstorm} from "iconoir-react";
 import {ToolDefinition} from "../types";
 import {extendSystemUserConfigSchema} from "../../../../../types/ollama.types";
 import {sanitizeStringInput} from "../utils/sanitize";
+import {FetchWeatherDataToolConfigFields} from "./components/FetchWeatherDataToolConfigFields";
 
 
 export const FetchWeatherDataToolDescriptor:ToolDefinition = {
@@ -37,6 +38,14 @@ export const FetchWeatherDataToolDescriptor:ToolDefinition = {
         apiKey: {type: "string", description: "WeatherAPI.com API Key", required: true}
     }),
     toSanitize: ["userConfig.apiKey"],
+    renderConfig: function ({userConfig, onConfigChange}) {
+        return (
+            <FetchWeatherDataToolConfigFields
+                userConfig={userConfig}
+                onConfigChange={onConfigChange}
+            />
+        );
+    },
     handlerFactory: (userConfig: { apiKey?: string }) => async ({city, date}: { city: string, date?: string }) => {
         if (!userConfig.apiKey) {
             return {error: "API key must be specified. Please set apiKey in the configuration."};

server/ws/webSocketManager.ts

@@ -17,6 +17,8 @@ export class WebSocketManager {
 
     private subscriptionFactories = new Map<string, SubscriptionFactory>();
 
+    private pingIntervals = new Map<WebSocket, ReturnType<typeof setInterval>>();
+
     private yogaHandler: ((request: Request) => Promise<Response>) | null = null;
 
     private socketSubscriptions = new Map<WebSocket, Map<string, string>>();
@@ -83,6 +85,17 @@ export class WebSocketManager {
     private setupSocketHandlers (socket: WebSocket): void {
         this.socketSubscriptions.set(socket, new Map());
 
+        const pingInterval = setInterval(() => {
+            if (socket.readyState === WebSocket.OPEN) {
+                socket.send(JSON.stringify({type: 'ping'}));
+            } else {
+                clearInterval(pingInterval);
+                this.pingIntervals.delete(socket);
+            }
+        }, 30_000);
+
+        this.pingIntervals.set(socket, pingInterval);
+
         socket.onopen = () => { };
 
         socket.onmessage = async (event) => {
@@ -126,6 +139,7 @@ export class WebSocketManager {
 
         socket.onerror = (error) => {
             console.error('[WebSocketManager] ❌ Socket error:', error);
+            this.clearPingInterval(socket);
         };
     }
 
@@ -212,7 +226,18 @@ export class WebSocketManager {
         socket.addEventListener('close', closeHandler, {once: true});
     }
 
+    private clearPingInterval (socket: WebSocket): void {
+        const interval = this.pingIntervals.get(socket);
+
+        if (interval !== undefined) {
+            clearInterval(interval);
+            this.pingIntervals.delete(socket);
+        }
+    }
+
     private cleanupSocket (socket: WebSocket): void {
+        this.clearPingInterval(socket);
+
         for (const [key, sockets] of this.activeSubscriptions.entries()) {
             sockets.delete(socket);