@prbe.ai/electron-sdk 0.1.13 → 0.1.14

package/dist/index.mjs

@@ -9,6 +9,7 @@ var WSMessageType = /* @__PURE__ */ ((WSMessageType2) => {
   WSMessageType2["START"] = "start";
   WSMessageType2["TOOL_RESULT"] = "tool_result";
   WSMessageType2["UPLOAD_REQUEST"] = "upload_request";
+  WSMessageType2["CONVERSATION_MESSAGE"] = "conversation_message";
   WSMessageType2["CANCEL"] = "cancel";
   WSMessageType2["PONG"] = "pong";
   WSMessageType2["THOUGHT"] = "thought";
@@ -17,11 +18,17 @@ var WSMessageType = /* @__PURE__ */ ((WSMessageType2) => {
   WSMessageType2["SERVER_OBSERVATION"] = "server_observation";
   WSMessageType2["UPLOAD_URL"] = "upload_url";
   WSMessageType2["SESSION_CONFIG"] = "session_config";
+  WSMessageType2["CONVERSATION_UPDATE"] = "conversation_update";
   WSMessageType2["COMPLETE"] = "complete";
   WSMessageType2["ERROR"] = "error";
   WSMessageType2["PING"] = "ping";
   return WSMessageType2;
 })(WSMessageType || {});
+var ConversationRole = /* @__PURE__ */ ((ConversationRole3) => {
+  ConversationRole3["User"] = "user";
+  ConversationRole3["Agent"] = "agent";
+  return ConversationRole3;
+})(ConversationRole || {});
 var ToolParamType = /* @__PURE__ */ ((ToolParamType2) => {
   ToolParamType2["STRING"] = "STRING";
   ToolParamType2["BOOLEAN"] = "BOOLEAN";
@@ -91,6 +98,83 @@ function redactPII(text) {
 var API_URL = "https://api.prbe.ai";
 var MIDDLEWARE_URL = "wss://middleware.prbe.ai";
 
+// src/connection.ts
+var InvestigationConnection = class {
+  constructor(url, apiKey, onMessage, onError, onClose) {
+    this.onMessage = onMessage;
+    this.onError = onError;
+    this.onClose = onClose;
+    this.ws = new WebSocket(url, {
+      headers: { "X-API-Key": apiKey }
+    });
+    this.ws.onmessage = (event) => {
+      const raw = typeof event.data === "string" ? event.data : event.data instanceof Buffer ? event.data.toString("utf-8") : null;
+      if (!raw) return;
+      try {
+        const msg = JSON.parse(raw);
+        this.onMessage(msg);
+      } catch {
+      }
+    };
+    this.ws.onerror = (event) => {
+      const errorEvent = event;
+      this.onError(errorEvent.message || "WebSocket connection error");
+    };
+    this.ws.onclose = () => {
+      this.onClose();
+    };
+  }
+  ws;
+  closed = false;
+  get isOpen() {
+    return !this.closed && this.ws.readyState === WebSocket.OPEN;
+  }
+  send(msg) {
+    if (!this.isOpen) return false;
+    try {
+      this.ws.send(JSON.stringify(msg));
+      return true;
+    } catch {
+      return false;
+    }
+  }
+  sendConversationMessage(content) {
+    this.send({
+      type: "conversation_message" /* CONVERSATION_MESSAGE */,
+      content
+    });
+  }
+  sendToolResult(callId, toolName, result, metadata) {
+    this.send({
+      type: "tool_result" /* TOOL_RESULT */,
+      id: callId,
+      name: toolName,
+      content: result,
+      metadata
+    });
+  }
+  sendCancel() {
+    this.send({ type: "cancel" /* CANCEL */ });
+  }
+  sendPong() {
+    this.send({ type: "pong" /* PONG */ });
+  }
+  close(code = 1e3, reason) {
+    if (this.closed) return;
+    this.closed = true;
+    try {
+      this.ws.close(code, reason);
+    } catch {
+    }
+  }
+  /**
+   * Set the onopen handler. Called when the connection is ready to send messages.
+   */
+  set onopen(handler) {
+    this.ws.onopen = handler;
+  }
+};
+
 // src/state.ts
 import { EventEmitter } from "events";
 import { randomUUID } from "crypto";
@@ -119,6 +203,7 @@ var PRBEAgentState = class extends EventEmitter {
   pendingInteraction;
   resolvedInteractions = [];
   agentMessage;
+  conversationHistory = [];
   // Completed user investigations (history)
   completedInvestigations = [];
   // Background context requests
@@ -142,6 +227,7 @@ var PRBEAgentState = class extends EventEmitter {
     this.isInvestigating = true;
     this.events = [];
     this.resolvedInteractions = [];
+    this.conversationHistory = [];
     this.report = "";
     this.summary = "";
     this.currentQuery = query;
@@ -149,10 +235,15 @@ var PRBEAgentState = class extends EventEmitter {
     this.agentMessage = void 0;
     this.emit("status" /* STATUS */);
   }
+  appendConversation(entry) {
+    this.conversationHistory.push(entry);
+    this.emit("status" /* STATUS */);
+  }
   resetInvestigation() {
     this.isInvestigating = false;
     this.events = [];
     this.resolvedInteractions = [];
+    this.conversationHistory = [];
     this.report = "";
     this.summary = "";
     this.currentQuery = "";
@@ -185,7 +276,7 @@ var PRBEAgentState = class extends EventEmitter {
       this.emit("status" /* STATUS */);
     }
   }
-  completeInvestigation(report, summary, ticketId) {
+  completeInvestigation(report, ticketId) {
     if (this.events.length > 0) {
       this.events[this.events.length - 1].isCompleted = true;
     }
@@ -194,16 +285,15 @@ var PRBEAgentState = class extends EventEmitter {
       id: randomUUID(),
       query: this.currentQuery,
       report,
-      summary,
       ticketId,
       events: [...this.events],
       resolvedInteractions: [...this.resolvedInteractions],
+      conversationHistory: [...this.conversationHistory],
       completedAt: /* @__PURE__ */ new Date()
     });
     this.report = report;
-    this.summary = summary;
     this.isInvestigating = false;
-    this.emit("complete" /* COMPLETE */, { report, summary });
+    this.emit("complete" /* COMPLETE */, { report });
     this.emit("status" /* STATUS */);
   }
   failInvestigation(message) {
@@ -328,7 +418,7 @@ var PRBEAgentState = class extends EventEmitter {
     cr.events[cr.events.length - 1].detail = text;
     this.emit("status" /* STATUS */);
   }
-  completeCR(id, report, summary) {
+  completeCR(id, report) {
     const cr = this.activeCRs.get(id);
     if (!cr) return;
     this.activeCRs.delete(id);
@@ -344,7 +434,6 @@ var PRBEAgentState = class extends EventEmitter {
     cr.isRunning = false;
     cr.isCompleted = true;
     cr.report = report;
-    cr.summary = summary;
     this.completedCRs.unshift(cr);
     this.emit("cr-complete" /* CR_COMPLETE */, cr);
     this.emit("status" /* STATUS */);
@@ -1281,9 +1370,9 @@ var AskUserTool = class {
           required: true
         },
         {
-          name: "context",
+          name: "reason",
           type: "STRING" /* STRING */,
-          description: "Optional context explaining why you need this information",
+          description: "Short reason displayed alongside the input prompt explaining why you need this information",
           required: false
         }
       ]
@@ -1295,12 +1384,11 @@ var AskUserTool = class {
     if (this.requester.investigationSource !== "user" /* USER */) {
       return "This is a context request investigation \u2014 you cannot ask the user questions. Use the available tools to answer the query autonomously.";
     }
-    const context = args["context"];
+    const reason = args["reason"] ?? "Waiting for your response";
     const response = await this.requester.requestUserInteraction({
       type: "ask_question" /* ASK_QUESTION */,
       interactionId: randomUUID3(),
-      question,
-      context
+      question: reason
     });
     const askResponse = response;
     return askResponse.answer;
@@ -1721,6 +1809,7 @@ var HistoryStore = class {
           ticketId: inv.ticketId,
           events: inv.events,
           resolvedInteractions: inv.resolvedInteractions,
+          conversationHistory: inv.conversationHistory,
           completedAt: inv.completedAt.toISOString()
         };
         fs2.writeFileSync(
@@ -1813,7 +1902,7 @@ var PRBEAgent = class _PRBEAgent {
   registry = new PRBEToolRegistry();
   grantedPaths = /* @__PURE__ */ new Set();
   userCancelled = false;
-  activeWS = null;
+  activeConnection = null;
   pollingTimer = null;
   persistedData;
   fetchAbortController = null;
@@ -1973,6 +2062,9 @@ var PRBEAgent = class _PRBEAgent {
   get investigationSource() {
     return this.currentInvestigationSource;
   }
+  sendConversationMessage(content) {
+    this.activeConnection?.sendConversationMessage(content);
+  }
   async requestUserInteraction(payload) {
     if (!this.interactionHandler) {
       throw new PRBEAgentError(
@@ -1992,6 +2084,13 @@ var PRBEAgent = class _PRBEAgent {
       } else {
         this.state.resolveInteraction(response);
       }
+      if (response.type === "request_permission" /* REQUEST_PERMISSION */) {
+        const approved = response.approved;
+        this.sendConversationMessage(approved ? "Approved" : "Denied");
+      } else if (response.type === "request_path_access" /* REQUEST_PATH_ACCESS */) {
+        const granted = response.granted;
+        this.sendConversationMessage(granted ? "Allowed" : "Denied");
+      }
       return response;
     } catch (err) {
       if (this.currentInvestigationSource === "context_request" /* CONTEXT_REQUEST */ && this.currentCRId) {
@@ -2044,7 +2143,7 @@ var PRBEAgent = class _PRBEAgent {
           this.state.appendEvent("Thinking", status.text);
           break;
         case "completed" /* COMPLETED */:
-          this.state.completeInvestigation(status.report, status.userSummary, status.ticketId);
+          this.state.completeInvestigation(status.report, status.ticketId);
           this.historyStore.save(this.state.completedInvestigations, this.state.completedCRs);
           break;
         case "error" /* ERROR */:
@@ -2074,9 +2173,9 @@ var PRBEAgent = class _PRBEAgent {
    */
   cancelInvestigation() {
     this.userCancelled = true;
-    if (this.activeWS) {
-      this.activeWS.close(1e3, "User cancelled");
-      this.activeWS = null;
+    if (this.activeConnection) {
+      this.activeConnection.close(1e3, "User cancelled");
+      this.activeConnection = null;
     }
   }
   /**
@@ -2084,9 +2183,9 @@ var PRBEAgent = class _PRBEAgent {
    */
   cancel() {
     this.userCancelled = true;
-    if (this.activeWS) {
-      this.activeWS.close(1e3, "User cancelled");
-      this.activeWS = null;
+    if (this.activeConnection) {
+      this.activeConnection.close(1e3, "User cancelled");
+      this.activeConnection = null;
     }
     this.abortInFlightRequests();
     this.stopPolling();
@@ -2225,7 +2324,7 @@ var PRBEAgent = class _PRBEAgent {
           this.state.appendCREvent(crID, "Thinking", status.text);
           break;
         case "completed" /* COMPLETED */:
-          this.state.completeCR(crID, status.report, status.userSummary);
+          this.state.completeCR(crID, status.report);
           this.historyStore.save(this.state.completedInvestigations, this.state.completedCRs);
           break;
         case "error" /* ERROR */:
@@ -2252,31 +2351,44 @@ var PRBEAgent = class _PRBEAgent {
   connectToProxy(query, contextRequestID, emit, isCancelled, ticketId) {
     return new Promise((resolve2) => {
       const wsUrl = `${MIDDLEWARE_URL}/api/agent/client/ws`;
-      let ws;
-      try {
-        ws = new WebSocket(wsUrl, {
-          headers: {
-            "X-API-Key": this.config.apiKey
-          }
-        });
-      } catch (err) {
-        emit({
-          type: "error" /* ERROR */,
-          message: `Failed to create WebSocket: ${err}`
-        });
-        resolve2(null);
-        return;
-      }
-      this.activeWS = ws;
       let uploadBaseUrl;
       let resolved = false;
       const finish = (result) => {
         if (resolved) return;
         resolved = true;
-        this.activeWS = null;
+        this.activeConnection = null;
         resolve2(result);
       };
-      ws.onopen = () => {
+      let conn;
+      try {
+        conn = new InvestigationConnection(
+          wsUrl,
+          this.config.apiKey,
+          (msg) => this.handleMessage(msg, conn, emit, isCancelled, finish, () => uploadBaseUrl, (url) => {
+            uploadBaseUrl = url;
+          }),
+          (message) => {
+            if (!isCancelled()) emit({ type: "error" /* ERROR */, message });
+            finish(null);
+          },
+          () => {
+            if (!resolved) {
+              if (isCancelled()) {
+                finish(null);
+              } else {
+                emit({ type: "error" /* ERROR */, message: "WebSocket connection closed unexpectedly" });
+                finish(null);
+              }
+            }
+          }
+        );
+      } catch (err) {
+        emit({ type: "error" /* ERROR */, message: `Failed to create WebSocket: ${err}` });
+        resolve2(null);
+        return;
+      }
+      this.activeConnection = conn;
+      conn.onopen = () => {
         const toolDeclarations = this.registry.allDeclarations().map((decl) => ({
           name: decl.name,
           description: decl.description,
@@ -2295,206 +2407,99 @@ var PRBEAgent = class _PRBEAgent {
           os_version: os2.release(),
           arch: os2.arch()
         };
-        if (contextRequestID) {
-          startMetadata["context_request_id"] = contextRequestID;
-        }
-        if (ticketId) {
-          startMetadata["ticket_id"] = ticketId;
-        }
-        if (this.appDataPath) {
-          startMetadata["app_data_path"] = this.appDataPath;
-        }
-        const startMsg = {
-          type: "start" /* START */,
-          content: query,
-          metadata: startMetadata
-        };
-        try {
-          ws.send(JSON.stringify(startMsg));
-        } catch (err) {
-          emit({
-            type: "error" /* ERROR */,
-            message: `Failed to send start message: ${err}`
-          });
+        if (contextRequestID) startMetadata["context_request_id"] = contextRequestID;
+        if (ticketId) startMetadata["ticket_id"] = ticketId;
+        if (this.appDataPath) startMetadata["app_data_path"] = this.appDataPath;
+        if (!conn.send({ type: "start" /* START */, content: query, metadata: startMetadata })) {
+          emit({ type: "error" /* ERROR */, message: "Failed to send start message" });
           finish(null);
           return;
         }
         emit({ type: "started" /* STARTED */ });
         this.pendingFlaggedFiles = [];
       };
-      ws.onmessage = async (event) => {
-        if (isCancelled()) {
-          this.sendCancel(ws);
-          ws.close(1e3, "Cancelled");
-          finish(null);
-          return;
-        }
-        let raw;
-        if (typeof event.data === "string") {
-          raw = event.data;
-        } else if (event.data instanceof Buffer) {
-          raw = event.data.toString("utf-8");
-        } else {
-          return;
-        }
-        let msg;
-        try {
-          msg = JSON.parse(raw);
-        } catch {
-          return;
-        }
-        switch (msg.type) {
-          case "thought" /* THOUGHT */:
-            emit({
-              type: "thought" /* THOUGHT */,
-              text: msg.content ?? ""
-            });
-            break;
-          case "tool_call" /* TOOL_CALL */: {
-            const toolName = msg.name ?? "";
-            const callId = msg.id ?? "";
-            const args = this.extractArgs(msg.metadata);
-            emit({
-              type: "tool_call" /* TOOL_CALL */,
-              name: toolName,
-              label: msg.content ?? `Running ${toolName}`
-            });
-            this.pendingFlaggedFiles = [];
-            const toolResult = redactPII(
-              await this.registry.execute(toolName, args)
-            );
-            emit({
-              type: "observation" /* OBSERVATION */,
-              text: toolResult.substring(0, 200)
-            });
-            let resultMetadata;
-            if (this.pendingFlaggedFiles.length > 0 && uploadBaseUrl) {
-              const uploadPrefix = this.extractUploadPrefix(uploadBaseUrl);
-              const uploadedRefs = [];
-              for (const file of this.pendingFlaggedFiles) {
-                const filename = path5.basename(file.originalPath);
-                const safeName = encodeURIComponent(filename);
-                const storagePath = `${uploadPrefix}/${safeName}`;
-                uploadedRefs.push({
-                  original_path: file.originalPath,
-                  reason: file.reason ?? "",
-                  storage_path: storagePath,
-                  file_size_bytes: file.data.length
-                });
-                const uploadUrl = `${uploadBaseUrl}/${safeName}`;
-                const fileData = file.data;
-                const contentType = file.isText ? "text/plain" : "application/octet-stream";
-                void _PRBEAgent.backgroundUpload(
-                  fileData,
-                  uploadUrl,
-                  this.config.apiKey,
-                  contentType,
-                  this.getFetchSignal()
-                );
-              }
-              resultMetadata = { flagged_files: uploadedRefs };
-              this.pendingFlaggedFiles = [];
-            }
-            const resultMsg = {
-              type: "tool_result" /* TOOL_RESULT */,
-              id: callId,
-              name: toolName,
-              content: toolResult,
-              metadata: resultMetadata
-            };
-            try {
-              ws.send(JSON.stringify(resultMsg));
-            } catch {
-            }
-            break;
-          }
-          case "server_tool_call" /* SERVER_TOOL_CALL */:
-            emit({
-              type: "tool_call" /* TOOL_CALL */,
-              name: msg.name ?? "",
-              label: msg.content ?? ""
-            });
-            break;
-          case "server_observation" /* SERVER_OBSERVATION */:
-            emit({
-              type: "observation" /* OBSERVATION */,
-              text: (msg.content ?? "").substring(0, 200)
-            });
-            break;
-          case "complete" /* COMPLETE */: {
-            const report = msg.content ?? "";
-            const userSummary = msg.metadata?.["user_summary"] ?? "";
-            const ticketId2 = msg.metadata?.["ticket_id"];
-            const sessionId = msg.metadata?.["session_id"];
-            emit({
-              type: "completed" /* COMPLETED */,
-              report,
-              userSummary,
-              ticketId: ticketId2
-            });
-            ws.close(1e3, "Complete");
-            finish({ report, userSummary, ticketId: ticketId2, sessionId });
-            break;
-          }
-          case "error" /* ERROR */:
-            emit({
-              type: "error" /* ERROR */,
-              message: msg.content || "Unknown error"
+    });
+  }
+  async handleMessage(msg, conn, emit, isCancelled, finish, getUploadBaseUrl, setUploadBaseUrl) {
+    if (isCancelled()) {
+      conn.sendCancel();
+      conn.close(1e3, "Cancelled");
+      finish(null);
+      return;
+    }
+    switch (msg.type) {
+      case "thought" /* THOUGHT */:
+        emit({ type: "thought" /* THOUGHT */, text: msg.content ?? "" });
+        break;
+      case "tool_call" /* TOOL_CALL */: {
+        const toolName = msg.name ?? "";
+        const callId = msg.id ?? "";
+        const args = this.extractArgs(msg.metadata);
+        emit({ type: "tool_call" /* TOOL_CALL */, name: toolName, label: msg.content ?? `Running ${toolName}` });
+        this.pendingFlaggedFiles = [];
+        const toolResult = redactPII(await this.registry.execute(toolName, args));
+        emit({ type: "observation" /* OBSERVATION */, text: toolResult.substring(0, 200) });
+        let resultMetadata;
+        const uploadBaseUrl = getUploadBaseUrl();
+        if (this.pendingFlaggedFiles.length > 0 && uploadBaseUrl) {
+          const uploadPrefix = this.extractUploadPrefix(uploadBaseUrl);
+          const uploadedRefs = [];
+          for (const file of this.pendingFlaggedFiles) {
+            const filename = path5.basename(file.originalPath);
+            const safeName = encodeURIComponent(filename);
+            const storagePath = `${uploadPrefix}/${safeName}`;
+            uploadedRefs.push({
+              original_path: file.originalPath,
+              reason: file.reason ?? "",
+              storage_path: storagePath,
+              file_size_bytes: file.data.length
             });
-            ws.close(1e3, "Error received");
-            finish(null);
-            break;
-          case "session_config" /* SESSION_CONFIG */:
-            uploadBaseUrl = msg.metadata?.["upload_url"];
-            break;
-          case "ping" /* PING */: {
-            const pongMsg = { type: "pong" /* PONG */ };
-            try {
-              ws.send(JSON.stringify(pongMsg));
-            } catch {
-            }
-            break;
+            const uploadUrl = `${uploadBaseUrl}/${safeName}`;
+            const contentType = file.isText ? "text/plain" : "application/octet-stream";
+            void _PRBEAgent.backgroundUpload(file.data, uploadUrl, this.config.apiKey, contentType, this.getFetchSignal());
           }
-          // SDK-originated types and upload_url — ignore
-          case "start" /* START */:
-          case "tool_result" /* TOOL_RESULT */:
-          case "upload_request" /* UPLOAD_REQUEST */:
-          case "cancel" /* CANCEL */:
-          case "pong" /* PONG */:
-          case "upload_url" /* UPLOAD_URL */:
-            break;
+          resultMetadata = { flagged_files: uploadedRefs };
+          this.pendingFlaggedFiles = [];
         }
-      };
-      ws.onerror = (event) => {
-        if (isCancelled()) {
-          finish(null);
-          return;
-        }
-        const errorEvent = event;
-        const message = errorEvent.message || "WebSocket connection error";
-        emit({ type: "error" /* ERROR */, message });
+        conn.sendToolResult(callId, toolName, toolResult, resultMetadata);
+        break;
+      }
+      case "server_tool_call" /* SERVER_TOOL_CALL */:
+        emit({ type: "tool_call" /* TOOL_CALL */, name: msg.name ?? "", label: msg.content ?? "" });
+        break;
+      case "server_observation" /* SERVER_OBSERVATION */:
+        emit({ type: "observation" /* OBSERVATION */, text: (msg.content ?? "").substring(0, 200) });
+        break;
+      case "complete" /* COMPLETE */: {
+        const report = msg.content ?? "";
+        const completedTicketId = msg.metadata?.["ticket_id"];
+        const sessionId = msg.metadata?.["session_id"];
+        emit({ type: "completed" /* COMPLETED */, report, ticketId: completedTicketId });
+        conn.close(1e3, "Complete");
+        finish({ report, ticketId: completedTicketId, sessionId });
+        break;
+      }
+      case "error" /* ERROR */:
+        emit({ type: "error" /* ERROR */, message: msg.content || "Unknown error" });
+        conn.close(1e3, "Error received");
         finish(null);
-      };
-      ws.onclose = (_event) => {
-        if (!resolved) {
-          if (isCancelled()) {
-            finish(null);
-          } else {
-            emit({
-              type: "error" /* ERROR */,
-              message: "WebSocket connection closed unexpectedly"
-            });
-            finish(null);
-          }
+        break;
+      case "session_config" /* SESSION_CONFIG */:
+        setUploadBaseUrl(msg.metadata?.["upload_url"]);
+        break;
+      case "conversation_update" /* CONVERSATION_UPDATE */: {
+        try {
+          const entry = JSON.parse(msg.content ?? "{}");
+          this.state.appendConversation(entry);
+        } catch {
         }
-      };
-    });
-  }
-  sendCancel(ws) {
-    try {
-      const cancelMsg = { type: "cancel" /* CANCEL */ };
-      ws.send(JSON.stringify(cancelMsg));
-    } catch {
+        break;
+      }
+      case "ping" /* PING */:
+        conn.sendPong();
+        break;
+      default:
+        break;
     }
   }
   extractArgs(metadata) {
@@ -2612,6 +2617,7 @@ var DEFAULT_PRBE_STATE = {
   summary: "",
   currentQuery: "",
   resolvedInteractions: [],
+  conversationHistory: [],
   completedInvestigations: [],
   activeCRs: [],
   completedCRs: [],
@@ -2649,6 +2655,7 @@ function serializePRBEState(state) {
     pendingInteraction: state.pendingInteraction,
     resolvedInteractions: state.resolvedInteractions,
     agentMessage: state.agentMessage,
+    conversationHistory: state.conversationHistory,
     completedInvestigations: state.completedInvestigations.map((inv) => ({
       id: inv.id,
       query: inv.query,
@@ -2657,6 +2664,7 @@ function serializePRBEState(state) {
       ticketId: inv.ticketId,
       events: inv.events,
       resolvedInteractions: inv.resolvedInteractions,
+      conversationHistory: inv.conversationHistory,
       completedAt: inv.completedAt.toISOString()
     })),
     activeCRs: Array.from(state.activeCRs.values()).map(serializeCR),
@@ -2678,6 +2686,7 @@ export {
   AskUserTool,
   BashExecuteTool,
   ClearAppLogsTool,
+  ConversationRole,
   DEFAULT_PRBE_STATE,
   FindFilesTool,
   FlagAppLogsTool,