ak-claude 0.0.5 → 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;