ak-gemini 1.0.6 → 1.0.7

package/index.cjs

@@ -29,6 +29,7 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // index.js
 var index_exports = {};
 __export(index_exports, {
+  ThinkingLevel: () => import_genai.ThinkingLevel,
   default: () => index_default,
   log: () => logger_default
 });
@@ -64,7 +65,7 @@ var DEFAULT_SAFETY_SETTINGS = [
 var DEFAULT_SYSTEM_INSTRUCTIONS = `
 You are an expert JSON transformation engine. Your task is to accurately convert data payloads from one format to another.
 
-You will be provided with example transformations (Source JSON -> Target JSON). 
+You will be provided with example transformations (Source JSON -> Target JSON).
 
 Learn the mapping rules from these examples.
 
@@ -72,8 +73,21 @@ When presented with new Source JSON, apply the learned transformation rules to p
 
 Always respond ONLY with a valid JSON object that strictly adheres to the expected output format.
 
-Do not include any additional text, explanations, or formatting before or after the JSON object. 
+Do not include any additional text, explanations, or formatting before or after the JSON object.
 `;
+var DEFAULT_THINKING_CONFIG = {
+  thinkingBudget: 0,
+  thinkingLevel: import_genai.ThinkingLevel.MINIMAL
+};
+var THINKING_SUPPORTED_MODELS = [
+  /^gemini-3-flash(-preview)?$/,
+  /^gemini-3-pro(-preview|-image-preview)?$/,
+  /^gemini-2\.5-pro/,
+  /^gemini-2\.5-flash(-preview)?$/,
+  /^gemini-2\.5-flash-lite(-preview)?$/,
+  /^gemini-2\.0-flash$/
+  // Experimental support, exact match only
+];
 var DEFAULT_CHAT_CONFIG = {
   responseMimeType: "application/json",
   temperature: 0.2,
@@ -152,6 +166,23 @@ function AITransformFactory(options = {}) {
     ...options.chatConfig,
     systemInstruction: this.systemInstructions
   };
+  const modelSupportsThinking = THINKING_SUPPORTED_MODELS.some(
+    (pattern) => pattern.test(this.modelName)
+  );
+  if (modelSupportsThinking && options.thinkingConfig) {
+    const thinkingConfig = {
+      ...DEFAULT_THINKING_CONFIG,
+      ...options.thinkingConfig
+    };
+    this.chatConfig.thinkingConfig = thinkingConfig;
+    if (logger_default.level !== "silent") {
+      logger_default.debug(`Model ${this.modelName} supports thinking. Applied thinkingConfig:`, thinkingConfig);
+    }
+  } else if (options.thinkingConfig && !modelSupportsThinking) {
+    if (logger_default.level !== "silent") {
+      logger_default.warn(`Model ${this.modelName} does not support thinking features. Ignoring thinkingConfig.`);
+    }
+  }
   if (options.responseSchema) {
     this.chatConfig.responseSchema = options.responseSchema;
   }
@@ -572,5 +603,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 = {
+  ThinkingLevel,
   log
 });

package/index.js

@@ -23,11 +23,12 @@ const { NODE_ENV = "unknown", GEMINI_API_KEY, LOG_LEVEL = "" } = process.env;
 
 
 //deps
-import { GoogleGenAI, HarmCategory, HarmBlockThreshold } from '@google/genai';
+import { GoogleGenAI, HarmCategory, HarmBlockThreshold, ThinkingLevel } from '@google/genai';
 import u from 'ak-tools';
 import path from 'path';
 import log from './logger.js';
 export { log };
+export { ThinkingLevel };
 
 
 
@@ -40,7 +41,7 @@ const DEFAULT_SAFETY_SETTINGS = [
 const DEFAULT_SYSTEM_INSTRUCTIONS = `
 You are an expert JSON transformation engine. Your task is to accurately convert data payloads from one format to another.
 
-You will be provided with example transformations (Source JSON -> Target JSON). 
+You will be provided with example transformations (Source JSON -> Target JSON).
 
 Learn the mapping rules from these examples.
 
@@ -48,9 +49,25 @@ When presented with new Source JSON, apply the learned transformation rules to p
 
 Always respond ONLY with a valid JSON object that strictly adheres to the expected output format.
 
-Do not include any additional text, explanations, or formatting before or after the JSON object. 
+Do not include any additional text, explanations, or formatting before or after the JSON object.
 `;
 
+const DEFAULT_THINKING_CONFIG = {
+	thinkingBudget: 0,
+	thinkingLevel: ThinkingLevel.MINIMAL
+};
+
+// Models that support thinking features (as of Dec 2024)
+// Using regex patterns for more precise matching
+const THINKING_SUPPORTED_MODELS = [
+	/^gemini-3-flash(-preview)?$/,
+	/^gemini-3-pro(-preview|-image-preview)?$/,
+	/^gemini-2\.5-pro/,
+	/^gemini-2\.5-flash(-preview)?$/,
+	/^gemini-2\.5-flash-lite(-preview)?$/,
+	/^gemini-2\.0-flash$/ // Experimental support, exact match only
+];
+
 const DEFAULT_CHAT_CONFIG = {
 	responseMimeType: 'application/json',
 	temperature: 0.2,
@@ -161,6 +178,7 @@ function AITransformFactory(options = {}) {
 
 	this.apiKey = options.apiKey !== undefined && options.apiKey !== null ? options.apiKey : GEMINI_API_KEY;
 	if (!this.apiKey) throw new Error("Missing Gemini API key. Provide via options.apiKey or GEMINI_API_KEY env var.");
+
 	// Build chat config, making sure systemInstruction uses the custom instructions
 	this.chatConfig = {
 		...DEFAULT_CHAT_CONFIG,
@@ -168,6 +186,28 @@ function AITransformFactory(options = {}) {
 		systemInstruction: this.systemInstructions
 	};
 
+	// Only add thinkingConfig if the model supports it
+	const modelSupportsThinking = THINKING_SUPPORTED_MODELS.some(pattern =>
+		pattern.test(this.modelName)
+	);
+
+	if (modelSupportsThinking && options.thinkingConfig) {
+		// Handle thinkingConfig - merge with defaults
+		const thinkingConfig = {
+			...DEFAULT_THINKING_CONFIG,
+			...options.thinkingConfig
+		};
+		this.chatConfig.thinkingConfig = thinkingConfig;
+
+		if (log.level !== 'silent') {
+			log.debug(`Model ${this.modelName} supports thinking. Applied thinkingConfig:`, thinkingConfig);
+		}
+	} else if (options.thinkingConfig && !modelSupportsThinking) {
+		if (log.level !== 'silent') {
+			log.warn(`Model ${this.modelName} does not support thinking features. Ignoring thinkingConfig.`);
+		}
+	}
+
 	// response schema is optional, but if provided, it should be a valid JSON schema
 	if (options.responseSchema) {
 		this.chatConfig.responseSchema = options.responseSchema;

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.6",
+	"version": "1.0.7",
 	"main": "index.js",
 	"files": [
 		"index.js",
@@ -33,7 +33,7 @@
 	"homepage": "https://github.com/ak--47/ak-gemini#readme",
 	"scripts": {
 		"prepublishOnly": "npm run typecheck && npm run build:cjs",
-		"post": "npm publish --access public",
+		"post": "npm publish",
 		"release": "npm version patch && npm publish --access public",
 		"update-deps": "npx npm-check-updates -u && npm install",
 		"prune": "rm -rf tmp/*",
@@ -41,7 +41,8 @@
 		"test:unit": "npm test -- tests/module.test.js",
 		"build:cjs": "esbuild index.js --bundle --platform=node --format=cjs --outfile=index.cjs --external:@google/genai --external:ak-tools --external:dotenv --external:pino-pretty --external:pino",
 		"coverage": "node --no-warnings --experimental-vm-modules node_modules/jest/bin/jest.js --coverage",
-		"typecheck": "tsc --noEmit"
+		"typecheck": "tsc --noEmit",
+		"update:gemini": "npm install @google/genai@latest"
 	},
 	"type": "module",
 	"keywords": [
@@ -51,7 +52,7 @@
 	],
 	"license": "ISC",
 	"dependencies": {
-		"@google/genai": "^1.25.0",
+		"@google/genai": "^1.34.0",
 		"ak-tools": "^1.1.12",
 		"dotenv": "^16.5.0",
 		"pino": "^9.7.0",

package/types.d.ts

@@ -1,4 +1,15 @@
-import type { GoogleGenAI } from '@google/genai';
+import type { GoogleGenAI, ThinkingLevel } from '@google/genai';
+
+export { ThinkingLevel };
+
+export interface ThinkingConfig {
+  /** Indicates whether to include thoughts in the response. If true, thoughts are returned only if the model supports thought and thoughts are available. */
+  includeThoughts?: boolean;
+  /** Indicates the thinking budget in tokens. 0 is DISABLED. -1 is AUTOMATIC. The default values and allowed ranges are model dependent. */
+  thinkingBudget?: number;
+  /** Optional. The number of thoughts tokens that the model should generate. */
+  thinkingLevel?: ThinkingLevel;
+}
 
 export interface SafetySetting {
   category: string; // The harm category
@@ -13,6 +24,7 @@ export interface ChatConfig {
   systemInstruction?: string; // System instruction for the model
   safetySettings?: SafetySetting[]; // Safety settings array
   responseSchema?: Object; // Schema for validating model responses
+  thinkingConfig?: ThinkingConfig; // Thinking features configuration
   [key: string]: any; // Additional properties for flexibility
 }
 
@@ -61,6 +73,7 @@ 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
+  thinkingConfig?: ThinkingConfig; // Thinking features configuration (defaults to thinkingBudget: 0, thinkingLevel: "MINIMAL")
   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 (alias for promptKey)