@prbe.ai/electron-sdk 0.1.16 → 0.1.17

package/dist/index.d.mts

@@ -1,8 +1,8 @@
-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-Cyfgy1uD.mjs';
-export { A as API_URL, h as AskQuestionPayload, i as AskQuestionResponse, C as ContextRequestOut, j as ConversationEntry, k as ConversationRole, D as DEFAULT_PRBE_STATE, l as InteractionType, m as InvestigationResult, M as MIDDLEWARE_URL, n as PRBEAgentConfigKey, o as PRBEAgentError, p as PRBEAgentErrorType, q as PRBEAgentStatus, r as PRBEAgentStatusType, s as PRBECRInvestigation, t as PRBECompletedInvestigation, u as PRBEInteractionHandler, v as PRBESerializedCR, w as PRBESerializedCompletedInvestigation, x as PRBESerializedState, y as PRBESerializedTicket, z as PRBEStateEvent, B as PRBEStatusEvent, E as PollRequest, R as RequestPathAccessPayload, G as RequestPathAccessResponse, H as RequestPermissionPayload, J as RequestPermissionResponse, K as ResolveSessionsRequest, L as ResolveSessionsResponse, N as ResolvedInteraction, O as ResolvedTicketOut, Q as TicketInfoRequest, S as TicketInfoResponse, U as TicketStatusOut, V as ToolName, W as ToolParamType, X as WSMessage, Y as WSMessageType, Z as redactPII, _ as serializePRBEState } from './types-Cyfgy1uD.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, U as UserIdentifierType, g as PollResponse, T as TicketInfoOut } from './types-BmH_CmsO.mjs';
+export { A as API_URL, h as AskQuestionPayload, i as AskQuestionResponse, C as ContextRequestOut, j as ConversationEntry, k as ConversationRole, D as DEFAULT_PRBE_STATE, l as InteractionType, m as InvestigationResult, M as MIDDLEWARE_URL, n as PRBEAgentConfigKey, o as PRBEAgentError, p as PRBEAgentErrorType, q as PRBEAgentStatus, r as PRBEAgentStatusType, s as PRBECRInvestigation, t as PRBECompletedInvestigation, u as PRBEInteractionHandler, v as PRBESerializedCR, w as PRBESerializedCompletedInvestigation, x as PRBESerializedState, y as PRBESerializedTicket, z as PRBEStateEvent, B as PRBEStatusEvent, E as PollRequest, R as RequestPathAccessPayload, G as RequestPathAccessResponse, H as RequestPermissionPayload, J as RequestPermissionResponse, K as ResolveSessionsRequest, L as ResolveSessionsResponse, N as ResolvedInteraction, O as ResolvedTicketOut, Q as TicketInfoRequest, S as TicketInfoResponse, V as TicketStatusOut, W as ToolName, X as ToolParamType, Y as WSMessage, Z as WSMessageType, _ as redactPII, $ as serializePRBEState } from './types-BmH_CmsO.mjs';
 import 'events';
 
-var version = "0.1.16";
+var version = "0.1.17";
 
 /**
  * tools/index.ts — Tool interface, registry, and closure tool
@@ -152,6 +152,8 @@ declare class PRBEAgent implements PRBEInteractionRequester {
     private readonly historyStore;
     /** Files flagged during the current tool call — uploaded immediately after the tool returns. */
     private pendingFlaggedFiles;
+    private userEmail?;
+    private userId?;
     private get agentID();
     private get trackedSessionIDs();
     private set trackedSessionIDs(value);
