taist 0.1.7 → 0.1.8

package/instrument.js

@@ -46,7 +46,8 @@ import {
   instrumentDirectory,
   instrumentModules,
   wrapWithContext,
-  instrumentClassWithContext
+  instrumentClassWithContext,
+  instrumentModule
 } from './lib/instrument-all.js';
 
 // Initialize global reporter for cross-process trace collection
@@ -277,6 +278,7 @@ export {
   instrumentAll,
   instrumentDirectory,
   instrumentModules,
+  instrumentModule,
   wrapWithContext,
   startTrace,
   getContext,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "taist",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "description": "Token-Optimized Testing Framework for AI-Assisted Development",
   "main": "index.js",
   "type": "module",

package/types/instrument.d.ts

@@ -199,6 +199,28 @@ export declare function instrumentModules(
   options?: InstrumentAllOptions
 ): Promise<Record<string, unknown>>;
 
+/**
+ * Instrument all exports from a module object
+ *
+ * Wraps all function exports with context-aware tracing.
+ * Classes are wrapped so new instances are automatically instrumented.
+ *
+ * @param moduleExports Module exports object (e.g., from `import * as mod from './mod.js'`)
+ * @param moduleName Module name prefix for trace names
+ * @returns Object with same keys but instrumented values
+ *
+ * @example
+ * import { instrumentModule } from 'taist/instrument';
+ * import * as orderServices from './services/order.js';
+ *
+ * export const Order = instrumentModule(orderServices, 'Order');
+ * // All functions in Order will now be traced
+ */
+export declare function instrumentModule<T extends Record<string, unknown>>(
+  moduleExports: T,
+  moduleName: string
+): T;
+
 /**
  * Wrap a function with trace context
  * @param fn Function to wrap