modality-kit 0.0.12 → 0.1.1

package/dist/index.js

@@ -188,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(fn, operation) {
+  return async (...args) => {
+    try {
+      return await fn(...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/schemas/schemas_tool_config.d.ts

@@ -1,6 +1,7 @@
 /**
  * TypeScript interfaces for tool configuration objects
  */
+export type AIToolExecutor = (...args: any[]) => Promise<string>;
 /**
  * Tool interface for AI SDK compatibility
  */
@@ -9,7 +10,7 @@ export interface AITool {
     annotations?: any;
     description: string;
     parameters: any;
-    execute: (parameters: any, options?: any) => Promise<any>;
+    execute: AIToolExecutor;
 }
 /**
  * Type for a collection of AI tools

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 any functions that adds consistent error handling
+ */
+export declare function withErrorHandling<T extends (...args: any[]) => Promise<any>>(fn: T, operation?: string): T;

package/package.json

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