donobu 2.16.0 → 2.17.0

package/dist/lib/fixtures/gptClients.js

@@ -0,0 +1,187 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getOrCreateDefaultGptClient = getOrCreateDefaultGptClient;
+exports.createGptClient = createGptClient;
+exports.anthropicClient = anthropicClient;
+exports.anthropicClientFixture = anthropicClientFixture;
+exports.anthropicAwsBedrockClient = anthropicAwsBedrockClient;
+exports.anthropicAwsBedrockClientFixture = anthropicAwsBedrockClientFixture;
+exports.googleGeminiClient = googleGeminiClient;
+exports.googleGeminiClientFixture = googleGeminiClientFixture;
+exports.openAiClient = openAiClient;
+exports.openAiClientFixture = openAiClientFixture;
+exports.vercelAiClientFixture = vercelAiClientFixture;
+exports.gptClientFixture = gptClientFixture;
+const GptClientFactory_1 = require("../../clients/GptClientFactory");
+const VercelAiGptClient_1 = require("../../clients/VercelAiGptClient");
+const envVars_1 = require("../../envVars");
+const donobuTestStack_1 = require("../utils/donobuTestStack");
+const DEFAULT_GPT_MODEL_FOR_PLATFORM = {
+    ANTHROPIC: process.env[envVars_1.ENV_VAR_NAMES.ANTHROPIC_MODEL_NAME] ||
+        'claude-3-5-sonnet-latest',
+    ANTHROPIC_AWS_BEDROCK: process.env[envVars_1.ENV_VAR_NAMES.AWS_BEDROCK_MODEL_NAME] || undefined,
+    GOOGLE_GEMINI: process.env[envVars_1.ENV_VAR_NAMES.GOOGLE_GENERATIVE_AI_MODEL_NAME] ||
+        'gemini-2.0-flash',
+    OPENAI: process.env[envVars_1.ENV_VAR_NAMES.OPENAI_API_MODEL_NAME] || 'gpt-4o',
+};
+let defaultGptClient = undefined;
+async function getOrCreateDefaultGptClient() {
+    if (!defaultGptClient) {
+        defaultGptClient = await createGptClient();
+        return defaultGptClient;
+    }
+    else {
+        return defaultGptClient;
+    }
+}
+/**
+ * This fixture resolves a GPT client based on environment variables.
+ * Supplying different environment variables will result in different GPT
+ * clients being created. Here is a mapping of client type to environment
+ * variables for how to create the various types of clients...
+ *
+ * Anthropic AWS Bedrock GPT client requires:
+ *  - AWS_BEDROCK_MODEL_NAME
+ *  - AWS_REGION
+ *  - AWS_ACCESS_KEY_ID
+ *  - AWS_SECRET_ACCESS_KEY
+ *
+ * Anthropic GPT client requires:
+ *  - ANTHROPIC_API_KEY
+ *
+ * Google GPT client requires:
+ *  - GOOGLE_GENERATIVE_AI_API_KEY
+ *
+ * OpenAI GPT client requires:
+ *  - OPENAI_API_KEY
+ *
+ * If the modelName is not specified, it will be defaulted according to the
+ * DEFAULT_GPT_MODEL_FOR_PLATFORM mapping.
+ */
+async function createGptClient(modelName) {
+    // Check which API keys are available
+    const hasAwsCredentials = !!DEFAULT_GPT_MODEL_FOR_PLATFORM.ANTHROPIC_AWS_BEDROCK &&
+        !!process.env[envVars_1.ENV_VAR_NAMES.AWS_ACCESS_KEY_ID] &&
+        !!process.env[envVars_1.ENV_VAR_NAMES.AWS_SECRET_ACCESS_KEY];
+    const hasAnthropicKey = !!process.env[envVars_1.ENV_VAR_NAMES.ANTHROPIC_API_KEY];
+    const hasGoogleKey = !!process.env[envVars_1.ENV_VAR_NAMES.GOOGLE_GENERATIVE_AI_API_KEY];
+    const hasOpenAiKey = !!process.env[envVars_1.ENV_VAR_NAMES.OPENAI_API_KEY];
+    // Priority order: Anthropic AWS Bedrock, Anthropic, Google, OpenAI
+    if (hasAwsCredentials) {
+        const model = modelName || DEFAULT_GPT_MODEL_FOR_PLATFORM.ANTHROPIC_AWS_BEDROCK;
+        const region = process.env[envVars_1.ENV_VAR_NAMES.AWS_REGION];
+        const accessKeyId = process.env[envVars_1.ENV_VAR_NAMES.AWS_ACCESS_KEY_ID];
+        const secretAccessKey = process.env[envVars_1.ENV_VAR_NAMES.AWS_SECRET_ACCESS_KEY];
+        return anthropicAwsBedrockClient(model, region, accessKeyId, secretAccessKey);
+    }
+    else if (hasAnthropicKey) {
+        const model = modelName || DEFAULT_GPT_MODEL_FOR_PLATFORM.ANTHROPIC;
+        return anthropicClient(model);
+    }
+    else if (hasGoogleKey) {
+        const model = modelName || DEFAULT_GPT_MODEL_FOR_PLATFORM.GOOGLE_GEMINI;
+        return googleGeminiClient(model);
+    }
+    else if (hasOpenAiKey) {
+        const model = modelName || DEFAULT_GPT_MODEL_FOR_PLATFORM.OPENAI;
+        return openAiClient(model);
+    }
+    else {
+        const donobu = await (0, donobuTestStack_1.getOrCreateDonobuStack)();
+        const defaultClient = (await donobu.flowsManager.createGptClient(null))
+            .gptClient;
+        if (defaultClient) {
+            return defaultClient;
+        }
+        else {
+            // No API keys are available and there is not flow-runner set up, throw an error.
+            throw new Error(`No API keys found for any supported GPT providers.
+Please set one of the following environment variables: ANTHROPIC_API_KEY, GOOGLE_GENERATIVE_AI_API_KEY, or OPENAI_API_KEY.
+Alternatively, set a flow-runner agent via POST /api/agents/flow-runner or via the Donobu app.`);
+        }
+    }
+}
+async function anthropicClient(modelName, apiKey) {
+    const divinedApiKey = apiKey || process.env[envVars_1.ENV_VAR_NAMES.ANTHROPIC_API_KEY];
+    if (!divinedApiKey) {
+        throw new Error('An API key is required to use the Anthropic API. Please provide one as an argument or set it using the ANTHROPIC_API_KEY environment variable.');
+    }
+    return GptClientFactory_1.GptClientFactory.createFromGptConfig({
+        type: 'ANTHROPIC',
+        apiKey: divinedApiKey,
+        modelName: modelName,
+    });
+}
+function anthropicClientFixture(modelName, apiKey) {
+    return async ({}, use) => {
+        const client = await anthropicClient(modelName, apiKey);
+        await use(client);
+    };
+}
+async function anthropicAwsBedrockClient(modelName, region, accessKeyId, secretAccessKey) {
+    const divinedAccessKeyId = accessKeyId || process.env[envVars_1.ENV_VAR_NAMES.AWS_ACCESS_KEY_ID];
+    const divinedSecretAccessKey = secretAccessKey || process.env[envVars_1.ENV_VAR_NAMES.AWS_SECRET_ACCESS_KEY];
+    if (!divinedAccessKeyId || !divinedSecretAccessKey) {
+        throw new Error('AWS access key ID and secret access key are required to use the Anthropic AWS Bedrock API. Please provide them as arguments or set them using the AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY environment variables.');
+    }
+    return GptClientFactory_1.GptClientFactory.createFromGptConfig({
+        type: 'ANTHROPIC_AWS_BEDROCK',
+        region: region,
+        modelName: modelName,
+        accessKeyId: divinedAccessKeyId,
+        secretAccessKey: divinedSecretAccessKey,
+    });
+}
+function anthropicAwsBedrockClientFixture(modelName, region, accessKeyId, secretAccessKey) {
+    return async ({}, use) => {
+        const client = await anthropicAwsBedrockClient(modelName, region, accessKeyId, secretAccessKey);
+        await use(client);
+    };
+}
+async function googleGeminiClient(modelName, apiKey) {
+    const divinedApiKey = apiKey || process.env[envVars_1.ENV_VAR_NAMES.GOOGLE_GENERATIVE_AI_API_KEY];
+    if (!divinedApiKey) {
+        throw new Error('An API key is required to use the Google Gemini API. Please provide one as an argument or set it using the GOOGLE_GENERATIVE_AI_API_KEY environment variable.');
+    }
+    return GptClientFactory_1.GptClientFactory.createFromGptConfig({
+        type: 'GOOGLE_GEMINI',
+        apiKey: divinedApiKey,
+        modelName: modelName,
+    });
+}
+function googleGeminiClientFixture(modelName, apiKey) {
+    return async ({}, use) => {
+        const client = await googleGeminiClient(modelName, apiKey);
+        await use(client);
+    };
+}
+async function openAiClient(modelName, apiKey) {
+    const divinedApiKey = apiKey || process.env[envVars_1.ENV_VAR_NAMES.OPENAI_API_KEY];
+    if (!divinedApiKey) {
+        throw new Error('An API key is required to use the OpenAI API. Please provide one as an argument or set it using the OPENAI_API_KEY environment variable.');
+    }
+    return GptClientFactory_1.GptClientFactory.createFromGptConfig({
+        type: 'OPENAI',
+        apiKey: divinedApiKey,
+        modelName: modelName,
+    });
+}
+function openAiClientFixture(modelName, apiKey) {
+    return async ({}, use) => {
+        const client = await openAiClient(modelName, apiKey);
+        await use(client);
+    };
+}
+function vercelAiClientFixture(model) {
+    return async ({}, use) => {
+        const client = new VercelAiGptClient_1.VercelAiGptClient(model);
+        await use(client);
+    };
+}
+function gptClientFixture(modelName) {
+    return async ({}, use) => {
+        const client = await createGptClient(modelName);
+        await use(client);
+    };
+}
+//# sourceMappingURL=gptClients.js.map

