@shareai-lab/kode-sdk 1.0.0-beta.0 → 1.0.0-beta.2

package/dist/core/agent.js

@@ -458,18 +458,75 @@ class Agent {
         this.events.emitEvent({ type: 'state', state: 'BUSY' });
         try {
             await this.hooks.runPreModel(this.messages);
-            const response = await this.provider.complete(this.messages, {
+            // Stream response
+            const contentBlocks = [];
+            let usage;
+            const stream = this.provider.stream(this.messages, {
                 tools: this.getToolSchemas(),
                 system: this.system,
                 maxTokens: this.maxTokens,
                 temperature: this.temperature,
             });
+            let currentTextIndex = -1;
+            let currentToolIndex = -1;
+            for await (const chunk of stream) {
+                if (chunk.type === 'content_block_start') {
+                    if (chunk.content_block?.type === 'text') {
+                        currentTextIndex = chunk.index;
+                        contentBlocks[currentTextIndex] = { type: 'text', text: '' };
+                    }
+                    else if (chunk.content_block?.type === 'tool_use') {
+                        currentToolIndex = chunk.index;
+                        contentBlocks[currentToolIndex] = {
+                            type: 'tool_use',
+                            id: chunk.content_block.id,
+                            name: chunk.content_block.name,
+                            input: {},
+                        };
+                    }
+                }
+                else if (chunk.type === 'content_block_delta') {
+                    if (chunk.delta?.type === 'text_delta') {
+                        const text = chunk.delta.text || '';
+                        if (currentTextIndex >= 0) {
+                            contentBlocks[currentTextIndex].text += text;
+                            // Emit text_chunk event
+                            this.events.emitEvent({ type: 'text_chunk', delta: text });
+                        }
+                    }
+                    else if (chunk.delta?.type === 'input_json_delta') {
+                        const json = chunk.delta.partial_json || '';
+                        if (currentToolIndex >= 0) {
+                            const block = contentBlocks[currentToolIndex];
+                            const currentJson = JSON.stringify(block.input || {});
+                            const prefix = currentJson === '{}' ? '' : currentJson.slice(0, -1);
+                            try {
+                                block.input = JSON.parse(prefix + json);
+                            }
+                            catch {
+                                // Partial JSON, will complete in next chunk
+                            }
+                        }
+                    }
+                }
+                else if (chunk.type === 'message_delta' && chunk.usage) {
+                    usage = {
+                        input_tokens: usage?.input_tokens || 0,
+                        output_tokens: (usage?.output_tokens || 0) + chunk.usage.output_tokens
+                    };
+                }
+            }
+            const response = {
+                role: 'assistant',
+                content: contentBlocks,
+                usage,
+            };
             await this.hooks.runPostModel(response);
             this.messages.push({
                 role: 'assistant',
                 content: response.content,
             });
-            // Emit text events
+            // Emit complete text events
             const textBlocks = response.content.filter((c) => c.type === 'text');
             for (const block of textBlocks) {
                 this.events.emitEvent({ type: 'text', text: block.text });

package/dist/core/events.js

@@ -55,16 +55,22 @@ class EventBus extends events_1.EventEmitter {
                 }
             }
         }
+        const self = this;
         return {
             [Symbol.asyncIterator]: () => ({
                 next: async () => {
                     const event = await subscriber.next();
                     if (!event) {
-                        this.subscribers.delete(subscriber);
+                        self.subscribers.delete(subscriber);
                         return { done: true, value: undefined };
                     }
                     return { done: false, value: event };
                 },
+                return: async () => {
+                    subscriber.close();
+                    self.subscribers.delete(subscriber);
+                    return { done: true, value: undefined };
+                },
             }),
         };
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shareai-lab/kode-sdk",
-  "version": "1.0.0-beta.0",
+  "version": "1.0.0-beta.2",
   "description": "Event-driven Agent Model Client SDK for building long-running, collaborative AI agents",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -32,6 +32,7 @@
   "author": "",
   "license": "MIT",
   "devDependencies": {
+    "@shareai-lab/kode-sdk": "^1.0.0-beta.1",
     "@types/node": "^20.0.0",
     "ts-node": "^10.9.0",
     "typescript": "^5.3.0"
@@ -39,6 +40,9 @@
   "engines": {
     "node": ">=18.0.0"
   },
+  "publishConfig": {
+    "access": "public"
+  },
   "files": [
     "dist",
     "README.md",