donobu 2.16.0 → 2.17.1

package/dist/lib/DonobuExtendedPage.d.ts

@@ -0,0 +1,106 @@
+import type { Page } from '@playwright/test';
+import { ToolCallResult } from '../models/ToolCallResult';
+import { GptClient } from '../clients/GptClient';
+import { LanguageModel } from 'ai';
+import { AnalyzePageTextToolCoreParameters } from '../tools/AnalyzePageTextTool';
+import { JSONSchema7 } from 'json-schema';
+import { FlowMetadata } from '../models/FlowMetadata';
+import { AssertPageTextToolCoreParameters } from '../tools/AssertPageTextTool';
+import { AssertToolCoreParameters } from '../tools/AssertTool';
+import { SelectorBasedChooseSelectOptionToolParameters } from '../tools/ChooseSelectOptionTool';
+import { SelectorBasedClickToolParameters } from '../tools/ClickTool';
+import { SelectorBasedHoverOverElementToolParameters } from '../tools/HoverOverElementTool';
+import { SelectorBasedInputRandomizedEmailAddressToolParameters } from '../tools/InputRandomizedEmailAddressTool';
+import { SelectorBasedInputTextToolParameters } from '../tools/InputTextTool';
+import { SelectorBasedPressKeyToolParameters } from '../tools/PressKeyTool';
+import { ScrollPageToolCoreParameters } from '../tools/ScrollPageTool';
+/**
+ * Extends the Page object with additional test-related methods.
+ */
+export interface DonobuExtendedPage extends Page {
+    /**
+     * Run an arbitrary Donobu tool by name.
+     */
+    run(toolName: string, toolParams?: any, gptClientOverride?: GptClient | LanguageModel): Promise<ToolCallResult>;
+    /**
+     * Run an analysis on the current webpage's full raw text. Note that since this runs an analysis on the raw text,
+     * rather than, the HTML of the page, the raw text may be a bit jumbled, have its styling lost, careful
+     * positioning lost, etc. This tool is useful if there happens to be a lot of text on a page, and it would be
+     * tedious to scroll through the entirety of the page in order to get enough context to run an analysis.
+     *
+     * If there is context relevant to running the analysis that would not be found in the webpage text itself,
+     * perhaps context given in another message, prompt, overall object, etc, then provide it in the
+     * "additionalRelevantContext" parameter.
+     */
+    analyzePageText(params: AnalyzePageTextToolCoreParameters, gptClientOverride?: GptClient | LanguageModel): Promise<ToolCallResult>;
+    /**
+     * Assert that given text is visible in the current page.
+     */
+    assertPageText(params: AssertPageTextToolCoreParameters, gptClientOverride?: GptClient | LanguageModel): Promise<ToolCallResult>;
+    /**
+     * Checks all of the pages in the current browser context and returns the one
+     * matching the given URL.
+     */
+    changeTab(url: string): Promise<Page>;
+    /**
+     * Choose <option> values for a particular <select> HTML element.
+     */
+    chooseSelectOption(params: SelectorBasedChooseSelectOptionToolParameters): Promise<ToolCallResult>;
+    /**
+     * Click an element on a webpage.
+     */
+    clickElement(params: SelectorBasedClickToolParameters): Promise<ToolCallResult>;
+    /**
+     * Perform an analysis of the cookies for the current web browsing session.
+     */
+    createCookieReport(gptClientOverride?: GptClient | LanguageModel): Promise<ToolCallResult>;
+    /**
+     * Return an object conforming to the given JSON-schema. The object will be
+     * generated considering the following:
+     * - The given instruction (if any).
+     * - A screenshot of the current state of the page.
+     * - The history of actions taken on the page.
+     */
+    extract(params: {
+        instruction?: string;
+        schema: JSONSchema7;
+    }, gptClientOverride?: GptClient | LanguageModel): Promise<Record<string, unknown>>;
+    /**
+     * Hover the mouse over a specified element on a webpage.
+     */
+    hoverOverElement(params: SelectorBasedHoverOverElementToolParameters): Promise<ToolCallResult>;
+    /**
+     * Create a new randomized email address based on a given email and inputs it to a specific input text field.
+     * For example if passed "foo@gmail.com", this tool will create "foo+SOMETHING_RANDOM@gmail.com", where
+     * SOMETHING_RANDOM is a random string of characters safe for use in an email address, and input it in the
+     * specified text field.
+     */
+    inputRandomizedEmailAddress(params: SelectorBasedInputRandomizedEmailAddressToolParameters): Promise<ToolCallResult>;
+    /**
+     * Input text to a webpage's text input box.
+     */
+    inputText(params: SelectorBasedInputTextToolParameters): Promise<ToolCallResult>;
+    /**
+     * Press a single key. Note that if there is existing text in the element, it is appended rather than cleared.
+     *
+     * Generally, prefer using the 'inputText' tool instead as it automatically clears existing text
+     * and it allows the input of multiple characters at a time. The only advantage this tool has is
+     * that it allows the passing of control keys like "Tab", "Backspace", etc.
+     */
+    pressKey(params: SelectorBasedPressKeyToolParameters): Promise<ToolCallResult>;
+    /**
+     * Run an web accessibility (i.e. axe-core) test on the current webpage.
+     */
+    runAccessibilityTest(): Promise<ToolCallResult>;
+    /**
+     * Scroll the current page up or down by one viewport.
+     */
+    scroll(params: ScrollPageToolCoreParameters): Promise<ToolCallResult>;
+    /**
+     * Visually assert that a given condition (i.e. the 'assertionToTestFor' field) holds true, using a screenshot of
+     * the current webpage.
+     */
+    visuallyAssert(params: AssertToolCoreParameters, gptClientOverride?: GptClient | LanguageModel): Promise<ToolCallResult>;
+    _donobuFlowMetadata: FlowMetadata;
+}
+//# sourceMappingURL=DonobuExtendedPage.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"DonobuExtendedPage.d.ts","sourceRoot":"","sources":["../../src/lib/DonobuExtendedPage.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACnC,OAAO,EAAE,iCAAiC,EAAE,MAAM,8BAA8B,CAAC;AACjF,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,gCAAgC,EAAE,MAAM,6BAA6B,CAAC;AAC/E,OAAO,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC;AAC/D,OAAO,EAAE,6CAA6C,EAAE,MAAM,iCAAiC,CAAC;AAChG,OAAO,EAAE,gCAAgC,EAAE,MAAM,oBAAoB,CAAC;AACtE,OAAO,EAAE,2CAA2C,EAAE,MAAM,+BAA+B,CAAC;AAC5F,OAAO,EAAE,sDAAsD,EAAE,MAAM,0CAA0C,CAAC;AAClH,OAAO,EAAE,oCAAoC,EAAE,MAAM,wBAAwB,CAAC;AAC9E,OAAO,EAAE,mCAAmC,EAAE,MAAM,uBAAuB,CAAC;AAC5E,OAAO,EAAE,4BAA4B,EAAE,MAAM,yBAAyB,CAAC;AAEvE;;GAEG;AACH,MAAM,WAAW,kBAAmB,SAAQ,IAAI;IAC9C;;OAEG;IACH,GAAG,CACD,QAAQ,EAAE,MAAM,EAChB,UAAU,CAAC,EAAE,GAAG,EAChB,iBAAiB,CAAC,EAAE,SAAS,GAAG,aAAa,GAC5C,OAAO,CAAC,cAAc,CAAC,CAAC;IAC3B;;;;;;;;;OASG;IACH,eAAe,CACb,MAAM,EAAE,iCAAiC,EACzC,iBAAiB,CAAC,EAAE,SAAS,GAAG,aAAa,GAC5C,OAAO,CAAC,cAAc,CAAC,CAAC;IAC3B;;OAEG;IACH,cAAc,CACZ,MAAM,EAAE,gCAAgC,EACxC,iBAAiB,CAAC,EAAE,SAAS,GAAG,aAAa,GAC5C,OAAO,CAAC,cAAc,CAAC,CAAC;IAC3B;;;OAGG;IACH,SAAS,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACtC;;OAEG;IACH,kBAAkB,CAChB,MAAM,EAAE,6CAA6C,GACpD,OAAO,CAAC,cAAc,CAAC,CAAC;IAC3B;;OAEG;IACH,YAAY,CACV,MAAM,EAAE,gCAAgC,GACvC,OAAO,CAAC,cAAc,CAAC,CAAC;IAC3B;;OAEG;IACH,kBAAkB,CAChB,iBAAiB,CAAC,EAAE,SAAS,GAAG,aAAa,GAC5C,OAAO,CAAC,cAAc,CAAC,CAAC;IAC3B;;;;;;OAMG;IACH,OAAO,CACL,MAAM,EAAE;QACN,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,MAAM,EAAE,WAAW,CAAC;KACrB,EACD,iBAAiB,CAAC,EAAE,SAAS,GAAG,aAAa,GAC5C,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IACpC;;OAEG;IACH,gBAAgB,CACd,MAAM,EAAE,2CAA2C,GAClD,OAAO,CAAC,cAAc,CAAC,CAAC;IAC3B;;;;;OAKG;IACH,2BAA2B,CACzB,MAAM,EAAE,sDAAsD,GAC7D,OAAO,CAAC,cAAc,CAAC,CAAC;IAC3B;;OAEG;IACH,SAAS,CACP,MAAM,EAAE,oCAAoC,GAC3C,OAAO,CAAC,cAAc,CAAC,CAAC;IAC3B;;;;;;OAMG;IACH,QAAQ,CACN,MAAM,EAAE,mCAAmC,GAC1C,OAAO,CAAC,cAAc,CAAC,CAAC;IAC3B;;OAEG;IACH,oBAAoB,IAAI,OAAO,CAAC,cAAc,CAAC,CAAC;IAChD;;OAEG;IACH,MAAM,CAAC,MAAM,EAAE,4BAA4B,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;IACtE;;;OAGG;IACH,cAAc,CACZ,MAAM,EAAE,wBAAwB,EAChC,iBAAiB,CAAC,EAAE,SAAS,GAAG,aAAa,GAC5C,OAAO,CAAC,cAAc,CAAC,CAAC;IAE3B,mBAAmB,EAAE,YAAY,CAAC;CACnC"}

