@prbe.ai/electron-sdk 0.1.6 → 0.1.8

package/dist/index.d.mts

@@ -1,5 +1,5 @@
-import { P as PRBEToolDeclaration, a as PRBEToolParameter, b as PRBEInteractionRequester, F as FlaggedFileIn, c as PRBEAgentState, d as PRBEAgentConfig, I as InvestigationSource, e as InteractionPayload, f as InteractionResponse, g as PollResponse, T as TicketInfoOut } from './types-DHT-JxMT.mjs';
-export { A as API_URL, h as AskQuestionPayload, i as AskQuestionResponse, C as ContextRequestOut, D as DEFAULT_PRBE_STATE, j as InteractionType, k as InvestigationResult, M as MIDDLEWARE_URL, l as PRBEAgentConfigKey, m as PRBEAgentError, n as PRBEAgentErrorType, o as PRBEAgentStatus, p as PRBEAgentStatusType, q as PRBECRInvestigation, r as PRBECompletedInvestigation, s as PRBEInteractionHandler, t as PRBESerializedCR, u as PRBESerializedCompletedInvestigation, v as PRBESerializedState, w as PRBESerializedTicket, x as PRBEStateEvent, y as PRBEStatusEvent, z as PollRequest, R as RequestPathAccessPayload, B as RequestPathAccessResponse, E as RequestPermissionPayload, G as RequestPermissionResponse, H as ResolvedInteraction, J as TicketInfoRequest, K as TicketInfoResponse, L as TicketStatusOut, N as ToolName, O as ToolParamType, W as WSMessage, Q as WSMessageType, S as redactPII, U as serializePRBEState } from './types-DHT-JxMT.mjs';
+import { P as PRBEToolDeclaration, a as PRBEToolParameter, b as PRBEInteractionRequester, F as FlaggedFileIn, c as PRBEAgentState, d as PRBEAgentConfig, I as InvestigationSource, e as InteractionPayload, f as InteractionResponse, g as PollResponse, T as TicketInfoOut } from './types-CwYRJCG0.mjs';
+export { A as API_URL, h as AskQuestionPayload, i as AskQuestionResponse, C as ContextRequestOut, D as DEFAULT_PRBE_STATE, j as InteractionType, k as InvestigationResult, M as MIDDLEWARE_URL, l as PRBEAgentConfigKey, m as PRBEAgentError, n as PRBEAgentErrorType, o as PRBEAgentStatus, p as PRBEAgentStatusType, q as PRBECRInvestigation, r as PRBECompletedInvestigation, s as PRBEInteractionHandler, t as PRBESerializedCR, u as PRBESerializedCompletedInvestigation, v as PRBESerializedState, w as PRBESerializedTicket, x as PRBEStateEvent, y as PRBEStatusEvent, z as PollRequest, R as RequestPathAccessPayload, B as RequestPathAccessResponse, E as RequestPermissionPayload, G as RequestPermissionResponse, H as ResolvedInteraction, J as TicketInfoRequest, K as TicketInfoResponse, L as TicketStatusOut, N as ToolName, O as ToolParamType, W as WSMessage, Q as WSMessageType, S as redactPII, U as serializePRBEState } from './types-CwYRJCG0.mjs';
 import 'events';
 
 /**
@@ -21,7 +21,9 @@ declare class PRBEToolRegistry {
 declare class PRBEClosureTool implements PRBETool {
     readonly declaration: PRBEToolDeclaration;
     private readonly handler;
-    constructor(name: string, description: string, parameters: PRBEToolParameter[], handler: (args: Record<string, unknown>) => Promise<string>);
+    constructor(name: string, description: string, parameters: PRBEToolParameter[], handler: (args: Record<string, unknown>) => Promise<string>, options?: {
+        interactive?: boolean;
+    });
     execute(args: Record<string, unknown>): Promise<string>;
 }
 /**
@@ -144,6 +146,7 @@ declare class PRBEAgent implements PRBEInteractionRequester {
     private fetchAbortController;
     private currentInvestigationSource;
     private currentCRId;
+    private readonly historyStore;
     /** Files flagged during the current tool call — uploaded immediately after the tool returns. */
     private pendingFlaggedFiles;
     private get agentID();
