Refactor: Rename and implement output schema transformation for tools

Author: behnamt

typescript/src/modelcontextprotocol/essentials.ts

@@ -14,7 +14,7 @@ import {Tool} from '../types/tools';
 import {contextToBulkTools} from '../shared/bulk/tools';
 import {DYNAMIC_TOOL_LOADING_THRESHOLD} from '../shared/constants';
 import {transformToolOutput} from './transform';
-import {transformTitle} from './transform/transformToolOutput';
+import {transfromOutputSchema} from './transform/transformToolOutput';
 
 class CommercetoolsAgentEssentials extends McpServer {
   private authConfig: AuthConfig;
@@ -174,14 +174,7 @@ class CommercetoolsAgentEssentials extends McpServer {
         description: tool.description,
         inputSchema: tool.parameters.shape,
         ...(this.configuration.context?.toolOutputFormat === 'json' &&
-          tool.outputSchema && {
-            outputSchema: z.object({
-              [transformTitle(tool.name)]: z.preprocess(
-                (val) => JSON.parse(val as string),
-                tool.outputSchema
-              ),
-            }).shape,
-          }),
+          transfromOutputSchema(tool)),
       },
       async (args: Record<string, unknown>) => {
         const result = await this.commercetoolsAPI.run(method, args, execute);

typescript/src/modelcontextprotocol/transform/transformToolOutput.ts

@@ -1,5 +1,6 @@
+import {Tool} from '../../types/tools';
 import {transformPropertyName} from './transformPropertyName';
-
+import {z} from 'zod';
 const emptyObjectTransformValue = 'no properties';
 const emptyArrayTransformValue = 'none';
 
@@ -13,6 +14,28 @@ const generateTabs = (tabCount: number) => {
 
 type Format = 'tabular' | 'json';
 
+/**
+ * Transforms the output schema of a tool to a JSON schema.
+ *
+ * @param {Tool} tool - The tool to transform the output schema of.
+ *
+ * @returns {Object} The transformed output schema.
+ */
+export const transfromOutputSchema = (tool: Tool) => {
+  if (!tool.outputSchema) {
+    return {};
+  }
+
+  return {
+    outputSchema: z.object({
+      [transformTitle(tool.name)]: z.preprocess(
+        (val) => JSON.parse(val as string),
+        tool.outputSchema
+      ),
+    }).shape,
+  };
+};
+
 /**
  * A method to strigify tool output into a LLM friendly and optimised format.
  *
@@ -58,7 +81,7 @@ export const transformToolOutput = (args: {
     transformedData ?? emptyObjectTransformValue);
 };
 
-export const transformTitle = (title: string) =>
+const transformTitle = (title: string) =>
   `${transformPropertyName(title).toUpperCase()}`;
 
 /**

typescript/src/shared/business-unit/output-schema.ts

@@ -1,8 +1,5 @@
 import {z} from 'zod';
 
-// Generated schemas for business-unit
-// DO NOT EDIT - This file is auto-generated
-
 // Base BusinessUnit schema
 export const businessUnitSchema = z.union([z.any(), z.any()]);
 

typescript/src/shared/cart-discount/output-schema.ts

@@ -1,8 +1,5 @@
 import {z} from 'zod';
 
-// Generated schemas for cart-discount
-// DO NOT EDIT - This file is auto-generated
-
 // Base CartDiscount schema
 export const cartDiscountSchema = z
   .object({

typescript/src/shared/cart/output-schema.ts

@@ -1,8 +1,5 @@
 import {z} from 'zod';
 
-// Generated schemas for cart
-// DO NOT EDIT - This file is auto-generated
-
 // Base Cart schema
 export const cartSchema = z
   .object({

typescript/src/shared/category/output-schema.ts

@@ -1,8 +1,5 @@
 import {z} from 'zod';
 
-// Generated schemas for category
-// DO NOT EDIT - This file is auto-generated
-
 // Base Category schema
 export const categorySchema = z
   .object({

typescript/src/shared/channel/output-schema.ts

@@ -1,8 +1,5 @@
 import {z} from 'zod';
 
-// Generated schemas for channel
-// DO NOT EDIT - This file is auto-generated
-
 // Base Channel schema
 export const channelSchema = z
   .object({

typescript/src/shared/custom-objects/output-schema.ts

@@ -1,8 +1,5 @@
 import {z} from 'zod';
 
-// Generated schemas for custom-objects
-// DO NOT EDIT - This file is auto-generated
-
 // Base CustomObject schema
 export const customObjectsSchema = z
   .object({

typescript/src/shared/customer-group/output-schema.ts

@@ -1,8 +1,5 @@
 import {z} from 'zod';
 
-// Generated schemas for customer-group
-// DO NOT EDIT - This file is auto-generated
-
 // Base CustomerGroup schema
 export const customerGroupSchema = z
   .object({

typescript/src/shared/customer/output-schema.ts

@@ -1,8 +1,5 @@
 import {z} from 'zod';
 
-// Generated schemas for customer
-// DO NOT EDIT - This file is auto-generated
-
 // Base Customer schema
 export const customerSchema = z
   .object({