package/dist/lib/DonobuExtendedPage.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=DonobuExtendedPage.js.map

package/dist/lib/DonobuExtendedPage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"DonobuExtendedPage.js","sourceRoot":"","sources":["../../src/lib/DonobuExtendedPage.ts"],"names":[],"mappings":""}

package/dist/lib/constants.d.ts

@@ -0,0 +1,8 @@
+export declare const PLACEHOLDER_FLOW_URL = "https://example.com";
+export declare const DEFAULT_GPT_MODEL_FOR_PLATFORM: {
+    ANTHROPIC: string;
+    ANTHROPIC_AWS_BEDROCK: string | undefined;
+    GOOGLE_GEMINI: string;
+    OPENAI: string;
+};
+//# sourceMappingURL=constants.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"constants.d.ts","sourceRoot":"","sources":["../../src/lib/constants.ts"],"names":[],"mappings":"AAEA,eAAO,MAAM,oBAAoB,wBAAwB,CAAC;AAC1D,eAAO,MAAM,8BAA8B;;;;;CAU1C,CAAC"}

package/dist/lib/constants.js

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DEFAULT_GPT_MODEL_FOR_PLATFORM = exports.PLACEHOLDER_FLOW_URL = void 0;
+const envVars_1 = require("../envVars");
+exports.PLACEHOLDER_FLOW_URL = 'https://example.com';
+exports.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',
+};
+//# sourceMappingURL=constants.js.map

