@prbe.ai/electron-sdk 0.1.15 → 0.1.17

package/dist/index.mjs

@@ -1,3 +1,6 @@
+// package.json
+var version = "0.1.17";
+
 // src/agent.ts
 import * as fs3 from "fs";
 import * as path5 from "path";
@@ -50,6 +53,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";
@@ -1913,6 +1921,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) {
@@ -1959,7 +1970,7 @@ var PRBEAgent = class _PRBEAgent {
     };
     this.interactionHandler = config.interactionHandler;
     this.appDataPath = config.appDataPath;
-    this.sessionMetadata = config.sessionMetadata ?? {};
+    this.sessionMetadata = { sdk_version: version, ...config.sessionMetadata };
     if (this.appDataPath && !this.config.autoApprovedDirs.includes(this.appDataPath)) {
       this.config.autoApprovedDirs.push(this.appDataPath);
     }
@@ -1967,6 +1978,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;
@@ -2120,6 +2132,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.
    */
@@ -2420,8 +2473,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" });
@@ -2721,10 +2781,12 @@ export {
   PROBE_MARK_SVG,
   ReadAppLogsTool,
   ReadFileTool,
+  version as SDK_VERSION,
   SearchAppLogsTool,
   SearchContentTool,
   ToolName,
   ToolParamType,
+  UserIdentifierType,
   WSMessageType,
   humanReadableSize,
   redactPII,