package/dist/lib/fixtures/gptClients.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gptClients.js","sourceRoot":"","sources":["../../../src/lib/fixtures/gptClients.ts"],"names":[],"mappings":";;AAsBA,kEAOC;AA0BD,0CAiDC;AAED,0CAiBC;AAED,wDAQC;AAED,8DAwBC;AAED,4EAeC;AAED,gDAkBC;AAED,8DAQC;AAED,oCAiBC;AAED,kDAQC;AAED,sDAOC;AAED,4CAOC;AA3PD,qEAAkE;AAClE,uEAAoE;AAEpE,2CAA8C;AAC9C,8DAAkE;AAElE,MAAM,8BAA8B,GAAG;IACrC,SAAS,EACP,OAAO,CAAC,GAAG,CAAC,uBAAa,CAAC,oBAAoB,CAAC;QAC/C,0BAA0B;IAC5B,qBAAqB,EACnB,OAAO,CAAC,GAAG,CAAC,uBAAa,CAAC,sBAAsB,CAAC,IAAI,SAAS;IAChE,aAAa,EACX,OAAO,CAAC,GAAG,CAAC,uBAAa,CAAC,+BAA+B,CAAC;QAC1D,kBAAkB;IACpB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,uBAAa,CAAC,qBAAqB,CAAC,IAAI,QAAQ;CACrE,CAAC;AAEF,IAAI,gBAAgB,GAA0B,SAAS,CAAC;AAEjD,KAAK,UAAU,2BAA2B;IAC/C,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACtB,gBAAgB,GAAG,MAAM,eAAe,EAAE,CAAC;QAC3C,OAAO,gBAAgB,CAAC;IAC1B,CAAC;SAAM,CAAC;QACN,OAAO,gBAAgB,CAAC;IAC1B,CAAC;AACH,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACI,KAAK,UAAU,eAAe,CAAC,SAAkB;IACtD,qCAAqC;IACrC,MAAM,iBAAiB,GACrB,CAAC,CAAC,8BAA8B,CAAC,qBAAqB;QACtD,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAa,CAAC,iBAAiB,CAAC;QAC9C,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAa,CAAC,qBAAqB,CAAC,CAAC;IACrD,MAAM,eAAe,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAa,CAAC,iBAAiB,CAAC,CAAC;IACvE,MAAM,YAAY,GAChB,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAa,CAAC,4BAA4B,CAAC,CAAC;IAC5D,MAAM,YAAY,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAa,CAAC,cAAc,CAAC,CAAC;IAEjE,mEAAmE;IACnE,IAAI,iBAAiB,EAAE,CAAC;QACtB,MAAM,KAAK,GACT,SAAS,IAAI,8BAA8B,CAAC,qBAAqB,CAAC;QACpE,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,uBAAa,CAAC,UAAU,CAAC,CAAC;QACrD,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,uBAAa,CAAC,iBAAiB,CAAC,CAAC;QACjE,MAAM,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,uBAAa,CAAC,qBAAqB,CAAC,CAAC;QACzE,OAAO,yBAAyB,CAC9B,KAAM,EACN,MAAM,EACN,WAAW,EACX,eAAe,CAChB,CAAC;IACJ,CAAC;SAAM,IAAI,eAAe,EAAE,CAAC;QAC3B,MAAM,KAAK,GAAG,SAAS,IAAI,8BAA8B,CAAC,SAAS,CAAC;QACpE,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;SAAM,IAAI,YAAY,EAAE,CAAC;QACxB,MAAM,KAAK,GAAG,SAAS,IAAI,8BAA8B,CAAC,aAAa,CAAC;QACxE,OAAO,kBAAkB,CAAC,KAAK,CAAC,CAAC;IACnC,CAAC;SAAM,IAAI,YAAY,EAAE,CAAC;QACxB,MAAM,KAAK,GAAG,SAAS,IAAI,8BAA8B,CAAC,MAAM,CAAC;QACjE,OAAO,YAAY,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC;SAAM,CAAC;QACN,MAAM,MAAM,GAAG,MAAM,IAAA,wCAAsB,GAAE,CAAC;QAC9C,MAAM,aAAa,GAAG,CAAC,MAAM,MAAM,CAAC,YAAY,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;aACpE,SAAS,CAAC;QAEb,IAAI,aAAa,EAAE,CAAC;YAClB,OAAO,aAAa,CAAC;QACvB,CAAC;aAAM,CAAC;YACN,iFAAiF;YACjF,MAAM,IAAI,KAAK,CACb;;+FAEuF,CACxF,CAAC;QACJ,CAAC;IACH,CAAC;AACH,CAAC;AAEM,KAAK,UAAU,eAAe,CACnC,SAAiB,EACjB,MAAe;IAEf,MAAM,aAAa,GAAG,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,uBAAa,CAAC,iBAAiB,CAAC,CAAC;IAE7E,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CACb,gJAAgJ,CACjJ,CAAC;IACJ,CAAC;IAED,OAAO,mCAAgB,CAAC,mBAAmB,CAAC;QAC1C,IAAI,EAAE,WAAW;QACjB,MAAM,EAAE,aAAa;QACrB,SAAS,EAAE,SAAS;KACrB,CAAC,CAAC;AACL,CAAC;AAED,SAAgB,sBAAsB,CACpC,SAAiB,EACjB,MAAe;IAEf,OAAO,KAAK,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE;QACvB,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QACxD,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAEM,KAAK,UAAU,yBAAyB,CAC7C,SAAiB,EACjB,MAAe,EACf,WAAoB,EACpB,eAAwB;IAExB,MAAM,kBAAkB,GACtB,WAAW,IAAI,OAAO,CAAC,GAAG,CAAC,uBAAa,CAAC,iBAAiB,CAAC,CAAC;IAC9D,MAAM,sBAAsB,GAC1B,eAAe,IAAI,OAAO,CAAC,GAAG,CAAC,uBAAa,CAAC,qBAAqB,CAAC,CAAC;IAEtE,IAAI,CAAC,kBAAkB,IAAI,CAAC,sBAAsB,EAAE,CAAC;QACnD,MAAM,IAAI,KAAK,CACb,sNAAsN,CACvN,CAAC;IACJ,CAAC;IAED,OAAO,mCAAgB,CAAC,mBAAmB,CAAC;QAC1C,IAAI,EAAE,uBAAuB;QAC7B,MAAM,EAAE,MAAM;QACd,SAAS,EAAE,SAAS;QACpB,WAAW,EAAE,kBAAkB;QAC/B,eAAe,EAAE,sBAAsB;KACxC,CAAC,CAAC;AACL,CAAC;AAED,SAAgB,gCAAgC,CAC9C,SAAiB,EACjB,MAAe,EACf,WAAoB,EACpB,eAAwB;IAExB,OAAO,KAAK,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE;QACvB,MAAM,MAAM,GAAG,MAAM,yBAAyB,CAC5C,SAAS,EACT,MAAM,EACN,WAAW,EACX,eAAe,CAChB,CAAC;QACF,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAEM,KAAK,UAAU,kBAAkB,CACtC,SAAiB,EACjB,MAAe;IAEf,MAAM,aAAa,GACjB,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,uBAAa,CAAC,4BAA4B,CAAC,CAAC;IAEpE,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CACb,+JAA+J,CAChK,CAAC;IACJ,CAAC;IAED,OAAO,mCAAgB,CAAC,mBAAmB,CAAC;QAC1C,IAAI,EAAE,eAAe;QACrB,MAAM,EAAE,aAAa;QACrB,SAAS,EAAE,SAAS;KACrB,CAAC,CAAC;AACL,CAAC;AAED,SAAgB,yBAAyB,CACvC,SAAiB,EACjB,MAAe;IAEf,OAAO,KAAK,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE;QACvB,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QAC3D,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAEM,KAAK,UAAU,YAAY,CAChC,SAAiB,EACjB,MAAe;IAEf,MAAM,aAAa,GAAG,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,uBAAa,CAAC,cAAc,CAAC,CAAC;IAE1E,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CACb,0IAA0I,CAC3I,CAAC;IACJ,CAAC;IAED,OAAO,mCAAgB,CAAC,mBAAmB,CAAC;QAC1C,IAAI,EAAE,QAAQ;QACd,MAAM,EAAE,aAAa;QACrB,SAAS,EAAE,SAAS;KACrB,CAAC,CAAC;AACL,CAAC;AAED,SAAgB,mBAAmB,CACjC,SAAiB,EACjB,MAAe;IAEf,OAAO,KAAK,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE;QACvB,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QACrD,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED,SAAgB,qBAAqB,CACnC,KAAoB;IAEpB,OAAO,KAAK,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE;QACvB,MAAM,MAAM,GAAG,IAAI,qCAAiB,CAAC,KAAK,CAAC,CAAC;QAC5C,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED,SAAgB,gBAAgB,CAC9B,SAAkB;IAElB,OAAO,KAAK,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE;QACvB,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,SAAS,CAAC,CAAC;QAChD,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC"}