@@ -166,7 +169,9 @@ declare class PRBEAgent implements PRBEInteractionRequester {
     /**
      * Register a custom tool that the middleware can invoke during investigations.
      */
-    registerTool(name: string, description: string, parameters: PRBEToolParameter[], handler: (args: Record<string, unknown>) => Promise<string>): void;
+    registerTool(name: string, description: string, parameters: PRBEToolParameter[], handler: (args: Record<string, unknown>) => Promise<string>, options?: {
+        interactive?: boolean;
+    }): void;
     /**
      * User-initiated investigation. Updates `state` events/report directly.
      */

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { P as PRBEToolDeclaration, a as PRBEToolParameter, b as PRBEInteractionRequester, F as FlaggedFileIn, c as PRBEAgentState, d as PRBEAgentConfig, I as InvestigationSource, e as InteractionPayload, f as InteractionResponse, g as PollResponse, T as TicketInfoOut } from './types-DHT-JxMT.js';
-export { A as API_URL, h as AskQuestionPayload, i as AskQuestionResponse, C as ContextRequestOut, D as DEFAULT_PRBE_STATE, j as InteractionType, k as InvestigationResult, M as MIDDLEWARE_URL, l as PRBEAgentConfigKey, m as PRBEAgentError, n as PRBEAgentErrorType, o as PRBEAgentStatus, p as PRBEAgentStatusType, q as PRBECRInvestigation, r as PRBECompletedInvestigation, s as PRBEInteractionHandler, t as PRBESerializedCR, u as PRBESerializedCompletedInvestigation, v as PRBESerializedState, w as PRBESerializedTicket, x as PRBEStateEvent, y as PRBEStatusEvent, z as PollRequest, R as RequestPathAccessPayload, B as RequestPathAccessResponse, E as RequestPermissionPayload, G as RequestPermissionResponse, H as ResolvedInteraction, J as TicketInfoRequest, K as TicketInfoResponse, L as TicketStatusOut, N as ToolName, O as ToolParamType, W as WSMessage, Q as WSMessageType, S as redactPII, U as serializePRBEState } from './types-DHT-JxMT.js';
+import { P as PRBEToolDeclaration, a as PRBEToolParameter, b as PRBEInteractionRequester, F as FlaggedFileIn, c as PRBEAgentState, d as PRBEAgentConfig, I as InvestigationSource, e as InteractionPayload, f as InteractionResponse, g as PollResponse, T as TicketInfoOut } from './types-CwYRJCG0.js';
+export { A as API_URL, h as AskQuestionPayload, i as AskQuestionResponse, C as ContextRequestOut, D as DEFAULT_PRBE_STATE, j as InteractionType, k as InvestigationResult, M as MIDDLEWARE_URL, l as PRBEAgentConfigKey, m as PRBEAgentError, n as PRBEAgentErrorType, o as PRBEAgentStatus, p as PRBEAgentStatusType, q as PRBECRInvestigation, r as PRBECompletedInvestigation, s as PRBEInteractionHandler, t as PRBESerializedCR, u as PRBESerializedCompletedInvestigation, v as PRBESerializedState, w as PRBESerializedTicket, x as PRBEStateEvent, y as PRBEStatusEvent, z as PollRequest, R as RequestPathAccessPayload, B as RequestPathAccessResponse, E as RequestPermissionPayload, G as RequestPermissionResponse, H as ResolvedInteraction, J as TicketInfoRequest, K as TicketInfoResponse, L as TicketStatusOut, N as ToolName, O as ToolParamType, W as WSMessage, Q as WSMessageType, S as redactPII, U as serializePRBEState } from './types-CwYRJCG0.js';
 import 'events';
 
 /**
@@ -21,7 +21,9 @@ declare class PRBEToolRegistry {
 declare class PRBEClosureTool implements PRBETool {
     readonly declaration: PRBEToolDeclaration;
     private readonly handler;
-    constructor(name: string, description: string, parameters: PRBEToolParameter[], handler: (args: Record<string, unknown>) => Promise<string>);
+    constructor(name: string, description: string, parameters: PRBEToolParameter[], handler: (args: Record<string, unknown>) => Promise<string>, options?: {
+        interactive?: boolean;
+    });
     execute(args: Record<string, unknown>): Promise<string>;
 }
 /**
@@ -144,6 +146,7 @@ declare class PRBEAgent implements PRBEInteractionRequester {
     private fetchAbortController;
     private currentInvestigationSource;
     private currentCRId;
+    private readonly historyStore;
     /** Files flagged during the current tool call — uploaded immediately after the tool returns. */
     private pendingFlaggedFiles;
     private get agentID();
@@ -166,7 +169,9 @@ declare class PRBEAgent implements PRBEInteractionRequester {
     /**
      * Register a custom tool that the middleware can invoke during investigations.
      */
-    registerTool(name: string, description: string, parameters: PRBEToolParameter[], handler: (args: Record<string, unknown>) => Promise<string>): void;
+    registerTool(name: string, description: string, parameters: PRBEToolParameter[], handler: (args: Record<string, unknown>) => Promise<string>, options?: {
+        interactive?: boolean;
+    }): void;
     /**
      * User-initiated investigation. Updates `state` events/report directly.
      */

package/dist/index.js

@@ -69,9 +69,9 @@ __export(src_exports, {
 module.exports = __toCommonJS(src_exports);
 
 // src/agent.ts
-var fs2 = __toESM(require("fs"));
-var path4 = __toESM(require("path"));
-var os = __toESM(require("os"));
+var fs3 = __toESM(require("fs"));
+var path5 = __toESM(require("path"));
+var os2 = __toESM(require("os"));
 var import_crypto5 = require("crypto");
 
 // src/models.ts
@@ -250,7 +250,7 @@ var PRBEAgentState = class extends import_events.EventEmitter {
       this.emit("status" /* STATUS */);
     }
   }
-  completeInvestigation(report, summary) {
+  completeInvestigation(report, summary, ticketId) {
     if (this.events.length > 0) {
       this.events[this.events.length - 1].isCompleted = true;
     }
@@ -260,6 +260,9 @@ var PRBEAgentState = class extends import_events.EventEmitter {
       query: this.currentQuery,
       report,
       summary,
+      ticketId,
+      events: [...this.events],
+      resolvedInteractions: [...this.resolvedInteractions],
       completedAt: /* @__PURE__ */ new Date()
     });
     this.report = report;
@@ -335,11 +338,12 @@ var PRBEAgentState = class extends import_events.EventEmitter {
     }
   }
   // ---------- CR state mutations ----------
-  beginCR(id, query, slug) {
+  beginCR(id, query, slug, ticketId) {
     const cr = {
       id,
       query,
       slug,
+      ticketId,
       events: [],
       resolvedInteractions: [],
       isRunning: true,
@@ -465,8 +469,13 @@ var PRBEToolRegistry = class {
 var PRBEClosureTool = class {
   declaration;
   handler;
-  constructor(name, description, parameters, handler) {
-    this.declaration = { name, description, parameters };
+  constructor(name, description, parameters, handler, options) {
+    this.declaration = {
+      name,
+      description,
+      parameters,
+      ...options?.interactive ? { interactive: true } : {}
+    };
     this.handler = handler;
   }
   async execute(args) {
@@ -1569,20 +1578,196 @@ var BashExecuteTool = class {
   }
 };
 
-// src/agent.ts
-function getPersistencePath() {
+// src/history.ts
+var fs2 = __toESM(require("fs"));
+var path4 = __toESM(require("path"));
+var os = __toESM(require("os"));
+var crypto = __toESM(require("crypto"));
+var HKDF_SALT = Buffer.from("prbe-history-encryption-salt", "utf-8");
+var HKDF_INFO = Buffer.from("prbe-history-v1", "utf-8");
+function getAppDataDir() {
   const appData = process.env["APPDATA"] || (process.platform === "darwin" ? path4.join(os.homedir(), "Library", "Application Support") : path4.join(os.homedir(), ".local", "share"));
-  const dir = path4.join(appData, "prbe-agent");
+  return path4.join(appData, "prbe-agent");
+}
+function getHistoryDir() {
+  const dir = path4.join(getAppDataDir(), "history" /* HISTORY_DIR */);
   if (!fs2.existsSync(dir)) {
     fs2.mkdirSync(dir, { recursive: true });
   }
-  return path4.join(dir, "agent-state.json");
+  return dir;
+}
+function deriveKey(apiKey) {
+  return Buffer.from(
+    crypto.hkdfSync(
+      "sha256",
+      Buffer.from(apiKey, "utf-8"),
+      HKDF_SALT,
+      HKDF_INFO,
+      32 /* KEY_LENGTH */
+    )
+  );
+}
+function encrypt(plaintext, key) {
+  const iv = crypto.randomBytes(12 /* IV_LENGTH */);
+  const cipher = crypto.createCipheriv(
+    "aes-256-gcm" /* ALGORITHM */,
+    key,
+    iv,
+    { authTagLength: 16 /* AUTH_TAG_LENGTH */ }
+  );
+  const encrypted = Buffer.concat([cipher.update(plaintext, "utf-8"), cipher.final()]);
+  const authTag = cipher.getAuthTag();
+  return Buffer.concat([iv, authTag, encrypted]);
+}
+function decrypt(data, key) {
+  const iv = data.subarray(0, 12 /* IV_LENGTH */);
+  const authTag = data.subarray(
+    12 /* IV_LENGTH */,
+    12 /* IV_LENGTH */ + 16 /* AUTH_TAG_LENGTH */
+  );
+  const ciphertext = data.subarray(
+    12 /* IV_LENGTH */ + 16 /* AUTH_TAG_LENGTH */
+  );
+  const decipher = crypto.createDecipheriv(
+    "aes-256-gcm" /* ALGORITHM */,
+    key,
+    iv,
+    { authTagLength: 16 /* AUTH_TAG_LENGTH */ }
+  );
+  decipher.setAuthTag(authTag);
+  return decipher.update(ciphertext) + decipher.final("utf-8");
+}
+var HistoryStore = class {
+  key;
+  constructor(apiKey) {
+    this.key = deriveKey(apiKey);
+  }
+  load() {
+    const investigations = [];
+    const crs = [];
+    try {
+      const dir = getHistoryDir();
+      let files;
+      try {
+        files = fs2.readdirSync(dir);
+      } catch {
+        return { investigations, crs };
+      }
+      for (const filename of files) {
+        try {
+          const filePath = path4.join(dir, filename);
+          const raw = fs2.readFileSync(filePath);
+          const json = decrypt(raw, this.key);
+          if (filename.startsWith("inv-") && filename.endsWith(".json")) {
+            const item = JSON.parse(json);
+            investigations.push({
+              ...item,
+              completedAt: new Date(item.completedAt)
+            });
+          } else if (filename.startsWith("cr-") && filename.endsWith(".json")) {
+            const item = JSON.parse(json);
+            crs.push({
+              ...item,
+              startedAt: new Date(item.startedAt),
+              resolvedInteractions: item.resolvedInteractions ?? []
+            });
+          }
+        } catch {
+          console.warn(`[PRBEAgent] Skipping unreadable history file: ${filename}`);
+        }
+      }
+    } catch {
+    }
+    investigations.sort(
+      (a, b) => b.completedAt.getTime() - a.completedAt.getTime()
+    );
+    crs.sort((a, b) => b.startedAt.getTime() - a.startedAt.getTime());
+    return { investigations, crs };
+  }
+  save(investigations, crs) {
+    try {
+      const dir = getHistoryDir();
+      const desiredFiles = /* @__PURE__ */ new Set();
+      for (const inv of investigations) {
+        const filename = `inv-${inv.id}.json`;
+        desiredFiles.add(filename);
+        const data = {
+          id: inv.id,
+          query: inv.query,
+          report: inv.report,
+          summary: inv.summary,
+          ticketId: inv.ticketId,
+          events: inv.events,
+          resolvedInteractions: inv.resolvedInteractions,
+          completedAt: inv.completedAt.toISOString()
+        };
+        fs2.writeFileSync(
+          path4.join(dir, filename),
+          encrypt(JSON.stringify(data), this.key)
+        );
+      }
+      for (const cr of crs) {
+        const filename = `cr-${cr.id}.json`;
+        desiredFiles.add(filename);
+        const data = {
+          id: cr.id,
+          query: cr.query,
+          slug: cr.slug,
+          ticketId: cr.ticketId,
+          events: cr.events,
+          isRunning: cr.isRunning,
+          isCompleted: cr.isCompleted,
+          isFailed: cr.isFailed,
+          report: cr.report,
+          summary: cr.summary,
+          errorMessage: cr.errorMessage,
+          startedAt: cr.startedAt.toISOString(),
+          pendingInteraction: cr.pendingInteraction,
+          resolvedInteractions: cr.resolvedInteractions ?? []
+        };
+        fs2.writeFileSync(
+          path4.join(dir, filename),
+          encrypt(JSON.stringify(data), this.key)
+        );
+      }
+      try {
+        const existing = fs2.readdirSync(dir);
+        for (const filename of existing) {
+          if (!desiredFiles.has(filename)) {
+            fs2.unlinkSync(path4.join(dir, filename));
+          }
+        }
+      } catch {
+      }
+    } catch {
+      console.error("[PRBEAgent] Failed to save investigation history");
+    }
+  }
+  static clear() {
+    try {
+      const dir = path4.join(getAppDataDir(), "history" /* HISTORY_DIR */);
+      if (fs2.existsSync(dir)) {
+        fs2.rmSync(dir, { recursive: true, force: true });
+      }
+    } catch {
+    }
+  }
+};
+
+// src/agent.ts
+function getPersistencePath() {
+  const appData = process.env["APPDATA"] || (process.platform === "darwin" ? path5.join(os2.homedir(), "Library", "Application Support") : path5.join(os2.homedir(), ".local", "share"));
+  const dir = path5.join(appData, "prbe-agent");
+  if (!fs3.existsSync(dir)) {
+    fs3.mkdirSync(dir, { recursive: true });
+  }
+  return path5.join(dir, "agent-state.json");
 }
 function loadPersistedData() {
   try {
     const filePath = getPersistencePath();
-    if (fs2.existsSync(filePath)) {
-      const raw = fs2.readFileSync(filePath, "utf-8");
+    if (fs3.existsSync(filePath)) {
+      const raw = fs3.readFileSync(filePath, "utf-8");
       return JSON.parse(raw);
     }
   } catch {
@@ -1592,7 +1777,7 @@ function loadPersistedData() {
 function savePersistedData(data) {
   try {
     const filePath = getPersistencePath();
-    fs2.writeFileSync(filePath, JSON.stringify(data, null, 2), "utf-8");
+    fs3.writeFileSync(filePath, JSON.stringify(data, null, 2), "utf-8");
   } catch {
     console.error("[PRBEAgent] Failed to save persisted data");
   }
@@ -1612,6 +1797,7 @@ var PRBEAgent = class _PRBEAgent {
   fetchAbortController = null;
   currentInvestigationSource = "user" /* USER */;
   currentCRId = null;
+  historyStore;
   /** Files flagged during the current tool call — uploaded immediately after the tool returns. */
   pendingFlaggedFiles = [];
   // ---------- Persistence ----------
@@ -1673,6 +1859,10 @@ var PRBEAgent = class _PRBEAgent {
     this.state = new PRBEAgentState();
     this.logCapture = new PRBELogCapture(this.config.maxLogEntries);
     this.persistedData = loadPersistedData();
+    this.historyStore = new HistoryStore(this.config.apiKey);
+    const history = this.historyStore.load();
+    this.state.completedInvestigations = history.investigations;
+    this.state.completedCRs = history.crs;
     const roots = this.config.autoApprovedDirs;
     const requester = this.interactionHandler ? this : void 0;
     const grantedPaths = this.grantedPaths;
@@ -1793,9 +1983,9 @@ var PRBEAgent = class _PRBEAgent {
   /**
    * Register a custom tool that the middleware can invoke during investigations.
    */
-  registerTool(name, description, parameters, handler) {
+  registerTool(name, description, parameters, handler, options) {
     this.registry.register(
-      new PRBEClosureTool(name, description, parameters, handler)
+      new PRBEClosureTool(name, description, parameters, handler, options)
     );
   }
   /**
@@ -1823,7 +2013,8 @@ var PRBEAgent = class _PRBEAgent {
           this.state.appendEvent("Thinking", status.text);
           break;
         case "completed" /* COMPLETED */:
-          this.state.completeInvestigation(status.report, status.userSummary);
+          this.state.completeInvestigation(status.report, status.userSummary, status.ticketId);
+          this.historyStore.save(this.state.completedInvestigations, this.state.completedCRs);
           break;
         case "error" /* ERROR */:
           this.state.failInvestigation(status.message);
@@ -1965,6 +2156,7 @@ var PRBEAgent = class _PRBEAgent {
     this.stopPolling();
     this.persistedData = {};
     savePersistedData(this.persistedData);
+    HistoryStore.clear();
     this.state.resetInvestigation();
     this.state.completedInvestigations = [];
     this.state.activeCRs.clear();
@@ -1979,9 +2171,10 @@ var PRBEAgent = class _PRBEAgent {
   static clearPersistedData() {
     try {
       const filePath = getPersistencePath();
-      if (fs2.existsSync(filePath)) {
-        fs2.unlinkSync(filePath);
+      if (fs3.existsSync(filePath)) {
+        fs3.unlinkSync(filePath);
       }
+      HistoryStore.clear();
     } catch {
     }
   }
@@ -1990,7 +2183,7 @@ var PRBEAgent = class _PRBEAgent {
     const crID = cr.id;
     this.currentInvestigationSource = "context_request" /* CONTEXT_REQUEST */;
     this.currentCRId = crID;
-    this.state.beginCR(crID, cr.query, cr.slug ?? void 0);
+    this.state.beginCR(crID, cr.query, cr.slug ?? void 0, ticketId);
     const emitter = (status) => {
       switch (status.type) {
         case "started" /* STARTED */:
@@ -2009,9 +2202,11 @@ var PRBEAgent = class _PRBEAgent {
           break;
         case "completed" /* COMPLETED */:
           this.state.completeCR(crID, status.report, status.userSummary);
+          this.historyStore.save(this.state.completedInvestigations, this.state.completedCRs);
           break;
         case "error" /* ERROR */:
           this.state.failCR(crID, status.message);
+          this.historyStore.save(this.state.completedInvestigations, this.state.completedCRs);
           break;
       }
     };
@@ -2150,7 +2345,7 @@ var PRBEAgent = class _PRBEAgent {
               const uploadPrefix = this.extractUploadPrefix(uploadBaseUrl);
               const uploadedRefs = [];
               for (const file of this.pendingFlaggedFiles) {
-                const filename = path4.basename(file.originalPath);
+                const filename = path5.basename(file.originalPath);
                 const safeName = encodeURIComponent(filename);
                 const storagePath = `${uploadPrefix}/${safeName}`;
                 uploadedRefs.push({
@@ -2206,7 +2401,8 @@ var PRBEAgent = class _PRBEAgent {
             emit({
               type: "completed" /* COMPLETED */,
               report,
-              userSummary
+              userSummary,
+              ticketId: ticketId2
             });
             ws.close(1e3, "Complete");
             finish({ report, userSummary, ticketId: ticketId2 });
@@ -2389,6 +2585,7 @@ function serializeCR(cr) {
     id: cr.id,
     query: cr.query,
     slug: cr.slug,
+    ticketId: cr.ticketId,
     events: cr.events,
     isRunning: cr.isRunning,
     isCompleted: cr.isCompleted,
@@ -2416,6 +2613,9 @@ function serializePRBEState(state) {
       query: inv.query,
       report: inv.report,
       summary: inv.summary,
+      ticketId: inv.ticketId,
+      events: inv.events,
+      resolvedInteractions: inv.resolvedInteractions,
       completedAt: inv.completedAt.toISOString()
     })),
     activeCRs: Array.from(state.activeCRs.values()).map(serializeCR),