@prbe.ai/electron-sdk 0.1.7 → 0.1.9

package/dist/index.mjs

@@ -122,7 +122,7 @@ var PRBEAgentState = class extends EventEmitter {
   activeCRs = /* @__PURE__ */ new Map();
   completedCRs = [];
   // Tracked tickets
-  trackedTicketIDs = [];
+  trackedSessionIDs = [];
   ticketInfo = [];
   // Computed
   get hasActiveWork() {
@@ -350,8 +350,8 @@ var PRBEAgentState = class extends EventEmitter {
     this.emit("status" /* STATUS */);
   }
   // ---------- Tickets ----------
-  updateTrackedTicketIDs(ids) {
-    this.trackedTicketIDs = ids;
+  updateTrackedSessionIDs(ids) {
+    this.trackedSessionIDs = ids;
     this.emit("tickets-changed" /* TICKETS_CHANGED */, ids);
     this.emit("status" /* STATUS */);
   }
@@ -399,8 +399,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) {
@@ -1735,13 +1740,13 @@ var PRBEAgent = class _PRBEAgent {
     savePersistedData(this.persistedData);
     return newID;
   }
-  get trackedTicketIDs() {
-    return this.persistedData.ticketIds ?? [];
+  get trackedSessionIDs() {
+    return this.persistedData.sessionIds ?? [];
   }
-  set trackedTicketIDs(ids) {
-    this.persistedData.ticketIds = ids;
+  set trackedSessionIDs(ids) {
+    this.persistedData.sessionIds = ids;
     savePersistedData(this.persistedData);
-    this.state.updateTrackedTicketIDs(ids);
+    this.state.updateTrackedSessionIDs(ids);
     this.syncPolling(ids.length > 0);
   }
   get respondedCRIDs() {
@@ -1751,19 +1756,19 @@ var PRBEAgent = class _PRBEAgent {
     this.persistedData.respondedCRIds = Array.from(ids);
     savePersistedData(this.persistedData);
   }
-  syncPolling(hasTickets) {
-    if (this.config.backgroundPolling && hasTickets) {
+  syncPolling(hasSessions) {
+    if (this.config.backgroundPolling && hasSessions) {
       if (this.pollingTimer === null) {
         this.startPolling();
       }
-    } else if (!hasTickets) {
+    } else if (!hasSessions) {
       this.stopPolling();
     }
   }
-  addTrackedTicket(id) {
-    const ids = this.trackedTicketIDs;
+  addTrackedSession(id) {
+    const ids = this.trackedSessionIDs;
     if (!ids.includes(id)) {
-      this.trackedTicketIDs = [...ids, id];
+      this.trackedSessionIDs = [...ids, id];
     }
   }
   // ---------- Constructor ----------
@@ -1821,9 +1826,9 @@ var PRBEAgent = class _PRBEAgent {
     if (config.ipcMain) {
       this.hookRendererLogs(config.ipcMain, config.rendererLogChannel ?? "prbe-renderer-log");
     }
-    const existingTickets = this.trackedTicketIDs;
-    if (existingTickets.length > 0) {
-      this.trackedTicketIDs = existingTickets;
+    const existingSessions = this.trackedSessionIDs;
+    if (existingSessions.length > 0) {
+      this.trackedSessionIDs = existingSessions;
     }
   }
   // ---------- Log integration ----------
@@ -1908,9 +1913,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)
     );
   }
   /**
@@ -1954,8 +1959,8 @@ var PRBEAgent = class _PRBEAgent {
     );
     this.currentInvestigationSource = "user" /* USER */;
     this.currentCRId = null;
-    if (result?.ticketId) {
-      this.addTrackedTicket(result.ticketId);
+    if (result?.sessionId) {
+      this.addTrackedSession(result.sessionId);
     } else if (!result) {
       if (this.state.isInvestigating) {
         const message = this.userCancelled ? "Investigation cancelled" : "Investigation ended unexpectedly";
@@ -1987,33 +1992,37 @@ var PRBEAgent = class _PRBEAgent {
   }
   /**
    * Poll the backend for context requests on tracked tickets.
+   * Resolves session IDs → ticket IDs first, then polls with ticket IDs.
    */
   async poll() {
-    const ticketIDs = this.trackedTicketIDs;
-    if (ticketIDs.length === 0) return null;
-    const request = {
-      agent_id: this.agentID,
-      ticket_ids: ticketIDs
-    };
+    const sessionIDs = this.trackedSessionIDs;
+    if (sessionIDs.length === 0) return null;
     try {
+      const resolved = await this.resolveSessions(sessionIDs);
+      const returnedSessionIDs = new Set(
+        resolved.tickets.flatMap((t) => t.session_ids)
+      );
+      const resolvedSessionIDs = new Set(
+        resolved.tickets.filter((t) => t.status === "resolved").flatMap((t) => t.session_ids)
+      );
+      const survivingSessions = sessionIDs.filter(
+        (id) => returnedSessionIDs.has(id) && !resolvedSessionIDs.has(id)
+      );
+      if (survivingSessions.length !== sessionIDs.length) {
+        this.trackedSessionIDs = survivingSessions;
+      }
+      const ticketIDs = resolved.tickets.filter((t) => t.status !== "resolved").map((t) => t.ticket_id);
+      if (ticketIDs.length === 0) return { tickets: [] };
+      const request = {
+        agent_id: this.agentID,
+        ticket_ids: ticketIDs
+      };
       const response = await this.post(
         "/api/agent/poll",
         request
       );
-      const returnedIDs = new Set(response.tickets.map((t) => t.ticket_id));
-      const orphaned = ticketIDs.filter((id) => !returnedIDs.has(id));
-      if (orphaned.length > 0) {
-        this.trackedTicketIDs = this.trackedTicketIDs.filter(
-          (id) => !orphaned.includes(id)
-        );
-      }
       const knownCRIDs = /* @__PURE__ */ new Set();
       for (const ticket of response.tickets) {
-        if (ticket.status === "resolved") {
-          this.trackedTicketIDs = this.trackedTicketIDs.filter(
-            (id) => id !== ticket.ticket_id
-          );
-        }
         for (const cr of ticket.context_requests) {
           knownCRIDs.add(cr.id);
           if (!cr.is_active || this.respondedCRIDs.has(cr.id)) continue;
@@ -2039,13 +2048,17 @@ var PRBEAgent = class _PRBEAgent {
     }
   }
   /**
-   * Fetch ticket display info for all tracked tickets.
+   * Fetch ticket display info for all tracked sessions.
+   * Resolves session IDs → ticket IDs first, then fetches ticket info.
    */
   async fetchTicketInfo() {
-    const ticketIDs = this.trackedTicketIDs;
-    if (ticketIDs.length === 0) return [];
-    const request = { ticket_ids: ticketIDs };
+    const sessionIDs = this.trackedSessionIDs;
+    if (sessionIDs.length === 0) return [];
     try {
+      const resolved = await this.resolveSessions(sessionIDs);
+      const ticketIDs = resolved.tickets.map((t) => t.ticket_id);
+      if (ticketIDs.length === 0) return [];
+      const request = { ticket_ids: ticketIDs };
       const response = await this.post(
         "/api/agent/tickets",
         request
@@ -2063,7 +2076,7 @@ var PRBEAgent = class _PRBEAgent {
     }
   }
   resumePolling() {
-    if (this.trackedTicketIDs.length === 0) return;
+    if (this.trackedSessionIDs.length === 0) return;
     this.startPolling();
   }
   /**
@@ -2086,7 +2099,7 @@ var PRBEAgent = class _PRBEAgent {
     this.state.completedInvestigations = [];
     this.state.activeCRs.clear();
     this.state.completedCRs = [];
-    this.state.trackedTicketIDs = [];
+    this.state.trackedSessionIDs = [];
     this.state.ticketInfo = [];
     this.state.emit("status" /* STATUS */);
   }
@@ -2145,8 +2158,8 @@ var PRBEAgent = class _PRBEAgent {
     );
     this.currentInvestigationSource = "user" /* USER */;
     this.currentCRId = null;
-    if (result?.ticketId) {
-      this.addTrackedTicket(result.ticketId);
+    if (result?.sessionId) {
+      this.addTrackedSession(result.sessionId);
     }
   }
   // ---------- WebSocket Investigation ----------
@@ -2323,6 +2336,7 @@ var PRBEAgent = class _PRBEAgent {
             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,
@@ -2330,7 +2344,7 @@ var PRBEAgent = class _PRBEAgent {
               ticketId: ticketId2
             });
             ws.close(1e3, "Complete");
-            finish({ report, userSummary, ticketId: ticketId2 });
+            finish({ report, userSummary, ticketId: ticketId2, sessionId });
             break;
           }
           case "error" /* ERROR */:
@@ -2441,12 +2455,23 @@ var PRBEAgent = class _PRBEAgent {
     this.stopPolling();
     this.pollingTimer = setInterval(() => {
       void this.poll().then(() => {
-        if (this.trackedTicketIDs.length === 0) {
+        if (this.trackedSessionIDs.length === 0) {
           this.stopPolling();
         }
       });
     }, this.config.pollingInterval);
   }
+  // ---------- Session Resolution ----------
+  async resolveSessions(sessionIDs) {
+    const request = {
+      agent_id: this.agentID,
+      session_ids: sessionIDs
+    };
+    return this.post(
+      "/api/agent/resolve-sessions",
+      request
+    );
+  }
   // ---------- Networking ----------
   /**
    * Abort all in-flight fetch requests to prevent orphaned DNS lookups
@@ -2501,7 +2526,7 @@ var DEFAULT_PRBE_STATE = {
   completedInvestigations: [],
   activeCRs: [],
   completedCRs: [],
-  trackedTicketIDs: [],
+  trackedSessionIDs: [],
   ticketInfo: [],
   hasActiveWork: false
 };
@@ -2545,7 +2570,7 @@ function serializePRBEState(state) {
     })),
     activeCRs: Array.from(state.activeCRs.values()).map(serializeCR),
     completedCRs: state.completedCRs.map(serializeCR),
-    trackedTicketIDs: state.trackedTicketIDs,
+    trackedSessionIDs: state.trackedSessionIDs,
     ticketInfo: state.ticketInfo,
     hasActiveWork: state.hasActiveWork
   };