@@ -173,6 +175,18 @@ declare class PRBEAgent implements PRBEInteractionRequester {
      * Call this to add user profile data, app version, etc. after async initialization.
      */
     updateSessionMetadata(metadata: Record<string, unknown>): void;
+    /**
+     * Set a user identifier for this agent. Persists to the backend agents table.
+     * Can be called multiple times with different types to set both email and ID.
+     * Also auto-injects user_email and/or user_id into session metadata for investigations.
+     */
+    setUserIdentifier(type: UserIdentifierType, value: string): void;
+    /**
+     * Clear user identifier(s) for this agent.
+     * Pass a specific type to clear only that identifier, or omit to clear both email and ID.
+     */
+    clearUserIdentifier(type?: UserIdentifierType): void;
+    private syncUserIdentifierToBackend;
     /**
      * Register a custom tool that the middleware can invoke during investigations.
      */
@@ -346,4 +360,4 @@ declare class BashExecuteTool implements PRBETool {
  */
 declare const PROBE_MARK_SVG = "<svg width=\"256\" height=\"256\" viewBox=\"0 0 256 256\" fill=\"none\" xmlns=\"http://www.w3.org/2000/svg\">\n  <path d=\"M50 112 L114 148 L50 184 Z\" fill=\"#111111\" opacity=\"0.07\"/>\n  <path d=\"M70 86 L148 134 L70 182 Z\" fill=\"#111111\" opacity=\"0.18\"/>\n  <path d=\"M92 56 L192 118 L92 180 Z\" fill=\"#111111\"/>\n</svg>";
 
-export { AskUserTool, BashExecuteTool, ClearAppLogsTool, FindFilesTool, FlagAppLogsTool, FlagFileTool, FlaggedFileIn, InteractionPayload, InteractionResponse, InvestigationSource, ListDirectoryTool, type LogEntry, PRBEAgent, PRBEAgentConfig, PRBEAgentState, PRBEClosureTool, PRBEInteractionRequester, PRBELogCapture, type PRBETool, PRBEToolDeclaration, PRBEToolParameter, PRBEToolRegistry, PROBE_MARK_SVG, PollResponse, ReadAppLogsTool, ReadFileTool, version as SDK_VERSION, SearchAppLogsTool, SearchContentTool, TicketInfoOut, humanReadableSize, resolveAndValidate, resolveWithAccessRequest };
+export { AskUserTool, BashExecuteTool, ClearAppLogsTool, FindFilesTool, FlagAppLogsTool, FlagFileTool, FlaggedFileIn, InteractionPayload, InteractionResponse, InvestigationSource, ListDirectoryTool, type LogEntry, PRBEAgent, PRBEAgentConfig, PRBEAgentState, PRBEClosureTool, PRBEInteractionRequester, PRBELogCapture, type PRBETool, PRBEToolDeclaration, PRBEToolParameter, PRBEToolRegistry, PROBE_MARK_SVG, PollResponse, ReadAppLogsTool, ReadFileTool, version as SDK_VERSION, SearchAppLogsTool, SearchContentTool, TicketInfoOut, UserIdentifierType, humanReadableSize, resolveAndValidate, resolveWithAccessRequest };

package/dist/index.d.ts

@@ -1,8 +1,8 @@
-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-Cyfgy1uD.js';
-export { A as API_URL, h as AskQuestionPayload, i as AskQuestionResponse, C as ContextRequestOut, j as ConversationEntry, k as ConversationRole, D as DEFAULT_PRBE_STATE, l as InteractionType, m as InvestigationResult, M as MIDDLEWARE_URL, n as PRBEAgentConfigKey, o as PRBEAgentError, p as PRBEAgentErrorType, q as PRBEAgentStatus, r as PRBEAgentStatusType, s as PRBECRInvestigation, t as PRBECompletedInvestigation, u as PRBEInteractionHandler, v as PRBESerializedCR, w as PRBESerializedCompletedInvestigation, x as PRBESerializedState, y as PRBESerializedTicket, z as PRBEStateEvent, B as PRBEStatusEvent, E as PollRequest, R as RequestPathAccessPayload, G as RequestPathAccessResponse, H as RequestPermissionPayload, J as RequestPermissionResponse, K as ResolveSessionsRequest, L as ResolveSessionsResponse, N as ResolvedInteraction, O as ResolvedTicketOut, Q as TicketInfoRequest, S as TicketInfoResponse, U as TicketStatusOut, V as ToolName, W as ToolParamType, X as WSMessage, Y as WSMessageType, Z as redactPII, _ as serializePRBEState } from './types-Cyfgy1uD.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, U as UserIdentifierType, g as PollResponse, T as TicketInfoOut } from './types-BmH_CmsO.js';
+export { A as API_URL, h as AskQuestionPayload, i as AskQuestionResponse, C as ContextRequestOut, j as ConversationEntry, k as ConversationRole, D as DEFAULT_PRBE_STATE, l as InteractionType, m as InvestigationResult, M as MIDDLEWARE_URL, n as PRBEAgentConfigKey, o as PRBEAgentError, p as PRBEAgentErrorType, q as PRBEAgentStatus, r as PRBEAgentStatusType, s as PRBECRInvestigation, t as PRBECompletedInvestigation, u as PRBEInteractionHandler, v as PRBESerializedCR, w as PRBESerializedCompletedInvestigation, x as PRBESerializedState, y as PRBESerializedTicket, z as PRBEStateEvent, B as PRBEStatusEvent, E as PollRequest, R as RequestPathAccessPayload, G as RequestPathAccessResponse, H as RequestPermissionPayload, J as RequestPermissionResponse, K as ResolveSessionsRequest, L as ResolveSessionsResponse, N as ResolvedInteraction, O as ResolvedTicketOut, Q as TicketInfoRequest, S as TicketInfoResponse, V as TicketStatusOut, W as ToolName, X as ToolParamType, Y as WSMessage, Z as WSMessageType, _ as redactPII, $ as serializePRBEState } from './types-BmH_CmsO.js';
 import 'events';
 
-var version = "0.1.16";
+var version = "0.1.17";
 
 /**
  * tools/index.ts — Tool interface, registry, and closure tool
@@ -152,6 +152,8 @@ declare class PRBEAgent implements PRBEInteractionRequester {
     private readonly historyStore;
     /** Files flagged during the current tool call — uploaded immediately after the tool returns. */
     private pendingFlaggedFiles;
+    private userEmail?;
+    private userId?;
     private get agentID();
     private get trackedSessionIDs();
     private set trackedSessionIDs(value);
@@ -173,6 +175,18 @@ declare class PRBEAgent implements PRBEInteractionRequester {
      * Call this to add user profile data, app version, etc. after async initialization.
      */
     updateSessionMetadata(metadata: Record<string, unknown>): void;
+    /**
+     * Set a user identifier for this agent. Persists to the backend agents table.
+     * Can be called multiple times with different types to set both email and ID.
+     * Also auto-injects user_email and/or user_id into session metadata for investigations.
+     */
+    setUserIdentifier(type: UserIdentifierType, value: string): void;
+    /**
+     * Clear user identifier(s) for this agent.
+     * Pass a specific type to clear only that identifier, or omit to clear both email and ID.
+     */
+    clearUserIdentifier(type?: UserIdentifierType): void;
+    private syncUserIdentifierToBackend;
     /**
      * Register a custom tool that the middleware can invoke during investigations.
      */
@@ -346,4 +360,4 @@ declare class BashExecuteTool implements PRBETool {
  */
 declare const PROBE_MARK_SVG = "<svg width=\"256\" height=\"256\" viewBox=\"0 0 256 256\" fill=\"none\" xmlns=\"http://www.w3.org/2000/svg\">\n  <path d=\"M50 112 L114 148 L50 184 Z\" fill=\"#111111\" opacity=\"0.07\"/>\n  <path d=\"M70 86 L148 134 L70 182 Z\" fill=\"#111111\" opacity=\"0.18\"/>\n  <path d=\"M92 56 L192 118 L92 180 Z\" fill=\"#111111\"/>\n</svg>";
 
-export { AskUserTool, BashExecuteTool, ClearAppLogsTool, FindFilesTool, FlagAppLogsTool, FlagFileTool, FlaggedFileIn, InteractionPayload, InteractionResponse, InvestigationSource, ListDirectoryTool, type LogEntry, PRBEAgent, PRBEAgentConfig, PRBEAgentState, PRBEClosureTool, PRBEInteractionRequester, PRBELogCapture, type PRBETool, PRBEToolDeclaration, PRBEToolParameter, PRBEToolRegistry, PROBE_MARK_SVG, PollResponse, ReadAppLogsTool, ReadFileTool, version as SDK_VERSION, SearchAppLogsTool, SearchContentTool, TicketInfoOut, humanReadableSize, resolveAndValidate, resolveWithAccessRequest };
+export { AskUserTool, BashExecuteTool, ClearAppLogsTool, FindFilesTool, FlagAppLogsTool, FlagFileTool, FlaggedFileIn, InteractionPayload, InteractionResponse, InvestigationSource, ListDirectoryTool, type LogEntry, PRBEAgent, PRBEAgentConfig, PRBEAgentState, PRBEClosureTool, PRBEInteractionRequester, PRBELogCapture, type PRBETool, PRBEToolDeclaration, PRBEToolParameter, PRBEToolRegistry, PROBE_MARK_SVG, PollResponse, ReadAppLogsTool, ReadFileTool, version as SDK_VERSION, SearchAppLogsTool, SearchContentTool, TicketInfoOut, UserIdentifierType, humanReadableSize, resolveAndValidate, resolveWithAccessRequest };

package/dist/index.js

@@ -61,6 +61,7 @@ __export(src_exports, {
   SearchContentTool: () => SearchContentTool,
   ToolName: () => ToolName,
   ToolParamType: () => ToolParamType,
+  UserIdentifierType: () => UserIdentifierType,
   WSMessageType: () => WSMessageType,
   humanReadableSize: () => humanReadableSize,
   redactPII: () => redactPII,
@@ -71,7 +72,7 @@ __export(src_exports, {
 module.exports = __toCommonJS(src_exports);
 
 // package.json
-var version = "0.1.16";
+var version = "0.1.17";
 
 // src/agent.ts
 var fs3 = __toESM(require("fs"));
@@ -125,6 +126,11 @@ var ToolName = /* @__PURE__ */ ((ToolName2) => {
   ToolName2["CLIENT_MESSAGE_USER"] = "client_message_user";
   return ToolName2;
 })(ToolName || {});
+var UserIdentifierType = /* @__PURE__ */ ((UserIdentifierType2) => {
+  UserIdentifierType2["EMAIL"] = "email";
+  UserIdentifierType2["ID"] = "id";
+  return UserIdentifierType2;
+})(UserIdentifierType || {});
 var PRBEAgentConfigKey = /* @__PURE__ */ ((PRBEAgentConfigKey3) => {
   PRBEAgentConfigKey3["API_KEY"] = "apiKey";
   PRBEAgentConfigKey3["AUTO_APPROVED_DIRS"] = "autoApprovedDirs";
@@ -1988,6 +1994,9 @@ var PRBEAgent = class _PRBEAgent {
   historyStore;
   /** Files flagged during the current tool call — uploaded immediately after the tool returns. */
   pendingFlaggedFiles = [];
+  // ---------- User Identifier ----------
+  userEmail;
+  userId;
   // ---------- Persistence ----------
   get agentID() {
     if (this.persistedData.agentId) {
@@ -2042,6 +2051,7 @@ var PRBEAgent = class _PRBEAgent {
     this.logCapture = new PRBELogCapture(this.config.maxLogEntries);
     this.persistedData = loadPersistedData();
     this.historyStore = new HistoryStore(this.config.apiKey);
+    void this.agentID;
     const history = this.historyStore.load();
     this.state.completedInvestigations = history.investigations;
     this.state.completedCRs = history.crs;
@@ -2195,6 +2205,47 @@ var PRBEAgent = class _PRBEAgent {
   updateSessionMetadata(metadata) {
     this.sessionMetadata = { ...this.sessionMetadata, ...metadata };
   }
+  /**
+   * Set a user identifier for this agent. Persists to the backend agents table.
+   * Can be called multiple times with different types to set both email and ID.
+   * Also auto-injects user_email and/or user_id into session metadata for investigations.
+   */
+  setUserIdentifier(type, value) {
+    if (type === "email" /* EMAIL */) {
+      this.userEmail = value;
+    } else {
+      this.userId = value;
+    }
+    this.syncUserIdentifierToBackend(type, value);
+  }
+  /**
+   * Clear user identifier(s) for this agent.
+   * Pass a specific type to clear only that identifier, or omit to clear both email and ID.
+   */
+  clearUserIdentifier(type) {
+    if (type) {
+      if (type === "email" /* EMAIL */) {
+        this.userEmail = void 0;
+      } else {
+        this.userId = void 0;
+      }
+      this.syncUserIdentifierToBackend(type, null);
+    } else {
+      this.userEmail = void 0;
+      this.userId = void 0;
+      this.syncUserIdentifierToBackend("email" /* EMAIL */, null);
+      this.syncUserIdentifierToBackend("id" /* ID */, null);
+    }
+  }
+  syncUserIdentifierToBackend(type, value) {
+    void this.post("/api/agent/update-user", {
+      agent_id: this.agentID,
+      identifier_type: type,
+      identifier_value: value
+    }).catch((err) => {
+      console.error(`[PRBEAgent] Failed to sync user identifier: ${err}`);
+    });
+  }
   /**
    * Register a custom tool that the middleware can invoke during investigations.
    */
@@ -2495,8 +2546,15 @@ var PRBEAgent = class _PRBEAgent {
         if (contextRequestID) startMetadata["context_request_id"] = contextRequestID;
         if (ticketId) startMetadata["ticket_id"] = ticketId;
         if (this.appDataPath) startMetadata["app_data_path"] = this.appDataPath;
-        if (Object.keys(this.sessionMetadata).length > 0) {
-          startMetadata["session_metadata"] = this.sessionMetadata;
+        const enrichedMetadata = { ...this.sessionMetadata };
+        if (this.userEmail) {
+          enrichedMetadata["user_email"] = this.userEmail;
+        }
+        if (this.userId) {
+          enrichedMetadata["user_id"] = this.userId;
+        }
+        if (Object.keys(enrichedMetadata).length > 0) {
+          startMetadata["session_metadata"] = enrichedMetadata;
         }
         if (!conn.send({ type: "start" /* START */, content: query, metadata: startMetadata })) {
           emit({ type: "error" /* ERROR */, message: "Failed to send start message" });
@@ -2802,6 +2860,7 @@ var PROBE_MARK_SVG = `<svg width="256" height="256" viewBox="0 0 256 256" fill="
   SearchContentTool,
   ToolName,
   ToolParamType,
+  UserIdentifierType,
   WSMessageType,
   humanReadableSize,
   redactPII,