donobu 2.12.2 → 2.13.0

package/dist/esm/utils/TemplateInterpolator.js

@@ -0,0 +1,206 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TEMPLATE_DATA_PATTERN = void 0;
+exports.resolveExpression = resolveExpression;
+exports.interpolateString = interpolateString;
+exports.interpolateObject = interpolateObject;
+const Logger_1 = require("./Logger");
+const jsonpath_1 = __importDefault(require("jsonpath"));
+/**
+ * Regular expression to match template JSON-path expressions inside of {{...}}
+ */
+exports.TEMPLATE_DATA_PATTERN = /\{\{([^}]+)\}\}/g;
+/**
+ * Regular expression to detect expressions that need to be handled specially
+ * Matches patterns like: $..path[?(@.condition)][0] or $..path[?(@.condition)][index] or $..path[?(@.condition)][n:m]
+ */
+const FILTER_WITH_INDEX_PATTERN = /(\[\?\(.*\)\])(\[-?\d+(\:\d+)?\])/;
+const STANDALONE_NEGATIVE_INDEX_PATTERN = /(\$\.calls)\[(-\d+)\](?!\:)/;
+/**
+ * Parse and modify JSONPath expressions to handle indexing after filter expressions
+ * @param expression - The JSONPath expression to process
+ * @returns A processed expression that works with the jsonpath library
+ */
+function processJSONPathExpression(expression) {
+    // First check for filter followed by index pattern
+    const filterMatch = expression.match(FILTER_WITH_INDEX_PATTERN);
+    if (filterMatch) {
+        // Extract just the number from indexPart (remove the brackets)
+        const indexMatch = filterMatch[2].match(/\[(-?\d+)(\:\d+)?\]/);
+        if (indexMatch) {
+            // If it's a simple index (not a slice)
+            if (!indexMatch[2]) {
+                // Remove the index part from the expression
+                const baseExpression = expression.replace(filterMatch[2], '');
+                // Return the modified expression
+                return baseExpression;
+            }
+        }
+    }
+    // Check for standalone negative index pattern and convert it to slice notation
+    // since the jsonpath library doesn't support negative indices directly.
+    const negativeIndexMatch = expression.match(STANDALONE_NEGATIVE_INDEX_PATTERN);
+    if (negativeIndexMatch && negativeIndexMatch[2].startsWith('-')) {
+        // Convert the standalone negative index to slice notation
+        // For example: $.calls[-1] becomes $.calls[-1:]
+        const convertedExpression = expression.replace(`[${negativeIndexMatch[2]}]`, `[${negativeIndexMatch[2]}:]`);
+        return convertedExpression;
+    }
+    // If it doesn't match our patterns, return the original expression
+    return expression;
+}
+/**
+ * Format a value to string with special handling for objects and arrays
+ * @param value - The value to format
+ * @returns A string representation of the value
+ */
+function formatValue(value) {
+    if (value === undefined || value === null) {
+        return '';
+    }
+    // Handle arrays by joining their string representations
+    if (Array.isArray(value)) {
+        return value.map(formatValue).join(',');
+    }
+    // Handle objects with proper JSON formatting
+    if (typeof value === 'object' && value !== null) {
+        return JSON.stringify(value);
+    }
+    // Default case
+    return String(value);
+}
+/**
+ * Resolve a JSONPath expression against the context
+ * @param expression - The JSONPath expression to evaluate
+ * @param dataSource - The data context to evaluate against
+ * @returns The resolved value or undefined if not found
+ */
+function resolveExpression(expression, dataSource) {
+    try {
+        // Process the expression to handle special cases
+        const processedExpression = processJSONPathExpression(expression);
+        // Handle expressions with wildcards or recursive searches that might return arrays
+        const isWildcardOrRecursive = expression.includes('..') ||
+            expression.includes('*') ||
+            expression.includes('[:]');
+        // Use JSONPath to evaluate the expression against the context
+        const results = jsonpath_1.default.query(dataSource, processedExpression);
+        // Handle the indexing ourselves if needed
+        if (expression !== processedExpression) {
+            const match = expression.match(FILTER_WITH_INDEX_PATTERN);
+            if (match) {
+                const indexMatch = match[2].match(/\[(-?\d+)(\:\d+)?\]/);
+                if (indexMatch) {
+                    const index = parseInt(indexMatch[1], 10);
+                    // If we have results, handle both positive and negative indices
+                    if (results && results.length > 0) {
+                        // For negative indices, count from the end of the array
+                        const actualIndex = index < 0 ? results.length + index : index;
+                        // Return the result if the calculated index is valid
+                        if (actualIndex >= 0 && actualIndex < results.length) {
+                            return results[actualIndex];
+                        }
+                    }
+                    return undefined;
+                }
+            }
+        }
+        // Return the entire array for expressions that clearly want multiple values
+        if (isWildcardOrRecursive && results.length > 1) {
+            return results;
+        }
+        // Default case - return the first result if available
+        if (results && results.length > 0) {
+            return results[0];
+        }
+        return undefined;
+    }
+    catch (error) {
+        throw new Error(`Failed to evaluate JSONPath expression: ${expression}. Error: ${error}`);
+    }
+}
+/**
+ * Interpolate template strings using JSONPath expressions
+ * @param template - The template string with {{...}} expressions
+ * @param dataSource - The data context to interpolate from
+ * @returns The interpolated string
+ */
+function interpolateString(template, dataSource) {
+    return template.replace(exports.TEMPLATE_DATA_PATTERN, (match, expression) => {
+        try {
+            const value = resolveExpression(expression.trim(), dataSource);
+            // If the value is undefined or null, keep the original expression
+            if (value === undefined) {
+                return match; // Return the original {{expression}} string
+            }
+            return formatValue(value);
+        }
+        catch (error) {
+            Logger_1.appLogger.warn(`Error interpolating template: ${template}, expression: ${expression}, error: ${error}`);
+            return match; // Return the original {{expression}} string on error
+        }
+    });
+}
+/**
+ * Interpolate the parameters object using JSONPath expressions. Any string values
+ * and keys in the object will be replaced with the corresponding values from the data source.
+ * Nested objects and arrays will be processed recursively.
+ *
+ * @param parameters - The original parameters object
+ * @param dataSource - The data context to interpolate from
+ * @returns A new parameters object with interpolated string values and keys
+ */
+function interpolateObject(parameters, dataSource) {
+    if (!parameters || typeof parameters !== 'object') {
+        return parameters;
+    }
+    // Create appropriate result container
+    const result = Array.isArray(parameters) ? [] : {};
+    // Process each key-value pair
+    if (Array.isArray(parameters)) {
+        // Handle array case
+        for (let i = 0; i < parameters.length; i++) {
+            const value = parameters[i];
+            if (typeof value === 'string') {
+                // Interpolate string values
+                result[i] = interpolateString(value, dataSource);
+            }
+            else if (typeof value === 'object' && value !== null) {
+                // Recursively process nested objects
+                result[i] = interpolateObject(value, dataSource);
+            }
+            else {
+                // Keep non-string values as is
+                result[i] = value;
+            }
+        }
+    }
+    else {
+        // Handle object case
+        for (const key in parameters) {
+            const value = parameters[key];
+            // Interpolate the key if it's a string and contains a template pattern
+            let processedKey = key;
+            if (typeof key === 'string' && exports.TEMPLATE_DATA_PATTERN.test(key)) {
+                processedKey = interpolateString(key, dataSource);
+            }
+            if (typeof value === 'string') {
+                // Interpolate string values
+                result[processedKey] = interpolateString(value, dataSource);
+            }
+            else if (typeof value === 'object' && value !== null) {
+                // Recursively process nested objects
+                result[processedKey] = interpolateObject(value, dataSource);
+            }
+            else {
+                // Keep non-string values as is
+                result[processedKey] = value;
+            }
+        }
+    }
+    return result;
+}
+//# sourceMappingURL=TemplateInterpolator.js.map