package/dist/lib/constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../src/lib/constants.ts"],"names":[],"mappings":";;;AAAA,wCAA2C;AAE9B,QAAA,oBAAoB,GAAG,qBAAqB,CAAC;AAC7C,QAAA,8BAA8B,GAAG;IAC5C,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"}

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

@@ -0,0 +1,6 @@
+import { TestFixture } from '@playwright/test';
+import { BrowserStorageState } from '../../models/BrowserStorageState';
+import { BrowserStateReference } from '../../models/BrowserStateFlowReference';
+export declare function getBrowserStorageState(flowIdOrName: BrowserStateReference): Promise<BrowserStorageState>;
+export declare function getBrowserStorageStateFixture(flowIdOrName: BrowserStateReference): TestFixture<BrowserStorageState, any>;
+//# sourceMappingURL=browserState.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"browserState.d.ts","sourceRoot":"","sources":["../../../src/lib/fixtures/browserState.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AACvE,OAAO,EAAE,qBAAqB,EAAE,MAAM,wCAAwC,CAAC;AAG/E,wBAAsB,sBAAsB,CAC1C,YAAY,EAAE,qBAAqB,GAClC,OAAO,CAAC,mBAAmB,CAAC,CAI9B;AAED,wBAAgB,6BAA6B,CAC3C,YAAY,EAAE,qBAAqB,GAClC,WAAW,CAAC,mBAAmB,EAAE,GAAG,CAAC,CAOvC"}

