@salesforce/agents 0.1.5 → 0.2.0

package/lib/agent.d.ts

@@ -1,10 +1,9 @@
 import { Connection, SfProject } from '@salesforce/core';
 import { type SfAgent, type AgentCreateConfig, type AgentCreateResponse, type AgentJobSpec, type AgentJobSpecCreateConfig } from './types.js';
 export declare class Agent implements SfAgent {
-    private connection;
     private project;
     private logger;
-    private mockDir?;
+    private maybeMock;
     constructor(connection: Connection, project: SfProject);
     create(config: AgentCreateConfig): Promise<AgentCreateResponse>;
     /**

package/lib/agent.js

@@ -7,22 +7,18 @@
  */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Agent = void 0;
-const node_path_1 = require("node:path");
-const node_fs_1 = require("node:fs");
 const node_util_1 = require("node:util");
 const core_1 = require("@salesforce/core");
 const kit_1 = require("@salesforce/kit");
-const mockDir_1 = require("./mockDir");
+const maybe_mock_1 = require("./maybe-mock");
 class Agent {
-    connection;
     project;
     logger;
-    mockDir;
+    maybeMock;
     constructor(connection, project) {
-        this.connection = connection;
         this.project = project;
         this.logger = core_1.Logger.childFromRoot(this.constructor.name);
-        this.mockDir = (0, mockDir_1.getMockDir)();
+        this.maybeMock = new maybe_mock_1.MaybeMock(connection);
     }
     async create(config) {
         this.logger.debug(`Creating Agent using config: ${(0, node_util_1.inspect)(config)} in project: ${this.project.getPath()}`);
@@ -40,51 +36,15 @@ class Agent {
     async createSpec(config) {
         this.verifyAgentSpecConfig(config);
         let agentSpec;
-        if (this.mockDir) {
-            const specFileName = `${config.name}.json`;
-            const specFilePath = (0, node_path_1.join)(this.mockDir, `${specFileName}`);
-            try {
-                this.logger.debug(`Using mock directory: ${this.mockDir} for agent job spec creation`);
-                (0, node_fs_1.statSync)(specFilePath);
-            }
-            catch (err) {
-                throw core_1.SfError.create({
-                    name: 'MissingMockFile',
-                    message: `SF_MOCK_DIR [${this.mockDir}] must contain a spec file with name ${specFileName}`,
-                    cause: err,
-                });
-            }
-            try {
-                this.logger.debug(`Returning mock agent spec file: ${specFilePath}`);
-                agentSpec = JSON.parse((0, node_fs_1.readFileSync)(specFilePath, 'utf8'));
-            }
-            catch (err) {
-                throw core_1.SfError.create({
-                    name: 'InvalidMockFile',
-                    message: `SF_MOCK_DIR [${this.mockDir}] must contain a valid spec file with name ${specFileName}`,
-                    cause: err,
-                    actions: [
-                        'Check that the file is readable',
-                        'Check that the file is a valid JSON array of jobTitle and jobDescription objects',
-                    ],
-                });
-            }
+        const response = await this.maybeMock.request('GET', this.buildAgentJobSpecUrl(config));
+        if (response.isSuccess && response.jobSpecs) {
+            agentSpec = response.jobSpecs;
         }
         else {
-            // TODO: We'll probably want to wrap this for better error handling but let's see
-            //       what it looks like first.
-            const response = await this.connection.requestGet(this.buildAgentJobSpecUrl(config), {
-                retry: { maxRetries: 3 },
+            throw core_1.SfError.create({
+                name: 'AgentJobSpecCreateError',
+                message: response.errorMessage ?? 'unknown',
             });
-            if (response.isSuccess) {
-                agentSpec = response?.jobSpecs;
-            }
-            else {
-                throw core_1.SfError.create({
-                    name: 'AgentJobSpecCreateError',
-                    message: response.errorMessage ?? 'unknown',
-                });
-            }
         }
         return agentSpec;
     }

package/lib/agent.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent.js","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAEH,yCAAiC;AACjC,qCAAiD;AACjD,yCAAoC;AACpC,2CAA0E;AAC1E,yCAAkD;AAClD,uCAAuC;AAUvC,MAAa,KAAK;IAIW;IAAgC;IAHnD,MAAM,CAAS;IACf,OAAO,CAAU;IAEzB,YAA2B,UAAsB,EAAU,OAAkB;QAAlD,eAAU,GAAV,UAAU,CAAY;QAAU,YAAO,GAAP,OAAO,CAAW;QAC3E,IAAI,CAAC,MAAM,GAAG,aAAM,CAAC,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAC1D,IAAI,CAAC,OAAO,GAAG,IAAA,oBAAU,GAAE,CAAC;IAC9B,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,MAAyB;QAC3C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gCAAgC,IAAA,mBAAO,EAAC,MAAM,CAAC,gBAAgB,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QAC3G,0DAA0D;QAE1D,+EAA+E;QAC/E,MAAM,IAAA,WAAK,EAAC,cAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;QAEjC,0CAA0C;QAE1C,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;IAC7B,CAAC;IAED;;;;OAIG;IACI,KAAK,CAAC,UAAU,CAAC,MAAgC;QACtD,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;QAEnC,IAAI,SAAuB,CAAC;QAE5B,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,YAAY,GAAG,GAAG,MAAM,CAAC,IAAI,OAAO,CAAC;YAC3C,MAAM,YAAY,GAAG,IAAA,gBAAI,EAAC,IAAI,CAAC,OAAO,EAAE,GAAG,YAAY,EAAE,CAAC,CAAC;YAC3D,IAAI,CAAC;gBACH,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,yBAAyB,IAAI,CAAC,OAAO,8BAA8B,CAAC,CAAC;gBACvF,IAAA,kBAAQ,EAAC,YAAY,CAAC,CAAC;YACzB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,cAAO,CAAC,MAAM,CAAC;oBACnB,IAAI,EAAE,iBAAiB;oBACvB,OAAO,EAAE,gBAAgB,IAAI,CAAC,OAAO,wCAAwC,YAAY,EAAE;oBAC3F,KAAK,EAAE,GAAG;iBACX,CAAC,CAAC;YACL,CAAC;YACD,IAAI,CAAC;gBACH,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,mCAAmC,YAAY,EAAE,CAAC,CAAC;gBACrE,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,IAAA,sBAAY,EAAC,YAAY,EAAE,MAAM,CAAC,CAAiB,CAAC;YAC7E,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,cAAO,CAAC,MAAM,CAAC;oBACnB,IAAI,EAAE,iBAAiB;oBACvB,OAAO,EAAE,gBAAgB,IAAI,CAAC,OAAO,8CAA8C,YAAY,EAAE;oBACjG,KAAK,EAAE,GAAG;oBACV,OAAO,EAAE;wBACP,iCAAiC;wBACjC,kFAAkF;qBACnF;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;aAAM,CAAC;YACN,iFAAiF;YACjF,kCAAkC;YAClC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAA6B,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,EAAE;gBAC/G,KAAK,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE;aACzB,CAAC,CAAC;YACH,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;gBACvB,SAAS,GAAG,QAAQ,EAAE,QAAwB,CAAC;YACjD,CAAC;iBAAM,CAAC;gBACN,MAAM,cAAO,CAAC,MAAM,CAAC;oBACnB,IAAI,EAAE,yBAAyB;oBAC/B,OAAO,EAAE,QAAQ,CAAC,YAAY,IAAI,SAAS;iBAC5C,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,kDAAkD;IAC1C,qBAAqB,CAAC,MAAgC;QAC5D,6EAA6E;QAC7E,IAAI,MAAM;YAAE,OAAO;IACrB,CAAC;IAED,kDAAkD;IAC1C,oBAAoB,CAAC,MAAgC;QAC3D,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,kBAAkB,EAAE,cAAc,EAAE,GAAG,MAAM,CAAC;QAC/E,MAAM,OAAO,GAAG,cAAc,CAAC,CAAC,CAAC,mBAAmB,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC1E,OAAO,qCAAqC,IAAI,SAAS,IAAI,gBAAgB,WAAW,uBAAuB,kBAAkB,GAAG,OAAO,EAAE,CAAC;IAChJ,CAAC;CACF;AAzFD,sBAyFC"}
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAEH,yCAAoC;AACpC,2CAA0E;AAC1E,yCAAkD;AAClD,6CAAyC;AAUzC,MAAa,KAAK;IAImC;IAH3C,MAAM,CAAS;IACf,SAAS,CAAY;IAE7B,YAAmB,UAAsB,EAAU,OAAkB;QAAlB,YAAO,GAAP,OAAO,CAAW;QACnE,IAAI,CAAC,MAAM,GAAG,aAAM,CAAC,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAC1D,IAAI,CAAC,SAAS,GAAG,IAAI,sBAAS,CAAC,UAAU,CAAC,CAAC;IAC7C,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,MAAyB;QAC3C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gCAAgC,IAAA,mBAAO,EAAC,MAAM,CAAC,gBAAgB,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QAC3G,0DAA0D;QAE1D,+EAA+E;QAC/E,MAAM,IAAA,WAAK,EAAC,cAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;QAEjC,0CAA0C;QAE1C,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;IAC7B,CAAC;IAED;;;;OAIG;IACI,KAAK,CAAC,UAAU,CAAC,MAAgC;QACtD,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;QAEnC,IAAI,SAAuB,CAAC;QAC5B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAA6B,KAAK,EAAE,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC,CAAC;QACpH,IAAI,QAAQ,CAAC,SAAS,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;YAC5C,SAAS,GAAG,QAAQ,CAAC,QAAQ,CAAC;QAChC,CAAC;aAAM,CAAC;YACN,MAAM,cAAO,CAAC,MAAM,CAAC;gBACnB,IAAI,EAAE,yBAAyB;gBAC/B,OAAO,EAAE,QAAQ,CAAC,YAAY,IAAI,SAAS;aAC5C,CAAC,CAAC;QACL,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,kDAAkD;IAC1C,qBAAqB,CAAC,MAAgC;QAC5D,6EAA6E;QAC7E,IAAI,MAAM;YAAE,OAAO;IACrB,CAAC;IAED,kDAAkD;IAC1C,oBAAoB,CAAC,MAAgC;QAC3D,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,kBAAkB,EAAE,cAAc,EAAE,GAAG,MAAM,CAAC;QAC/E,MAAM,OAAO,GAAG,cAAc,CAAC,CAAC,CAAC,mBAAmB,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC1E,OAAO,qCAAqC,IAAI,SAAS,IAAI,gBAAgB,WAAW,uBAAuB,kBAAkB,GAAG,OAAO,EAAE,CAAC;IAChJ,CAAC;CACF;AAvDD,sBAuDC"}

package/lib/agentTester.d.ts

@@ -0,0 +1,77 @@
+import { Connection } from '@salesforce/core';
+import { Duration } from '@salesforce/kit';
+type Format = 'human' | 'json';
+type TestStatus = 'NEW' | 'IN_PROGRESS' | 'COMPLETED' | 'ERROR';
+type AgentTestStatusResponse = {
+    status: TestStatus;
+    startTime: string;
+    endTime?: string;
+    errorMessage?: string;
+};
+type TestCaseResult = {
+    status: TestStatus;
+    number: string;
+    startTime: string;
+    endTime?: string;
+    generatedData: {
+        type: 'AGENT';
+        actionsSequence: string[];
+        outcome: 'Success' | 'Failure';
+        topic: string;
+        inputTokensCount: string;
+        outputTokensCount: string;
+    };
+    expectationResults: Array<{
+        name: string;
+        actualValue: string;
+        expectedValue: string;
+        score: number;
+        result: 'Passed' | 'Failed';
+        metricLabel: 'Accuracy' | 'Precision';
+        metricExplainability: string;
+        status: TestStatus;
+        startTime: string;
+        endTime?: string;
+        errorCode?: string;
+        errorMessage?: string;
+    }>;
+};
+type AgentTestDetailsResponse = {
+    status: TestStatus;
+    startTime: string;
+    endTime?: string;
+    errorMessage?: string;
+    testCases: TestCaseResult[];
+};
+export declare class AgentTester {
+    private maybeMock;
+    constructor(connection: Connection);
+    /**
+     * Starts an AI evaluation run based on the provided name or ID.
+     *
+     * @param nameOrId - The name or ID of the AI evaluation definition.
+     * @param type - Specifies whether the provided identifier is a 'name' or 'id'. Defaults to 'name'. If 'name' is provided, nameOrId is treated as the name of the AiEvaluationDefinition. If 'id' is provided, nameOrId is treated as the unique ID of the AiEvaluationDefinition.
+     * @returns A promise that resolves to an object containing the ID of the started AI evaluation run.
+     */
+    start(nameOrId: string, type?: 'name' | 'id'): Promise<{
+        aiEvaluationId: string;
+    }>;
+    status(jobId: string): Promise<AgentTestStatusResponse>;
+    poll(jobId: string, { format, timeout, }?: {
+        format?: Format;
+        timeout?: Duration;
+    }): Promise<{
+        response: AgentTestDetailsResponse;
+        formatted: string;
+    }>;
+    details(jobId: string, format?: Format): Promise<{
+        response: AgentTestDetailsResponse;
+        formatted: string;
+    }>;
+    cancel(jobId: string): Promise<{
+        success: boolean;
+    }>;
+}
+export declare function humanFormat(name: string, details: AgentTestDetailsResponse): Promise<string>;
+export declare function jsonFormat(details: AgentTestDetailsResponse): Promise<string>;
+export {};

package/lib/agentTester.js

@@ -0,0 +1,114 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgentTester = void 0;
+exports.humanFormat = humanFormat;
+exports.jsonFormat = jsonFormat;
+/*
+ * Copyright (c) 2024, salesforce.com, inc.
+ * All rights reserved.
+ * Licensed under the BSD 3-Clause license.
+ * For full license text, see LICENSE.txt file in the repo root or https://opensource.org/licenses/BSD-3-Clause
+ */
+const core_1 = require("@salesforce/core");
+const kit_1 = require("@salesforce/kit");
+const maybe_mock_1 = require("./maybe-mock");
+class AgentTester {
+    maybeMock;
+    constructor(connection) {
+        this.maybeMock = new maybe_mock_1.MaybeMock(connection);
+    }
+    /**
+     * Starts an AI evaluation run based on the provided name or ID.
+     *
+     * @param nameOrId - The name or ID of the AI evaluation definition.
+     * @param type - Specifies whether the provided identifier is a 'name' or 'id'. Defaults to 'name'. If 'name' is provided, nameOrId is treated as the name of the AiEvaluationDefinition. If 'id' is provided, nameOrId is treated as the unique ID of the AiEvaluationDefinition.
+     * @returns A promise that resolves to an object containing the ID of the started AI evaluation run.
+     */
+    async start(nameOrId, type = 'name') {
+        const url = '/einstein/ai-evaluations/runs';
+        return this.maybeMock.request('POST', url, {
+            [type === 'name' ? 'aiEvaluationDefinitionName' : 'aiEvaluationDefinitionVersionId']: nameOrId,
+        });
+    }
+    async status(jobId) {
+        const url = `/einstein/ai-evaluations/runs/${jobId}`;
+        return this.maybeMock.request('GET', url);
+    }
+    async poll(jobId, { format = 'human', timeout = kit_1.Duration.minutes(5), } = {
+        format: 'human',
+        timeout: kit_1.Duration.minutes(5),
+    }) {
+        const lifecycle = core_1.Lifecycle.getInstance();
+        const client = await core_1.PollingClient.create({
+            poll: async () => {
+                // NOTE: we don't actually need to call the status API here since all the same information is present on the
+                // details API. We could just call the details API and check the status there.
+                const [detailsResponse, statusResponse] = await Promise.all([this.details(jobId, format), this.status(jobId)]);
+                const totalTestCases = detailsResponse.response.testCases.length;
+                const failingTestCases = detailsResponse.response.testCases.filter((tc) => tc.status === 'ERROR').length;
+                const passingTestCases = detailsResponse.response.testCases.filter((tc) => tc.status === 'COMPLETED' && tc.expectationResults.every((r) => r.result === 'Passed')).length;
+                if (statusResponse.status.toLowerCase() === 'completed') {
+                    await lifecycle.emit('AGENT_TEST_POLLING_EVENT', {
+                        jobId,
+                        status: statusResponse.status,
+                        totalTestCases,
+                        failingTestCases,
+                        passingTestCases,
+                    });
+                    return { payload: await this.details(jobId, format), completed: true };
+                }
+                await lifecycle.emit('AGENT_TEST_POLLING_EVENT', {
+                    jobId,
+                    status: statusResponse.status,
+                    totalTestCases,
+                    failingTestCases,
+                    passingTestCases,
+                });
+                return { completed: false };
+            },
+            frequency: kit_1.Duration.seconds(1),
+            timeout,
+        });
+        const result = await client.subscribe();
+        return result;
+    }
+    async details(jobId, format = 'human') {
+        const url = `/einstein/ai-evaluations/runs/${jobId}/details`;
+        const response = await this.maybeMock.request('GET', url);
+        return {
+            response,
+            formatted: format === 'human' ? await humanFormat(jobId, response) : await jsonFormat(response),
+        };
+    }
+    async cancel(jobId) {
+        const url = `/einstein/ai-evaluations/runs/${jobId}/cancel`;
+        return this.maybeMock.request('POST', url);
+    }
+}
+exports.AgentTester = AgentTester;
+async function humanFormat(name, details) {
+    const { Ux } = await import('@salesforce/sf-plugins-core');
+    const ux = new Ux();
+    const tables = [];
+    for (const testCase of details.testCases) {
+        const table = ux.makeTable({
+            title: `Test Case #${testCase.number}`,
+            data: testCase.expectationResults.map((r) => ({
+                name: r.name,
+                outcome: r.result === 'Passed' ? 'Pass' : 'Fail',
+                actualValue: r.actualValue,
+                expectedValue: r.expectedValue,
+                score: r.score,
+                'metric label': r.metricLabel,
+                message: r.errorMessage ?? '',
+                'runtime (MS)': r.endTime ? new Date(r.endTime).getTime() - new Date(r.startTime).getTime() : 0,
+            })),
+        });
+        tables.push(table);
+    }
+    return tables.join('\n');
+}
+async function jsonFormat(details) {
+    return Promise.resolve(JSON.stringify(details, null, 2));
+}
+//# sourceMappingURL=agentTester.js.map

package/lib/agentTester.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agentTester.js","sourceRoot":"","sources":["../src/agentTester.ts"],"names":[],"mappings":";;;AAmKA,kCAsBC;AAED,gCAEC;AA7LD;;;;;GAKG;AACH,2CAAsF;AACtF,yCAA2C;AAC3C,6CAAyC;AAuDzC,MAAa,WAAW;IACd,SAAS,CAAY;IAC7B,YAAmB,UAAsB;QACvC,IAAI,CAAC,SAAS,GAAG,IAAI,sBAAS,CAAC,UAAU,CAAC,CAAC;IAC7C,CAAC;IAED;;;;;;OAMG;IACI,KAAK,CAAC,KAAK,CAAC,QAAgB,EAAE,OAAsB,MAAM;QAC/D,MAAM,GAAG,GAAG,+BAA+B,CAAC;QAE5C,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAyB,MAAM,EAAE,GAAG,EAAE;YACjE,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,4BAA4B,CAAC,CAAC,CAAC,iCAAiC,CAAC,EAAE,QAAQ;SAC/F,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,KAAa;QAC/B,MAAM,GAAG,GAAG,iCAAiC,KAAK,EAAE,CAAC;QAErD,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAA0B,KAAK,EAAE,GAAG,CAAC,CAAC;IACrE,CAAC;IAEM,KAAK,CAAC,IAAI,CACf,KAAa,EACb,EACE,MAAM,GAAG,OAAO,EAChB,OAAO,GAAG,cAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,MAI3B;QACF,MAAM,EAAE,OAAO;QACf,OAAO,EAAE,cAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;KAC7B;QAED,MAAM,SAAS,GAAG,gBAAS,CAAC,WAAW,EAAE,CAAC;QAC1C,MAAM,MAAM,GAAG,MAAM,oBAAa,CAAC,MAAM,CAAC;YACxC,IAAI,EAAE,KAAK,IAA2B,EAAE;gBACtC,4GAA4G;gBAC5G,8EAA8E;gBAC9E,MAAM,CAAC,eAAe,EAAE,cAAc,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBAC/G,MAAM,cAAc,GAAG,eAAe,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,CAAC;gBACjE,MAAM,gBAAgB,GAAG,eAAe,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,MAAM,KAAK,OAAO,CAAC,CAAC,MAAM,CAAC;gBACzG,MAAM,gBAAgB,GAAG,eAAe,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,CAChE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,MAAM,KAAK,WAAW,IAAI,EAAE,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,QAAQ,CAAC,CAC/F,CAAC,MAAM,CAAC;gBAET,IAAI,cAAc,CAAC,MAAM,CAAC,WAAW,EAAE,KAAK,WAAW,EAAE,CAAC;oBACxD,MAAM,SAAS,CAAC,IAAI,CAAC,0BAA0B,EAAE;wBAC/C,KAAK;wBACL,MAAM,EAAE,cAAc,CAAC,MAAM;wBAC7B,cAAc;wBACd,gBAAgB;wBAChB,gBAAgB;qBACjB,CAAC,CAAC;oBACH,OAAO,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;gBACzE,CAAC;gBAED,MAAM,SAAS,CAAC,IAAI,CAAC,0BAA0B,EAAE;oBAC/C,KAAK;oBACL,MAAM,EAAE,cAAc,CAAC,MAAM;oBAC7B,cAAc;oBACd,gBAAgB;oBAChB,gBAAgB;iBACjB,CAAC,CAAC;gBACH,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;YAC9B,CAAC;YACD,SAAS,EAAE,cAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;YAC9B,OAAO;SACR,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,EAA6D,CAAC;QACnG,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,OAAO,CAClB,KAAa,EACb,SAAiB,OAAO;QAExB,MAAM,GAAG,GAAG,iCAAiC,KAAK,UAAU,CAAC;QAE7D,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAA2B,KAAK,EAAE,GAAG,CAAC,CAAC;QACpF,OAAO;YACL,QAAQ;YACR,SAAS,EAAE,MAAM,KAAK,OAAO,CAAC,CAAC,CAAC,MAAM,WAAW,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,MAAM,UAAU,CAAC,QAAQ,CAAC;SAChG,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,KAAa;QAC/B,MAAM,GAAG,GAAG,iCAAiC,KAAK,SAAS,CAAC;QAE5D,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAuB,MAAM,EAAE,GAAG,CAAC,CAAC;IACnE,CAAC;CACF;AAlGD,kCAkGC;AAEM,KAAK,UAAU,WAAW,CAAC,IAAY,EAAE,OAAiC;IAC/E,MAAM,EAAE,EAAE,EAAE,GAAG,MAAM,MAAM,CAAC,6BAA6B,CAAC,CAAC;IAC3D,MAAM,EAAE,GAAG,IAAI,EAAE,EAAE,CAAC;IAEpB,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,KAAK,MAAM,QAAQ,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,EAAE,CAAC,SAAS,CAAC;YACzB,KAAK,EAAE,cAAc,QAAQ,CAAC,MAAM,EAAE;YACtC,IAAI,EAAE,QAAQ,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC5C,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,OAAO,EAAE,CAAC,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM;gBAChD,WAAW,EAAE,CAAC,CAAC,WAAW;gBAC1B,aAAa,EAAE,CAAC,CAAC,aAAa;gBAC9B,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,cAAc,EAAE,CAAC,CAAC,WAAW;gBAC7B,OAAO,EAAE,CAAC,CAAC,YAAY,IAAI,EAAE;gBAC7B,cAAc,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;aAChG,CAAC,CAAC;SACJ,CAAC,CAAC;QACH,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACrB,CAAC;IACD,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC3B,CAAC;AAEM,KAAK,UAAU,UAAU,CAAC,OAAiC;IAChE,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;AAC3D,CAAC"}

package/lib/index.d.ts

@@ -1,2 +1,3 @@
 export { AgentCreateConfig, AgentCreateResponse, AgentJobSpec, AgentJobSpecCreateConfig, AgentJobSpecCreateResponse, SfAgent, } from './types';
 export { Agent } from './agent';
+export { AgentTester } from './agentTester';

package/lib/index.js

@@ -6,7 +6,9 @@
  * For full license text, see LICENSE.txt file in the repo root or https://opensource.org/licenses/BSD-3-Clause
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Agent = void 0;
+exports.AgentTester = exports.Agent = void 0;
 var agent_1 = require("./agent");
 Object.defineProperty(exports, "Agent", { enumerable: true, get: function () { return agent_1.Agent; } });
+var agentTester_1 = require("./agentTester");
+Object.defineProperty(exports, "AgentTester", { enumerable: true, get: function () { return agentTester_1.AgentTester; } });
 //# sourceMappingURL=index.js.map

package/lib/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAUH,iCAAgC;AAAvB,8FAAA,KAAK,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAUH,iCAAgC;AAAvB,8FAAA,KAAK,OAAA;AACd,6CAA4C;AAAnC,0GAAA,WAAW,OAAA"}

package/lib/maybe-mock.d.ts

@@ -0,0 +1,10 @@
+import { Connection } from '@salesforce/core';
+import nock from 'nock';
+export declare class MaybeMock {
+    private connection;
+    private mockDir;
+    private scopes;
+    private logger;
+    constructor(connection: Connection);
+    request<T extends nock.Body>(method: 'GET' | 'POST', url: string, body?: nock.RequestBodyMatcher): Promise<T>;
+}

package/lib/maybe-mock.js

@@ -0,0 +1,154 @@
+"use strict";
+/*
+ * Copyright (c) 2024, salesforce.com, inc.
+ * All rights reserved.
+ * Licensed under the BSD 3-Clause license.
+ * For full license text, see LICENSE.txt file in the repo root or https://opensource.org/licenses/BSD-3-Clause
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MaybeMock = void 0;
+const node_path_1 = require("node:path");
+const node_fs_1 = require("node:fs");
+const promises_1 = require("node:fs/promises");
+const core_1 = require("@salesforce/core");
+const kit_1 = require("@salesforce/kit");
+const nock_1 = __importDefault(require("nock"));
+/**
+ * If the `SF_MOCK_DIR` environment variable is set, resolve to an absolute path
+ * and ensure the directory exits, then return the path.
+ *
+ * NOTE: THIS SHOULD BE MOVED TO SOME OTHER LIBRARY LIKE `@salesforce/kit`.
+ *
+ * @returns the absolute path to an existing directory used for mocking behavior
+ */
+const getMockDir = () => {
+    const mockDir = kit_1.env.getString('SF_MOCK_DIR');
+    if (mockDir) {
+        let mockDirStat;
+        try {
+            mockDirStat = (0, node_fs_1.statSync)((0, node_path_1.resolve)(mockDir));
+        }
+        catch (err) {
+            throw core_1.SfError.create({
+                name: 'InvalidMockDir',
+                message: `SF_MOCK_DIR [${mockDir}] not found`,
+                cause: err,
+                actions: [
+                    "If you're trying to mock agent behavior you must create the mock directory and add expected mock files to it.",
+                ],
+            });
+        }
+        if (!mockDirStat.isDirectory()) {
+            throw core_1.SfError.create({
+                name: 'InvalidMockDir',
+                message: `SF_MOCK_DIR [${mockDir}] is not a directory`,
+                actions: [
+                    "If you're trying to mock agent behavior you must create the mock directory and add expected mock files to it.",
+                ],
+            });
+        }
+        return mockDir;
+    }
+};
+async function readJson(path) {
+    return JSON.parse(await (0, promises_1.readFile)(path, 'utf-8'));
+}
+async function readPlainText(path) {
+    return (0, promises_1.readFile)(path, 'utf-8');
+}
+async function readDirectory(path) {
+    const files = await (0, promises_1.readdir)(path);
+    const promises = files.map((file) => {
+        if (file.endsWith('.json')) {
+            return readJson((0, node_path_1.join)(path, file));
+        }
+        else {
+            return readPlainText((0, node_path_1.join)(path, file));
+        }
+    });
+    return (await Promise.all(promises)).filter((r) => !!r);
+}
+async function readResponses(mockDir, url, logger) {
+    const mockResponseName = url.replace(/\//g, '_').replace(/^_/, '').split('?')[0];
+    const mockResponsePath = (0, node_path_1.join)(mockDir, mockResponseName);
+    // Try all possibilities for the mock response file
+    const responses = (await Promise.all([
+        readJson(`${mockResponsePath}.json`)
+            .then((r) => {
+            logger.debug(`Found JSON mock file: ${mockResponsePath}.json`);
+            return r;
+        })
+            .catch(() => undefined),
+        readPlainText(mockResponsePath)
+            .then((r) => {
+            logger.debug(`Found plain text mock file: ${mockResponsePath}`);
+            return r;
+        })
+            .catch(() => undefined),
+        readDirectory(mockResponsePath)
+            .then((r) => {
+            logger.debug(`Found directory of mock files: ${mockResponsePath}`);
+            return r;
+        })
+            .catch(() => undefined),
+    ]))
+        .filter((r) => !!r)
+        .flat();
+    if (responses.length === 0) {
+        throw core_1.SfError.create({
+            name: 'MissingMockFile',
+            message: `SF_MOCK_DIR [${mockDir}] must contain a spec file with name ${mockResponsePath} or ${mockResponsePath}.json`,
+        });
+    }
+    logger.debug(`Using responses: ${responses.map((r) => JSON.stringify(r)).join(', ')}`);
+    return responses;
+}
+class MaybeMock {
+    connection;
+    mockDir = getMockDir();
+    scopes = new Map();
+    logger;
+    constructor(connection) {
+        this.connection = connection;
+        this.logger = core_1.Logger.childFromRoot(this.constructor.name);
+    }
+    async request(method, url, body = {}) {
+        if (this.mockDir) {
+            this.logger.debug(`Mocking ${method} request to ${url} using ${this.mockDir}`);
+            const responses = await readResponses(this.mockDir, url, this.logger);
+            const baseUrl = this.connection.baseUrl();
+            const scope = this.scopes.get(baseUrl) ?? (0, nock_1.default)(baseUrl);
+            this.scopes.set(baseUrl, scope);
+            switch (method) {
+                case 'GET':
+                    for (const response of responses) {
+                        scope.get(url).reply(200, response);
+                    }
+                    break;
+                case 'POST':
+                    for (const response of responses) {
+                        scope.post(url, body).reply(200, response);
+                    }
+                    break;
+            }
+        }
+        this.logger.debug(`Making ${method} request to ${url}`);
+        switch (method) {
+            case 'GET':
+                return this.connection.requestGet(url, { retry: { maxRetries: 3 } });
+            case 'POST':
+                if (!body) {
+                    throw core_1.SfError.create({
+                        name: 'InvalidBody',
+                        message: 'POST requests must include a body',
+                    });
+                }
+                return this.connection.requestPost(url, body, { retry: { maxRetries: 3 } });
+        }
+    }
+}
+exports.MaybeMock = MaybeMock;
+//# sourceMappingURL=maybe-mock.js.map

package/lib/maybe-mock.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"maybe-mock.js","sourceRoot":"","sources":["../src/maybe-mock.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;;;;AAEH,yCAA0C;AAC1C,qCAA+C;AAC/C,+CAAqD;AACrD,2CAA+D;AAC/D,yCAAsC;AACtC,gDAAwB;AAExB;;;;;;;GAOG;AACH,MAAM,UAAU,GAAG,GAAuB,EAAE;IAC1C,MAAM,OAAO,GAAG,SAAG,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;IAC7C,IAAI,OAAO,EAAE,CAAC;QACZ,IAAI,WAAkB,CAAC;QACvB,IAAI,CAAC;YACH,WAAW,GAAG,IAAA,kBAAQ,EAAC,IAAA,mBAAO,EAAC,OAAO,CAAC,CAAC,CAAC;QAC3C,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,cAAO,CAAC,MAAM,CAAC;gBACnB,IAAI,EAAE,gBAAgB;gBACtB,OAAO,EAAE,gBAAgB,OAAO,aAAa;gBAC7C,KAAK,EAAE,GAAG;gBACV,OAAO,EAAE;oBACP,+GAA+G;iBAChH;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,EAAE,CAAC;YAC/B,MAAM,cAAO,CAAC,MAAM,CAAC;gBACnB,IAAI,EAAE,gBAAgB;gBACtB,OAAO,EAAE,gBAAgB,OAAO,sBAAsB;gBACtD,OAAO,EAAE;oBACP,+GAA+G;iBAChH;aACF,CAAC,CAAC;QACL,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;AACH,CAAC,CAAC;AAEF,KAAK,UAAU,QAAQ,CAAsB,IAAY;IACvD,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,IAAA,mBAAQ,EAAC,IAAI,EAAE,OAAO,CAAC,CAAM,CAAC;AACxD,CAAC;AAED,KAAK,UAAU,aAAa,CAAC,IAAY;IACvC,OAAO,IAAA,mBAAQ,EAAC,IAAI,EAAE,OAAO,CAAC,CAAC;AACjC,CAAC;AAED,KAAK,UAAU,aAAa,CAAsB,IAAY;IAC5D,MAAM,KAAK,GAAG,MAAM,IAAA,kBAAO,EAAC,IAAI,CAAC,CAAC;IAClC,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QAClC,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3B,OAAO,QAAQ,CAAC,IAAA,gBAAI,EAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;QACpC,CAAC;aAAM,CAAC;YACN,OAAO,aAAa,CAAC,IAAA,gBAAI,EAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;QACzC,CAAC;IACH,CAAC,CAAC,CAAC;IACH,OAAO,CAAC,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAU,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAClE,CAAC;AAED,KAAK,UAAU,aAAa,CAAsB,OAAe,EAAE,GAAW,EAAE,MAAc;IAC5F,MAAM,gBAAgB,GAAG,GAAG,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACjF,MAAM,gBAAgB,GAAG,IAAA,gBAAI,EAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;IAEzD,mDAAmD;IACnD,MAAM,SAAS,GAAG,CAChB,MAAM,OAAO,CAAC,GAAG,CAAC;QAChB,QAAQ,CAAC,GAAG,gBAAgB,OAAO,CAAC;aACjC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE;YACV,MAAM,CAAC,KAAK,CAAC,yBAAyB,gBAAgB,OAAO,CAAC,CAAC;YAC/D,OAAO,CAAC,CAAC;QACX,CAAC,CAAC;aACD,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;QACzB,aAAa,CAAC,gBAAgB,CAAC;aAC5B,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE;YACV,MAAM,CAAC,KAAK,CAAC,+BAA+B,gBAAgB,EAAE,CAAC,CAAC;YAChE,OAAO,CAAC,CAAC;QACX,CAAC,CAAC;aACD,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;QACzB,aAAa,CAAC,gBAAgB,CAAC;aAC5B,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE;YACV,MAAM,CAAC,KAAK,CAAC,kCAAkC,gBAAgB,EAAE,CAAC,CAAC;YACnE,OAAO,CAAC,CAAC;QACX,CAAC,CAAC;aACD,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;KAC1B,CAAC,CACH;SACE,MAAM,CAAC,CAAC,CAAC,EAAY,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;SAC5B,IAAI,EAAE,CAAC;IACV,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC3B,MAAM,cAAO,CAAC,MAAM,CAAC;YACnB,IAAI,EAAE,iBAAiB;YACvB,OAAO,EAAE,gBAAgB,OAAO,wCAAwC,gBAAgB,OAAO,gBAAgB,OAAO;SACvH,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,oBAAoB,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEvF,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAa,SAAS;IAKO;IAJnB,OAAO,GAAG,UAAU,EAAE,CAAC;IACvB,MAAM,GAAG,IAAI,GAAG,EAAsB,CAAC;IACvC,MAAM,CAAS;IAEvB,YAA2B,UAAsB;QAAtB,eAAU,GAAV,UAAU,CAAY;QAC/C,IAAI,CAAC,MAAM,GAAG,aAAM,CAAC,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;IAC5D,CAAC;IAEM,KAAK,CAAC,OAAO,CAClB,MAAsB,EACtB,GAAW,EACX,OAAgC,EAAE;QAElC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,MAAM,eAAe,GAAG,UAAU,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;YAC/E,MAAM,SAAS,GAAG,MAAM,aAAa,CAAI,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YACzE,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,IAAA,cAAI,EAAC,OAAO,CAAC,CAAC;YACxD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YAChC,QAAQ,MAAM,EAAE,CAAC;gBACf,KAAK,KAAK;oBACR,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;wBACjC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;oBACtC,CAAC;oBACD,MAAM;gBACR,KAAK,MAAM;oBACT,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;wBACjC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;oBAC7C,CAAC;oBACD,MAAM;YACV,CAAC;QACH,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,MAAM,eAAe,GAAG,EAAE,CAAC,CAAC;QACxD,QAAQ,MAAM,EAAE,CAAC;YACf,KAAK,KAAK;gBACR,OAAO,IAAI,CAAC,UAAU,CAAC,UAAU,CAAI,GAAG,EAAE,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAC1E,KAAK,MAAM;gBACT,IAAI,CAAC,IAAI,EAAE,CAAC;oBACV,MAAM,cAAO,CAAC,MAAM,CAAC;wBACnB,IAAI,EAAE,aAAa;wBACnB,OAAO,EAAE,mCAAmC;qBAC7C,CAAC,CAAC;gBACL,CAAC;gBACD,OAAO,IAAI,CAAC,UAAU,CAAC,WAAW,CAAI,GAAG,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QACnF,CAAC;IACH,CAAC;CACF;AAhDD,8BAgDC"}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@salesforce/agents",
   "description": "Client side APIs for working with Salesforce agents",
-  "version": "0.1.5",
+  "version": "0.2.0",
   "license": "BSD-3-Clause",
   "author": "Salesforce",
   "main": "lib/index",
@@ -11,11 +11,14 @@
     "url": "https://github.com/forcedotcom/agents.git"
   },
   "dependencies": {
+    "@oclif/table": "^0.3.3",
     "@salesforce/core": "^8.8.0",
-    "@salesforce/kit": "^3.2.3"
+    "@salesforce/kit": "^3.2.3",
+    "@salesforce/sf-plugins-core": "^12.1.0",
+    "nock": "^13.5.6"
   },
   "devDependencies": {
-    "@salesforce/cli-plugins-testkit": "^5.3.20",
+    "@salesforce/cli-plugins-testkit": "^5.3.38",
     "@salesforce/dev-scripts": "^10.2.10",
     "ts-node": "^10.9.2",
     "typescript": "^5.5.4"

package/lib/mockDir.d.ts

@@ -1,9 +0,0 @@
-/**
- * If the `SF_MOCK_DIR` environment variable is set, resolve to an absolue path
- * and ensure the directory exits, then return the path.
- *
- * NOTE: THIS SHOULD BE MOVED TO SOME OTHER LIBRARY LIKE `@salesforce/kit`.
- *
- * @returns the absolute path to an existing directory used for mocking behavior
- */
-export declare const getMockDir: () => string | undefined;

package/lib/mockDir.js

@@ -1,48 +0,0 @@
-"use strict";
-/*
- * Copyright (c) 2024, salesforce.com, inc.
- * All rights reserved.
- * Licensed under the BSD 3-Clause license.
- * For full license text, see LICENSE.txt file in the repo root or https://opensource.org/licenses/BSD-3-Clause
- */
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.getMockDir = void 0;
-const node_path_1 = require("node:path");
-const node_fs_1 = require("node:fs");
-const core_1 = require("@salesforce/core");
-const kit_1 = require("@salesforce/kit");
-/**
- * If the `SF_MOCK_DIR` environment variable is set, resolve to an absolue path
- * and ensure the directory exits, then return the path.
- *
- * NOTE: THIS SHOULD BE MOVED TO SOME OTHER LIBRARY LIKE `@salesforce/kit`.
- *
- * @returns the absolute path to an existing directory used for mocking behavior
- */
-const getMockDir = () => {
-    const mockDir = kit_1.env.getString('SF_MOCK_DIR');
-    if (mockDir) {
-        let mockDirStat;
-        try {
-            mockDirStat = (0, node_fs_1.statSync)((0, node_path_1.resolve)(mockDir));
-        }
-        catch (err) {
-            throw core_1.SfError.create({
-                name: 'InvalidMockDir',
-                message: `SF_MOCK_DIR [${mockDir}] not found`,
-                cause: err,
-                actions: ['If you\'re trying to mock agent behavior you must create the mock directory and add expected mock files to it.']
-            });
-        }
-        if (!mockDirStat.isDirectory()) {
-            throw core_1.SfError.create({
-                name: 'InvalidMockDir',
-                message: `SF_MOCK_DIR [${mockDir}] is not a directory`,
-                actions: ['If you\'re trying to mock agent behavior you must create the mock directory and add expected mock files to it.']
-            });
-        }
-        return mockDir;
-    }
-};
-exports.getMockDir = getMockDir;
-//# sourceMappingURL=mockDir.js.map

package/lib/mockDir.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"mockDir.js","sourceRoot":"","sources":["../src/mockDir.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAEH,yCAAoC;AACpC,qCAA+C;AAC/C,2CAA2C;AAC3C,yCAAsC;AAEtC;;;;;;;GAOG;AACI,MAAM,UAAU,GAAG,GAAuB,EAAE;IACjD,MAAM,OAAO,GAAG,SAAG,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;IAC7C,IAAI,OAAO,EAAE,CAAC;QACZ,IAAI,WAAkB,CAAC;QACvB,IAAI,CAAC;YACH,WAAW,GAAG,IAAA,kBAAQ,EAAC,IAAA,mBAAO,EAAC,OAAO,CAAC,CAAC,CAAC;QAC3C,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,cAAO,CAAC,MAAM,CAAC;gBACnB,IAAI,EAAE,gBAAgB;gBACtB,OAAO,EAAE,gBAAgB,OAAO,aAAa;gBAC7C,KAAK,EAAE,GAAG;gBACV,OAAO,EAAE,CAAC,gHAAgH,CAAC;aAC5H,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,EAAE,CAAC;YAC/B,MAAM,cAAO,CAAC,MAAM,CAAC;gBACnB,IAAI,EAAE,gBAAgB;gBACtB,OAAO,EAAE,gBAAgB,OAAO,sBAAsB;gBACtD,OAAO,EAAE,CAAC,gHAAgH,CAAC;aAC5H,CAAC,CAAC;QACL,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;AACH,CAAC,CAAA;AAxBY,QAAA,UAAU,cAwBtB"}