@adminforth/agent 2.0.9 → 2.0.10

package/apiBasedTools.ts

@@ -635,6 +635,14 @@ function stringifyToolOutput(output: unknown): string {
   return YAML.stringify(output === undefined ? EMPTY_HANDLER_RESPONSE_ERROR : output);
 }
 
+function attachDurationMs(output: unknown, durationMs: number): unknown {
+  if (output !== null && typeof output === 'object' && !Array.isArray(output)) {
+    return { ...output, durationMs };
+  }
+
+  return { result: output, durationMs };
+}
+
 export function prepareApiBasedTools(
   adminforth: IAdminForth,
   hiddenResourceIds: Iterable<string> = [],
@@ -666,11 +674,13 @@ export function prepareApiBasedTools(
       input_schema: schema.request_schema,
       agent: schema.agent,
       call: async ({ adminUser, adminuser, abortSignal, inputs, userTimeZone, acceptLanguage } = {}) => {
+        const startedAt = Date.now();
+
         if (isHiddenResourceCall(hiddenResourceIdSet, inputs)) {
-          return stringifyToolOutput({
+          return stringifyToolOutput(attachDurationMs({
             error: 'RESOURCE_NOT_AVAILABLE',
             message: 'This resource is not available to the agent.',
-          });
+          }, Date.now() - startedAt));
         }
 
         const output = await callOpenApiSchema({
@@ -693,7 +703,7 @@ export function prepareApiBasedTools(
           userTimeZone,
         });
 
-        return stringifyToolOutput(processedOutput);
+        return stringifyToolOutput(attachDurationMs(processedOutput, Date.now() - startedAt));
       },
     };
   }

package/dist/apiBasedTools.js

@@ -432,6 +432,12 @@ function callOpenApiSchema(params) {
 function stringifyToolOutput(output) {
     return YAML.stringify(output === undefined ? EMPTY_HANDLER_RESPONSE_ERROR : output);
 }
+function attachDurationMs(output, durationMs) {
+    if (output !== null && typeof output === 'object' && !Array.isArray(output)) {
+        return Object.assign(Object.assign({}, output), { durationMs });
+    }
+    return { result: output, durationMs };
+}
 export function prepareApiBasedTools(adminforth, hiddenResourceIds = []) {
     const apiBasedTools = {};
     const openApiSchemas = adminforth.openApi.registeredSchemas;
@@ -451,11 +457,12 @@ export function prepareApiBasedTools(adminforth, hiddenResourceIds = []) {
             input_schema: schema.request_schema,
             agent: schema.agent,
             call: (...args_1) => __awaiter(this, [...args_1], void 0, function* ({ adminUser, adminuser, abortSignal, inputs, userTimeZone, acceptLanguage } = {}) {
+                const startedAt = Date.now();
                 if (isHiddenResourceCall(hiddenResourceIdSet, inputs)) {
-                    return stringifyToolOutput({
+                    return stringifyToolOutput(attachDurationMs({
                         error: 'RESOURCE_NOT_AVAILABLE',
                         message: 'This resource is not available to the agent.',
-                    });
+                    }, Date.now() - startedAt));
                 }
                 const output = yield callOpenApiSchema({
                     adminforth,
@@ -475,7 +482,7 @@ export function prepareApiBasedTools(adminforth, hiddenResourceIds = []) {
                     toolName,
                     userTimeZone,
                 });
-                return stringifyToolOutput(processedOutput);
+                return stringifyToolOutput(attachDurationMs(processedOutput, Date.now() - startedAt));
             }),
         };
     }

package/package.json

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