package/dist/lib/fixtures/browserState.js

@@ -0,0 +1,15 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getBrowserStorageState = getBrowserStorageState;
+exports.getBrowserStorageStateFixture = getBrowserStorageStateFixture;
+const donobuTestStack_1 = require("../utils/donobuTestStack");
+async function getBrowserStorageState(flowIdOrName) {
+    return (await (0, donobuTestStack_1.getOrCreateDonobuStack)()).flowsManager.getBrowserStorageState(flowIdOrName);
+}
+function getBrowserStorageStateFixture(flowIdOrName) {
+    return async ({}, use) => {
+        const storage = await (await (0, donobuTestStack_1.getOrCreateDonobuStack)()).flowsManager.getBrowserStorageState(flowIdOrName);
+        await use(storage);
+    };
+}
+//# sourceMappingURL=browserState.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"browserState.js","sourceRoot":"","sources":["../../../src/lib/fixtures/browserState.ts"],"names":[],"mappings":";;AAKA,wDAMC;AAED,sEASC;AAnBD,8DAAkE;AAE3D,KAAK,UAAU,sBAAsB,CAC1C,YAAmC;IAEnC,OAAO,CAAC,MAAM,IAAA,wCAAsB,GAAE,CAAC,CAAC,YAAY,CAAC,sBAAsB,CACzE,YAAY,CACb,CAAC;AACJ,CAAC;AAED,SAAgB,6BAA6B,CAC3C,YAAmC;IAEnC,OAAO,KAAK,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE;QACvB,MAAM,OAAO,GAAG,MAAM,CACpB,MAAM,IAAA,wCAAsB,GAAE,CAC/B,CAAC,YAAY,CAAC,sBAAsB,CAAC,YAAY,CAAC,CAAC;QACpD,MAAM,GAAG,CAAC,OAAO,CAAC,CAAC;IACrB,CAAC,CAAC;AACJ,CAAC"}

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