package/dist/lib/fixtures/testExtension.d.ts

@@ -0,0 +1,10 @@
+import { DonobuExtendedPage } from '../types';
+import { GptClient } from '../../clients/GptClient';
+import { BrowserStorageState } from '../../models/BrowserStorageState';
+export * from 'playwright/test';
+export declare const test: import("@playwright/test").TestType<import("@playwright/test").PlaywrightTestArgs & import("@playwright/test").PlaywrightTestOptions & {
+    storageState?: BrowserStorageState | Promise<BrowserStorageState>;
+    gptClient?: GptClient;
+    page: DonobuExtendedPage;
+}, import("@playwright/test").PlaywrightWorkerArgs & import("@playwright/test").PlaywrightWorkerOptions>;
+//# sourceMappingURL=testExtension.d.ts.map

package/dist/lib/fixtures/testExtension.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"testExtension.d.ts","sourceRoot":"","sources":["../../../src/lib/fixtures/testExtension.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAC9C,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AAmEpD,OAAO,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AAEvE,cAAc,iBAAiB,CAAC;AAIhC,eAAO,MAAM,IAAI;mBACA,mBAAmB,GAAG,OAAO,CAAC,mBAAmB,CAAC;gBACrD,SAAS;UACf,kBAAkB;wGAqaxB,CAAC"}

package/dist/lib/fixtures/testExtension.js

