@adminforth/agent 1.26.7 → 1.26.8

package/apiBasedTools.ts

@@ -23,6 +23,7 @@ type CookieItem = {
 type ToolOverrideCallParams = Pick<ApiBasedToolCallParams, 'httpExtra' | 'inputs' | 'userTimeZone'>;
 
 type ToolOverrideContext = {
+  adminforth: IAdminForth;
   output?: unknown;
   adminUser?: AdminUser;
   httpExtra?: Partial<HttpExtra>;
@@ -38,15 +39,6 @@ type ToolOverride = {
   post_process_response?: (params: ToolOverrideContext) => Promise<unknown> | unknown;
 };
 
-type GetResourceToolResponse = {
-  resource: {
-    columns: Array<{
-      name: string;
-      type?: string;
-    }>;
-  };
-};
-
 type GetResourceDataToolResponse = {
   data: Array<Record<string, unknown>>;
   total?: number;
@@ -107,16 +99,12 @@ const TOOL_OVERRIDES: Record<string, ToolOverride> = {
     format_tool: ({ inputs, resourceLabel }) => (
       `Get ${getDataPrefix(inputs)}${resourceLabel}`
     ),
-    post_process_response: async ({ output, inputs, invokeTool, userTimeZone }) => {
+    post_process_response: async ({ adminforth, output, inputs, userTimeZone }) => {
       if (hasToolError(output)) {
         return output;
       }
 
-      const resourceId = inputs?.resourceId as string;
-      const getResourceOutput = await invokeTool('get_resource', {
-        inputs: { resourceId },
-      });
-      const dateTimeColumnNames = getDateTimeColumnNames(getResourceOutput);
+      const dateTimeColumnNames = getDateTimeColumnNames(adminforth, inputs);
 
       if (dateTimeColumnNames.length === 0) {
         return output;
@@ -300,10 +288,18 @@ function hasToolError(output: unknown): output is { error: unknown } {
   return typeof output === 'object' && output !== null && 'error' in output;
 }
 
-function getDateTimeColumnNames(output: unknown): string[] {
-  const resource = (output as GetResourceToolResponse).resource;
+function getDateTimeColumnNames(
+  adminforth: IAdminForth,
+  inputs: Record<string, unknown> | undefined,
+): string[] {
+  const resourceId = getInputString(inputs, 'resourceId');
+  const resource = adminforth.config.resources.find((res) => res.resourceId === resourceId);
+
+  if (!resource) {
+    return [];
+  }
 
-  return resource.columns
+  return resource.dataSourceColumns
     .filter((column) => column.type === AdminForthDataTypes.DATETIME)
     .map((column) => column.name);
 }
@@ -378,6 +374,7 @@ async function applyToolOverride(params: {
   }
 
   const postProcessedOutput = await override.post_process_response({
+    adminforth,
     output: sanitizedOutput,
     adminUser,
     httpExtra,
@@ -450,6 +447,7 @@ export async function formatApiBasedToolCall(params: {
   const formatTool = TOOL_OVERRIDES[params.toolName]?.format_tool;
 
   return await formatTool?.({
+    adminforth: params.adminforth,
     adminUser: params.adminUser,
     httpExtra: params.httpExtra,
     inputs: params.inputs,

package/dist/apiBasedTools.js

@@ -54,15 +54,11 @@ const TOOL_OVERRIDES = {
     },
     get_resource_data: {
         format_tool: ({ inputs, resourceLabel }) => (`Get ${getDataPrefix(inputs)}${resourceLabel}`),
-        post_process_response: (_a) => __awaiter(void 0, [_a], void 0, function* ({ output, inputs, invokeTool, userTimeZone }) {
+        post_process_response: (_a) => __awaiter(void 0, [_a], void 0, function* ({ adminforth, output, inputs, userTimeZone }) {
             if (hasToolError(output)) {
                 return output;
             }
-            const resourceId = inputs === null || inputs === void 0 ? void 0 : inputs.resourceId;
-            const getResourceOutput = yield invokeTool('get_resource', {
-                inputs: { resourceId },
-            });
-            const dateTimeColumnNames = getDateTimeColumnNames(getResourceOutput);
+            const dateTimeColumnNames = getDateTimeColumnNames(adminforth, inputs);
             if (dateTimeColumnNames.length === 0) {
                 return output;
             }
@@ -196,9 +192,13 @@ function wipePath(target, pathParts) {
 function hasToolError(output) {
     return typeof output === 'object' && output !== null && 'error' in output;
 }
-function getDateTimeColumnNames(output) {
-    const resource = output.resource;
-    return resource.columns
+function getDateTimeColumnNames(adminforth, inputs) {
+    const resourceId = getInputString(inputs, 'resourceId');
+    const resource = adminforth.config.resources.find((res) => res.resourceId === resourceId);
+    if (!resource) {
+        return [];
+    }
+    return resource.dataSourceColumns
         .filter((column) => column.type === AdminForthDataTypes.DATETIME)
         .map((column) => column.name);
 }
@@ -242,6 +242,7 @@ function applyToolOverride(params) {
             return sanitizedOutput;
         }
         const postProcessedOutput = yield override.post_process_response({
+            adminforth,
             output: sanitizedOutput,
             adminUser,
             httpExtra,
@@ -301,6 +302,7 @@ export function formatApiBasedToolCall(params) {
         var _a;
         const formatTool = (_a = TOOL_OVERRIDES[params.toolName]) === null || _a === void 0 ? void 0 : _a.format_tool;
         return yield (formatTool === null || formatTool === void 0 ? void 0 : formatTool({
+            adminforth: params.adminforth,
             adminUser: params.adminUser,
             httpExtra: params.httpExtra,
             inputs: params.inputs,

package/package.json

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