package/dist/esm/utils/TemplateInterpolator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"TemplateInterpolator.js","sourceRoot":"","sources":["../../../src/utils/TemplateInterpolator.ts"],"names":[],"mappings":";;;;;;AAyFA,8CAsDC;AAQD,8CAqBC;AAWD,8CAoDC;AA3OD,qCAAqC;AACrC,wDAAgC;AAEhC;;GAEG;AACU,QAAA,qBAAqB,GAAG,kBAAkB,CAAC;AAExD;;;GAGG;AACH,MAAM,yBAAyB,GAAG,mCAAmC,CAAC;AACtE,MAAM,iCAAiC,GAAG,6BAA6B,CAAC;AAExE;;;;GAIG;AACH,SAAS,yBAAyB,CAAC,UAAkB;IACnD,mDAAmD;IACnD,MAAM,WAAW,GAAG,UAAU,CAAC,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAEhE,IAAI,WAAW,EAAE,CAAC;QAChB,+DAA+D;QAC/D,MAAM,UAAU,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;QAE/D,IAAI,UAAU,EAAE,CAAC;YACf,uCAAuC;YACvC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;gBACnB,4CAA4C;gBAC5C,MAAM,cAAc,GAAG,UAAU,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC9D,iCAAiC;gBACjC,OAAO,cAAc,CAAC;YACxB,CAAC;QACH,CAAC;IACH,CAAC;IAED,+EAA+E;IAC/E,wEAAwE;IACxE,MAAM,kBAAkB,GAAG,UAAU,CAAC,KAAK,CACzC,iCAAiC,CAClC,CAAC;IAEF,IAAI,kBAAkB,IAAI,kBAAkB,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QAChE,0DAA0D;QAC1D,gDAAgD;QAChD,MAAM,mBAAmB,GAAG,UAAU,CAAC,OAAO,CAC5C,IAAI,kBAAkB,CAAC,CAAC,CAAC,GAAG,EAC5B,IAAI,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAC9B,CAAC;QACF,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IAED,mEAAmE;IACnE,OAAO,UAAU,CAAC;AACpB,CAAC;AAED;;;;GAIG;AACH,SAAS,WAAW,CAAC,KAAU;IAC7B,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QAC1C,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,wDAAwD;IACxD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC1C,CAAC;IAED,6CAA6C;IAC7C,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QAChD,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;IAED,eAAe;IACf,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;AACvB,CAAC;AAED;;;;;GAKG;AACH,SAAgB,iBAAiB,CAC/B,UAAkB,EAClB,UAAmC;IAEnC,IAAI,CAAC;QACH,iDAAiD;QACjD,MAAM,mBAAmB,GAAG,yBAAyB,CAAC,UAAU,CAAC,CAAC;QAElE,mFAAmF;QACnF,MAAM,qBAAqB,GACzB,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC;YACzB,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC;YACxB,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAE7B,8DAA8D;QAC9D,MAAM,OAAO,GAAG,kBAAQ,CAAC,KAAK,CAAC,UAAU,EAAE,mBAAmB,CAAC,CAAC;QAEhE,0CAA0C;QAC1C,IAAI,UAAU,KAAK,mBAAmB,EAAE,CAAC;YACvC,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,yBAAyB,CAAC,CAAC;YAC1D,IAAI,KAAK,EAAE,CAAC;gBACV,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;gBACzD,IAAI,UAAU,EAAE,CAAC;oBACf,MAAM,KAAK,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBAC1C,gEAAgE;oBAChE,IAAI,OAAO,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAClC,wDAAwD;wBACxD,MAAM,WAAW,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC;wBAC/D,qDAAqD;wBACrD,IAAI,WAAW,IAAI,CAAC,IAAI,WAAW,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;4BACrD,OAAO,OAAO,CAAC,WAAW,CAAC,CAAC;wBAC9B,CAAC;oBACH,CAAC;oBACD,OAAO,SAAS,CAAC;gBACnB,CAAC;YACH,CAAC;QACH,CAAC;QAED,4EAA4E;QAC5E,IAAI,qBAAqB,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChD,OAAO,OAAO,CAAC;QACjB,CAAC;QAED,sDAAsD;QACtD,IAAI,OAAO,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAClC,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC;QACpB,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CACb,2CAA2C,UAAU,YAAY,KAAK,EAAE,CACzE,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;GAKG;AACH,SAAgB,iBAAiB,CAC/B,QAAgB,EAChB,UAAmC;IAEnC,OAAO,QAAQ,CAAC,OAAO,CAAC,6BAAqB,EAAE,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE;QACnE,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,iBAAiB,CAAC,UAAU,CAAC,IAAI,EAAE,EAAE,UAAU,CAAC,CAAC;YAE/D,kEAAkE;YAClE,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,OAAO,KAAK,CAAC,CAAC,4CAA4C;YAC5D,CAAC;YAED,OAAO,WAAW,CAAC,KAAK,CAAC,CAAC;QAC5B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,kBAAS,CAAC,IAAI,CACZ,iCAAiC,QAAQ,iBAAiB,UAAU,YAAY,KAAK,EAAE,CACxF,CAAC;YACF,OAAO,KAAK,CAAC,CAAC,qDAAqD;QACrE,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;;;;;GAQG;AACH,SAAgB,iBAAiB,CAC/B,UAAe,EACf,UAAmC;IAEnC,IAAI,CAAC,UAAU,IAAI,OAAO,UAAU,KAAK,QAAQ,EAAE,CAAC;QAClD,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,sCAAsC;IACtC,MAAM,MAAM,GAAQ,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAExD,8BAA8B;IAC9B,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QAC9B,oBAAoB;QACpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3C,MAAM,KAAK,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;YAC5B,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gBAC9B,4BAA4B;gBAC5B,MAAM,CAAC,CAAC,CAAC,GAAG,iBAAiB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;YACnD,CAAC;iBAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;gBACvD,qCAAqC;gBACrC,MAAM,CAAC,CAAC,CAAC,GAAG,iBAAiB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;YACnD,CAAC;iBAAM,CAAC;gBACN,+BAA+B;gBAC/B,MAAM,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC;YACpB,CAAC;QACH,CAAC;IACH,CAAC;SAAM,CAAC;QACN,qBAAqB;QACrB,KAAK,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;YAC7B,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;YAE9B,uEAAuE;YACvE,IAAI,YAAY,GAAG,GAAG,CAAC;YACvB,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,6BAAqB,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC/D,YAAY,GAAG,iBAAiB,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;YACpD,CAAC;YAED,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gBAC9B,4BAA4B;gBAC5B,MAAM,CAAC,YAAY,CAAC,GAAG,iBAAiB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;YAC9D,CAAC;iBAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;gBACvD,qCAAqC;gBACrC,MAAM,CAAC,YAAY,CAAC,GAAG,iBAAiB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;YAC9D,CAAC;iBAAM,CAAC;gBACN,+BAA+B;gBAC/B,MAAM,CAAC,YAAY,CAAC,GAAG,KAAK,CAAC;YAC/B,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/managers/CodeGenerator.d.ts

@@ -1,11 +1,6 @@
 import { FlowMetadata } from '../models/FlowMetadata';
 import { ProposedToolCall } from '../models/ProposedToolCall';