@@ -0,0 +1,383 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.test = void 0;
+const test_1 = require("@playwright/test");
+const uuid_1 = require("uuid");
+const GptClient_1 = require("../../clients/GptClient");
+const VercelAiGptClient_1 = require("../../clients/VercelAiGptClient");
+const PlaywrightUtils_1 = require("../../utils/PlaywrightUtils");
+const donobuTestStack_1 = require("../utils/donobuTestStack");
+const selfHealing_1 = require("../utils/selfHealing");
+const envVars_1 = require("../../envVars");
+const MiscUtils_1 = require("../../utils/MiscUtils");
+const DonobuFlowsManager_1 = require("../../managers/DonobuFlowsManager");
+const gptClients_1 = require("./gptClients");
+const GoToWebpageTool_1 = require("../../tools/GoToWebpageTool");
+const PageInspector_1 = require("../../managers/PageInspector");
+const ControlPanel_1 = require("../../managers/ControlPanel");
+const InteractionVisualizer_1 = require("../../managers/InteractionVisualizer");
+const ToolManager_1 = require("../../managers/ToolManager");
+const AnalyzePageTextTool_1 = require("../../tools/AnalyzePageTextTool");
+const AssertPageTextTool_1 = require("../../tools/AssertPageTextTool");
+const Logger_1 = require("../../utils/Logger");
+const DonobuFlow_1 = require("../../managers/DonobuFlow");
+const HoverOverElementTool_1 = require("../../tools/HoverOverElementTool");
+const InputRandomizedEmailAddressTool_1 = require("../../tools/InputRandomizedEmailAddressTool");
+const AssertTool_1 = require("../../tools/AssertTool");
+const ChangeWebBrowserTabTool_1 = require("../../tools/ChangeWebBrowserTabTool");
+const ChooseSelectOptionTool_1 = require("../../tools/ChooseSelectOptionTool");
+const ClickTool_1 = require("../../tools/ClickTool");
+const CreateBrowserCookieReportTool_1 = require("../../tools/CreateBrowserCookieReportTool");
+const InputTextTool_1 = require("../../tools/InputTextTool");
+const PressKeyTool_1 = require("../../tools/PressKeyTool");
+const RunAccessibilityTestTool_1 = require("../../tools/RunAccessibilityTestTool");
+const ScrollPageTool_1 = require("../../tools/ScrollPageTool");
+__exportStar(require("playwright/test"), exports);
+const PLACEHOLDER_FLOW_URL = 'https://example.com';
+exports.test = test_1.test.extend({
+    // 1) Declare `gptClient` as an "option" fixture with a default of `undefined`.
+    gptClient: [
+        undefined, // default
+        { option: true }, // so that test.use({ gptClient: ... }) can override
+    ],
+    // Override the default page fixture
+    page: async ({ page, gptClient }, use, testInfo) => {
+        if (!gptClient) {
+            try {
+                gptClient = await (0, gptClients_1.getOrCreateDefaultGptClient)();
+            }
+            catch (_error) {
+                // Do nothing since we may not need a GPT client at all anyway.
+                // If something later on requires a GPT client, it will throw an error.
+            }
+        }
+        // Patch the page with "run", but do it lazily
+        const donobu = await (0, donobuTestStack_1.getOrCreateDonobuStack)();
+        await PlaywrightUtils_1.PlaywrightUtils.setupBasicBrowserContext(page.context());
+        let isTestRunInitialized = false;
+        let isTestRunFailed = false;
+        async function maybeInitDonobu() {
+            if (isTestRunInitialized) {
+                return;
+            }
+            const overallObjective = testInfo.annotations.find((v) => {
+                return v.type === 'objective';
+            })?.description ?? null;
+            // Get the implied allowed environment variables by inspecting the objective.
+            const allowedEnvVarsByName = DonobuFlowsManager_1.DonobuFlowsManager.distillAllowedEnvVariableNames(overallObjective, null);
+            const donobuTestFlowMetadata = {
+                id: (0, uuid_1.v4)(),
+                name: testInfo.title,
+                createdWithDonobuVersion: MiscUtils_1.MiscUtils.DONOBU_VERSION,
+                browser: {
+                    using: {
+                        type: 'device',
+                        deviceName: 'Desktop Chromium',
+                        headless: true,
+                    },
+                },
+                envVars: allowedEnvVarsByName,
+                gptConfigName: null,
+                hasGptConfigNameOverride: false,
+                customTools: null,
+                defaultMessageDuration: 0,
+                runMode: 'DETERMINISTIC',
+                isControlPanelEnabled: false,
+                callbackUrl: null,
+                targetWebsite: PLACEHOLDER_FLOW_URL,
+                overallObjective: overallObjective,
+                allowedTools: [],
+                resultJsonSchema: null,
+                result: null,
+                inputTokensUsed: 0,
+                completionTokensUsed: 0,
+                iterations: 0,
+                maxIterations: 0,
+                startedAt: new Date().getTime(),
+                completedAt: null,
+                state: 'RUNNING_ACTION',
+                nextState: null,
+            };
+            const persistenceLayer = await donobu.flowsPersistenceFactory.createPersistenceLayer();
+            await persistenceLayer.setMetadata(donobuTestFlowMetadata);
+            page._donobuFlowMetadata = donobuTestFlowMetadata;
+            isTestRunInitialized = true;
+        }
+        const originalGoto = page.goto;
+        // Intercept 'page.goto' so that Donobu can later re-run it as a standard
+        // Donobu action if necessary. This can happen if you run a Playwright
+        // test using npx, but later decide to rerun that run using the Donobu UI.
+        page.goto = async (url, options) => {
+            await maybeInitDonobu();
+            const startedAt = new Date().getTime();
+            const flowId = page._donobuFlowMetadata.id;
+            const persistenceLayer = await donobu.flowsPersistenceFactory.createPersistenceLayer();
+            // Update the target website to whatever the first place we are
+            // navigating to is.
+            if (page._donobuFlowMetadata.targetWebsite === PLACEHOLDER_FLOW_URL) {
+                page._donobuFlowMetadata = {
+                    ...page._donobuFlowMetadata,
+                    targetWebsite: url,
+                };
+                await persistenceLayer.setMetadata(page._donobuFlowMetadata);
+            }
+            try {
+                const resp = await originalGoto.call(page, url, options);
+                await PlaywrightUtils_1.PlaywrightUtils.waitForPageStability(page);
+                const pageTitle = await page.title();
+                const postCallImage = await PlaywrightUtils_1.PlaywrightUtils.takeScreenshot(page);
+                const postCallImageId = await persistenceLayer.saveScreenShot(flowId, postCallImage);
+                const completedAt = new Date().getTime();
+                await persistenceLayer.saveToolCall(flowId, {
+                    id: MiscUtils_1.MiscUtils.createAdHocToolCallId(),
+                    toolName: GoToWebpageTool_1.GoToWebpageTool.NAME,
+                    parameters: {
+                        url: url,
+                    },
+                    outcome: {
+                        isSuccessful: true,
+                        forLlm: `Successfully navigated to ${url}`,
+                        metadata: {
+                            pageTitle: pageTitle,
+                            resolvedUrl: page.url(),
+                        },
+                    },
+                    postCallImageId: postCallImageId,
+                    page: url,
+                    startedAt: startedAt,
+                    completedAt: completedAt,
+                });
+                return resp;
+            }
+            catch (error) {
+                const postCallImage = await PlaywrightUtils_1.PlaywrightUtils.takeScreenshot(page);
+                const postCallImageId = await persistenceLayer.saveScreenShot(flowId, postCallImage);
+                const completedAt = new Date().getTime();
+                await persistenceLayer.saveToolCall(flowId, {
+                    id: MiscUtils_1.MiscUtils.createAdHocToolCallId(),
+                    toolName: 'goToWebpage',
+                    parameters: {
+                        url: url,
+                    },
+                    outcome: {
+                        isSuccessful: false,
+                        forLlm: `FAILED! ${typeof error}: ${error.message}`,
+                        metadata: null,
+                    },
+                    postCallImageId: postCallImageId,
+                    page: url,
+                    startedAt: startedAt,
+                    completedAt: completedAt,
+                });
+                // Mark test as failed for later callback
+                isTestRunFailed = true;
+                throw error;
+            }
+        };
+        page.run = async (toolName, toolParams, gptClientOverride) => {
+            await maybeInitDonobu();
+            let finalGptClient = null;
+            if (gptClientOverride) {
+                if (!(gptClientOverride instanceof GptClient_1.GptClient)) {
+                    finalGptClient = new VercelAiGptClient_1.VercelAiGptClient(gptClientOverride);
+                }
+                else {
+                    finalGptClient = gptClientOverride;
+                }
+            }
+            else if (gptClient) {
+                finalGptClient = gptClient;
+            }
+            const pageInspector = new PageInspector_1.PageInspector();
+            const controlPanel = new ControlPanel_1.ControlPanel(page._donobuFlowMetadata);
+            const persistenceLayer = await donobu.flowsPersistenceFactory.createPersistenceLayer();
+            const clonedParams = toolParams
+                ? JSON.parse(JSON.stringify(toolParams))
+                : {};
+            clonedParams.rationale || (clonedParams.rationale = '');
+            const envData = await donobu.flowsManager.buildEnvData(page._donobuFlowMetadata.envVars ?? []);
+            const toolCallContext = {
+                flowsManager: donobu.flowsManager,
+                envData: envData,
+                pageInspector: pageInspector,
+                controlPanel: controlPanel,
+                persistence: persistenceLayer,
+                gptClient: finalGptClient,
+                interactionVisualizer: new InteractionVisualizer_1.InteractionVisualizer(0),
+                proposedToolCalls: [],
+                invokedToolCalls: [],
+                page: { current: page },
+                metadata: page._donobuFlowMetadata,
+                toolCallId: clonedParams.toolCallId ?? MiscUtils_1.MiscUtils.createAdHocToolCallId(),
+            };
+            try {
+                const result = await new ToolManager_1.ToolManager(ToolManager_1.ToolManager.ALL_TOOLS).invokeTool(toolCallContext, toolName, clonedParams, false);
+                await persistenceLayer.setMetadata(page._donobuFlowMetadata);
+                await persistenceLayer.saveToolCall(page._donobuFlowMetadata.id, result);
+                // Check if tool execution was successful
+                if (!result.outcome.isSuccessful) {
+                    throw new Error(`${toolName} execution failed: ${JSON.stringify(result, null, 2)}`);
+                }
+                return result.outcome;
+            }
+            catch (error) {
+                // Mark test as failed if an exception occurs during tool execution
+                isTestRunFailed = true;
+                throw error;
+            }
+        };
+        page.analyzePageText = async (params, gptClientOverride) => {
+            return page.run(AnalyzePageTextTool_1.AnalyzePageTextTool.NAME, params, gptClientOverride);
+        };
+        page.assertPageText = async (params, gptClientOverride) => {
+            return page.run(AssertPageTextTool_1.AssertPageTextTool.NAME, params, gptClientOverride);
+        };
+        page.changeTab = async (url) => {
+            // The underlying tool call is mostly performative. We do it so we can
+            // record the operation occurred, but the actual page reassignment happens
+            // at the call site of this function.
+            const result = await page.run(ChangeWebBrowserTabTool_1.ChangeWebBrowserTabTool.NAME, {
+                tabUrl: url,
+            });
+            if (result.isSuccessful) {
+                // We need to return a real Page object, not just the raw URL, so we
+                // loop through the pages of the current context, knowing we will find
+                // a match.
+                return page
+                    .context()
+                    .pages()
+                    .find((tab) => tab.url() === url);
+            }
+            else {
+                throw Error(result.forLlm);
+            }
+        };
+        page.chooseSelectOption = async (params) => {
+            return page.run(ChooseSelectOptionTool_1.ChooseSelectOptionTool.NAME, params);
+        };
+        page.clickElement = async (params) => {
+            return page.run(ClickTool_1.ClickTool.NAME, params);
+        };
+        page.createCookieReport = async (gptClientOverride) => {
+            return page.run(CreateBrowserCookieReportTool_1.CreateBrowserCookieReportTool.NAME, gptClientOverride);
+        };
+        page.extract = async (params, gptClientOverride) => {
+            await maybeInitDonobu();
+            let finalGptClient = null;
+            if (gptClientOverride) {
+                if (!(gptClientOverride instanceof GptClient_1.GptClient)) {
+                    finalGptClient = new VercelAiGptClient_1.VercelAiGptClient(gptClientOverride);
+                }
+                else {
+                    finalGptClient = gptClientOverride;
+                }
+            }
+            else if (gptClient) {
+                finalGptClient = gptClient;
+            }
+            if (!finalGptClient) {
+                isTestRunFailed = true;
+                throw Error('Cannot extract an object from page without a GPT client set up!');
+            }
+            try {
+                const persistenceLayer = await donobu.flowsPersistenceFactory.createPersistenceLayer();
+                const toolCallHistory = await persistenceLayer.getToolCalls(page._donobuFlowMetadata.id);
+                const structuredOutputMessage = await (0, DonobuFlow_1.extractFromPage)(params.instruction ??
+                    'Generate an object conforming to the given JSON-schema', params.schema, page, toolCallHistory, finalGptClient);
+                page._donobuFlowMetadata.resultJsonSchema = params.schema;
+                page._donobuFlowMetadata.result = structuredOutputMessage.output;
+                await persistenceLayer.setMetadata(page._donobuFlowMetadata);
+                return structuredOutputMessage.output;
+            }
+            catch (error) {
+                isTestRunFailed = true;
+                throw error;
+            }
+        };
+        page.hoverOverElement = async (params) => {
+            return page.run(HoverOverElementTool_1.HoverOverElementTool.NAME, params);
+        };
+        page.inputRandomizedEmailAddress = async (params) => {
+            return page.run(InputRandomizedEmailAddressTool_1.InputRandomizedEmailAddressTool.NAME, params);
+        };
+        page.inputText = async (params) => {
+            return page.run(InputTextTool_1.InputTextTool.NAME, params);
+        };
+        page.pressKey = async (params) => {
+            return page.run(PressKeyTool_1.PressKeyTool.NAME, params);
+        };
+        page.runAccessibilityTest = async () => {
+            return page.run(RunAccessibilityTestTool_1.RunAccessibilityTestTool.NAME);
+        };
+        page.scroll = async (params) => {
+            return page.run(ScrollPageTool_1.ScrollPageTool.NAME, params);
+        };
+        page.visuallyAssert = async (params, gptClientOverride) => {
+            return page.run(AssertTool_1.AssertTool.NAME, params, gptClientOverride);
+        };
+        try {
+            // Let Playwright continue and give tests the patched page.
+            await use(page);
+            // Check if the test status is failed in the testInfo
+            // This catches failures that were marked by Playwright itself
+            if (testInfo.status === 'failed' || testInfo.status === 'timedOut') {
+                isTestRunFailed = true;
+            }
+        }
+        catch (error) {
+            // Mark test as failed if an uncaught exception occurs during the test
+            isTestRunFailed = true;
+            throw error;
+        }
+        finally {
+            try {
+                const persistenceLayer = await donobu.flowsPersistenceFactory.createPersistenceLayer();
+                if (page._donobuFlowMetadata) {
+                    // Update metadata with final status
+                    page._donobuFlowMetadata.state = isTestRunFailed
+                        ? 'FAILED'
+                        : 'SUCCESS';
+                    page._donobuFlowMetadata.completedAt = new Date().getTime();
+                    await persistenceLayer.setMetadata(page._donobuFlowMetadata);
+                    await testInfo.attach('test-flow-metadata.json', {
+                        body: JSON.stringify(page._donobuFlowMetadata, null, 2),
+                        contentType: 'application/json',
+                    });
+                }
+            }
+            catch (error) {
+                // Log but don't throw, to ensure cleanup continues.
+                Logger_1.appLogger.error('Error during test cleanup:', error);
+            }
+            const isSelfHealEnabledEnvVar = process.env[envVars_1.ENV_VAR_NAMES.SELF_HEAL_TESTS_ENABLED]?.trim();
+            if (isTestRunFailed &&
+                (isSelfHealEnabledEnvVar === 'true' || isSelfHealEnabledEnvVar === '1')) {
+                if (!gptClient) {
+                    Logger_1.appLogger.warn('Will not self-heal due to no GPT client being set up.');
+                    return;
+                }
+                try {
+                    await (0, selfHealing_1.selfHeal)(gptClient, testInfo, page._donobuFlowMetadata);
+                }
+                catch (selfHealError) {
+                    Logger_1.appLogger.error('Error when attempting to self heal:', selfHealError);
+                }
+            }
+        }
+    },
+});
+//# sourceMappingURL=testExtension.js.map

