@salesforce/agents 0.20.1-beta.0 → 0.21.1

package/lib/{productionAgent.js → agents/productionAgent.js}

@@ -19,19 +19,19 @@ exports.ProductionAgent = void 0;
 const node_crypto_1 = require("node:crypto");
 const core_1 = require("@salesforce/core");
 const kit_1 = require("@salesforce/kit");
-const maybe_mock_1 = require("./maybe-mock");
-const utils_1 = require("./utils");
-const apexUtils_1 = require("./apexUtils");
-const agentInteractionBase_1 = require("./agentInteractionBase");
+const maybe_mock_1 = require("../maybe-mock");
+const utils_1 = require("../utils");
+const apexUtils_1 = require("../apexUtils");
+const agentBase_1 = require("./agentBase");
 ;
 const messages = new core_1.Messages('@salesforce/agents', 'agents', new Map([["invalidAgentSpecConfig", "Missing one or more of the required agent spec arguments: type, role, companyName, companyDescription"], ["missingAgentName", "The \"agentName\" configuration property is required when saving an agent."], ["agentRetrievalError", "Unable to retrieve newly created Agent metadata. Due to: %s"], ["agentRetrievalErrorActions", "Retrieve the agent metadata using the \"project retrieve start\" command."], ["missingAgentNameOrId", "The \"nameOrId\" agent option is required when creating an Agent instance."], ["agentIsDeleted", "The %s agent has been deleted and its activation state can't be changed."], ["agentActivationError", "Changing the agent's activation status was unsuccessful due to %s."]]));
