ak-gemini 1.0.9 → 1.0.10

package/index.cjs

@@ -134,6 +134,7 @@ var AITransformer = class {
     this.transformWithValidation = prepareAndValidateMessage.bind(this);
     this.estimate = estimateTokenUsage.bind(this);
     this.estimateTokenUsage = estimateTokenUsage.bind(this);
+    this.updateSystemInstructions = updateSystemInstructions.bind(this);
   }
 };
 var index_default = AITransformer;
@@ -500,6 +501,15 @@ function getChatHistory() {
   }
   return this.chat.getHistory();
 }
+async function updateSystemInstructions(newInstructions) {
+  if (!newInstructions || typeof newInstructions !== "string") {
+    throw new Error("System instructions must be a non-empty string");
+  }
+  this.systemInstructions = newInstructions.trim();
+  this.chatConfig.systemInstruction = this.systemInstructions;
+  logger_default.debug("Updating system instructions and reinitializing chat...");
+  await this.init(true);
+}
 function attemptJSONRecovery(text, maxAttempts = 100) {
   if (!text || typeof text !== "string") return null;
   try {

package/index.js

@@ -134,6 +134,7 @@ class AITransformer {
 		this.transformWithValidation = prepareAndValidateMessage.bind(this);
 		this.estimate = estimateTokenUsage.bind(this);
 		this.estimateTokenUsage = estimateTokenUsage.bind(this);
+		this.updateSystemInstructions = updateSystemInstructions.bind(this);
 	}
 }
 
@@ -714,6 +715,24 @@ function getChatHistory() {
 	return this.chat.getHistory();
 }
 
+/**
+ * Updates system instructions and reinitializes the chat session
+ * @this {ExportedAPI}
+ * @param {string} newInstructions - The new system instructions
+ * @returns {Promise<void>}
+ */
+async function updateSystemInstructions(newInstructions) {
+	if (!newInstructions || typeof newInstructions !== 'string') {
+		throw new Error('System instructions must be a non-empty string');
+	}
+
+	this.systemInstructions = newInstructions.trim();
+	this.chatConfig.systemInstruction = this.systemInstructions;
+
+	log.debug('Updating system instructions and reinitializing chat...');
+	await this.init(true); // Force reinitialize with new instructions
+}
+
 
 /*
 ----

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.9",
+	"version": "1.0.10",
 	"main": "index.js",
 	"files": [
 		"index.js",

package/types.d.ts

@@ -139,8 +139,8 @@ export declare class AITransformer {
   getHistory(): Array<any>;
   estimateTokenUsage(nextPayload: Record<string, unknown> | string): Promise<{ totalTokens: number; breakdown?: any }>;
   estimate(nextPayload: Record<string, unknown> | string): Promise<{ totalTokens: number; breakdown?: any }>;
+  updateSystemInstructions(newInstructions: string): Promise<void>;
 }
 
 // Default export
-declare const _default: typeof AITransformer;
-export default _default;
+export default AITransformer;