neuro-vault-mcp 3.3.0 → 3.3.1

package/dist/cli.js

@@ -14275,7 +14275,7 @@ function unwrap(schema) {
 function isPlainObject2(value) {
   return typeof value === "object" && value !== null && !Array.isArray(value);
 }
-function coerceField(schema, value) {
+function coerceFieldValue(schema, value) {
   if (value === null || value === void 0) return value;
   const inner = unwrap(schema);
   if (inner instanceof external_exports.ZodNumber) {
@@ -14303,18 +14303,26 @@ function coerceField(schema, value) {
   }
   return value;
 }
-function coerceInput(schema, value) {
-  const inner = unwrap(schema);
-  if (!(inner instanceof external_exports.ZodObject)) return value;
-  if (!isPlainObject2(value)) return value;
-  const shape = inner.shape;
-  const out = { ...value };
-  for (const key of Object.keys(shape)) {
-    if (key in out) {
-      out[key] = coerceField(shape[key], out[key]);
-    }
+function wrapField(field) {
+  let inner = field;
+  let isOptional = false;
+  while (inner instanceof external_exports.ZodOptional || inner instanceof external_exports.ZodNullable || inner instanceof external_exports.ZodDefault) {
+    if (inner instanceof external_exports.ZodOptional) isOptional = true;
+    const next = inner.unwrap();
+    if (!next) break;
+    inner = next;
   }
-  return out;
+  const wrapped = external_exports.preprocess((v) => coerceFieldValue(inner, v), inner);
+  return isOptional ? wrapped.optional() : wrapped;
+}
+function wrapSchemaWithCoercion(schema) {
+  if (!(schema instanceof external_exports.ZodObject)) return schema;
+  const shape = schema.shape;
+  const newShape = {};
+  for (const [key, field] of Object.entries(shape)) {
+    newShape[key] = wrapField(field);
+  }
+  return external_exports.object(newShape);
 }
 
 // src/lib/tool-response.ts
@@ -14385,18 +14393,18 @@ function formatZodError(error48) {
   return { message, issues };
 }
 function registerTool(tool) {
+  const coercingSchema = wrapSchemaWithCoercion(tool.inputSchema);
   return {
     name: tool.name,
     spec: {
       ...tool.title !== void 0 ? { title: tool.title } : {},
       description: tool.description,
-      inputSchema: tool.inputSchema,
+      inputSchema: coercingSchema,
       ...tool.outputSchema !== void 0 ? { outputSchema: tool.outputSchema } : {},
       ...tool.annotations !== void 0 ? { annotations: tool.annotations } : {}
     },
     handler: async (args) => invokeTool(async () => {
-      const coerced = coerceInput(tool.inputSchema, args);
-      const result = tool.inputSchema.safeParse(coerced);
+      const result = coercingSchema.safeParse(args);
       if (!result.success) {
         const { message, issues } = formatZodError(result.error);
         throw new ToolHandlerError("INVALID_PARAMS", message, { details: { issues } });