ak-gemini 2.1.1 → 2.1.3

package/chat.js

@@ -9,6 +9,7 @@ import log from './logger.js';
 /**
  * @typedef {import('./types').ChatOptions} ChatOptions
  * @typedef {import('./types').ChatResponse} ChatResponse
+ * @typedef {import('./types').ChatStreamEvent} ChatStreamEvent
  */
 
 /**
@@ -82,6 +83,34 @@ class Chat extends BaseGemini {
 			usage: this.getLastUsage()
 		};
 	}
+
+	/**
+	 * Send a message and stream the response as events.
+	 *
+	 * @param {string} message - The user's message
+	 * @param {Object} [opts={}] - Per-message options
+	 * @yields {ChatStreamEvent}
+	 */
+	async *stream(message, opts = {}) {
+		if (!this.chatSession) await this.init();
+
+		let fullText = '';
+		const streamResponse = await this._withRetry(() => this.chatSession.sendMessageStream({ message }));
+
+		for await (const chunk of streamResponse) {
+			if (chunk.candidates?.[0]?.content?.parts?.[0]?.text) {
+				const text = chunk.candidates[0].content.parts[0].text;
+				fullText += text;
+				yield { type: 'text', text };
+			}
+		}
+
+		yield {
+			type: 'done',
+			fullText,
+			usage: this.getLastUsage()
+		};
+	}
 }
 
 export default Chat;

package/index.cjs

@@ -1216,6 +1216,30 @@ var Chat = class extends base_default {
       usage: this.getLastUsage()
     };
   }
+  /**
+   * Send a message and stream the response as events.
+   *
+   * @param {string} message - The user's message
+   * @param {Object} [opts={}] - Per-message options
+   * @yields {ChatStreamEvent}
+   */
+  async *stream(message, opts = {}) {
+    if (!this.chatSession) await this.init();
+    let fullText = "";
+    const streamResponse = await this._withRetry(() => this.chatSession.sendMessageStream({ message }));
+    for await (const chunk of streamResponse) {
+      if (chunk.candidates?.[0]?.content?.parts?.[0]?.text) {
+        const text = chunk.candidates[0].content.parts[0].text;
+        fullText += text;
+        yield { type: "text", text };
+      }
+    }
+    yield {
+      type: "done",
+      fullText,
+      usage: this.getLastUsage()
+    };
+  }
 };
 var chat_default = Chat;
 

package/package.json

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

package/types.d.ts

@@ -421,6 +421,13 @@ export interface ChatResponse {
   usage: UsageData | null;
 }
 
+export interface ChatStreamEvent {
+  type: 'text' | 'done';
+  text?: string;
+  fullText?: string;
+  usage?: UsageData | null;
+}
+
 export interface MessageResponse {
   /** The model's text response */
   text: string;
@@ -548,6 +555,7 @@ export declare class Chat extends BaseGemini {
   constructor(options?: ChatOptions);
 
   send(message: string, opts?: { labels?: Record<string, string> }): Promise<ChatResponse>;
+  stream(message: string, opts?: { labels?: Record<string, string> }): AsyncGenerator<ChatStreamEvent, void, unknown>;
 }
 
 export declare class Message extends BaseGemini {