ak-gemini 1.0.5 → 1.0.52

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

@@ -64,9 +64,16 @@ const DEFAULT_CHAT_CONFIG = {
 	safetySettings: DEFAULT_SAFETY_SETTINGS
 };
 
+/**
+ * @typedef {import('./types').AITransformer}
+ */
+
+
+
 /**
  * main export class for AI Transformer
  * @class AITransformer
+ * @type {AITransformer}
  * @description A class that provides methods to initialize, seed, transform, and manage AI-based transformations using Google Gemini API.
  * @implements {ExportedAPI}
  */
@@ -99,7 +106,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.52",
 	"main": "index.js",
 	"files": [
 		"index.js",
@@ -10,7 +10,7 @@
 		"types.ts",
 		"logger.js"
 	],
-	"types": "types.ts",
+	"types": "types.d.ts",
 	"exports": {
 		".": {
 			"import": "./index.js",
@@ -59,4 +59,4 @@
 		"jest": "^29.7.0",
 		"nodemon": "^3.1.10"
 	}
-}
+}

package/types.ts

@@ -1,65 +0,0 @@
-import type { GoogleGenAI } from '@google/genai';
-
-export interface SafetySetting {
-  category: string; // The harm category
-  threshold: string; // The blocking threshold
-}
-
-export interface ChatConfig {
-  responseMimeType?: string; // MIME type for responses
-  temperature?: number; // Controls randomness (0.0 to 1.0)
-  topP?: number; // Controls diversity via nucleus sampling
-  topK?: number; // Controls diversity by limiting top-k tokens
-  systemInstruction?: string; // System instruction for the model
-  safetySettings?: SafetySetting[]; // Safety settings array
-  responseSchema?: Object; // Schema for validating model responses
-}
-
-export interface AITransformerContext {
-  modelName?: string;
-  systemInstructions?: string;
-  chatConfig?: ChatConfig;
-  genAI?: any;
-  chat?: any;
-  examplesFile?: string | null;
-  exampleData?: TransformationExample[] | null;
-  promptKey?: string;
-  answerKey?: string;
-  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
-  genAIClient?: GoogleGenAI; // Google GenAI client instance
-  
-}
-
-export interface TransformationExample {
-  CONTEXT?: Record<string, unknown>; // optional context for the transformation
-  PROMPT?: Record<string, unknown>; // what the user provides as input
-  ANSWER?: Record<string, unknown>; // what the model should return as output
-}
-
-export interface ExampleFileContent {
-  examples: TransformationExample[];
-}
-
-export interface AITransformerOptions {
-  modelName?: string; // The Gemini model to use
-  systemInstructions?: string; // Custom system instructions for the model
-  chatConfig?: ChatConfig; // Configuration object for the chat session
-  examplesFile?: string; // Path to JSON file containing transformation examples
-  exampleData?: TransformationExample[]; // Inline examples to seed the transformer
-  sourceKey?: string; // Key name for source data in examples
-  targetKey?: string; // Key name for target data in examples
-  contextKey?: string; // Key name for context data in examples
-  maxRetries?: number; // Maximum retry attempts for auto-retry functionality
-  retryDelay?: number; // Initial retry delay in milliseconds
-  // ? https://ai.google.dev/gemini-api/docs/structured-output
-  responseSchema?: Object; // Schema for validating model responses 
-  apiKey?: string; // API key for Google GenAI
-}
-
-// Async validator function type
-export type AsyncValidatorFunction = (payload: Record<string, unknown>) => Promise<Record<string, unknown>>;