@prbe.ai/electron-sdk 0.1.7 → 0.1.9

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-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 { 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-CeIldtlI.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 ResolveSessionsRequest, J as ResolveSessionsResponse, K as ResolvedInteraction, L as ResolvedTicketOut, N as TicketInfoRequest, O as TicketInfoResponse, Q as TicketStatusOut, S as ToolName, U as ToolParamType, W as WSMessage, V as WSMessageType, X as redactPII, Y as serializePRBEState } from './types-CeIldtlI.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>;
 }
 /**
@@ -148,12 +150,12 @@ declare class PRBEAgent implements PRBEInteractionRequester {
     /** Files flagged during the current tool call — uploaded immediately after the tool returns. */
     private pendingFlaggedFiles;
     private get agentID();
-    private get trackedTicketIDs();
-    private set trackedTicketIDs(value);
+    private get trackedSessionIDs();
+    private set trackedSessionIDs(value);
     private get respondedCRIDs();
     private set respondedCRIDs(value);
     private syncPolling;
-    private addTrackedTicket;
+    private addTrackedSession;
     constructor(config: PRBEAgentConfig);
     private hookElectronLog;
     private hookRendererLogs;
@@ -167,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.
      */
@@ -182,10 +186,12 @@ declare class PRBEAgent implements PRBEInteractionRequester {
     cancel(): void;
     /**
      * Poll the backend for context requests on tracked tickets.
+     * Resolves session IDs → ticket IDs first, then polls with ticket IDs.
      */
     poll(): Promise<PollResponse | null>;
     /**
-     * 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.
      */
     fetchTicketInfo(): Promise<TicketInfoOut[]>;
     stopPolling(): void;
@@ -210,6 +216,7 @@ declare class PRBEAgent implements PRBEInteractionRequester {
     private extractUploadPrefix;
     private static backgroundUpload;
     private startPolling;
+    private resolveSessions;
     /**
      * Abort all in-flight fetch requests to prevent orphaned DNS lookups
      * that can crash the c-ares resolver during process shutdown.

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-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 { 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-CeIldtlI.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 ResolveSessionsRequest, J as ResolveSessionsResponse, K as ResolvedInteraction, L as ResolvedTicketOut, N as TicketInfoRequest, O as TicketInfoResponse, Q as TicketStatusOut, S as ToolName, U as ToolParamType, W as WSMessage, V as WSMessageType, X as redactPII, Y as serializePRBEState } from './types-CeIldtlI.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>;
 }
 /**
@@ -148,12 +150,12 @@ declare class PRBEAgent implements PRBEInteractionRequester {
     /** Files flagged during the current tool call — uploaded immediately after the tool returns. */
     private pendingFlaggedFiles;
     private get agentID();
-    private get trackedTicketIDs();
-    private set trackedTicketIDs(value);
+    private get trackedSessionIDs();
+    private set trackedSessionIDs(value);
     private get respondedCRIDs();
     private set respondedCRIDs(value);
     private syncPolling;
-    private addTrackedTicket;
+    private addTrackedSession;
     constructor(config: PRBEAgentConfig);
     private hookElectronLog;
     private hookRendererLogs;
@@ -167,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.
      */
@@ -182,10 +186,12 @@ declare class PRBEAgent implements PRBEInteractionRequester {
     cancel(): void;
     /**
      * Poll the backend for context requests on tracked tickets.
+     * Resolves session IDs → ticket IDs first, then polls with ticket IDs.
      */
     poll(): Promise<PollResponse | null>;
     /**
-     * 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.
      */
     fetchTicketInfo(): Promise<TicketInfoOut[]>;
     stopPolling(): void;
@@ -210,6 +216,7 @@ declare class PRBEAgent implements PRBEInteractionRequester {
     private extractUploadPrefix;
     private static backgroundUpload;
     private startPolling;
+    private resolveSessions;
     /**
      * Abort all in-flight fetch requests to prevent orphaned DNS lookups
      * that can crash the c-ares resolver during process shutdown.

package/dist/index.js

@@ -192,7 +192,7 @@ var PRBEAgentState = class extends import_events.EventEmitter {
   activeCRs = /* @__PURE__ */ new Map();
   completedCRs = [];
   // Tracked tickets
-  trackedTicketIDs = [];
+  trackedSessionIDs = [];
   ticketInfo = [];
   // Computed
   get hasActiveWork() {
@@ -420,8 +420,8 @@ var PRBEAgentState = class extends import_events.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 */);
   }
@@ -469,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) {
@@ -1805,13 +1810,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() {
@@ -1821,19 +1826,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 ----------
@@ -1891,9 +1896,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 ----------
@@ -1978,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)
     );
   }
   /**
@@ -2024,8 +2029,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";
@@ -2057,33 +2062,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;
@@ -2109,13 +2118,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
@@ -2133,7 +2146,7 @@ var PRBEAgent = class _PRBEAgent {
     }
   }
   resumePolling() {
-    if (this.trackedTicketIDs.length === 0) return;
+    if (this.trackedSessionIDs.length === 0) return;
     this.startPolling();
   }
   /**
@@ -2156,7 +2169,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 */);
   }
@@ -2215,8 +2228,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 ----------
@@ -2393,6 +2406,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,
@@ -2400,7 +2414,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 */:
@@ -2511,12 +2525,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
@@ -2571,7 +2596,7 @@ var DEFAULT_PRBE_STATE = {
   completedInvestigations: [],
   activeCRs: [],
   completedCRs: [],
-  trackedTicketIDs: [],
+  trackedSessionIDs: [],
   ticketInfo: [],
   hasActiveWork: false
 };
@@ -2615,7 +2640,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
   };