@@ -0,0 +1,40 @@
+import { TestFixture } from '@playwright/test';
+import { GptClient } from '../../clients/GptClient';
+import { LanguageModel } from 'ai';
+export declare function getOrCreateDefaultGptClient(): Promise<GptClient>;
+/**
+ * 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.
+ */
+export declare function createGptClient(modelName?: string): Promise<GptClient>;
+export declare function anthropicClient(modelName: string, apiKey?: string): Promise<GptClient>;
+export declare function anthropicClientFixture(modelName: string, apiKey?: string): TestFixture<GptClient, any>;
+export declare function anthropicAwsBedrockClient(modelName: string, region?: string, accessKeyId?: string, secretAccessKey?: string): Promise<GptClient>;
+export declare function anthropicAwsBedrockClientFixture(modelName: string, region?: string, accessKeyId?: string, secretAccessKey?: string): TestFixture<GptClient, any>;
+export declare function googleGeminiClient(modelName: string, apiKey?: string): Promise<GptClient>;
+export declare function googleGeminiClientFixture(modelName: string, apiKey?: string): TestFixture<GptClient, any>;
+export declare function openAiClient(modelName: string, apiKey?: string): Promise<GptClient>;
+export declare function openAiClientFixture(modelName: string, apiKey?: string): TestFixture<GptClient, any>;
+export declare function vercelAiClientFixture(model: LanguageModel): TestFixture<GptClient, any>;
+export declare function gptClientFixture(modelName?: string): TestFixture<GptClient, any>;
+//# sourceMappingURL=gptClients.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"gptClients.d.ts","sourceRoot":"","sources":["../../../src/lib/fixtures/gptClients.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AAGpD,OAAO,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AAkBnC,wBAAsB,2BAA2B,IAAI,OAAO,CAAC,SAAS,CAAC,CAOtE;AAED;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,wBAAsB,eAAe,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,CAiD5E;AAED,wBAAsB,eAAe,CACnC,SAAS,EAAE,MAAM,EACjB,MAAM,CAAC,EAAE,MAAM,GACd,OAAO,CAAC,SAAS,CAAC,CAcpB;AAED,wBAAgB,sBAAsB,CACpC,SAAS,EAAE,MAAM,EACjB,MAAM,CAAC,EAAE,MAAM,GACd,WAAW,CAAC,SAAS,EAAE,GAAG,CAAC,CAK7B;AAED,wBAAsB,yBAAyB,CAC7C,SAAS,EAAE,MAAM,EACjB,MAAM,CAAC,EAAE,MAAM,EACf,WAAW,CAAC,EAAE,MAAM,EACpB,eAAe,CAAC,EAAE,MAAM,GACvB,OAAO,CAAC,SAAS,CAAC,CAmBpB;AAED,wBAAgB,gCAAgC,CAC9C,SAAS,EAAE,MAAM,EACjB,MAAM,CAAC,EAAE,MAAM,EACf,WAAW,CAAC,EAAE,MAAM,EACpB,eAAe,CAAC,EAAE,MAAM,GACvB,WAAW,CAAC,SAAS,EAAE,GAAG,CAAC,CAU7B;AAED,wBAAsB,kBAAkB,CACtC,SAAS,EAAE,MAAM,EACjB,MAAM,CAAC,EAAE,MAAM,GACd,OAAO,CAAC,SAAS,CAAC,CAepB;AAED,wBAAgB,yBAAyB,CACvC,SAAS,EAAE,MAAM,EACjB,MAAM,CAAC,EAAE,MAAM,GACd,WAAW,CAAC,SAAS,EAAE,GAAG,CAAC,CAK7B;AAED,wBAAsB,YAAY,CAChC,SAAS,EAAE,MAAM,EACjB,MAAM,CAAC,EAAE,MAAM,GACd,OAAO,CAAC,SAAS,CAAC,CAcpB;AAED,wBAAgB,mBAAmB,CACjC,SAAS,EAAE,MAAM,EACjB,MAAM,CAAC,EAAE,MAAM,GACd,WAAW,CAAC,SAAS,EAAE,GAAG,CAAC,CAK7B;AAED,wBAAgB,qBAAqB,CACnC,KAAK,EAAE,aAAa,GACnB,WAAW,CAAC,SAAS,EAAE,GAAG,CAAC,CAK7B;AAED,wBAAgB,gBAAgB,CAC9B,SAAS,CAAC,EAAE,MAAM,GACjB,WAAW,CAAC,SAAS,EAAE,GAAG,CAAC,CAK7B"}

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"}