pentesting 0.7.38 → 0.7.39

package/dist/index.js

@@ -5216,23 +5216,8 @@ ${this.state.findings.filter((f) => f.severity !== "info").map((f) => `- Address
         }
       }
       if (hasToolCalls && response.stop_reason === "tool_use") {
-        const followUp = await this.client.messages.create({
-          model: LLM_MODEL,
-          max_tokens: LLM_MAX_TOKENS,
-          system: systemPrompt,
-          messages: this.state.history,
-          tools: this.tools
-        });
-        for (const block of followUp.content) {
-          if (block.type === "text") {
-            textResponse += block.text;
-            this.emit(AGENT_EVENT.RESPONSE, block.text);
-          }
-        }
-        this.state.history.push({
-          role: "assistant",
-          content: followUp.content
-        });
+        const continuedResponse = await this.processChatToolLoop(systemPrompt);
+        textResponse += continuedResponse;
       } else if (!hasToolCalls) {
         this.state.history.push({
           role: "assistant",
@@ -5248,6 +5233,55 @@ ${this.state.findings.filter((f) => f.severity !== "info").map((f) => `- Address
       return `Error: ${errMsg}`;
     }
   }
+  /**
+   * Process chat tool loop recursively until we get a text response
+   */
+  async processChatToolLoop(systemPrompt, maxIterations = 5) {
+    let textResponse = "";
+    let iteration = 0;
+    while (iteration < maxIterations) {
+      iteration++;
+      const response = await this.client.messages.create({
+        model: LLM_MODEL,
+        max_tokens: LLM_MAX_TOKENS,
+        system: systemPrompt,
+        messages: this.state.history,
+        tools: this.tools
+      });
+      let hasToolCalls = false;
+      for (const block of response.content) {
+        if (block.type === "text") {
+          textResponse += block.text;
+          this.emit(AGENT_EVENT.RESPONSE, block.text);
+        } else if (block.type === "tool_use") {
+          hasToolCalls = true;
+          this.emit(AGENT_EVENT.TOOL_CALL, { name: block.name, input: block.input });
+          const result = await executeToolCall(block.name, block.input);
+          this.emit(AGENT_EVENT.TOOL_RESULT, { name: block.name, result });
+          this.state.history.push({
+            role: "assistant",
+            content: response.content
+          });
+          this.state.history.push({
+            role: "user",
+            content: [{
+              type: "tool_result",
+              tool_use_id: block.id,
+              content: typeof result === "string" ? result : JSON.stringify(result)
+            }]
+          });
+        }
+      }
+      if (!hasToolCalls || response.stop_reason !== "tool_use") {
+        this.state.history.push({
+          role: "assistant",
+          content: response.content
+        });
+        break;
+      }
+    }
+    return textResponse;
+  }
   /**
    * Build a context-aware prompt for chat interactions
    */

package/package.json

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