package/dist/lib/fixtures/testExtension.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"testExtension.js","sourceRoot":"","sources":["../../../src/lib/fixtures/testExtension.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAAA,2CAAsD;AACtD,+BAAkC;AAElC,uDAAoD;AAEpD,uEAAoE;AACpE,iEAA8D;AAC9D,8DAAkE;AAClE,sDAAgD;AAChD,2CAA8C;AAC9C,qDAAkD;AAClD,0EAAuE;AAEvE,6CAA2D;AAC3D,iEAGqC;AAErC,gEAA6D;AAC7D,8DAA2D;AAC3D,gFAA6E;AAE7E,4DAAyD;AACzD,yEAGyC;AACzC,uEAGwC;AACxC,+CAA+C;AAC/C,0DAA4D;AAE5D,2EAG0C;AAC1C,iGAGqD;AACrD,uDAA8E;AAC9E,iFAG6C;AAC7C,+EAG4C;AAC5C,qDAG+B;AAC/B,6FAA0F;AAC1F,6DAGmC;AACnC,2DAGkC;AAClC,mFAAgF;AAChF,+DAGoC;AAGpC,kDAAgC;AAEhC,MAAM,oBAAoB,GAAG,qBAAqB,CAAC;AAEtC,QAAA,IAAI,GAAG,WAAI,CAAC,MAAM,CAI5B;IACD,+EAA+E;IAC/E,SAAS,EAAE;QACT,SAAS,EAAE,UAAU;QACrB,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,oDAAoD;KACvE;IACD,oCAAoC;IACpC,IAAI,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,EAAE;QACjD,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,IAAI,CAAC;gBACH,SAAS,GAAG,MAAM,IAAA,wCAA2B,GAAE,CAAC;YAClD,CAAC;YAAC,OAAO,MAAM,EAAE,CAAC;gBAChB,+DAA+D;gBAC/D,uEAAuE;YACzE,CAAC;QACH,CAAC;QACD,8CAA8C;QAC9C,MAAM,MAAM,GAAG,MAAM,IAAA,wCAAsB,GAAE,CAAC;QAC9C,MAAM,iCAAe,CAAC,wBAAwB,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAC/D,IAAI,oBAAoB,GAAG,KAAK,CAAC;QACjC,IAAI,eAAe,GAAG,KAAK,CAAC;QAE5B,KAAK,UAAU,eAAe;YAC5B,IAAI,oBAAoB,EAAE,CAAC;gBACzB,OAAO;YACT,CAAC;YAED,MAAM,gBAAgB,GACpB,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE;gBAC9B,OAAO,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC;YAChC,CAAC,CAAC,EAAE,WAAW,IAAI,IAAI,CAAC;YAC1B,6EAA6E;YAC7E,MAAM,oBAAoB,GACxB,uCAAkB,CAAC,8BAA8B,CAC/C,gBAAgB,EAChB,IAAI,CACL,CAAC;YACJ,MAAM,sBAAsB,GAAiB;gBAC3C,EAAE,EAAE,IAAA,SAAI,GAAE;gBACV,IAAI,EAAE,QAAQ,CAAC,KAAK;gBACpB,wBAAwB,EAAE,qBAAS,CAAC,cAAc;gBAClD,OAAO,EAAE;oBACP,KAAK,EAAE;wBACL,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,kBAAkB;wBAC9B,QAAQ,EAAE,IAAI;qBACf;iBACF;gBACD,OAAO,EAAE,oBAAoB;gBAC7B,aAAa,EAAE,IAAI;gBACnB,wBAAwB,EAAE,KAAK;gBAC/B,WAAW,EAAE,IAAI;gBACjB,sBAAsB,EAAE,CAAC;gBACzB,OAAO,EAAE,eAAe;gBACxB,qBAAqB,EAAE,KAAK;gBAC5B,WAAW,EAAE,IAAI;gBACjB,aAAa,EAAE,oBAAoB;gBACnC,gBAAgB,EAAE,gBAAgB;gBAClC,YAAY,EAAE,EAAE;gBAChB,gBAAgB,EAAE,IAAI;gBACtB,MAAM,EAAE,IAAI;gBACZ,eAAe,EAAE,CAAC;gBAClB,oBAAoB,EAAE,CAAC;gBACvB,UAAU,EAAE,CAAC;gBACb,aAAa,EAAE,CAAC;gBAChB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE;gBAC/B,WAAW,EAAE,IAAI;gBACjB,KAAK,EAAE,gBAAgB;gBACvB,SAAS,EAAE,IAAI;aAChB,CAAC;YACF,MAAM,gBAAgB,GACpB,MAAM,MAAM,CAAC,uBAAuB,CAAC,sBAAsB,EAAE,CAAC;YAChE,MAAM,gBAAgB,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC;YAC3D,IAAI,CAAC,mBAAmB,GAAG,sBAAsB,CAAC;YAClD,oBAAoB,GAAG,IAAI,CAAC;QAC9B,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC;QAC/B,yEAAyE;QACzE,sEAAsE;QACtE,0EAA0E;QAC1E,IAAI,CAAC,IAAI,GAAG,KAAK,EACf,GAAW,EACX,OAIC,EACD,EAAE;YACF,MAAM,eAAe,EAAE,CAAC;YACxB,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,CAAC;YACvC,MAAM,MAAM,GAAG,IAAI,CAAC,mBAAmB,CAAC,EAAE,CAAC;YAC3C,MAAM,gBAAgB,GACpB,MAAM,MAAM,CAAC,uBAAuB,CAAC,sBAAsB,EAAE,CAAC;YAChE,+DAA+D;YAC/D,oBAAoB;YACpB,IAAI,IAAI,CAAC,mBAAmB,CAAC,aAAa,KAAK,oBAAoB,EAAE,CAAC;gBACpE,IAAI,CAAC,mBAAmB,GAAG;oBACzB,GAAG,IAAI,CAAC,mBAAmB;oBAC3B,aAAa,EAAE,GAAG;iBACnB,CAAC;gBACF,MAAM,gBAAgB,CAAC,WAAW,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAC/D,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;gBACzD,MAAM,iCAAe,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;gBACjD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;gBACrC,MAAM,aAAa,GAAG,MAAM,iCAAe,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;gBACjE,MAAM,eAAe,GAAG,MAAM,gBAAgB,CAAC,cAAc,CAC3D,MAAM,EACN,aAAa,CACd,CAAC;gBACF,MAAM,WAAW,GAAG,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,CAAC;gBACzC,MAAM,gBAAgB,CAAC,YAAY,CAAC,MAAM,EAAE;oBAC1C,EAAE,EAAE,qBAAS,CAAC,qBAAqB,EAAE;oBACrC,QAAQ,EAAE,iCAAe,CAAC,IAAI;oBAC9B,UAAU,EAAE;wBACV,GAAG,EAAE,GAAG;qBAC+B;oBACzC,OAAO,EAAE;wBACP,YAAY,EAAE,IAAI;wBAClB,MAAM,EAAE,6BAA6B,GAAG,EAAE;wBAC1C,QAAQ,EAAE;4BACR,SAAS,EAAE,SAAS;4BACpB,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;yBACxB;qBACF;oBACD,eAAe,EAAE,eAAe;oBAChC,IAAI,EAAE,GAAG;oBACT,SAAS,EAAE,SAAS;oBACpB,WAAW,EAAE,WAAW;iBACN,CAAC,CAAC;gBACtB,OAAO,IAAI,CAAC;YACd,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,aAAa,GAAG,MAAM,iCAAe,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;gBACjE,MAAM,eAAe,GAAG,MAAM,gBAAgB,CAAC,cAAc,CAC3D,MAAM,EACN,aAAa,CACd,CAAC;gBACF,MAAM,WAAW,GAAG,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,CAAC;gBACzC,MAAM,gBAAgB,CAAC,YAAY,CAAC,MAAM,EAAE;oBAC1C,EAAE,EAAE,qBAAS,CAAC,qBAAqB,EAAE;oBACrC,QAAQ,EAAE,aAAa;oBACvB,UAAU,EAAE;wBACV,GAAG,EAAE,GAAG;qBAC+B;oBACzC,OAAO,EAAE;wBACP,YAAY,EAAE,KAAK;wBACnB,MAAM,EAAE,WAAW,OAAO,KAAK,KAAM,KAAa,CAAC,OAAO,EAAE;wBAC5D,QAAQ,EAAE,IAAI;qBACf;oBACD,eAAe,EAAE,eAAe;oBAChC,IAAI,EAAE,GAAG;oBACT,SAAS,EAAE,SAAS;oBACpB,WAAW,EAAE,WAAW;iBACN,CAAC,CAAC;gBAEtB,yCAAyC;gBACzC,eAAe,GAAG,IAAI,CAAC;gBACvB,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC,CAAC;QAEF,IAAI,CAAC,GAAG,GAAG,KAAK,EACd,QAAgB,EAChB,UAAgB,EAChB,iBAA6C,EAC7C,EAAE;YACF,MAAM,eAAe,EAAE,CAAC;YACxB,IAAI,cAAc,GAAqB,IAAI,CAAC;YAE5C,IAAI,iBAAiB,EAAE,CAAC;gBACtB,IAAI,CAAC,CAAC,iBAAiB,YAAY,qBAAS,CAAC,EAAE,CAAC;oBAC9C,cAAc,GAAG,IAAI,qCAAiB,CAAC,iBAAiB,CAAC,CAAC;gBAC5D,CAAC;qBAAM,CAAC;oBACN,cAAc,GAAG,iBAAiB,CAAC;gBACrC,CAAC;YACH,CAAC;iBAAM,IAAI,SAAS,EAAE,CAAC;gBACrB,cAAc,GAAG,SAAS,CAAC;YAC7B,CAAC;YAED,MAAM,aAAa,GAAG,IAAI,6BAAa,EAAE,CAAC;YAC1C,MAAM,YAAY,GAAG,IAAI,2BAAY,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAChE,MAAM,gBAAgB,GACpB,MAAM,MAAM,CAAC,uBAAuB,CAAC,sBAAsB,EAAE,CAAC;YAChE,MAAM,YAAY,GAAG,UAAU;gBAC7B,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;gBACxC,CAAC,CAAC,EAAE,CAAC;YACP,YAAY,CAAC,SAAS,KAAtB,YAAY,CAAC,SAAS,GAAK,EAAE,EAAC;YAC9B,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,YAAY,CACpD,IAAI,CAAC,mBAAmB,CAAC,OAAO,IAAI,EAAE,CACvC,CAAC;YACF,MAAM,eAAe,GAAG;gBACtB,YAAY,EAAE,MAAM,CAAC,YAAY;gBACjC,OAAO,EAAE,OAAO;gBAChB,aAAa,EAAE,aAAa;gBAC5B,YAAY,EAAE,YAAY;gBAC1B,WAAW,EAAE,gBAAgB;gBAC7B,SAAS,EAAE,cAAc;gBACzB,qBAAqB,EAAE,IAAI,6CAAqB,CAAC,CAAC,CAAC;gBACnD,iBAAiB,EAAE,EAAE;gBACrB,gBAAgB,EAAE,EAAE;gBACpB,IAAI,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;gBACvB,QAAQ,EAAE,IAAI,CAAC,mBAAmB;gBAClC,UAAU,EACR,YAAY,CAAC,UAAU,IAAI,qBAAS,CAAC,qBAAqB,EAAE;aACrC,CAAC;YAE5B,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,IAAI,yBAAW,CAAC,yBAAW,CAAC,SAAS,CAAC,CAAC,UAAU,CACpE,eAAe,EACf,QAAQ,EACR,YAAY,EACZ,KAAK,CACN,CAAC;gBAEF,MAAM,gBAAgB,CAAC,WAAW,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;gBAC7D,MAAM,gBAAgB,CAAC,YAAY,CACjC,IAAI,CAAC,mBAAmB,CAAC,EAAE,EAC3B,MAAM,CACP,CAAC;gBAEF,yCAAyC;gBACzC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;oBACjC,MAAM,IAAI,KAAK,CACb,GAAG,QAAQ,sBAAsB,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CACnE,CAAC;gBACJ,CAAC;gBAED,OAAO,MAAM,CAAC,OAAO,CAAC;YACxB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,mEAAmE;gBACnE,eAAe,GAAG,IAAI,CAAC;gBACvB,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC,CAAC;QAEF,IAAI,CAAC,eAAe,GAAG,KAAK,EAC1B,MAAyC,EACzC,iBAA6C,EAC7C,EAAE;YACF,OAAO,IAAI,CAAC,GAAG,CAAC,yCAAmB,CAAC,IAAI,EAAE,MAAM,EAAE,iBAAiB,CAAC,CAAC;QACvE,CAAC,CAAC;QACF,IAAI,CAAC,cAAc,GAAG,KAAK,EACzB,MAAwC,EACxC,iBAA6C,EAC7C,EAAE;YACF,OAAO,IAAI,CAAC,GAAG,CAAC,uCAAkB,CAAC,IAAI,EAAE,MAAM,EAAE,iBAAiB,CAAC,CAAC;QACtE,CAAC,CAAC;QACF,IAAI,CAAC,SAAS,GAAG,KAAK,EAAE,GAAW,EAAiB,EAAE;YACpD,sEAAsE;YACtE,0EAA0E;YAC1E,qCAAqC;YACrC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,iDAAuB,CAAC,IAAI,EAAE;gBAC1D,MAAM,EAAE,GAAG;aACoC,CAAC,CAAC;YAEnD,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;gBACxB,oEAAoE;gBACpE,sEAAsE;gBACtE,WAAW;gBACX,OAAO,IAAI;qBACR,OAAO,EAAE;qBACT,KAAK,EAAE;qBACP,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,GAAG,CAAE,CAAC;YACvC,CAAC;iBAAM,CAAC;gBACN,MAAM,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC,CAAC;QACF,IAAI,CAAC,kBAAkB,GAAG,KAAK,EAC7B,MAAqD,EACrD,EAAE;YACF,OAAO,IAAI,CAAC,GAAG,CAAC,+CAAsB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACvD,CAAC,CAAC;QACF,IAAI,CAAC,YAAY,GAAG,KAAK,EAAE,MAAwC,EAAE,EAAE;YACrE,OAAO,IAAI,CAAC,GAAG,CAAC,qBAAS,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC1C,CAAC,CAAC;QACF,IAAI,CAAC,kBAAkB,GAAG,KAAK,EAC7B,iBAA6C,EAC7C,EAAE;YACF,OAAO,IAAI,CAAC,GAAG,CAAC,6DAA6B,CAAC,IAAI,EAAE,iBAAiB,CAAC,CAAC;QACzE,CAAC,CAAC;QACF,IAAI,CAAC,OAAO,GAAG,KAAK,EAClB,MAGC,EACD,iBAA6C,EACX,EAAE;YACpC,MAAM,eAAe,EAAE,CAAC;YAExB,IAAI,cAAc,GAAqB,IAAI,CAAC;YAE5C,IAAI,iBAAiB,EAAE,CAAC;gBACtB,IAAI,CAAC,CAAC,iBAAiB,YAAY,qBAAS,CAAC,EAAE,CAAC;oBAC9C,cAAc,GAAG,IAAI,qCAAiB,CAAC,iBAAiB,CAAC,CAAC;gBAC5D,CAAC;qBAAM,CAAC;oBACN,cAAc,GAAG,iBAAiB,CAAC;gBACrC,CAAC;YACH,CAAC;iBAAM,IAAI,SAAS,EAAE,CAAC;gBACrB,cAAc,GAAG,SAAS,CAAC;YAC7B,CAAC;YAED,IAAI,CAAC,cAAc,EAAE,CAAC;gBACpB,eAAe,GAAG,IAAI,CAAC;gBACvB,MAAM,KAAK,CACT,iEAAiE,CAClE,CAAC;YACJ,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,gBAAgB,GACpB,MAAM,MAAM,CAAC,uBAAuB,CAAC,sBAAsB,EAAE,CAAC;gBAChE,MAAM,eAAe,GAAG,MAAM,gBAAgB,CAAC,YAAY,CACzD,IAAI,CAAC,mBAAmB,CAAC,EAAE,CAC5B,CAAC;gBACF,MAAM,uBAAuB,GAAG,MAAM,IAAA,4BAAe,EACnD,MAAM,CAAC,WAAW;oBAChB,wDAAwD,EAC1D,MAAM,CAAC,MAAM,EACb,IAAI,EACJ,eAAe,EACf,cAAc,CACf,CAAC;gBACF,IAAI,CAAC,mBAAmB,CAAC,gBAAgB,GAAG,MAAM,CAAC,MAAM,CAAC;gBAC1D,IAAI,CAAC,mBAAmB,CAAC,MAAM,GAAG,uBAAuB,CAAC,MAAM,CAAC;gBACjE,MAAM,gBAAgB,CAAC,WAAW,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;gBAC7D,OAAO,uBAAuB,CAAC,MAAM,CAAC;YACxC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,eAAe,GAAG,IAAI,CAAC;gBACvB,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC,CAAC;QACF,IAAI,CAAC,gBAAgB,GAAG,KAAK,EAC3B,MAAmD,EACnD,EAAE;YACF,OAAO,IAAI,CAAC,GAAG,CAAC,2CAAoB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACrD,CAAC,CAAC;QACF,IAAI,CAAC,2BAA2B,GAAG,KAAK,EACtC,MAA8D,EAC9D,EAAE;YACF,OAAO,IAAI,CAAC,GAAG,CAAC,iEAA+B,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAChE,CAAC,CAAC;QACF,IAAI,CAAC,SAAS,GAAG,KAAK,EAAE,MAA4C,EAAE,EAAE;YACtE,OAAO,IAAI,CAAC,GAAG,CAAC,6BAAa,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC9C,CAAC,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,KAAK,EAAE,MAA2C,EAAE,EAAE;YACpE,OAAO,IAAI,CAAC,GAAG,CAAC,2BAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC7C,CAAC,CAAC;QACF,IAAI,CAAC,oBAAoB,GAAG,KAAK,IAAI,EAAE;YACrC,OAAO,IAAI,CAAC,GAAG,CAAC,mDAAwB,CAAC,IAAI,CAAC,CAAC;QACjD,CAAC,CAAC;QACF,IAAI,CAAC,MAAM,GAAG,KAAK,EAAE,MAAoC,EAAE,EAAE;YAC3D,OAAO,IAAI,CAAC,GAAG,CAAC,+BAAc,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC/C,CAAC,CAAC;QACF,IAAI,CAAC,cAAc,GAAG,KAAK,EACzB,MAAgC,EAChC,iBAA6B,EAC7B,EAAE;YACF,OAAO,IAAI,CAAC,GAAG,CAAC,uBAAU,CAAC,IAAI,EAAE,MAAM,EAAE,iBAAiB,CAAC,CAAC;QAC9D,CAAC,CAAC;QAEF,IAAI,CAAC;YACH,2DAA2D;YAC3D,MAAM,GAAG,CAAC,IAAI,CAAC,CAAC;YAEhB,qDAAqD;YACrD,8DAA8D;YAC9D,IAAI,QAAQ,CAAC,MAAM,KAAK,QAAQ,IAAI,QAAQ,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;gBACnE,eAAe,GAAG,IAAI,CAAC;YACzB,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,sEAAsE;YACtE,eAAe,GAAG,IAAI,CAAC;YACvB,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC;gBACH,MAAM,gBAAgB,GACpB,MAAM,MAAM,CAAC,uBAAuB,CAAC,sBAAsB,EAAE,CAAC;gBAEhE,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;oBAC7B,oCAAoC;oBACpC,IAAI,CAAC,mBAAmB,CAAC,KAAK,GAAG,eAAe;wBAC9C,CAAC,CAAC,QAAQ;wBACV,CAAC,CAAC,SAAS,CAAC;oBACd,IAAI,CAAC,mBAAmB,CAAC,WAAW,GAAG,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,CAAC;oBAC5D,MAAM,gBAAgB,CAAC,WAAW,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;oBAC7D,MAAM,QAAQ,CAAC,MAAM,CAAC,yBAAyB,EAAE;wBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,mBAAmB,EAAE,IAAI,EAAE,CAAC,CAAC;wBACvD,WAAW,EAAE,kBAAkB;qBAChC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,oDAAoD;gBACpD,kBAAS,CAAC,KAAK,CAAC,4BAA4B,EAAE,KAAK,CAAC,CAAC;YACvD,CAAC;YAED,MAAM,uBAAuB,GAC3B,OAAO,CAAC,GAAG,CAAC,uBAAa,CAAC,uBAAuB,CAAC,EAAE,IAAI,EAAE,CAAC;YAE7D,IACE,eAAe;gBACf,CAAC,uBAAuB,KAAK,MAAM,IAAI,uBAAuB,KAAK,GAAG,CAAC,EACvE,CAAC;gBACD,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,kBAAS,CAAC,IAAI,CACZ,uDAAuD,CACxD,CAAC;oBACF,OAAO;gBACT,CAAC;gBAED,IAAI,CAAC;oBACH,MAAM,IAAA,sBAAQ,EAAC,SAAS,EAAE,QAAQ,EAAE,IAAI,CAAC,mBAAmB,CAAC,CAAC;gBAChE,CAAC;gBAAC,OAAO,aAAa,EAAE,CAAC;oBACvB,kBAAS,CAAC,KAAK,CAAC,qCAAqC,EAAE,aAAa,CAAC,CAAC;gBACxE,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;CACF,CAAC,CAAC"}