-class ProductionAgent extends agentInteractionBase_1.AgentInteractionBase {
+class ProductionAgent extends agentBase_1.AgentBase {
     options;
     preview;
     botMetadata;
     id;
     apiName;
-    apiBase = `https://${kit_1.env.getBoolean('SF_TEST_API') ? 'test.' : ''}api.salesforce.com/einstein/ai-agent/v1`;
+    apiBase = `https://${(0, utils_1.getEndpoint)()}api.salesforce.com/einstein/ai-agent/v1`;
     constructor(options) {
         super(options.connection);
         this.options = options;
@@ -43,7 +43,7 @@ class ProductionAgent extends agentInteractionBase_1.AgentInteractionBase {
             send: (message) => this.sendMessage(message),
             getAllTraces: () => this.getAllTracesFromDisc(),
             end: (reason) => this.endSession(reason),
-            saveSession: (outputDir) => this.saveSessionToDisc(outputDir),
+            saveSession: (outputDir) => this.saveSessionTo(outputDir),
             setApexDebugging: (apexDebugging) => this.setApexDebugging(apexDebugging),
         };
         if (options.apiNameOrId.startsWith('0Xx') && [15, 18].includes(options.apiNameOrId.length)) {
@@ -78,16 +78,13 @@ class ProductionAgent extends agentInteractionBase_1.AgentInteractionBase {
     }
     // eslint-disable-next-line @typescript-eslint/require-await,class-methods-use-this,@typescript-eslint/no-unused-vars
     async getTrace(planId) {
-        // return this.connection.request<PlannerResponse>({
-        //   method: 'GET',
-        //   url: `${this.connection.getConnectionOptions().instanceUrl!}:9443/proxy/worker/internal/sessions/${this
-        //     .sessionId!}/plans/${planId}`,
-        //   headers: {
-        //     'x-client-name': 'afdx',
-        //   },
-        // });
         return undefined;
     }
+    getHistoryFromDisc(sessionId) {
+        // Use provided sessionId, or fall back to this.sessionId, or let getAllHistory find the most recent
+        const actualSessionId = sessionId ?? this.sessionId;
+        return (0, utils_1.getAllHistory)(this.getAgentIdForStorage(), actualSessionId);
+    }
     /**
      * Returns the ID for this agent.
      *
@@ -134,11 +131,90 @@ class ProductionAgent extends agentInteractionBase_1.AgentInteractionBase {
             }
         }
     }
-    getSendMessageUrl() {
+    async sendMessage(message) {
         if (!this.sessionId) {
-            throw core_1.SfError.create({ name: 'noSessionId', message: 'Session not started' });
+            throw core_1.SfError.create({ name: 'noSessionId', message: 'Agent not started, please call .start() first' });
+        }
+        const url = `${this.apiBase}/sessions/${this.sessionId}/messages`;
+        const body = {
+            message: {
+                sequenceId: Date.now(),
+                type: 'Text',
+                text: message,
+            },
+            variables: [],
+        };
+        try {
+            const start = Date.now();
+            // Handle Apex debugging setup if needed
+            if (this.apexDebugging && this.canApexDebug()) {
+                await this.handleApexDebuggingSetup();
+            }
+            const agentId = this.getAgentIdForStorage();
+            // Ensure session directory exists
+            if (!this.historyDir) {
+                this.historyDir = await (0, utils_1.getHistoryDir)(agentId, this.sessionId);
+            }
+            void (0, utils_1.appendTranscriptToHistory)({
+                timestamp: new Date().toISOString(),
+                agentId,
+                sessionId: this.sessionId,
+                role: 'user',
+                text: message,
+            }, this.historyDir);
+            let response;
+            try {
+                response = await this.connection.request({
+                    method: 'POST',
+                    url,
+                    body: JSON.stringify(body),
+                    headers: {
+                        'x-client-name': 'afdx',
+                    },
+                });
+            }
+            catch (error) {
+                const errorName = error?.name ?? '';
+                if (errorName.includes('404')) {
+                    throw core_1.SfError.create({
+                        name: 'AgentApiNotFound',
+                        message: `Preview Send API returned 404. SF_TEST_API=${kit_1.env.getBoolean('SF_TEST_API') ? 'true' : 'false'} If targeting a test.api environment, set SF_TEST_API=true, otherwise it's false.`,
+                        cause: error,
+                    });
+                }
+                throw core_1.SfError.wrap(error);
+            }
+            const planId = response.messages.at(0).planId;
+            this.planIds.add(planId);
+            await (0, utils_1.appendTranscriptToHistory)({
+                timestamp: new Date().toISOString(),
+                agentId,
+                sessionId: this.sessionId,
+                role: 'agent',
+                text: response.messages.at(0)?.message,
+                raw: response.messages,
+            }, this.historyDir);
+            // Fetch and write trace immediately if available
+            if (planId) {
+                try {
+                    const trace = await this.getTrace(planId);
+                    await (0, utils_1.writeTraceToHistory)(planId, trace, this.historyDir);
+                }
+                catch (error) {
+                    throw core_1.SfError.wrap(error);
+                }
+            }
+            if (this.apexDebugging && this.canApexDebug()) {
+                const apexLog = await (0, apexUtils_1.getDebugLog)(this.connection, start, Date.now());
+                if (apexLog) {
+                    response.apexDebugLog = apexLog;
+                }
+            }
+            return response;
+        }
+        catch (err) {
+            throw core_1.SfError.wrap(err);
         }
-        return `${this.apiBase}/sessions/${this.sessionId}/messages`;
     }
     async setAgentStatus(desiredState) {
         const botMetadata = await this.getBotMetadata();
@@ -179,35 +255,41 @@ class ProductionAgent extends agentInteractionBase_1.AgentInteractionBase {
             bypassUser: true,
         };
         try {
-            const response = await this.connection.request({
-                method: 'POST',
-                url,
-                body: JSON.stringify(body),
-                headers: {
-                    'x-client-name': 'afdx',
-                },
-            });
+            let response;
+            try {
+                response = await this.connection.request({
+                    method: 'POST',
+                    url,
+                    body: JSON.stringify(body),
+                    headers: {
+                        'x-client-name': 'afdx',
+                    },
+                });
+            }
+            catch (error) {
+                const errorName = error?.name ?? '';
+                if (errorName.includes('404')) {
+                    throw core_1.SfError.create({
+                        name: 'AgentApiNotFound',
+                        message: `Preview Start API returned 404. SF_TEST_API=${kit_1.env.getBoolean('SF_TEST_API') ? 'true' : 'false'} If targeting a test.api environment, set SF_TEST_API=true, otherwise it's false.`,
+                        cause: error,
+                    });
+                }
+                throw core_1.SfError.wrap(error);
+            }
             this.sessionId = response.sessionId;
-            // Initialize session directory and write initial data
-            // Session directory structure:
-            // .sfdx/agents/<agentId>/sessions/<sessionId>/
-            // ├── transcript.jsonl    # All transcript entries (one per line)
-            // ├── traces/             # Individual trace files
-            // │   ├── <planId1>.json
-            // │   └── <planId2>.json
-            // └── metadata.json       # Session metadata (start time, end time, planIds, etc.)
             const agentId = this.id;
-            this.sessionDir = await (0, utils_1.getSessionDir)(agentId, response.sessionId);
-            await (0, utils_1.appendTranscriptEntryToSession)({
+            this.historyDir = await (0, utils_1.getHistoryDir)(agentId, response.sessionId);
+            await (0, utils_1.appendTranscriptToHistory)({
                 timestamp: new Date().toISOString(),
                 agentId,
                 sessionId: response.sessionId,
                 role: 'agent',
                 text: response.messages.map((m) => m.message).join('\n'),
                 raw: response.messages,
-            }, this.sessionDir);
+            }, this.historyDir);
             // Write initial metadata
-            await (0, utils_1.writeMetadataToSession)(this.sessionDir, {
+            await (0, utils_1.writeMetaFileToHistory)(this.historyDir, {
                 sessionId: response.sessionId,
                 agentId,
                 startTime: new Date().toISOString(),
@@ -237,29 +319,43 @@ class ProductionAgent extends agentInteractionBase_1.AgentInteractionBase {
         const url = `${this.apiBase}/v1.1/sessions/${this.sessionId}`;
         try {
             // https://developer.salesforce.com/docs/einstein/genai/guide/agent-api-examples.html#end-session
-            const response = await this.connection.request({
-                method: 'DELETE',
-                url,
-                headers: {
-                    'x-session-end-reason': reason,
-                },
-            });
+            let response;
+            try {
+                response = await this.connection.request({
+                    method: 'DELETE',
+                    url,
+                    headers: {
+                        'x-session-end-reason': reason,
+                    },
+                });
+            }
+            catch (error) {
+                const errorName = error?.name ?? '';
+                if (errorName.includes('404')) {
+                    throw core_1.SfError.create({
+                        name: 'AgentApiNotFound',
+                        message: `Preview End API returned 404. SF_TEST_API=${kit_1.env.getBoolean('SF_TEST_API') ? 'true' : 'false'} If targeting a test.api environment, set SF_TEST_API=true, otherwise it's false.`,
+                        cause: error,
+                    });
+                }
+                throw core_1.SfError.wrap(error);
+            }
             // Write end entry immediately
-            if (this.sessionDir) {
-                await (0, utils_1.appendTranscriptEntryToSession)({
+            if (this.historyDir) {
+                await (0, utils_1.appendTranscriptToHistory)({
                     timestamp: new Date().toISOString(),
                     agentId: this.id,
                     sessionId: this.sessionId,
                     role: 'agent',
                     reason,
                     raw: response.messages,
-                }, this.sessionDir);
+                }, this.historyDir);
                 // Update metadata with end time
-                await (0, utils_1.updateMetadataEndTime)(this.sessionDir, new Date().toISOString(), this.planIds);
+                await (0, utils_1.updateMetadataEndTime)(this.historyDir, new Date().toISOString(), this.planIds);
             }
             // Clear session data for next session
             this.sessionId = undefined;
-            this.sessionDir = undefined;
+            this.historyDir = undefined;
             this.planIds = new Set();
             return response;
         }

package/lib/agents/productionAgent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"productionAgent.js","sourceRoot":"","sources":["../../src/agents/productionAgent.ts"],"names":[],"mappings":";;;AAAA;;;;;;;;;;;;;;GAcG;AACH,6CAAyC;AACzC,2CAAqD;AACrD,yCAAsC;AActC,8CAA0C;AAC1C,oCASkB;AAClB,4CAA2E;AAC3E,2CAAwC;;AAExC,MAAM,QAAQ,OAAG,eAAQ,CAAc,oBAAoB,EAAE,QAAQ,kuBAAC,CAAC;AAEvE,MAAa,eAAgB,SAAQ,qBAAS;IAOjB;IANpB,OAAO,CAAwB;IAC9B,WAAW,CAA0B;IACrC,EAAE,CAAqB;IACvB,OAAO,CAAqB;IAC5B,OAAO,GAAG,WAAW,IAAA,mBAAW,GAAE,yCAAyC,CAAC;IAEpF,YAA2B,OAA+B;QACxD,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QADD,YAAO,GAAP,OAAO,CAAwB;QAExD,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;YACzB,MAAM,QAAQ,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,CAAC,OAAO,GAAG;YACb,KAAK,EAAE,CAAC,aAAuB,EAAsC,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC;YACxG,IAAI,EAAE,CAAC,OAAe,EAAqC,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC;YACvF,YAAY,EAAE,GAA+B,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE;YAC3E,GAAG,EAAE,CAAC,MAAiB,EAAoC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;YACrF,WAAW,EAAE,CAAC,SAAiB,EAAmB,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC;YAClF,gBAAgB,EAAE,CAAC,aAAsB,EAAQ,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC;SAChE,CAAC;QAE3B,IAAI,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3F,IAAI,CAAC,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC;QAChC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC;QACrC,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,cAAc;QACzB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,WAAW,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,oBAAoB,IAAI,CAAC,OAAQ,GAAG,CAAC;YACzF,IAAI,CAAC,WAAW,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,iBAAiB,CACxD,+FAA+F,WAAW,UAAU,CACrH,CAAC;YACF,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,aAAa,CAAC;YAC9C,wCAAwC;YACxC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC;QAC3C,CAAC;QACD,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED;;;;OAIG;IACI,KAAK,CAAC,2BAA2B;QACtC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,IAAI,CAAC,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QACjD,CAAC;QACD,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,OAAO,CAAC;QACzD,OAAO,WAAW,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAC7C,CAAC;IAED,qHAAqH;IAC9G,KAAK,CAAC,QAAQ,CAAC,MAAc;QAClC,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,kBAAkB,CAAC,SAAkB;QAK1C,oGAAoG;QACpG,MAAM,eAAe,GAAG,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACpD,OAAO,IAAA,qBAAa,EAAC,IAAI,CAAC,oBAAoB,EAAE,EAAE,eAAe,CAAC,CAAC;IACrE,CAAC;IAED;;;;OAIG;IACI,KAAK,CAAC,KAAK;QAChB,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;YACb,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC9B,CAAC;QACD,OAAO,IAAI,CAAC,EAAG,CAAC,CAAC,oDAAoD;IACvE,CAAC;IACD;;;;OAIG;IACI,KAAK,CAAC,QAAQ;QACnB,OAAO,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;IACvC,CAAC;IAED;;;;OAIG;IACI,KAAK,CAAC,UAAU;QACrB,OAAO,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;IACzC,CAAC;IAEM,oBAAoB;QACzB,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;YACb,MAAM,cAAO,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,mDAAmD,EAAE,CAAC,CAAC;QACvG,CAAC;QACD,OAAO,IAAI,CAAC,EAAE,CAAC;IACjB,CAAC;IAED,kDAAkD;IACxC,YAAY;QACpB,OAAO,IAAI,CAAC;IACd,CAAC;IAES,KAAK,CAAC,wBAAwB;QACtC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAChD,IAAI,WAAW,CAAC,SAAS,EAAE,CAAC;YAC1B,MAAM,SAAS,GAAG,MAAM,IAAA,yBAAa,EAAC,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,SAAS,CAAC,CAAC;YAC9E,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,IAAA,2BAAe,EAAC,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,SAAS,CAAC,CAAC;YAChE,CAAC;QACH,CAAC;IACH,CAAC;IAES,KAAK,CAAC,WAAW,CAAC,OAAe;QACzC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,cAAO,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,+CAA+C,EAAE,CAAC,CAAC;QAC1G,CAAC;QAED,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,aAAa,IAAI,CAAC,SAAS,WAAW,CAAC;QAElE,MAAM,IAAI,GAAG;YACX,OAAO,EAAE;gBACP,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE;gBACtB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,OAAO;aACd;YACD,SAAS,EAAE,EAAE;SACd,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAEzB,wCAAwC;YACxC,IAAI,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC;gBAC9C,MAAM,IAAI,CAAC,wBAAwB,EAAE,CAAC;YACxC,CAAC;YAED,MAAM,OAAO,GAAG,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAE5C,kCAAkC;YAClC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;gBACrB,IAAI,CAAC,UAAU,GAAG,MAAM,IAAA,qBAAa,EAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;YACjE,CAAC;YAED,KAAK,IAAA,iCAAyB,EAC5B;gBACE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,OAAO;gBACP,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,OAAO;aACd,EACD,IAAI,CAAC,UAAU,CAChB,CAAC;YAEF,IAAI,QAAkC,CAAC;YACvC,IAAI,CAAC;gBACH,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAA2B;oBACjE,MAAM,EAAE,MAAM;oBACd,GAAG;oBACH,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;oBAC1B,OAAO,EAAE;wBACP,eAAe,EAAE,MAAM;qBACxB;iBACF,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,SAAS,GAAI,KAA2B,EAAE,IAAI,IAAI,EAAE,CAAC;gBAC3D,IAAI,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;oBAC9B,MAAM,cAAO,CAAC,MAAM,CAAC;wBACnB,IAAI,EAAE,kBAAkB;wBACxB,OAAO,EAAE,8CACP,SAAG,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAC3C,mFAAmF;wBACnF,KAAK,EAAE,KAAK;qBACb,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM,cAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC5B,CAAC;YAED,MAAM,MAAM,GAAG,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAE,CAAC,MAAM,CAAC;YAC/C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAEzB,MAAM,IAAA,iCAAyB,EAC7B;gBACE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,OAAO;gBACP,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO;gBACtC,GAAG,EAAE,QAAQ,CAAC,QAAQ;aACvB,EACD,IAAI,CAAC,UAAU,CAChB,CAAC;YAEF,iDAAiD;YACjD,IAAI,MAAM,EAAE,CAAC;gBACX,IAAI,CAAC;oBACH,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;oBAC1C,MAAM,IAAA,2BAAmB,EAAC,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;gBAC5D,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,cAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBAC5B,CAAC;YACH,CAAC;YAED,IAAI,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC;gBAC9C,MAAM,OAAO,GAAG,MAAM,IAAA,uBAAW,EAAC,IAAI,CAAC,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;gBACtE,IAAI,OAAO,EAAE,CAAC;oBACZ,QAAQ,CAAC,YAAY,GAAG,OAAO,CAAC;gBAClC,CAAC;YACH,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,cAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,YAAmC;QAC9D,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAChD,MAAM,kBAAkB,GAAG,MAAM,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAEpE,IAAI,WAAW,CAAC,SAAS,EAAE,CAAC;YAC1B,MAAM,QAAQ,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,CAAC;QAC5E,CAAC;QAED,IAAI,kBAAkB,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;YAC/C,OAAO;QACT,CAAC;QAED,MAAM,GAAG,GAAG,yBAAyB,kBAAkB,CAAC,EAAE,aAAa,CAAC;QACxE,MAAM,SAAS,GAAG,IAAI,sBAAS,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,OAAO,CAAwB,MAAM,EAAE,GAAG,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAC;QACvG,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACrB,IAAI,CAAC,WAAY,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC;QACjG,CAAC;aAAM,CAAC;YACN,MAAM,QAAQ,CAAC,WAAW,CAAC,sBAAsB,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,QAAQ,EAAE,IAAI,SAAS,CAAC,CAAC,CAAC;QACnG,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,aAAuB;QAChD,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;YACb,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QACrB,CAAC;QACD,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,WAAW,IAAI,CAAC,EAAG,WAAW,CAAC;QAC1D,2FAA2F;QAC3F,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;YAChC,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACrC,CAAC;QAED,MAAM,IAAI,GAAG;YACX,kBAAkB,EAAE,IAAA,wBAAU,GAAE;YAChC,cAAc,EAAE;gBACd,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,WAAW;aAC9C;YACD,qBAAqB,EAAE;gBACrB,UAAU,EAAE,CAAC,MAAM,CAAC;aACrB;YACD,UAAU,EAAE,IAAI;SACjB,CAAC;QAEF,IAAI,CAAC;YACH,IAAI,QAAmC,CAAC;YACxC,IAAI,CAAC;gBACH,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAA4B;oBAClE,MAAM,EAAE,MAAM;oBACd,GAAG;oBACH,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;oBAC1B,OAAO,EAAE;wBACP,eAAe,EAAE,MAAM;qBACxB;iBACF,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,SAAS,GAAI,KAA2B,EAAE,IAAI,IAAI,EAAE,CAAC;gBAC3D,IAAI,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;oBAC9B,MAAM,cAAO,CAAC,MAAM,CAAC;wBACnB,IAAI,EAAE,kBAAkB;wBACxB,OAAO,EAAE,+CACP,SAAG,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAC3C,mFAAmF;wBACnF,KAAK,EAAE,KAAK;qBACb,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM,cAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC5B,CAAC;YACD,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC;YAEpC,MAAM,OAAO,GAAG,IAAI,CAAC,EAAG,CAAC;YACzB,IAAI,CAAC,UAAU,GAAG,MAAM,IAAA,qBAAa,EAAC,OAAO,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;YAEnE,MAAM,IAAA,iCAAyB,EAC7B;gBACE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,OAAO;gBACP,SAAS,EAAE,QAAQ,CAAC,SAAS;gBAC7B,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;gBACxD,GAAG,EAAE,QAAQ,CAAC,QAAQ;aACvB,EACD,IAAI,CAAC,UAAU,CAChB,CAAC;YAEF,yBAAyB;YACzB,MAAM,IAAA,8BAAsB,EAAC,IAAI,CAAC,UAAU,EAAE;gBAC5C,SAAS,EAAE,QAAQ,CAAC,SAAS;gBAC7B,OAAO;gBACP,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,aAAa,EAAE,IAAI,CAAC,aAAa;gBACjC,OAAO,EAAE,EAAE;aACZ,CAAC,CAAC;YAEH,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,cAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACK,KAAK,CAAC,UAAU,CAAC,MAAiB;QACxC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,cAAO,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,4BAA4B,EAAE,CAAC,CAAC;QACvF,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;YACb,MAAM,cAAO,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,4BAA4B,EAAE,CAAC,CAAC;QAChF,CAAC;QACD,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,kBAAkB,IAAI,CAAC,SAAS,EAAE,CAAC;QAC9D,IAAI,CAAC;YACH,iGAAiG;YACjG,IAAI,QAAiC,CAAC;YACtC,IAAI,CAAC;gBACH,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAA0B;oBAChE,MAAM,EAAE,QAAQ;oBAChB,GAAG;oBACH,OAAO,EAAE;wBACP,sBAAsB,EAAE,MAAM;qBAC/B;iBACF,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,SAAS,GAAI,KAA2B,EAAE,IAAI,IAAI,EAAE,CAAC;gBAC3D,IAAI,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;oBAC9B,MAAM,cAAO,CAAC,MAAM,CAAC;wBACnB,IAAI,EAAE,kBAAkB;wBACxB,OAAO,EAAE,6CACP,SAAG,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAC3C,mFAAmF;wBACnF,KAAK,EAAE,KAAK;qBACb,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM,cAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC5B,CAAC;YAED,8BAA8B;YAC9B,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBACpB,MAAM,IAAA,iCAAyB,EAC7B;oBACE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;oBACnC,OAAO,EAAE,IAAI,CAAC,EAAE;oBAChB,SAAS,EAAE,IAAI,CAAC,SAAS;oBACzB,IAAI,EAAE,OAAO;oBACb,MAAM;oBACN,GAAG,EAAE,QAAQ,CAAC,QAAQ;iBACvB,EACD,IAAI,CAAC,UAAU,CAChB,CAAC;gBACF,gCAAgC;gBAChC,MAAM,IAAA,6BAAqB,EAAC,IAAI,CAAC,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YACvF,CAAC;YAED,sCAAsC;YACtC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;YAC3B,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;YAC5B,IAAI,CAAC,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;YAEjC,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,cAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;CACF;AAxYD,0CAwYC"}

package/lib/{scriptAgent.d.ts → agents/scriptAgent.d.ts}

@@ -1,16 +1,18 @@
 import { SfProject } from '@salesforce/core';
-import { type AgentPreviewEndResponse, type AgentPreviewSendResponse, type CompileAgentScriptResponse, ExtendedAgentJobSpec, PlannerResponse, PublishAgent, ScriptAgentOptions } from './types';
-import { AgentInteractionBase, type AgentPreviewInterface } from './agentInteractionBase';
-export declare class ScriptAgent extends AgentInteractionBase {
+import { type AgentPreviewEndResponse, AgentPreviewInterface, type AgentPreviewSendResponse, type CompileAgentScriptResponse, ExtendedAgentJobSpec, PlannerResponse, PreviewMetadata, PublishAgent, ScriptAgentOptions } from '../types';
+import { TranscriptEntry } from '../utils';
+import { AgentBase } from './agentBase';
+export declare class ScriptAgent extends AgentBase {
     private options;
     preview: AgentPreviewInterface & {
         setMockMode: (mockMode: 'Mock' | 'Live Test') => void;
     };
     private mockMode;
     private agentScriptContent;
-    private metaContent;
     private agentJson;
     private apiBase;
+    private readonly aabDirectory;
+    private readonly metaContent;
     constructor(options: ScriptAgentOptions);
     /**
      * Creates an AiAuthoringBundle directory, .script file, and -meta.xml file
@@ -46,16 +48,20 @@ export declare class ScriptAgent extends AgentInteractionBase {
      * @returns {Promise<PublishAgentJsonResponse>} The publish response
      */
     publish(): Promise<PublishAgent>;
+    getHistoryFromDisc(sessionId?: string): Promise<{
+        metadata: PreviewMetadata | null;
+        transcript: TranscriptEntry[];
+        traces: PlannerResponse[];
+    }>;
     /**
      * Ending is not required
-     * this will save all of the transcripts to disc
+     * this will save all the transcripts to disc
      *
      * @returns `AgentPreviewEndResponse`
      */
     endSession(): Promise<AgentPreviewEndResponse>;
-    protected getAgentIdForStorage(): string;
+    getAgentIdForStorage(): string;
     protected canApexDebug(): boolean;
-    protected getSendMessageUrl(): string;
     protected handleApexDebuggingSetup(): Promise<void>;
     protected sendMessage(message: string): Promise<AgentPreviewSendResponse>;
     private setMockMode;