modality-kit 0.0.11 → 0.1.0

package/dist/index.js

@@ -117,6 +117,8 @@ class ModalityLogger {
     if (categroy) {
       prefix += ` [${categroy}]`;
     }
+    console.log(`
+`);
     console.log(prefix);
     return payload;
   }
@@ -162,18 +164,17 @@ class ModalityLogger {
         break;
     }
     console.log(`
-
 `);
   }
   debug(message, error) {
     this.log("debug", { message, error });
   }
   info(message, data) {
-    const log = { message };
+    const payload = { message };
     if (data) {
-      log.data = data;
+      payload.data = data;
     }
-    this.log("info", { message, data });
+    this.log("info", payload);
   }
   warn(message, resolution) {
     this.log("warn", { message, resolution });
@@ -187,10 +188,42 @@ class ModalityLogger {
   }
 }
 var loggerInstance = ModalityLogger.getInstance.bind(ModalityLogger);
+// src/util_error.ts
+var logger = loggerInstance("Error");
+
+class ErrorCode extends Error {
+  constructor(message) {
+    super(message);
+    this.name = this.constructor.name;
+  }
+}
+function withErrorHandling(taskFn, operation) {
+  return async (...args) => {
+    try {
+      return await taskFn(...args);
+    } catch (error) {
+      if (error instanceof ErrorCode) {
+        logger.error(`${operation || "Task operation"} failed: ${error.code}`, {
+          message: error.message,
+          code: error.code
+        });
+      } else if (error instanceof Error) {
+        logger.error(`${operation || "Error"} failed`, {
+          message: error.message
+        });
+      } else {
+        logger.error(`${operation || "Exception"} unexpected error:`, error);
+      }
+      return formatErrorResponse(error, operation);
+    }
+  };
+}
 export {
+  withErrorHandling,
   setupAITools,
   loggerInstance,
   formatSuccessResponse,
   formatErrorResponse,
-  ModalityLogger
+  ModalityLogger,
+  ErrorCode
 };

package/dist/types/index.d.ts

@@ -2,3 +2,4 @@ export { setupAITools } from "./util_mcp_tools_converter";
 export type { AITools, AITool } from "./schemas/schemas_tool_config";
 export { formatErrorResponse, formatSuccessResponse } from "./util_response";
 export { ModalityLogger, loggerInstance } from "./util_logger";
+export { withErrorHandling, ErrorCode } from "./util_error";

package/dist/types/util_error.d.ts

@@ -0,0 +1,11 @@
+/**
+ * Base class for task-related errors
+ */
+export declare abstract class ErrorCode extends Error {
+    abstract readonly code: string;
+    constructor(message: string);
+}
+/**
+ * Wrapper for MCP task functions that adds consistent error handling
+ */
+export declare function withErrorHandling<T extends (...args: any[]) => Promise<string>>(taskFn: T, operation?: string): T;

package/package.json

@@ -1,5 +1,5 @@
 {
-  "version": "0.0.11",
+  "version": "0.1.0",
   "name": "modality-kit",
   "repository": {
     "type": "git",