package/dist/lib/testExtension.d.ts

@@ -0,0 +1,10 @@
+import { DonobuExtendedPage } from './DonobuExtendedPage';
+import { GptClient } from '../clients/GptClient';
+import { BrowserStorageState } from '../models/BrowserStorageState';
+export * from 'playwright/test';
+export declare const test: import("@playwright/test").TestType<import("@playwright/test").PlaywrightTestArgs & import("@playwright/test").PlaywrightTestOptions & {
+    storageState?: BrowserStorageState | Promise<BrowserStorageState>;
+    gptClient?: GptClient;
+    page: DonobuExtendedPage;
+}, import("@playwright/test").PlaywrightWorkerArgs & import("@playwright/test").PlaywrightWorkerOptions>;
+//# sourceMappingURL=testExtension.d.ts.map

package/dist/lib/testExtension.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"testExtension.d.ts","sourceRoot":"","sources":["../../src/lib/testExtension.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AAmEjD,OAAO,EAAE,mBAAmB,EAAE,MAAM,+BAA+B,CAAC;AAEpE,cAAc,iBAAiB,CAAC;AAIhC,eAAO,MAAM,IAAI;mBACA,mBAAmB,GAAG,OAAO,CAAC,mBAAmB,CAAC;gBACrD,SAAS;UACf,kBAAkB;wGAqaxB,CAAC"}