wagent 1.0.1 → 1.0.2

package/dist/index.js

@@ -1,4 +1,5 @@
 #!/usr/bin/env node
+import "./quiet-console.js";
 import { InstanceManager } from "./services/instance-manager.js";
 import { createChildLogger } from "./utils/logger.js";
 import { generateText, tool } from "ai";
@@ -12,33 +13,6 @@ const MAX_HISTORY = 30;
 const processedMessages = new Set();
 const agentSentMessages = new Set();
 const typingUntil = new Map();
-const originalConsole = {
-    log: console.log.bind(console),
-    warn: console.warn.bind(console),
-    error: console.error.bind(console),
-};
-function shouldSuppressConsole(args) {
-    if (process.argv.includes("--debug"))
-        return false;
-    const text = args.map((arg) => (typeof arg === "string" ? arg : "")).join(" ");
-    return (text.startsWith("Closing session:") ||
-        text.includes("Decrypted message with closed session") ||
-        text.includes("stream errored out") ||
-        text.includes("no name present") ||
-        text.includes("blocked on missing key"));
-}
-console.log = (...args) => {
-    if (!shouldSuppressConsole(args))
-        originalConsole.log(...args);
-};
-console.warn = (...args) => {
-    if (!shouldSuppressConsole(args))
-        originalConsole.warn(...args);
-};
-console.error = (...args) => {
-    if (!shouldSuppressConsole(args))
-        originalConsole.error(...args);
-};
 function getOwnJid(adapter) {
     return adapter.getMyJid();
 }
