@adminforth/agent 1.48.0 → 1.49.0

package/agent/middleware/apiBasedTools.ts

@@ -47,6 +47,70 @@ function getEnabledApiToolNames(messages: unknown[]) {
   return enabledToolNames;
 }
 
+const loadedApiToolNamesBySession = new Map<string, Set<string>>();
+
+function getSessionLoadedApiToolNames(sessionId: string) {
+  let toolNames = loadedApiToolNamesBySession.get(sessionId);
+
+  if (!toolNames) {
+    toolNames = new Set<string>();
+    loadedApiToolNamesBySession.set(sessionId, toolNames);
+  }
+
+  return toolNames;
+}
+
+function getEnabledApiToolNamesForSession(messages: unknown[], sessionId?: string) {
+  const enabledToolNames = getEnabledApiToolNames(messages);
+
+  if (!sessionId) {
+    return enabledToolNames;
+  }
+
+  for (const toolName of getSessionLoadedApiToolNames(sessionId)) {
+    enabledToolNames.add(toolName);
+  }
+
+  return enabledToolNames;
+}
+
+function getToolMessageContent(message: unknown) {
+  if (!ToolMessage.isInstance(message)) {
+    return "";
+  }
+
+  return typeof message.content === "string"
+    ? message.content
+    : Array.isArray(message.content)
+      ? message.content
+          .map((block) =>
+            typeof block === "string"
+              ? block
+              : "text" in block
+                ? block.text
+                : "",
+          )
+          .join("")
+      : "";
+}
+
+function rememberLoadedToolFromFetchResult(sessionId: string, result: unknown) {
+  if (!ToolMessage.isInstance(result) || result.name !== "fetch_tool_schema") {
+    return;
+  }
+
+  try {
+    const parsed = JSON.parse(getToolMessageContent(result)) as {
+      status?: number;
+      name?: string;
+    };
+
+    if (parsed.status === 200 && parsed.name) {
+      getSessionLoadedApiToolNames(sessionId).add(parsed.name);
+    }
+  } catch {}
+}
+
 export function createApiBasedToolsMiddleware(
   apiBasedTools: Record<string, ApiBasedTool>,
   adminforth: IAdminForth,
@@ -62,7 +126,11 @@ export function createApiBasedToolsMiddleware(
   return createMiddleware({
     name: "ApiBasedToolsMiddleware",
     async wrapModelCall(request, handler) {
-      const enabledApiToolNames = getEnabledApiToolNames(request.state.messages);
+      const { sessionId } = request.runtime.context as { sessionId?: string };
+      const enabledApiToolNames = getEnabledApiToolNamesForSession(
+        request.state.messages,
+        sessionId,
+      );
       const tools = [...enabledApiToolNames]
         .filter((toolName) => !alwaysAvailableApiToolNames.has(toolName))
         .map((toolName) => dynamicTools[toolName]);
@@ -80,9 +148,10 @@ export function createApiBasedToolsMiddleware(
     async wrapToolCall(request, handler) {
       const startedAt = Date.now();
       const toolInput = JSON.stringify(request.toolCall.args ?? {});
-      const { adminUser, emitToolCallEvent, userTimeZone } = request.runtime.context as {
+      const { adminUser, emitToolCallEvent, sessionId, userTimeZone } = request.runtime.context as {
         adminUser: AdminUser;
         emitToolCallEvent: ToolCallEventSink;
+        sessionId: string;
         userTimeZone: string;
       };
       const toolArgs = (request.toolCall.args ?? {}) as Record<string, unknown>;
@@ -120,7 +189,10 @@ export function createApiBasedToolsMiddleware(
         if (request.tool) {
           result = await handler(request);
         } else {
-          const enabledApiToolNames = getEnabledApiToolNames(request.state.messages);
+          const enabledApiToolNames = getEnabledApiToolNamesForSession(
+            request.state.messages,
+            sessionId,
+          );
 
           if (enabledApiToolNames.has(request.toolCall.name)) {
             result = await handler({
@@ -137,6 +209,10 @@ export function createApiBasedToolsMiddleware(
           }
         }
 
+        if (sessionId) {
+          rememberLoadedToolFromFetchResult(sessionId, result);
+        }
+
         toolCallTracker.finishSuccess(result);
         return result;
       } catch (error) {

package/build.log

@@ -62,5 +62,5 @@ custom/speech_recognition_frontend/voiceActivityDetection.ts
 custom/speech_recognition_frontend/types/
 custom/speech_recognition_frontend/types/voice-activity-detection.d.ts
 
-sent 1,667,891 bytes  received 921 bytes  3,337,624.00 bytes/sec
-total size is 1,663,796  speedup is 1.00
+sent 1,668,700 bytes  received 921 bytes  3,339,242.00 bytes/sec
+total size is 1,664,547  speedup is 1.00

package/custom/conversation_area/ConversationArea.vue

@@ -118,10 +118,13 @@ const THRESHOLD_TO_SHOW_BUTTON = 10;
 let messageResizeObserver: ResizeObserver | null = null;
 let observedLastUserMessageElement: HTMLElement | null = null;
 let observedLastAgentMessageElement: HTMLElement | null = null;
+let updateSpacerFrameId: number | null = null;
+let pendingSpacerUpdate: Promise<void> | null = null;
+let spacerUpdateQueued = false;
 
 onMounted(async () => {
   messageResizeObserver = new ResizeObserver(() => {
-    updateSpacerHeight();
+    scheduleSpacerHeightUpdate();
   });
 
   await import('@incremark/theme/styles.css')
@@ -136,6 +139,10 @@ onUnmounted(() => {
 
   stopObservingLastMessages();
   messageResizeObserver?.disconnect();
+  if (updateSpacerFrameId !== null) {
+    cancelAnimationFrame(updateSpacerFrameId);
+    updateSpacerFrameId = null;
+  }
   agentStore.stopPlaceholderAnimation();
 });
 
@@ -221,6 +228,30 @@ async function updateSpacerHeight() {
   spacerHeight.value = Math.max(0, clientHeight - (lastUserMessageHeight + MASK_HEIGHT + lastAgentMessageHeight));
 }
 
+function scheduleSpacerHeightUpdate() {
+  spacerUpdateQueued = true;
+
+  if (updateSpacerFrameId !== null) {
+    return;
+  }
+
+  updateSpacerFrameId = requestAnimationFrame(() => {
+    updateSpacerFrameId = null;
+
+    if (pendingSpacerUpdate) {
+      return;
+    }
+
+    spacerUpdateQueued = false;
+    pendingSpacerUpdate = updateSpacerHeight().finally(() => {
+      pendingSpacerUpdate = null;
+      if (spacerUpdateQueued) {
+        scheduleSpacerHeightUpdate();
+      }
+    });
+  });
+}
+
 function stopObservingLastMessages() {
   if (!messageResizeObserver) {
     return;
@@ -285,4 +316,4 @@ function recalculateScroll() {
   }
 }
 
-</script>
+</script>

package/dist/agent/middleware/apiBasedTools.js

@@ -41,6 +41,53 @@ function getEnabledApiToolNames(messages) {
     }
     return enabledToolNames;
 }
+const loadedApiToolNamesBySession = new Map();
+function getSessionLoadedApiToolNames(sessionId) {
+    let toolNames = loadedApiToolNamesBySession.get(sessionId);
+    if (!toolNames) {
+        toolNames = new Set();
+        loadedApiToolNamesBySession.set(sessionId, toolNames);
+    }
+    return toolNames;
+}
+function getEnabledApiToolNamesForSession(messages, sessionId) {
+    const enabledToolNames = getEnabledApiToolNames(messages);
+    if (!sessionId) {
+        return enabledToolNames;
+    }
+    for (const toolName of getSessionLoadedApiToolNames(sessionId)) {
+        enabledToolNames.add(toolName);
+    }
+    return enabledToolNames;
+}
+function getToolMessageContent(message) {
+    if (!ToolMessage.isInstance(message)) {
+        return "";
+    }
+    return typeof message.content === "string"
+        ? message.content
+        : Array.isArray(message.content)
+            ? message.content
+                .map((block) => typeof block === "string"
+                ? block
+                : "text" in block
+                    ? block.text
+                    : "")
+                .join("")
+            : "";
+}
+function rememberLoadedToolFromFetchResult(sessionId, result) {
+    if (!ToolMessage.isInstance(result) || result.name !== "fetch_tool_schema") {
+        return;
+    }
+    try {
+        const parsed = JSON.parse(getToolMessageContent(result));
+        if (parsed.status === 200 && parsed.name) {
+            getSessionLoadedApiToolNames(sessionId).add(parsed.name);
+        }
+    }
+    catch (_a) { }
+}
 export function createApiBasedToolsMiddleware(apiBasedTools, adminforth) {
     const alwaysAvailableApiToolNames = new Set(ALWAYS_AVAILABLE_API_TOOL_NAMES);
     const dynamicTools = Object.fromEntries(Object.entries(apiBasedTools).map(([toolName, apiBasedTool]) => [
@@ -51,7 +98,8 @@ export function createApiBasedToolsMiddleware(apiBasedTools, adminforth) {
         name: "ApiBasedToolsMiddleware",
         wrapModelCall(request, handler) {
             return __awaiter(this, void 0, void 0, function* () {
-                const enabledApiToolNames = getEnabledApiToolNames(request.state.messages);
+                const { sessionId } = request.runtime.context;
+                const enabledApiToolNames = getEnabledApiToolNamesForSession(request.state.messages, sessionId);
                 const tools = [...enabledApiToolNames]
                     .filter((toolName) => !alwaysAvailableApiToolNames.has(toolName))
                     .map((toolName) => dynamicTools[toolName]);
@@ -65,7 +113,7 @@ export function createApiBasedToolsMiddleware(apiBasedTools, adminforth) {
                 var _a, _b, _c, _d;
                 const startedAt = Date.now();
                 const toolInput = JSON.stringify((_a = request.toolCall.args) !== null && _a !== void 0 ? _a : {});
-                const { adminUser, emitToolCallEvent, userTimeZone } = request.runtime.context;
+                const { adminUser, emitToolCallEvent, sessionId, userTimeZone } = request.runtime.context;
                 const toolArgs = ((_b = request.toolCall.args) !== null && _b !== void 0 ? _b : {});
                 let toolInfo;
                 if (request.toolCall.name === "fetch_skill") {
@@ -99,7 +147,7 @@ export function createApiBasedToolsMiddleware(apiBasedTools, adminforth) {
                         result = yield handler(request);
                     }
                     else {
-                        const enabledApiToolNames = getEnabledApiToolNames(request.state.messages);
+                        const enabledApiToolNames = getEnabledApiToolNamesForSession(request.state.messages, sessionId);
                         if (enabledApiToolNames.has(request.toolCall.name)) {
                             result = yield handler(Object.assign(Object.assign({}, request), { tool: dynamicTools[request.toolCall.name] }));
                         }
@@ -112,6 +160,9 @@ export function createApiBasedToolsMiddleware(apiBasedTools, adminforth) {
                             });
                         }
                     }
+                    if (sessionId) {
+                        rememberLoadedToolFromFetchResult(sessionId, result);
+                    }
                     toolCallTracker.finishSuccess(result);
                     return result;
                 }

package/dist/custom/conversation_area/ConversationArea.vue

@@ -118,10 +118,13 @@ const THRESHOLD_TO_SHOW_BUTTON = 10;
 let messageResizeObserver: ResizeObserver | null = null;
 let observedLastUserMessageElement: HTMLElement | null = null;
 let observedLastAgentMessageElement: HTMLElement | null = null;
+let updateSpacerFrameId: number | null = null;
+let pendingSpacerUpdate: Promise<void> | null = null;
+let spacerUpdateQueued = false;
 
 onMounted(async () => {
   messageResizeObserver = new ResizeObserver(() => {
-    updateSpacerHeight();
+    scheduleSpacerHeightUpdate();
   });
 
   await import('@incremark/theme/styles.css')
@@ -136,6 +139,10 @@ onUnmounted(() => {
 
   stopObservingLastMessages();
   messageResizeObserver?.disconnect();
+  if (updateSpacerFrameId !== null) {
+    cancelAnimationFrame(updateSpacerFrameId);
+    updateSpacerFrameId = null;
+  }
   agentStore.stopPlaceholderAnimation();
 });
 
@@ -221,6 +228,30 @@ async function updateSpacerHeight() {
   spacerHeight.value = Math.max(0, clientHeight - (lastUserMessageHeight + MASK_HEIGHT + lastAgentMessageHeight));
 }
 
+function scheduleSpacerHeightUpdate() {
+  spacerUpdateQueued = true;
+
+  if (updateSpacerFrameId !== null) {
+    return;
+  }
+
+  updateSpacerFrameId = requestAnimationFrame(() => {
+    updateSpacerFrameId = null;
+
+    if (pendingSpacerUpdate) {
+      return;
+    }
+
+    spacerUpdateQueued = false;
+    pendingSpacerUpdate = updateSpacerHeight().finally(() => {
+      pendingSpacerUpdate = null;
+      if (spacerUpdateQueued) {
+        scheduleSpacerHeightUpdate();
+      }
+    });
+  });
+}
+
 function stopObservingLastMessages() {
   if (!messageResizeObserver) {
     return;
@@ -285,4 +316,4 @@ function recalculateScroll() {
   }
 }
 
-</script>
+</script>

package/dist/surfaces/web-sse/createSseEventEmitter.js

@@ -11,6 +11,7 @@ import { randomUUID } from "crypto";
 function createAgentEventStream(res, options = {}) {
     let isStreamClosed = false;
     let activeBlock = null;
+    const isAiUiMessageStream = options.vercelAiUiMessageStream === true;
     res.writeHead(200, Object.assign({ "Content-Type": "text/event-stream", "Cache-Control": "no-cache", "Connection": "keep-alive" }, (options.vercelAiUiMessageStream
         ? { "x-vercel-ai-ui-message-stream": "v1" }
         : {})));
@@ -89,7 +90,7 @@ function createAgentEventStream(res, options = {}) {
         },
         transcript(text, language) {
             stream.send({
-                type: "transcript",
+                type: isAiUiMessageStream ? "data-transcript" : "transcript",
                 data: {
                     text,
                     language,
@@ -98,7 +99,7 @@ function createAgentEventStream(res, options = {}) {
         },
         response(text, sessionId, turnId) {
             stream.send({
-                type: "response",
+                type: isAiUiMessageStream ? "data-response" : "response",
                 data: {
                     text,
                     sessionId,
@@ -108,7 +109,7 @@ function createAgentEventStream(res, options = {}) {
         },
         speechResponse(transcript, response, sessionId, turnId) {
             stream.send({
-                type: "speech-response",
+                type: isAiUiMessageStream ? "data-speech-response" : "speech-response",
                 data: {
                     transcript,
                     response,
@@ -119,7 +120,7 @@ function createAgentEventStream(res, options = {}) {
         },
         audioStart(mimeType, format, sampleRate, channelCount, bitsPerSample) {
             stream.send({
-                type: "audio-start",
+                type: isAiUiMessageStream ? "data-audio-start" : "audio-start",
                 data: {
                     mimeType,
                     format,
@@ -131,22 +132,25 @@ function createAgentEventStream(res, options = {}) {
         },
         audioDelta(value) {
             stream.send({
-                type: "audio-delta",
+                type: isAiUiMessageStream ? "data-audio-delta" : "audio-delta",
                 data: {
                     base64: Buffer.from(value).toString("base64"),
                 },
             });
         },
         audioDone() {
-            stream.send({
-                type: "audio-done",
-            });
+            stream.send(Object.assign({ type: isAiUiMessageStream ? "data-audio-done" : "audio-done" }, (isAiUiMessageStream ? { data: {} } : {})));
         },
         error(error) {
-            stream.send({
-                type: "error",
-                error,
-            });
+            stream.send(isAiUiMessageStream
+                ? {
+                    type: "error",
+                    errorText: error,
+                }
+                : {
+                    type: "error",
+                    error,
+                });
         },
         end() {
             if (isStreamClosed || res.writableEnded || res.destroyed) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@adminforth/agent",
-  "version": "1.48.0",
+  "version": "1.49.0",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "type": "module",

package/surfaces/web-sse/createSseEventEmitter.ts

@@ -22,6 +22,7 @@ function createAgentEventStream(
 ) {
   let isStreamClosed = false;
   let activeBlock: { type: "text" | "reasoning"; id: string } | null = null;
+  const isAiUiMessageStream = options.vercelAiUiMessageStream === true;
 
   res.writeHead(200, {
     "Content-Type": "text/event-stream",
@@ -124,7 +125,7 @@ function createAgentEventStream(
 
     transcript(text: string, language?: string) {
       stream.send({
-        type: "transcript",
+        type: isAiUiMessageStream ? "data-transcript" : "transcript",
         data: {
           text,
           language,
@@ -134,7 +135,7 @@ function createAgentEventStream(
 
     response(text: string, sessionId: string, turnId: string) {
       stream.send({
-        type: "response",
+        type: isAiUiMessageStream ? "data-response" : "response",
         data: {
           text,
           sessionId,
@@ -150,7 +151,7 @@ function createAgentEventStream(
       turnId: string,
     ) {
       stream.send({
-        type: "speech-response",
+        type: isAiUiMessageStream ? "data-speech-response" : "speech-response",
         data: {
           transcript,
           response,
@@ -168,7 +169,7 @@ function createAgentEventStream(
       bitsPerSample: number,
     ) {
       stream.send({
-        type: "audio-start",
+        type: isAiUiMessageStream ? "data-audio-start" : "audio-start",
         data: {
           mimeType,
           format,
@@ -181,7 +182,7 @@ function createAgentEventStream(
 
     audioDelta(value: Uint8Array) {
       stream.send({
-        type: "audio-delta",
+        type: isAiUiMessageStream ? "data-audio-delta" : "audio-delta",
         data: {
           base64: Buffer.from(value).toString("base64"),
         },
@@ -190,15 +191,21 @@ function createAgentEventStream(
 
     audioDone() {
       stream.send({
-        type: "audio-done",
+        type: isAiUiMessageStream ? "data-audio-done" : "audio-done",
+        ...(isAiUiMessageStream ? { data: {} } : {}),
       });
     },
 
     error(error: string) {
-      stream.send({
-        type: "error",
-        error,
-      });
+      stream.send(isAiUiMessageStream
+        ? {
+            type: "error",
+            errorText: error,
+          }
+        : {
+            type: "error",
+            error,
+          });
     },
 
     end() {