@empiricalrun/test-gen 0.31.10 → 0.31.12

package/CHANGELOG.md

@@ -1,5 +1,20 @@
 # @empiricalrun/test-gen
 
+## 0.31.12
+
+### Patch Changes
+
+- a8f2ea6: chore: add test for master agent
+- Updated dependencies [10f554d]
+  - @empiricalrun/reporter@0.20.2
+
+## 0.31.11
+
+### Patch Changes
+
+- Updated dependencies [9623b74]
+  - @empiricalrun/reporter@0.20.1
+
 ## 0.31.10
 
 ### Patch Changes

package/dist/agent/master/run.d.ts

@@ -2,6 +2,15 @@ import { LLM, TraceClient } from "@empiricalrun/llm";
 import { Page } from "playwright";
 import { TestGenConfigOptions } from "../../types";
 type BrowsingAgentOptions = Partial<TestGenConfigOptions>;
+export declare function getNextAction({ task, executedActions, page, trace, llm, options, pageScreenshot, }: {
+    task: string;
+    executedActions: string[];
+    page: Page;
+    trace?: TraceClient;
+    llm: LLM;
+    options: BrowsingAgentOptions;
+    pageScreenshot: string;
+}): Promise<import("openai/resources/index.mjs").ChatCompletionMessageToolCall | undefined>;
 export declare function masterAgent(task: string, page: Page, executedActions: string[], trace: TraceClient, llm: LLM, options: BrowsingAgentOptions): Promise<any>;
 export {};
 //# sourceMappingURL=run.d.ts.map

package/dist/agent/master/run.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"run.d.ts","sourceRoot":"","sources":["../../../src/agent/master/run.ts"],"names":[],"mappings":"AAAA,OAAO,EAAa,GAAG,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAGhE,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AASlC,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAEnD,KAAK,oBAAoB,GAAG,OAAO,CAAC,oBAAoB,CAAC,CAAC;AAE1D,wBAAsB,WAAW,CAC/B,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,IAAI,EACV,eAAe,EAAE,MAAM,EAAE,EACzB,KAAK,EAAE,WAAW,EAClB,GAAG,EAAE,GAAG,EACR,OAAO,EAAE,oBAAoB,gBA0E9B"}
+{"version":3,"file":"run.d.ts","sourceRoot":"","sources":["../../../src/agent/master/run.ts"],"names":[],"mappings":"AAAA,OAAO,EAAa,GAAG,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAGhE,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AASlC,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAEnD,KAAK,oBAAoB,GAAG,OAAO,CAAC,oBAAoB,CAAC,CAAC;AAE1D,wBAAsB,aAAa,CAAC,EAClC,IAAI,EACJ,eAAe,EACf,IAAI,EACJ,KAAK,EACL,GAAG,EACH,OAAO,EACP,cAAc,GACf,EAAE;IACD,IAAI,EAAE,MAAM,CAAC;IACb,eAAe,EAAE,MAAM,EAAE,CAAC;IAC1B,IAAI,EAAE,IAAI,CAAC;IACX,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,GAAG,EAAE,GAAG,CAAC;IACT,OAAO,EAAE,oBAAoB,CAAC;IAC9B,cAAc,EAAE,MAAM,CAAC;CACxB,2FA+CA;AAED,wBAAsB,WAAW,CAC/B,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,IAAI,EACV,eAAe,EAAE,MAAM,EAAE,EACzB,KAAK,EAAE,WAAW,EAClB,GAAG,EAAE,GAAG,EACR,OAAO,EAAE,oBAAoB,gBAsC9B"}

package/dist/agent/master/run.js

@@ -1,19 +1,14 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.masterAgent = void 0;
+exports.masterAgent = exports.getNextAction = void 0;
 const llm_1 = require("@empiricalrun/llm");
 const vision_1 = require("@empiricalrun/llm/vision");
 const done_1 = require("../../actions/done");
 const next_task_1 = require("../../actions/next-task");
 const constants_1 = require("../../constants");
 const reporter_1 = require("../../reporter");
-async function masterAgent(task, page, executedActions, trace, llm, options) {
-    trace.update({ input: { task } });
-    const promptSpan = trace.span({ name: "page-prompt" });
-    const buffer = await page.screenshot({ fullPage: true });
-    const testGenReporter = new reporter_1.TestGenUpdatesReporter();
-    const testGenSnapshotUpdatePromise = testGenReporter.sendCurrentView(buffer);
-    const pageScreenshot = buffer.toString("base64");
+async function getNextAction({ task, executedActions, page, trace, llm, options, pageScreenshot, }) {
+    const promptSpan = trace?.span({ name: "page-prompt" });
     const promptMessages = await (0, llm_1.getPrompt)("test-gen", {
         task,
         executedActions: executedActions.map((a) => a).join("\n"),
@@ -39,7 +34,7 @@ async function masterAgent(task, page, executedActions, trace, llm, options) {
         userMessage,
     ];
     const tools = [next_task_1.NextTaskAction.schema, (0, done_1.doneActionGenerator)(page).schema];
-    promptSpan.end({ output: { messages } });
+    promptSpan?.end({ output: { messages } });
     const completion = await llm.createChatCompletion({
         messages,
         modelParameters: {
@@ -52,8 +47,26 @@ async function masterAgent(task, page, executedActions, trace, llm, options) {
         // @ts-ignore
         tools,
     });
-    let output;
     const toolCall = completion?.tool_calls?.[0];
+    return toolCall;
+}
+exports.getNextAction = getNextAction;
+async function masterAgent(task, page, executedActions, trace, llm, options) {
+    trace.update({ input: { task } });
+    const buffer = await page.screenshot({ fullPage: true });
+    const testGenReporter = new reporter_1.TestGenUpdatesReporter();
+    const testGenSnapshotUpdatePromise = testGenReporter.sendCurrentView(buffer);
+    const pageScreenshot = buffer.toString("base64");
+    let output;
+    const toolCall = await getNextAction({
+        task,
+        executedActions,
+        page,
+        trace,
+        llm,
+        options,
+        pageScreenshot,
+    });
     if (toolCall) {
         if (toolCall.function.name === "task_done") {
             output = {

package/dist/bin/index.js

@@ -69,10 +69,12 @@ async function runAgent(testGenConfig) {
         testGenFailed = true;
         new logger_1.CustomLogger().error(`Failed to generate test for the scenario. ${process.env.LOG_URL ? `[view log](${process.env.LOG_URL})` : ""}`);
     }
-    await new reporter_1.TestGenUpdatesReporter().reportGenAssets({
-        projectRepoName: testGenConfig.options.metadata.projectRepoName,
-        testName: testGenConfig.testCase.name,
-    });
+    if (testGenConfig.options?.agent !== "code") {
+        await new reporter_1.TestGenUpdatesReporter().reportGenAssets({
+            projectRepoName: testGenConfig.options.metadata.projectRepoName,
+            testName: testGenConfig.testCase.name,
+        });
+    }
     // TODO: move these reporters to a better lifecycle
     await (0, ci_1.reportOnCI)(testGenConfig.testCase);
     await (0, llm_1.flushAllTraces)();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@empiricalrun/test-gen",
-  "version": "0.31.10",
+  "version": "0.31.12",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
     "access": "public"
@@ -46,7 +46,7 @@
     "typescript": "^5.3.3",
     "@empiricalrun/llm": "^0.9.3",
     "@empiricalrun/r2-uploader": "^0.3.3",
-    "@empiricalrun/reporter": "^0.20.0"
+    "@empiricalrun/reporter": "^0.20.2"
   },
   "devDependencies": {
     "@types/detect-port": "^1.3.5",