-import { AgentsManager } from './AgentsManager';
-import { GptConfigsManager } from './GptConfigsManager';
 export declare class CodeGenerator {
-    private readonly gptConfigsManager;
-    private readonly agentsManager;
-    constructor(gptConfigsManager: GptConfigsManager, agentsManager: AgentsManager);
     /** Creates a Node.js Microsoft Playwright script to replay the given flow. */
     getFlowAsPlaywrightScript(flowMetadata: FlowMetadata, toolCalls: ProposedToolCall<any>[]): Promise<string>;
     /**

package/dist/managers/CodeGenerator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"CodeGenerator.d.ts","sourceRoot":"","sources":["../../src/managers/CodeGenerator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AAEtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAuB9D,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAGxD,qBAAa,aAAa;IAEtB,OAAO,CAAC,QAAQ,CAAC,iBAAiB;IAClC,OAAO,CAAC,QAAQ,CAAC,aAAa;gBADb,iBAAiB,EAAE,iBAAiB,EACpC,aAAa,EAAE,aAAa;IAG/C,8EAA8E;IACjE,yBAAyB,CACpC,YAAY,EAAE,YAAY,EAC1B,SAAS,EAAE,gBAAgB,CAAC,GAAG,CAAC,EAAE,GACjC,OAAO,CAAC,MAAM,CAAC;IAsIlB;;;OAGG;WACW,uCAAuC,CACnD,gBAAgB,EAAE,gBAAgB,CAAC,GAAG,CAAC,GACtC,MAAM;YAmEK,YAAY;IAY1B;;;;;;OAMG;IACH,OAAO,CAAC,0BAA0B;CASnC"}
+{"version":3,"file":"CodeGenerator.d.ts","sourceRoot":"","sources":["../../src/managers/CodeGenerator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAyB9D,qBAAa,aAAa;IACxB,8EAA8E;IACjE,yBAAyB,CACpC,YAAY,EAAE,YAAY,EAC1B,SAAS,EAAE,gBAAgB,CAAC,GAAG,CAAC,EAAE,GACjC,OAAO,CAAC,MAAM,CAAC;IA0FlB;;;OAGG;WACW,uCAAuC,CACnD,gBAAgB,EAAE,gBAAgB,CAAC,GAAG,CAAC,GACtC,MAAM;YAmEK,YAAY;IAY1B;;;;;;OAMG;IACH,OAAO,CAAC,0BAA0B;CASnC"}

package/dist/managers/CodeGenerator.js

@@ -25,10 +25,6 @@ const WaitTool_1 = require("../tools/WaitTool");
 const JsonUtils_1 = require("../utils/JsonUtils");
 const ToolManager_1 = require("./ToolManager");
 class CodeGenerator {
-    constructor(gptConfigsManager, agentsManager) {
-        this.gptConfigsManager = gptConfigsManager;
-        this.agentsManager = agentsManager;
-    }
     /** Creates a Node.js Microsoft Playwright script to replay the given flow. */
     async getFlowAsPlaywrightScript(flowMetadata, toolCalls) {
         // These tools are not supported in the generated script because they have
@@ -51,38 +47,11 @@ class CodeGenerator {
                 .filter((tool) => {
                 return tool?.requiresGpt;
             }).length > 0;
-        let gptConfig = null;
-        // Check if the flow metadata has a specific GPT config name.
-        // If it does, try to get it from the GptConfigsManager.
-        // If it doesn't, try to get the default GPT config from the AgentsManager.
-        //
-        // This is done so that we can generate code using the preferred GPT client.
-        if (isGptClientRequired) {
-            try {
-                if (flowMetadata.gptConfigName) {
-                    gptConfig = await this.gptConfigsManager.get(flowMetadata.gptConfigName);
-                }
-            }
-            catch (_error) {
-                // This is expected if the config does not exist.
-            }
-            if (!gptConfig) {
-                const gptConfigName = await this.agentsManager.get('flow-runner');
-                if (gptConfigName) {
-                    try {
-                        gptConfig = await this.gptConfigsManager.get(gptConfigName);
-                    }
-                    catch (_error) {
-                        // This is expected if the config does not exist.
-                    }
-                }
-            }
-        }
         const gptSetupNote = isGptClientRequired
             ? ` *
  * Also, since this test is using tools that require the usage of an LLM, be
  * sure to have an appropriate LLM API key available as an environment variable
- * (e.g. OPENAI_API_KEY, ANTHROPIC_API_KEY, etc).
+ * (e.g. OPENAI_API_KEY, ANTHROPIC_API_KEY, or GOOGLE_GENERATIVE_AI_API_KEY).
  `
             : ' ';
         const hasObjective = (flowMetadata.overallObjective?.trim().length ?? 0) > 0;
@@ -108,22 +77,9 @@ class CodeGenerator {
         const useStorageState = flowMetadata.browser && flowMetadata.browser.initialState
             ? `storageState: getBrowserStorageStateFixture(${JSON.stringify(flowMetadata.browser.initialState, null, 2)}),`
             : undefined;
-        let useGptClientImport = undefined;
-        let useGptClient = undefined;
-        if (isGptClientRequired) {
-            if (gptConfig && gptConfig.type === 'ANTHROPIC_AWS_BEDROCK') {
-                useGptClientImport =
-                    "import { anthropicAwsBedrockClientFixture } from 'donobu';";
-                useGptClient = `gptClient: anthropicAwsBedrockClientFixture('${gptConfig.modelName}', '${gptConfig.region}'),`;
-            }
-            else {
-                useGptClientImport = "import { gptClientFixture } from 'donobu';";
-                useGptClient = `gptClient: gptClientFixture(),`;
-            }
-        }
         let testExtension = '';
-        if (useStorageState || useGptClient) {
-            testExtension = `.extend({${useStorageState ?? ''}${useGptClient ?? ''}})`;
+        if (useStorageState) {
+            testExtension = `.extend({${useStorageState}})`;
         }
         const scriptedToolCalls = toolCalls
             .filter((toolCall) => !unsupportedToolsByName.has(toolCall.name))
@@ -143,7 +99,7 @@ testInfo.attach('extracted-object', { body: JSON.stringify(extractedObject), con
  * Also, be sure that Playwright's browsers are installed...
  *    'npx playwright install'
 ${gptSetupNote}*/
-import { test } from 'donobu';${useGptClientImport ? '\n' + useGptClientImport : ''}${useStorageState ? "\nimport { getBrowserStorageStateFixture } from 'donobu';" : ''}
+import { test } from 'donobu';${useStorageState ? "\nimport { getBrowserStorageStateFixture } from 'donobu';" : ''}
 
 const testTitle = '${testName}';
 const testDetails = ${testDetails};

package/dist/managers/CodeGenerator.js.map

@@ -1 +1 @@
-{"version":3,"file":"CodeGenerator.js","sourceRoot":"","sources":["../../src/managers/CodeGenerator.ts"],"names":[],"mappings":";;;AAGA,sEAAmE;AACnE,oEAAiE;AACjE,oDAAiD;AACjD,8EAA2E;AAC3E,4EAAyE;AACzE,kDAA+C;AAC/C,0FAAuF;AACvF,sEAAmE;AACnE,8DAA2D;AAC3D,wEAAqE;AACrE,8FAA2F;AAC3F,0DAAuD;AACvD,8DAA2D;AAC3D,kFAA+E;AAC/E,8FAA2F;AAC3F,wDAAqD;AACrD,4DAAyD;AACzD,gFAA6E;AAC7E,4DAAyD;AACzD,4EAAyE;AACzE,gDAA6C;AAC7C,kDAA+C;AAG/C,+CAA4C;AAE5C,MAAa,aAAa;IACxB,YACmB,iBAAoC,EACpC,aAA4B;QAD5B,sBAAiB,GAAjB,iBAAiB,CAAmB;QACpC,kBAAa,GAAb,aAAa,CAAe;IAC5C,CAAC;IAEJ,8EAA8E;IACvE,KAAK,CAAC,yBAAyB,CACpC,YAA0B,EAC1B,SAAkC;QAElC,0EAA0E;QAC1E,qEAAqE;QACrE,kBAAkB;QAClB,MAAM,sBAAsB,GAAG,IAAI,GAAG,CAAC;YACrC,iCAAe,CAAC,IAAI;YACpB,qDAAyB,CAAC,IAAI;YAC9B,iEAA+B,CAAC,IAAI;YACpC,+CAAsB,CAAC,IAAI;SAC5B,CAAC,CAAC;QACH,MAAM,mBAAmB,GACvB,YAAY,CAAC,gBAAgB;YAC7B,SAAS;iBACN,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAChB,OAAO,QAAQ,CAAC,IAAI,CAAC;YACvB,CAAC,CAAC;iBACD,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE;gBACpB,OAAO,yBAAW,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC;YACpE,CAAC,CAAC;iBACD,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE;gBACf,OAAO,IAAI,EAAE,WAAW,CAAC;YAC3B,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;QAClB,IAAI,SAAS,GAAqB,IAAI,CAAC;QACvC,6DAA6D;QAC7D,wDAAwD;QACxD,2EAA2E;QAC3E,EAAE;QACF,4EAA4E;QAC5E,IAAI,mBAAmB,EAAE,CAAC;YACxB,IAAI,CAAC;gBACH,IAAI,YAAY,CAAC,aAAa,EAAE,CAAC;oBAC/B,SAAS,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAC1C,YAAY,CAAC,aAAa,CAC3B,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,OAAO,MAAM,EAAE,CAAC;gBAChB,iDAAiD;YACnD,CAAC;YAED,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;gBAElE,IAAI,aAAa,EAAE,CAAC;oBAClB,IAAI,CAAC;wBACH,SAAS,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;oBAC9D,CAAC;oBAAC,OAAO,MAAM,EAAE,CAAC;wBAChB,iDAAiD;oBACnD,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,YAAY,GAAG,mBAAmB;YACtC,CAAC,CAAC;;;;EAIN;YACI,CAAC,CAAC,GAAG,CAAC;QACR,MAAM,YAAY,GAChB,CAAC,YAAY,CAAC,gBAAgB,EAAE,IAAI,EAAE,CAAC,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QAC1D,MAAM,WAAW,GAAG,YAAY;YAC9B,CAAC,CAAC;;;2BAGmB,IAAI,CAAC,0BAA0B,CAAC,YAAY,CAAC,gBAAiB,CAAC;;UAEhF;YACJ,CAAC,CAAC,IAAI,CAAC;QACT,MAAM,QAAQ,GAAG,YAAY,CAAC,IAAI;YAChC,CAAC,CAAC,YAAY,CAAC,IAAI;gBACf,4BAA4B;iBAC3B,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC;gBACvB,wBAAwB;iBACvB,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC;gBACrB,mBAAmB;iBAClB,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC;gBACtB,2BAA2B;iBAC1B,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC;YAC1B,CAAC,CAAC,YAAY,YAAY,CAAC,aAAa,EAAE,CAAC;QAC7C,MAAM,eAAe,GACnB,YAAY,CAAC,OAAO,IAAI,YAAY,CAAC,OAAO,CAAC,YAAY;YACvD,CAAC,CAAC,+CAA+C,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI;YAC/G,CAAC,CAAC,SAAS,CAAC;QAChB,IAAI,kBAAkB,GAAuB,SAAS,CAAC;QACvD,IAAI,YAAY,GAAuB,SAAS,CAAC;QAEjD,IAAI,mBAAmB,EAAE,CAAC;YACxB,IAAI,SAAS,IAAI,SAAS,CAAC,IAAI,KAAK,uBAAuB,EAAE,CAAC;gBAC5D,kBAAkB;oBAChB,4DAA4D,CAAC;gBAC/D,YAAY,GAAG,gDAAgD,SAAS,CAAC,SAAS,OAAO,SAAS,CAAC,MAAM,KAAK,CAAC;YACjH,CAAC;iBAAM,CAAC;gBACN,kBAAkB,GAAG,4CAA4C,CAAC;gBAClE,YAAY,GAAG,gCAAgC,CAAC;YAClD,CAAC;QACH,CAAC;QAED,IAAI,aAAa,GAAG,EAAE,CAAC;QAEvB,IAAI,eAAe,IAAI,YAAY,EAAE,CAAC;YACpC,aAAa,GAAG,YAAY,eAAe,IAAI,EAAE,GAAG,YAAY,IAAI,EAAE,IAAI,CAAC;QAC7E,CAAC;QAED,MAAM,iBAAiB,GAAG,SAAS;aAChC,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,sBAAsB,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;aAChE,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;YAChB,OAAO,aAAa,CAAC,uCAAuC,CAAC,QAAQ,CAAC,CAAC;QACzE,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC,CAAC;QACd,MAAM,UAAU,GAAG,YAAY,CAAC,gBAAgB;YAC9C,CAAC,CAAC;sDAC8C,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,gBAAgB,CAAC;iHACc;YAC3G,CAAC,CAAC,EAAE,CAAC;QACP,MAAM,MAAM,GAAG;;;;;;EAMjB,YAAY;gCACkB,kBAAkB,CAAC,CAAC,CAAC,IAAI,GAAG,kBAAkB,CAAC,CAAC,CAAC,EAAE,GAAG,eAAe,CAAC,CAAC,CAAC,2DAA2D,CAAC,CAAC,CAAC,EAAE;;qBAEnJ,QAAQ;sBACP,WAAW;MAC3B,aAAa,2CAA2C,YAAY,CAAC,gBAAgB,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE;EAC7G,iBAAiB;EACjB,UAAU;;CAEX,CAAC;QACE,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;IAED;;;OAGG;IACI,MAAM,CAAC,uCAAuC,CACnD,gBAAuC;QAEvC,MAAM,SAAS,GAAG,qBAAS,CAAC,YAAY,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QACtE,MAAM,SAAS,GACb,SAAS,CAAC,SAAS,IAAK,SAAS,CAAC,SAAoB,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC;YACtE,CAAC,CAAE,SAAS,CAAC,SAAoB;iBAC5B,KAAK,CAAC,IAAI,CAAC;iBACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;iBAClC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI;YACtB,CAAC,CAAC,EAAE,CAAC;QACT,oDAAoD;QACpD,OAAO,SAAS,CAAC,SAAS,CAAC;QAC3B,OAAO,SAAS,CAAC,iBAAiB,CAAC;QACnC,OAAO,SAAS,CAAC,UAAU,CAAC;QAC5B,MAAM,qBAAqB,GAAG,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;QAChE,MAAM,gBAAgB,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAE5D,QAAQ,gBAAgB,CAAC,IAAI,EAAE,CAAC;YAC9B,KAAK,uBAAU,CAAC,IAAI,CAAC,CAAC,CAAC;gBACrB,OAAO,GAAG,SAAS,6BAA6B,gBAAgB,IAAI,CAAC;YACvE,CAAC;YACD,KAAK,iDAAuB,CAAC,IAAI,CAAC,CAAC,CAAC;gBAClC,OAAO,GAAG,SAAS,8DAA8D,SAAS,CAAC,MAAM,KAAK,CAAC;YACzG,CAAC;YACD,KAAK,+CAAsB,CAAC,IAAI,CAAC,CAAC,CAAC;gBACjC,OAAO,GAAG,SAAS,iCAAiC,gBAAgB,IAAI,CAAC;YAC3E,CAAC;YACD,KAAK,qBAAS,CAAC,IAAI,CAAC,CAAC,CAAC;gBACpB,OAAO,GAAG,SAAS,2BAA2B,gBAAgB,IAAI,CAAC;YACrE,CAAC;YACD,KAAK,yCAAmB,CAAC,IAAI;gBAC3B,IAAI,SAAS,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;oBACtC,OAAO,GAAG,SAAS,yBAAyB,CAAC;gBAC/C,CAAC;qBAAM,IAAI,SAAS,CAAC,SAAS,KAAK,MAAM,EAAE,CAAC;oBAC1C,OAAO,GAAG,SAAS,sBAAsB,CAAC;gBAC5C,CAAC;qBAAM,CAAC;oBACN,MAAM,IAAI,KAAK,CACb,WAAW,yCAAmB,CAAC,IAAI,YAAY,gBAAgB,EAAE,CAClE,CAAC;gBACJ,CAAC;YACH,KAAK,iCAAe,CAAC,IAAI;gBACvB,OAAO,GAAG,SAAS,oBAAoB,SAAS,CAAC,GAAG,KAAK,CAAC;YAC5D,KAAK,+BAAc,CAAC,IAAI;gBACtB,OAAO,GAAG,SAAS,sBAAsB,CAAC;YAC5C,KAAK,+BAAc,CAAC,IAAI;gBACtB,OAAO,GAAG,SAAS,qBAAqB,gBAAgB,IAAI,CAAC;YAC/D,KAAK,mBAAQ,CAAC,IAAI;gBAChB,OAAO,GAAG,SAAS,6BAA6B,SAAS,CAAC,OAAO,GAAG,IAAI,IAAI,CAAC;YAC/E,8CAA8C;YAC9C,KAAK,yCAAmB,CAAC,IAAI,CAAC;YAC9B,KAAK,uCAAkB,CAAC,IAAI,CAAC;YAC7B,KAAK,6DAA6B,CAAC,IAAI,CAAC;YACxC,KAAK,2CAAoB,CAAC,IAAI,CAAC;YAC/B,KAAK,iEAA+B,CAAC,IAAI,CAAC;YAC1C,KAAK,6BAAa,CAAC,IAAI,CAAC;YACxB,KAAK,2BAAY,CAAC,IAAI,CAAC;YACvB,KAAK,mDAAwB,CAAC,IAAI,CAAC,CAAC,CAAC;gBACnC,OAAO,GAAG,SAAS,cAAc,gBAAgB,CAAC,IAAI,IAAI,qBAAqB,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;YAC9G,CAAC;YACD;gBACE,MAAM,QAAQ,GAAG,gBAAgB,CAAC,IAAI,CAAC;gBACvC,MAAM,cAAc,GAAG,qBAAqB;oBAC1C,CAAC,CAAC,GAAG,SAAS,mBAAmB,QAAQ,MAAM,gBAAgB,IAAI;oBACnE,CAAC,CAAC,GAAG,SAAS,mBAAmB,QAAQ,KAAK,CAAC;gBACjD,OAAO,cAAc,CAAC;QAC1B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,IAAY;QACrC,iEAAiE;QACjE,kDAAkD;QAClD,MAAM,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;QACrC,MAAM,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE;YAC1C,MAAM,EAAE,YAAY;YACpB,IAAI,EAAE,IAAI;YACV,WAAW,EAAE,IAAI;SAClB,CAAC,CAAC;QACH,OAAO,aAAa,CAAC;IACvB,CAAC;IAED;;;;;;OAMG;IACK,0BAA0B,CAAC,UAAkB;QACnD,OAAO,CACL,UAAU;YACR,2DAA2D;aAC1D,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC;YACrB,yDAAyD;aACxD,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAC3B,CAAC;IACJ,CAAC;CACF;AArPD,sCAqPC"}
+{"version":3,"file":"CodeGenerator.js","sourceRoot":"","sources":["../../src/managers/CodeGenerator.ts"],"names":[],"mappings":";;;AAEA,sEAAmE;AACnE,oEAAiE;AACjE,oDAAiD;AACjD,8EAA2E;AAC3E,4EAAyE;AACzE,kDAA+C;AAC/C,0FAAuF;AACvF,sEAAmE;AACnE,8DAA2D;AAC3D,wEAAqE;AACrE,8FAA2F;AAC3F,0DAAuD;AACvD,8DAA2D;AAC3D,kFAA+E;AAC/E,8FAA2F;AAC3F,wDAAqD;AACrD,4DAAyD;AACzD,gFAA6E;AAC7E,4DAAyD;AACzD,4EAAyE;AACzE,gDAA6C;AAC7C,kDAA+C;AAC/C,+CAA4C;AAE5C,MAAa,aAAa;IACxB,8EAA8E;IACvE,KAAK,CAAC,yBAAyB,CACpC,YAA0B,EAC1B,SAAkC;QAElC,0EAA0E;QAC1E,qEAAqE;QACrE,kBAAkB;QAClB,MAAM,sBAAsB,GAAG,IAAI,GAAG,CAAC;YACrC,iCAAe,CAAC,IAAI;YACpB,qDAAyB,CAAC,IAAI;YAC9B,iEAA+B,CAAC,IAAI;YACpC,+CAAsB,CAAC,IAAI;SAC5B,CAAC,CAAC;QACH,MAAM,mBAAmB,GACvB,YAAY,CAAC,gBAAgB;YAC7B,SAAS;iBACN,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAChB,OAAO,QAAQ,CAAC,IAAI,CAAC;YACvB,CAAC,CAAC;iBACD,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE;gBACpB,OAAO,yBAAW,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC;YACpE,CAAC,CAAC;iBACD,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE;gBACf,OAAO,IAAI,EAAE,WAAW,CAAC;YAC3B,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;QAClB,MAAM,YAAY,GAAG,mBAAmB;YACtC,CAAC,CAAC;;;;EAIN;YACI,CAAC,CAAC,GAAG,CAAC;QACR,MAAM,YAAY,GAChB,CAAC,YAAY,CAAC,gBAAgB,EAAE,IAAI,EAAE,CAAC,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QAC1D,MAAM,WAAW,GAAG,YAAY;YAC9B,CAAC,CAAC;;;2BAGmB,IAAI,CAAC,0BAA0B,CAAC,YAAY,CAAC,gBAAiB,CAAC;;UAEhF;YACJ,CAAC,CAAC,IAAI,CAAC;QACT,MAAM,QAAQ,GAAG,YAAY,CAAC,IAAI;YAChC,CAAC,CAAC,YAAY,CAAC,IAAI;gBACf,4BAA4B;iBAC3B,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC;gBACvB,wBAAwB;iBACvB,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC;gBACrB,mBAAmB;iBAClB,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC;gBACtB,2BAA2B;iBAC1B,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC;YAC1B,CAAC,CAAC,YAAY,YAAY,CAAC,aAAa,EAAE,CAAC;QAC7C,MAAM,eAAe,GACnB,YAAY,CAAC,OAAO,IAAI,YAAY,CAAC,OAAO,CAAC,YAAY;YACvD,CAAC,CAAC,+CAA+C,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI;YAC/G,CAAC,CAAC,SAAS,CAAC;QAChB,IAAI,aAAa,GAAG,EAAE,CAAC;QAEvB,IAAI,eAAe,EAAE,CAAC;YACpB,aAAa,GAAG,YAAY,eAAe,IAAI,CAAC;QAClD,CAAC;QAED,MAAM,iBAAiB,GAAG,SAAS;aAChC,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,sBAAsB,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;aAChE,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;YAChB,OAAO,aAAa,CAAC,uCAAuC,CAAC,QAAQ,CAAC,CAAC;QACzE,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC,CAAC;QACd,MAAM,UAAU,GAAG,YAAY,CAAC,gBAAgB;YAC9C,CAAC,CAAC;sDAC8C,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,gBAAgB,CAAC;iHACc;YAC3G,CAAC,CAAC,EAAE,CAAC;QACP,MAAM,MAAM,GAAG;;;;;;EAMjB,YAAY;gCACkB,eAAe,CAAC,CAAC,CAAC,2DAA2D,CAAC,CAAC,CAAC,EAAE;;qBAE7F,QAAQ;sBACP,WAAW;MAC3B,aAAa,2CAA2C,YAAY,CAAC,gBAAgB,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE;EAC7G,iBAAiB;EACjB,UAAU;;CAEX,CAAC;QACE,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;IAED;;;OAGG;IACI,MAAM,CAAC,uCAAuC,CACnD,gBAAuC;QAEvC,MAAM,SAAS,GAAG,qBAAS,CAAC,YAAY,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QACtE,MAAM,SAAS,GACb,SAAS,CAAC,SAAS,IAAK,SAAS,CAAC,SAAoB,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC;YACtE,CAAC,CAAE,SAAS,CAAC,SAAoB;iBAC5B,KAAK,CAAC,IAAI,CAAC;iBACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;iBAClC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI;YACtB,CAAC,CAAC,EAAE,CAAC;QACT,oDAAoD;QACpD,OAAO,SAAS,CAAC,SAAS,CAAC;QAC3B,OAAO,SAAS,CAAC,iBAAiB,CAAC;QACnC,OAAO,SAAS,CAAC,UAAU,CAAC;QAC5B,MAAM,qBAAqB,GAAG,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;QAChE,MAAM,gBAAgB,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAE5D,QAAQ,gBAAgB,CAAC,IAAI,EAAE,CAAC;YAC9B,KAAK,uBAAU,CAAC,IAAI,CAAC,CAAC,CAAC;gBACrB,OAAO,GAAG,SAAS,6BAA6B,gBAAgB,IAAI,CAAC;YACvE,CAAC;YACD,KAAK,iDAAuB,CAAC,IAAI,CAAC,CAAC,CAAC;gBAClC,OAAO,GAAG,SAAS,8DAA8D,SAAS,CAAC,MAAM,KAAK,CAAC;YACzG,CAAC;YACD,KAAK,+CAAsB,CAAC,IAAI,CAAC,CAAC,CAAC;gBACjC,OAAO,GAAG,SAAS,iCAAiC,gBAAgB,IAAI,CAAC;YAC3E,CAAC;YACD,KAAK,qBAAS,CAAC,IAAI,CAAC,CAAC,CAAC;gBACpB,OAAO,GAAG,SAAS,2BAA2B,gBAAgB,IAAI,CAAC;YACrE,CAAC;YACD,KAAK,yCAAmB,CAAC,IAAI;gBAC3B,IAAI,SAAS,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;oBACtC,OAAO,GAAG,SAAS,yBAAyB,CAAC;gBAC/C,CAAC;qBAAM,IAAI,SAAS,CAAC,SAAS,KAAK,MAAM,EAAE,CAAC;oBAC1C,OAAO,GAAG,SAAS,sBAAsB,CAAC;gBAC5C,CAAC;qBAAM,CAAC;oBACN,MAAM,IAAI,KAAK,CACb,WAAW,yCAAmB,CAAC,IAAI,YAAY,gBAAgB,EAAE,CAClE,CAAC;gBACJ,CAAC;YACH,KAAK,iCAAe,CAAC,IAAI;gBACvB,OAAO,GAAG,SAAS,oBAAoB,SAAS,CAAC,GAAG,KAAK,CAAC;YAC5D,KAAK,+BAAc,CAAC,IAAI;gBACtB,OAAO,GAAG,SAAS,sBAAsB,CAAC;YAC5C,KAAK,+BAAc,CAAC,IAAI;gBACtB,OAAO,GAAG,SAAS,qBAAqB,gBAAgB,IAAI,CAAC;YAC/D,KAAK,mBAAQ,CAAC,IAAI;gBAChB,OAAO,GAAG,SAAS,6BAA6B,SAAS,CAAC,OAAO,GAAG,IAAI,IAAI,CAAC;YAC/E,8CAA8C;YAC9C,KAAK,yCAAmB,CAAC,IAAI,CAAC;YAC9B,KAAK,uCAAkB,CAAC,IAAI,CAAC;YAC7B,KAAK,6DAA6B,CAAC,IAAI,CAAC;YACxC,KAAK,2CAAoB,CAAC,IAAI,CAAC;YAC/B,KAAK,iEAA+B,CAAC,IAAI,CAAC;YAC1C,KAAK,6BAAa,CAAC,IAAI,CAAC;YACxB,KAAK,2BAAY,CAAC,IAAI,CAAC;YACvB,KAAK,mDAAwB,CAAC,IAAI,CAAC,CAAC,CAAC;gBACnC,OAAO,GAAG,SAAS,cAAc,gBAAgB,CAAC,IAAI,IAAI,qBAAqB,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;YAC9G,CAAC;YACD;gBACE,MAAM,QAAQ,GAAG,gBAAgB,CAAC,IAAI,CAAC;gBACvC,MAAM,cAAc,GAAG,qBAAqB;oBAC1C,CAAC,CAAC,GAAG,SAAS,mBAAmB,QAAQ,MAAM,gBAAgB,IAAI;oBACnE,CAAC,CAAC,GAAG,SAAS,mBAAmB,QAAQ,KAAK,CAAC;gBACjD,OAAO,cAAc,CAAC;QAC1B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,IAAY;QACrC,iEAAiE;QACjE,kDAAkD;QAClD,MAAM,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;QACrC,MAAM,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE;YAC1C,MAAM,EAAE,YAAY;YACpB,IAAI,EAAE,IAAI;YACV,WAAW,EAAE,IAAI;SAClB,CAAC,CAAC;QACH,OAAO,aAAa,CAAC;IACvB,CAAC;IAED;;;;;;OAMG;IACK,0BAA0B,CAAC,UAAkB;QACnD,OAAO,CACL,UAAU;YACR,2DAA2D;aAC1D,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC;YACrB,yDAAyD;aACxD,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAC3B,CAAC;IACJ,CAAC;CACF;AApMD,sCAoMC"}

package/dist/managers/DonobuFlowsManager.d.ts

@@ -106,8 +106,21 @@ export declare class DonobuFlowsManager {
     private validateFlowParams;
     private setupAllowedTools;
     private prepareInitialToolCalls;
+    /**
+     * This method creates a temporary directory for the flow with the given ID,
+     * returning the path to the directory.
+     */
     private createTempDirectoryForFlow;
+    /**
+     * Searches the given directory for the largest video file and sets it as the
+     * flow's video. This should be called after the flow has completed.
+     */
+    private setFlowVideo;
     private runFlow;
+    /**
+     * Returns the largest video file in the given directory, or null if no
+     * video files are found.
+     */
     private findLargestVideo;
     private isLocallyRunning;
     private checkIfAnyToolsRequireGpt;

package/dist/managers/DonobuFlowsManager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"DonobuFlowsManager.d.ts","sourceRoot":"","sources":["../../src/managers/DonobuFlowsManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AASjD,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC1C,OAAO,EAAE,YAAY,EAAuB,MAAM,wBAAwB,CAAC;AAE3E,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAQ9D,OAAO,EAAE,uBAAuB,EAAE,MAAM,wCAAwC,CAAC;AACjF,OAAO,EAEL,oBAAoB,EACrB,MAAM,iCAAiC,CAAC;AAUzC,OAAO,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAC9C,OAAO,EAAE,mBAAmB,EAAE,MAAM,+BAA+B,CAAC;AAIpE,OAAO,EAAE,qBAAqB,EAAE,MAAM,qCAAqC,CAAC;AAC5E,OAAO,EAAE,2BAA2B,EAAE,MAAM,uCAAuC,CAAC;AAEpF,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AACxD,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAKlD;;GAEG;AACH,MAAM,WAAW,UAAU;IAEzB,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC;IAGhC,QAAQ,CAAC,GAAG,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;CAC7B;AAwBD,qBAAa,kBAAkB;IAkB3B,OAAO,CAAC,QAAQ,CAAC,qBAAqB;IACtC,OAAO,CAAC,QAAQ,CAAC,iBAAiB;IAClC,OAAO,CAAC,QAAQ,CAAC,aAAa;IAC9B,OAAO,CAAC,QAAQ,CAAC,uBAAuB;IApB1C,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAQ;IACzD,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB,CAAM;IACpD,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB,CAQpB;IAC1B,gBAAuB,8BAA8B,uBAAuB;IAE5E,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAsC;IAClE,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAgB;gBAG3B,qBAAqB,EAAE,2BAA2B,EAClD,iBAAiB,EAAE,iBAAiB,EACpC,aAAa,EAAE,aAAa,EAC5B,uBAAuB,EAAE,uBAAuB;IAKnE;;;OAGG;IACU,UAAU,CACrB,UAAU,EAAE,gBAAgB,EAC5B,SAAS,CAAC,EAAE,SAAS,GACpB,OAAO,CAAC,UAAU,CAAC;IAyItB;;;;;;OAMG;IACU,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAwBtE,6EAA6E;IAChE,eAAe,CAC1B,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAClC,OAAO,CAAC,IAAI,CAAC;IAqBhB;;;;;;OAMG;IACI,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,UAAU;IAahD;;OAEG;IACU,gBAAgB,CAC3B,KAAK,EAAE,UAAU,GAChB,OAAO,CAAC,oBAAoB,CAAC;IAqFhC;;;;;OAKG;IACU,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAsBnE;;OAEG;IACU,aAAa,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAcnE,+DAA+D;IAClD,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;IA4B9D;;;;OAIG;IACU,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAwB1D;;;;OAIG;IACU,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAmB9D,8EAA8E;IACjE,yBAAyB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAavE;;;;;;;QAOI;IACS,eAAe,CAAC,aAAa,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC;QAClE,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;QAC7B,SAAS,EAAE,WAAW,GAAG,IAAI,CAAC;QAC9B,SAAS,EAAE,SAAS,GAAG,IAAI,CAAC;KAC7B,CAAC;IAoCF;;;OAGG;IACU,sBAAsB,CACjC,eAAe,EAAE,qBAAqB,GACrC,OAAO,CAAC,mBAAmB,CAAC;IAsC/B,OAAO,CAAC,kBAAkB;IA8F1B,OAAO,CAAC,iBAAiB;IAoCzB,OAAO,CAAC,uBAAuB;IAsB/B,OAAO,CAAC,0BAA0B;YAMpB,OAAO;IAmBrB,OAAO,CAAC,gBAAgB;IAaxB,OAAO,CAAC,gBAAgB;IAIxB,OAAO,CAAC,yBAAyB;IAyBjC,OAAO,CAAC,wBAAwB;IA0DhC;;OAEG;IACH,OAAO,CAAC,MAAM,CAAC,QAAQ;IAYvB;;OAEG;IACH,OAAO,CAAC,uBAAuB;IAqB/B;;OAEG;IACH,OAAO,CAAC,wBAAwB;CAGjC"}
+{"version":3,"file":"DonobuFlowsManager.d.ts","sourceRoot":"","sources":["../../src/managers/DonobuFlowsManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AASjD,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC1C,OAAO,EAAE,YAAY,EAAuB,MAAM,wBAAwB,CAAC;AAE3E,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAQ9D,OAAO,EAAE,uBAAuB,EAAE,MAAM,wCAAwC,CAAC;AACjF,OAAO,EAEL,oBAAoB,EACrB,MAAM,iCAAiC,CAAC;AAUzC,OAAO,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAC9C,OAAO,EAAE,mBAAmB,EAAE,MAAM,+BAA+B,CAAC;AAIpE,OAAO,EAAE,qBAAqB,EAAE,MAAM,qCAAqC,CAAC;AAC5E,OAAO,EAAE,2BAA2B,EAAE,MAAM,uCAAuC,CAAC;AAEpF,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AACxD,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAOlD;;GAEG;AACH,MAAM,WAAW,UAAU;IAEzB,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC;IAGhC,QAAQ,CAAC,GAAG,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;CAC7B;AAgBD,qBAAa,kBAAkB;IAkB3B,OAAO,CAAC,QAAQ,CAAC,qBAAqB;IACtC,OAAO,CAAC,QAAQ,CAAC,iBAAiB;IAClC,OAAO,CAAC,QAAQ,CAAC,aAAa;IAC9B,OAAO,CAAC,QAAQ,CAAC,uBAAuB;IApB1C,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAQ;IACzD,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB,CAAM;IACpD,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB,CAQpB;IAC1B,gBAAuB,8BAA8B,uBAAuB;IAE5E,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAsC;IAClE,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAgB;gBAG3B,qBAAqB,EAAE,2BAA2B,EAClD,iBAAiB,EAAE,iBAAiB,EACpC,aAAa,EAAE,aAAa,EAC5B,uBAAuB,EAAE,uBAAuB;IAKnE;;;OAGG;IACU,UAAU,CACrB,UAAU,EAAE,gBAAgB,EAC5B,SAAS,CAAC,EAAE,SAAS,GACpB,OAAO,CAAC,UAAU,CAAC;IAmItB;;;;;;OAMG;IACU,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAwBtE,6EAA6E;IAChE,eAAe,CAC1B,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAClC,OAAO,CAAC,IAAI,CAAC;IAqBhB;;;;;;OAMG;IACI,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,UAAU;IAahD;;OAEG;IACU,gBAAgB,CAC3B,KAAK,EAAE,UAAU,GAChB,OAAO,CAAC,oBAAoB,CAAC;IAwFhC;;;;;OAKG;IACU,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAsBnE;;OAEG;IACU,aAAa,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAcnE,+DAA+D;IAClD,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;IA4B9D;;;;OAIG;IACU,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAwB1D;;;;OAIG;IACU,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAmB9D,8EAA8E;IACjE,yBAAyB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAavE;;;;;;;QAOI;IACS,eAAe,CAAC,aAAa,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC;QAClE,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;QAC7B,SAAS,EAAE,WAAW,GAAG,IAAI,CAAC;QAC9B,SAAS,EAAE,SAAS,GAAG,IAAI,CAAC;KAC7B,CAAC;IAmGF;;;OAGG;IACU,sBAAsB,CACjC,eAAe,EAAE,qBAAqB,GACrC,OAAO,CAAC,mBAAmB,CAAC;IAsC/B,OAAO,CAAC,kBAAkB;IA8F1B,OAAO,CAAC,iBAAiB;IAoCzB,OAAO,CAAC,uBAAuB;IAsB/B;;;OAGG;IACH,OAAO,CAAC,0BAA0B;IAMlC;;;OAGG;YACW,YAAY;YAaZ,OAAO;IA2BrB;;;OAGG;IACH,OAAO,CAAC,gBAAgB;IAaxB,OAAO,CAAC,gBAAgB;IAIxB,OAAO,CAAC,yBAAyB;IAyBjC,OAAO,CAAC,wBAAwB;IA0DhC;;OAEG;IACH,OAAO,CAAC,MAAM,CAAC,QAAQ;IAYvB;;OAEG;IACH,OAAO,CAAC,uBAAuB;IAqB/B;;OAEG;IACH,OAAO,CAAC,wBAAwB;CAGjC"}

package/dist/managers/DonobuFlowsManager.js

@@ -63,6 +63,7 @@ const DonobuDeploymentEnvironment_1 = require("../models/DonobuDeploymentEnviron
 const Logger_1 = require("../utils/Logger");
 const CodeGenerator_1 = require("./CodeGenerator");
 const uuid_1 = require("uuid");
+const envVars_1 = require("../envVars");
 class DonobuFlowsManager {
     constructor(deploymentEnvironment, gptConfigsManager, agentsManager, flowsPersistenceFactory) {
         this.deploymentEnvironment = deploymentEnvironment;
@@ -70,7 +71,7 @@ class DonobuFlowsManager {
         this.agentsManager = agentsManager;
         this.flowsPersistenceFactory = flowsPersistenceFactory;
         this.activeFlows = new Map();
-        this.codeGenerator = new CodeGenerator_1.CodeGenerator(gptConfigsManager, agentsManager);
+        this.codeGenerator = new CodeGenerator_1.CodeGenerator();
     }
     /**
      * Create a flow with the given parameters and invoke its `DonobuFlow#run`
@@ -132,18 +133,18 @@ class DonobuFlowsManager {
             state: 'UNSTARTED',
             nextState: null,
         };
-        const tempDirectoryForVideos = this.createTempDirectoryForFlow(flowMetadata.id);
+        const flowTempDir = this.createTempDirectoryForFlow(flowMetadata.id);
         const flowsPersistence = await this.flowsPersistenceFactory.createPersistenceLayer();
         const browserStorageState = flowParams.browser?.initialState
             ? await this.getBrowserStorageState(flowParams.browser.initialState)
             : undefined;
-        const browserFramework = await BrowserFramework_1.BrowserFramework.create(browserConfig, tempDirectoryForVideos, browserStorageState);
+        const browserFramework = await BrowserFramework_1.BrowserFramework.create(browserConfig, flowTempDir, browserStorageState);
         try {
             const donobuFlow = new DonobuFlow_1.DonobuFlow(this, browserFramework, flowsPersistence, gptClientData.gptClient, toolManager, tooltipper, toolCallsOnStart, [], [], {
                 current: null,
             }, flowMetadata);
             await flowsPersistence.saveMetadata(flowMetadata);
-            const job = this.runFlow(donobuFlow, tempDirectoryForVideos, flowsPersistence);
+            const job = this.runFlow(donobuFlow, flowTempDir, flowsPersistence);
             const flowHandle = { donobuFlow, job };
             this.activeFlows.set(flowMetadata.id, flowHandle);
             return flowHandle;
@@ -424,6 +425,51 @@ class DonobuFlowsManager {
                 Logger_1.appLogger.warn(`Failed to find GPT configuration: ${gptConfigName}, will default to the 'flow-runner' agent config (if it exists).`);
             }
         }
+        let gptConfigFromEnv = null;
+        if (process.env[envVars_1.ENV_VAR_NAMES.AWS_BEDROCK_MODEL_NAME]) {
+            gptConfigFromEnv = {
+                type: 'ANTHROPIC_AWS_BEDROCK',
+                modelName: process.env[envVars_1.ENV_VAR_NAMES.AWS_BEDROCK_MODEL_NAME],
+            };
+        }
+        else if (process.env[envVars_1.ENV_VAR_NAMES.ANTHROPIC_MODEL_NAME]) {
+            if (!process.env[envVars_1.ENV_VAR_NAMES.ANTHROPIC_API_KEY]) {
+                throw new InvalidParamValueException_1.InvalidParamValueException(envVars_1.ENV_VAR_NAMES.ANTHROPIC_API_KEY, undefined, `the ${envVars_1.ENV_VAR_NAMES.ANTHROPIC_MODEL_NAME} environment variable is set but the ${envVars_1.ENV_VAR_NAMES.ANTHROPIC_API_KEY} is not`);
+            }
+            gptConfigFromEnv = {
+                type: 'ANTHROPIC',
+                modelName: process.env[envVars_1.ENV_VAR_NAMES.ANTHROPIC_MODEL_NAME],
+                apiKey: process.env[envVars_1.ENV_VAR_NAMES.ANTHROPIC_API_KEY],
+            };
+        }
+        else if (process.env[envVars_1.ENV_VAR_NAMES.GOOGLE_GENERATIVE_AI_MODEL_NAME]) {
+            if (!process.env[envVars_1.ENV_VAR_NAMES.GOOGLE_GENERATIVE_AI_API_KEY]) {
+                throw new InvalidParamValueException_1.InvalidParamValueException(envVars_1.ENV_VAR_NAMES.GOOGLE_GENERATIVE_AI_API_KEY, undefined, `the ${envVars_1.ENV_VAR_NAMES.GOOGLE_GENERATIVE_AI_MODEL_NAME} environment variable is set but the ${envVars_1.ENV_VAR_NAMES.GOOGLE_GENERATIVE_AI_API_KEY} is not`);
+            }
+            gptConfigFromEnv = {
+                type: 'GOOGLE_GEMINI',
+                modelName: process.env[envVars_1.ENV_VAR_NAMES.GOOGLE_GENERATIVE_AI_MODEL_NAME],
+                apiKey: process.env[envVars_1.ENV_VAR_NAMES.GOOGLE_GENERATIVE_AI_API_KEY],
+            };
+        }
+        else if (process.env[envVars_1.ENV_VAR_NAMES.OPENAI_API_MODEL_NAME]) {
+            if (!process.env[envVars_1.ENV_VAR_NAMES.OPENAI_API_KEY]) {
+                throw new InvalidParamValueException_1.InvalidParamValueException(envVars_1.ENV_VAR_NAMES.OPENAI_API_KEY, undefined, `the ${envVars_1.ENV_VAR_NAMES.OPENAI_API_MODEL_NAME} environment variable is set but the ${envVars_1.ENV_VAR_NAMES.OPENAI_API_KEY} is not`);
+            }
+            gptConfigFromEnv = {
+                type: 'OPENAI',
+                modelName: process.env[envVars_1.ENV_VAR_NAMES.OPENAI_API_MODEL_NAME],
+                apiKey: process.env[envVars_1.ENV_VAR_NAMES.OPENAI_API_KEY],
+            };
+        }
+        if (gptConfigFromEnv) {
+            const gptClient = await GptClientFactory_1.GptClientFactory.createFromGptConfig(gptConfigFromEnv);
+            return {
+                gptConfigName: null,
+                agentName: null,
+                gptClient: gptClient,
+            };
+        }
         const defaultGptConfigName = await this.agentsManager.get('flow-runner');
         if (!defaultGptConfigName) {
             return {
@@ -432,8 +478,8 @@ class DonobuFlowsManager {
                 gptClient: null,
             };
         }
-        const gptConfig = await this.gptConfigsManager.get(defaultGptConfigName);
-        const gptClient = await GptClientFactory_1.GptClientFactory.createFromGptConfig(gptConfig);
+        const defaultGptConfig = await this.gptConfigsManager.get(defaultGptConfigName);
+        const gptClient = await GptClientFactory_1.GptClientFactory.createFromGptConfig(defaultGptConfig);
         return {
             gptConfigName: defaultGptConfigName,
             agentName: 'flow-runner',
@@ -561,25 +607,50 @@ class DonobuFlowsManager {
             return [];
         }
     }
+    /**
+     * This method creates a temporary directory for the flow with the given ID,
+     * returning the path to the directory.
+     */
     createTempDirectoryForFlow(flowId) {
         const tempDir = path.join((0, os_1.tmpdir)(), flowId);
         fs.mkdirSync(tempDir);
         return tempDir;
     }
-    async runFlow(donobuFlow, tempDirectoryForVideos, flowsPersistence) {
+    /**
+     * Searches the given directory for the largest video file and sets it as the
+     * flow's video. This should be called after the flow has completed.
+     */
+    async setFlowVideo(flowId, flowTempDir, flowsPersistence) {
+        const videoPath = this.findLargestVideo(flowTempDir);
+        if (videoPath) {
+            const videoBytes = fs.readFileSync(videoPath);
+            await flowsPersistence.setVideo(flowId, videoBytes);
+        }
+    }
+    async runFlow(donobuFlow, flowTempDir, flowsPersistence) {
         try {
             await donobuFlow.run();
-            this.activeFlows.delete(donobuFlow.metadata.id);
-            const videoPath = this.findLargestVideo(tempDirectoryForVideos);
-            if (videoPath) {
-                const videoBytes = fs.readFileSync(videoPath);
-                await flowsPersistence.setVideo(donobuFlow.metadata.id, videoBytes);
-            }
+            await this.setFlowVideo(donobuFlow.metadata.id, flowTempDir, flowsPersistence);
         }
         finally {
-            fs.rmSync(tempDirectoryForVideos, { recursive: true, force: true });
+            try {
+                this.activeFlows.delete(donobuFlow.metadata.id);
+            }
+            catch (error) {
+                Logger_1.appLogger.error('Failed to delete active flow:', error);
+            }
+            try {
+                fs.rmSync(flowTempDir, { recursive: true, force: true });
+            }
+            catch (error) {
+                Logger_1.appLogger.error('Failed to remove temporary directory:', error);
+            }
         }
     }
+    /**
+     * Returns the largest video file in the given directory, or null if no
+     * video files are found.
+     */
     findLargestVideo(directory) {
         const files = fs
             .readdirSync(directory)