pentesting 0.8.23 → 0.8.25

package/dist/index.js

@@ -5079,24 +5079,49 @@ ${this.currentSpec.systemPrompt}`;
       systemPrompt = buildAgentSystemPrompt(systemPrompt, this.currentAgent);
     }
     this.emit(AGENT_EVENT.LLM_START, { model: LLM_MODEL });
-    const response = await withRetry(
-      () => this.client.messages.create({
+    let response;
+    let streamBuffer = "";
+    try {
+      const stream = this.client.messages.stream({
         model: LLM_MODEL,
         max_tokens: LLM_MAX_TOKENS,
         system: systemPrompt,
         tools: this.tools,
         messages
-      }),
-      {
-        maxRetries: 3,
-        onRetry: (attempt, error, delay) => {
-          this.think(
-            THOUGHT_TYPE.STUCK,
-            `API retry ${attempt}/3 after ${delay}ms: ${error.message}`
-          );
+      });
+      stream.on("text", (text) => {
+        streamBuffer += text;
+        this.emit(AGENT_EVENT.THOUGHT, { type: "thinking", content: text });
+      });
+      stream.on("contentBlock", (block) => {
+        if (block.type === "tool_use") {
+          this.emit(AGENT_EVENT.THOUGHT, {
+            type: "action",
+            content: `Calling: ${block.name}`
+          });
         }
-      }
-    );
+      });
+      response = await stream.finalMessage();
+    } catch (error) {
+      response = await withRetry(
+        () => this.client.messages.create({
+          model: LLM_MODEL,
+          max_tokens: LLM_MAX_TOKENS,
+          system: systemPrompt,
+          tools: this.tools,
+          messages
+        }),
+        {
+          maxRetries: 3,
+          onRetry: (attempt, err, delay) => {
+            this.think(
+              THOUGHT_TYPE.STUCK,
+              `API retry ${attempt}/3 after ${delay}ms: ${err.message}`
+            );
+          }
+        }
+      );
+    }
     this.emit(AGENT_EVENT.LLM_END, { model: LLM_MODEL });
     if (response.usage) {
       this.tokenUsage.input += response.usage.input_tokens;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pentesting",
-  "version": "0.8.23",
+  "version": "0.8.25",
   "description": "Autonomous Penetration Testing AI Agent",
   "type": "module",
   "main": "dist/index.js",