ak-gemini 1.0.5 → 1.0.51

package/index.cjs

@@ -29,6 +29,7 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // index.js
 var index_exports = {};
 __export(index_exports, {
+  AITransformer: () => AITransformer,
   default: () => AITransformer,
   log: () => logger_default
 });
@@ -374,5 +375,6 @@ if (import_meta.url === new URL(`file://${process.argv[1]}`).href) {
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  AITransformer,
   log
 });

package/index.js

@@ -99,7 +99,7 @@ export default class AITransformer {
 		this.estimate = estimateTokenUsage.bind(this);
 	}
 }
-
+export { AITransformer };
 /**
  * factory function to create an AI Transformer instance
  * @param {AITransformerOptions} [options={}] - Configuration options for the transformer

package/package.json

@@ -2,7 +2,7 @@
 	"name": "ak-gemini",
 	"author": "ak@mixpanel.com",
 	"description": "AK's Generative AI Helper for doing... transforms",
-	"version": "1.0.5",
+	"version": "1.0.51",
 	"main": "index.js",
 	"files": [
 		"index.js",

package/types.ts

@@ -28,9 +28,9 @@ export interface AITransformerContext {
   contextKey?: string;
   maxRetries?: number;
   retryDelay?: number;
-  init: () => Promise<void>; // Initialization function
-  seed: () => Promise<void>; // Function to seed the transformer with examples  
-  message: (payload: Record<string, unknown>) => Promise<Record<string, unknown>>; // Function to send messages to the model
+  init?: () => Promise<void>; // Initialization function
+  seed?: () => Promise<void>; // Function to seed the transformer with examples  
+  message?: (payload: Record<string, unknown>) => Promise<Record<string, unknown>>; // Function to send messages to the model
   genAIClient?: GoogleGenAI; // Google GenAI client instance
   
 }
@@ -62,4 +62,7 @@ export interface AITransformerOptions {
 }
 
 // Async validator function type
-export type AsyncValidatorFunction = (payload: Record<string, unknown>) => Promise<Record<string, unknown>>;
+export type AsyncValidatorFunction = (payload: Record<string, unknown>) => Promise<Record<string, unknown>>;
+
+
+export declare class AITransformer implements AITransformerContext {}