@@ -179,7 +153,7 @@ function buildTools(adapter) {
             execute: async ({ to, text, quotedMessageId }, { experimental_context }) => {
                 const ctx = experimental_context;
                 const target = await resolveTarget(adapter, to, ctx);
-                const content = { type: "text", text };
+                const content = { type: "text", text: withAgentPrefix(text) };
                 if (quotedMessageId)
                     content.quotedMessageId = quotedMessageId;
                 await delayBeforeReply(ctx.config);
@@ -710,9 +684,15 @@ function messageCanReachAgent(config, event, isSelf, text) {
 function isConfirmation(text) {
     return /\b(confirm|confirmed|approve|approved|yes|ship it|looks good)\b/i.test(text);
 }
-function isReadyForPromptDraft(text, notes) {
-    const total = notes.join(" ").length;
-    return /\b(enough|draft|write.*prompt|make.*prompt|done|that's it|that is it)\b/i.test(text) || notes.length >= 3 || total >= 500;
+function isPromptDraftRequest(text) {
+    return /\b(enough|draft|write.*prompt|make.*prompt|system prompt|ready|done|that's it|that is it)\b/i.test(text);
+}
+function maybeExtractSayCommand(text) {
+    const match = text.match(/^\s*say\s+["']?(.+?)["']?\s*$/i);
+    return match?.[1]?.trim() ?? null;
+}
+function withAgentPrefix(text) {
+    return text.trim().toLowerCase().startsWith("wagent:") ? text : `wagent: ${text}`;
 }
 function nextBootstrapQuestion(notes) {
     const questions = [
@@ -724,14 +704,14 @@ function nextBootstrapQuestion(notes) {
     return questions[Math.min(Math.max(notes.length - 1, 0), questions.length - 1)];
 }
 async function sendTracked(adapter, to, text) {
-    const res = await adapter.sendMessage(to, { type: "text", text });
+    const res = await adapter.sendMessage(to, { type: "text", text: withAgentPrefix(text) });
     rememberAgentMessage(res.messageId);
 }
 async function startBootstrap(adapter, config, myJid) {
     if (config.agent.bootstrapStarted || config.agent.mode !== "bootstrap")
         return config;
     await sendTracked(adapter, myJid, "I'm online. I don't know who I am yet.");
-    await sendTracked(adapter, myJid, "Tell me my purpose, your relationship to me, my vibe, and who I can read or reply to.");
+    await sendTracked(adapter, myJid, "Tell me my purpose, what to call you, my vibe, and who I can read/reply to. Say 'draft prompt' when ready.");
     await sendTracked(adapter, myJid, "I can message contacts, manage WhatsApp tools, remember context, compact memory, and enforce read/reply permissions.");
     config.agent.bootstrapStarted = true;
     saveConfig(config);
@@ -747,8 +727,20 @@ async function handleBootstrapMessage(adapter, config, chatId, text) {
         await sendTracked(adapter, chatId, "Confirmed. I compacted setup context and will follow this system prompt from now on.");
         return config;
     }
+    if (config.agent.pendingSystemPrompt && !isConfirmation(text)) {
+        config.agent.pendingSystemPrompt = undefined;
+        config.agent.bootstrapNotes.push(`Feedback on rejected draft: ${text}`);
+        saveConfig(config);
+        await sendTracked(adapter, chatId, "Okay, I won't use that draft. Tell me what to change, or say 'draft prompt' when ready.");
+        return config;
+    }
+    const say = maybeExtractSayCommand(text);
+    if (say) {
+        await sendTracked(adapter, chatId, say);
+        return config;
+    }
     config.agent.bootstrapNotes.push(text);
-    if (!isReadyForPromptDraft(text, config.agent.bootstrapNotes)) {
+    if (!isPromptDraftRequest(text)) {
         saveConfig(config);
         await sendTracked(adapter, chatId, nextBootstrapQuestion(config.agent.bootstrapNotes));
         return config;
@@ -794,7 +786,7 @@ async function processMessage(adapter, config, instId, event, text, isSelf, send
         if (responseText) {
             await adapter.sendPresence(chatId, "composing");
             await delayBeforeReply(config);
-            const content = { type: "text", text: responseText };
+            const content = { type: "text", text: withAgentPrefix(responseText) };
             const res = await adapter.sendMessage(chatId, content);
             rememberAgentMessage(res.messageId);
             await adapter.sendPresence(chatId, "paused");
@@ -802,7 +794,7 @@ async function processMessage(adapter, config, instId, event, text, isSelf, send
         }
         else if (allToolCalls.length > 0) {
             logger.info({ chatId, toolCount: allToolCalls.length }, "Tools executed silently");
-            const res = await adapter.sendMessage(chatId, { type: "text", text: "Done." });
+            const res = await adapter.sendMessage(chatId, { type: "text", text: withAgentPrefix("Done.") });
             rememberAgentMessage(res.messageId);
         }
     }
@@ -810,7 +802,7 @@ async function processMessage(adapter, config, instId, event, text, isSelf, send
         const msg = err instanceof Error ? err.message : String(err);
         logger.error({ err, chatId }, "Agent error");
         try {
-            const fallback = "Sorry, I hit an error processing that. Please try again.";
+            const fallback = withAgentPrefix("Sorry, I hit an error processing that. Please try again.");
             const res = await adapter.sendMessage(chatId, { type: "text", text: fallback });
             rememberAgentMessage(res.messageId);
         }

package/dist/quiet-console.js

@@ -0,0 +1,31 @@
+const originalConsole = {
+    log: console.log.bind(console),
+    warn: console.warn.bind(console),
+    error: console.error.bind(console),
+};
+function shouldSuppressConsole(args) {
+    if (process.argv.includes("--debug"))
+        return false;
+    const text = args.map((arg) => (typeof arg === "string" ? arg : "")).join(" ");
+    return (text.startsWith("Closing session:") ||
+        text.includes("Decrypted message with closed session") ||
+        text.includes("stream errored out") ||
+        text.includes("no name present") ||
+        text.includes("blocked on missing key") ||
+        text.includes("failed to find key") ||
+        text.includes("transaction failed") ||
+        text.includes("failed to decrypt message"));
+}
+console.log = (...args) => {
+    if (!shouldSuppressConsole(args))
+        originalConsole.log(...args);
+};
+console.warn = (...args) => {
+    if (!shouldSuppressConsole(args))
+        originalConsole.warn(...args);
+};
+console.error = (...args) => {
+    if (!shouldSuppressConsole(args))
+        originalConsole.error(...args);
+};
+export {};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wagent",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "WhatsApp Agent — AI-powered WhatsApp assistant using WAMCP + Groq",
   "type": "module",
   "bin": {