ak-claude 0.0.7 → 0.0.9

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
  */
 
 /**
@@ -71,6 +72,37 @@ class Chat extends BaseClaude {
 			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._initialized) await this.init();
+
+		let fullText = '';
+		const stream = await this._streamMessage(message, opts);
+		const finalMessage = await stream.finalMessage();
+
+		for (const block of finalMessage.content) {
+			if (block.type === 'text') {
+				fullText += block.text;
+				yield { type: 'text', text: block.text };
+			}
+		}
+
+		this.history.push({ role: 'assistant', content: finalMessage.content });
+		this._captureMetadata(finalMessage);
+
+		yield {
+			type: 'done',
+			fullText,
+			usage: this.getLastUsage()
+		};
+	}
 }
 
 export default Chat;

package/index.cjs

@@ -1166,6 +1166,32 @@ 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._initialized) await this.init();
+    let fullText = "";
+    const stream = await this._streamMessage(message, opts);
+    const finalMessage = await stream.finalMessage();
+    for (const block of finalMessage.content) {
+      if (block.type === "text") {
+        fullText += block.text;
+        yield { type: "text", text: block.text };
+      }
+    }
+    this.history.push({ role: "assistant", content: finalMessage.content });
+    this._captureMetadata(finalMessage);
+    yield {
+      type: "done",
+      fullText,
+      usage: this.getLastUsage()
+    };
+  }
 };
 var chat_default = Chat;
 

package/package.json

@@ -2,7 +2,7 @@
 	"name": "ak-claude",
 	"author": "ak@mixpanel.com",
 	"description": "AK's Claude AI Helper for doing... everything",
-	"version": "0.0.7",
+	"version": "0.0.9",
 	"main": "index.js",
 	"files": [
 		"index.js",

package/types.d.ts

@@ -394,6 +394,13 @@ export interface ChatResponse {
   usage: UsageData | null;
 }
 
+export interface ChatStreamEvent {
+  type: 'text' | 'done';
+  text?: string;
+  fullText?: string;
+  usage?: UsageData | null;
+}
+
 export interface MessageResponse {
   text: string;
   data?: any;
@@ -527,6 +534,7 @@ export declare class Chat extends BaseClaude {
   constructor(options?: ChatOptions);
 
   send(message: string, opts?: Record<string, any>): Promise<ChatResponse>;
+  stream(message: string, opts?: Record<string, any>): AsyncGenerator<ChatStreamEvent, void, unknown>;
 }
 
 export declare class Message extends BaseClaude {