@empiricalrun/test-gen 0.73.1 → 0.74.1

package/CHANGELOG.md

@@ -1,5 +1,99 @@
 # @empiricalrun/test-gen
 
+## 0.74.1
+
+### Patch Changes
+
+- Updated dependencies [6f2aa95]
+  - @empiricalrun/ast-parser@0.0.10
+
+## 0.74.0
+
+### Minor Changes
+
+- de7be1a: feat: added console traces to trace.zip tool
+- 5fc3b85: feat: add fetchLastSuccessfulTestRun tool & api endpoint
+- 8956b5c: refactor: simplify default model selection by delegating to a single function
+- 7c4057e: feat: Made the listIssues tool call inline
+- 17fa058: feat: add model mode, mode based tool & system prompt
+- b52ec03: feat: triage summary tool with slack messages
+- 3613b5f: feat: updated tool defs, runtTest, fetchLastSuccessfulTestRun, viewFailedTestRunReport, updated last-successful-run endpoint
+- c0c97cf: feat: add chunking and dedup fs to avoid memory issues, add LLM usage to toolResponse
+- a03391a: feat: upload unique frames to video-analysis bucket with video hash
+- b404912: feat: added trace.zip tool for network traces (based on existing approach from inngest)
+- 1d90369: feat: permalink support in issues, updated IDashboardClient to getBaseUrl
+- 4b45e85: feat: split upload b/w frames and summary to decrease wait
+- cbaf687: feat: Added auto option in CLI (empty --model results in default fallback)
+- 5a8af12: feat: tools array based on agent mode or env
+- ce54ca8: feat: video analysis working in CLI & UI minor cleanup
+- 565a820: refactor: converted PUT calls to PATCH in update issue api
+- 681aebb: fix: moving trace dot zip tool to not be inline so it gets executed on tool execute
+- 773d1f8: feat: issue queries & tools update with new test metadata fields
+- 6585338: feat: added inline viewTestRunReport tool for triage
+- c0931db: feat: add issue(s) list & add tools
+- 2718944: fix: trim viewTestRunReportTool result size by removing steps from results array
+- aa14384: feat: add updateIssue tool
+- c112011: feat: add reset-chat flag to cli
+- 50c05ff: feat: added send triage summary tool
+- 9635632: feat: issues with health stats
+- 8eeb121: fix: project id filter fix, use-triage in cli
+
+### Patch Changes
+
+- 1600c5d: feat: improvements to tool call telemetry
+- 01b693e: feat: moved file view commands to inline exec
+- 63a7b56: chore: remove some default model indirection
+- af74c99: fix: rm codeframes, steps, snippet from the test viewTestRunReport output
+- 9f33a7a: chore: fetch r2 files with suffix, add created_at to summary
+- 04940d6: fix: fix chunking error in ffmpeg local client with error handling
+- 028d845: chore: Added more tools to the inline execution roster
+- 01b693e: fix: added branch name field to the GitHub API and minor refactors in the test-gen package
+- 035febd: feat: runTest tool is now a common tool
+- 7fecf0a: feat: enable streaming on openai chat models
+- 4f598eb: feat: add gpt-5
+- 4c8249a: test: fix agent-worker tests for parallel execution
+- ad91c42: fix: allow insert_text in str_replace_editor tool for claude 4
+- 68ce1d1: fix: fetchVideoAnalysis error handling, minor cleanup
+- 2de74d0: chore: bump openai to 5.x
+- 41816c3: chore: enabled inline tool execution for fetch diagnosis.
+- 5651643: feat: recorder can pause before failing line when given a report url
+- 3d738f4: chore: split text editor tool call telemetry for commands
+- de01455: fix: moved project filter before inngest event send, added project id to log, separate test-gen utils exports
+- 40ef060: fix: support dir paths in inline file view
+- 5da0a60: feat: internal openai model supports multi-modal tool responses
+- 6d80d21: feat: video analysis review UI under admin route
+- fdc2ee8: feat: working video analysis with gemini-2.5-flash
+- 612e383: fix: parallel tool calling claude chat model
+- fe95cd7: chore: update prompts for triage session
+- 4b898bd: fix: elevate usage summary to agent loop
+- 7b3c12b: chore: simplify agent loop interface
+- 91d856c: fix: mode in agent-loop mandatroy, and being used
+- 5c91897: chore: Completed TODOs and refactored the inline tool identification logic"
+- ac01228: chore: upgrade to eslint 9
+- Updated dependencies [cb503a6]
+- Updated dependencies [29eb6fb]
+- Updated dependencies [a6e6141]
+- Updated dependencies [63271bd]
+- Updated dependencies [9f33a7a]
+- Updated dependencies [c0c97cf]
+- Updated dependencies [7fecf0a]
+- Updated dependencies [4f598eb]
+- Updated dependencies [ad91c42]
+- Updated dependencies [ce54ca8]
+- Updated dependencies [2de74d0]
+- Updated dependencies [5651643]
+- Updated dependencies [5da0a60]
+- Updated dependencies [fdc2ee8]
+- Updated dependencies [612e383]
+- Updated dependencies [4b898bd]
+- Updated dependencies [89858b4]
+- Updated dependencies [f83a0b0]
+- Updated dependencies [7b3c12b]
+- Updated dependencies [98853ca]
+  - @empiricalrun/llm@0.21.0
+  - @empiricalrun/r2-uploader@0.3.10
+  - @empiricalrun/ast-parser@0.0.9
+
 ## 0.73.1
 
 ### Patch Changes

package/dist/actions/utils/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/actions/utils/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAElC,OAAO,CAAC,MAAM,CAAC;IACb,UAAU,MAAM;QACd,UAAU,EAAE,GAAG,CAAC;QAChB,MAAM,EAAE,GAAG,CAAC;KACb;CACF;AAED,wBAAsB,oCAAoC,CACxD,WAAW,EAAE,MAAM,EACnB,KAAK,EAAE,MAAM,EACb,IAAI,EAAE,IAAI,EACV,iBAAiB,CAAC,EAAE,MAAM,gBAkH3B;AAED,wBAAgB,cAAc,WAG7B"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/actions/utils/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAElC,OAAO,CAAC,MAAM,CAAC;IACb,UAAU,MAAM;QACd,UAAU,EAAE,GAAG,CAAC;QAChB,MAAM,EAAE,GAAG,CAAC;KACb;CACF;AAED,wBAAsB,oCAAoC,CACxD,WAAW,EAAE,MAAM,EACnB,KAAK,EAAE,MAAM,EACb,IAAI,EAAE,IAAI,EACV,iBAAiB,CAAC,EAAE,MAAM,gBAkH3B;AAED,wBAAgB,cAAc,WAE7B"}

package/dist/actions/utils/index.js

@@ -71,7 +71,7 @@ async function getPlaywrightLocatorUsingCssSelector(cssSelector, xpath, page, el
                 });
                 selectedElem = leastNestedElements[0];
             }
-            catch (e) {
+            catch {
                 // ignore error
             }
         }
@@ -103,6 +103,5 @@ async function getPlaywrightLocatorUsingCssSelector(cssSelector, xpath, page, el
     });
 }
 function getPageVarName() {
-    // eslint-disable-next-line turbo/no-undeclared-env-vars
     return process.env.PAGE_VAR_NAME || "page";
 }

package/dist/agent/browsing/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/browsing/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiB,GAAG,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACpE,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAElC,OAAO,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAKlD,MAAM,MAAM,kBAAkB,GAAG;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,GAAG,SAAS,CAAC;CAC1B,CAAC;AAEF,wBAAsB,6BAA6B,CAAC,EAClD,MAAM,EACN,IAAI,EACJ,OAAO,EACP,GAAG,EACH,KAAK,GACN,EAAE;IACD,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,IAAI,CAAC;IACX,OAAO,EAAE,iBAAiB,CAAC;IAC3B,GAAG,CAAC,EAAE,GAAG,CAAC;IACV,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB,GAAG,OAAO,CAAC,kBAAkB,GAAG,SAAS,CAAC,CAwD1C"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/browsing/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiB,GAAG,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACpE,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAElC,OAAO,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAKlD,MAAM,MAAM,kBAAkB,GAAG;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,GAAG,SAAS,CAAC;CAC1B,CAAC;AAEF,wBAAsB,6BAA6B,CAAC,EAClD,MAAM,EACN,IAAI,EACJ,OAAO,EACP,GAAG,EACH,KAAK,GACN,EAAE;IACD,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,IAAI,CAAC;IACX,OAAO,EAAE,iBAAiB,CAAC;IAC3B,GAAG,CAAC,EAAE,GAAG,CAAC;IACV,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB,GAAG,OAAO,CAAC,kBAAkB,GAAG,SAAS,CAAC,CA0D1C"}

package/dist/agent/browsing/index.js

@@ -45,13 +45,15 @@ async function executeTaskUsingBrowsingAgent({ action, page, actions, llm, trace
     const toolCallsSpan = browsingAgentSpan?.span({ name: "tool-calls" });
     for (const i in toolCalls) {
         const toolCall = toolCalls[i];
-        const args = JSON.parse(toolCall.function.arguments);
-        const code = await actions.executeAction(toolCall.function.name, JSON.parse(toolCall.function.arguments), toolCallsSpan);
-        generatedCodeSteps = {
-            // Passing reason as action, in order to pass the correct action that took place to runtime planner
-            action: args.reason,
-            code,
-        };
+        if ("function" in toolCall) {
+            const args = JSON.parse(toolCall.function.arguments);
+            const code = await actions.executeAction(toolCall.function.name, JSON.parse(toolCall.function.arguments), toolCallsSpan);
+            generatedCodeSteps = {
+                // Passing reason as action, in order to pass the correct action that took place to runtime planner
+                action: args.reason,
+                code,
+            };
+        }
     }
     toolCallsSpan?.end({ output: { toolCalls, generatedCodeSteps } });
     return generatedCodeSteps;

package/dist/agent/browsing/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/agent/browsing/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAI3D,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAClC,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAsBvD,wBAAgB,QAAQ,CAAC,GAAG,EAAE,GAAG,GAAG,GAAG,IAAI,MAAM,CAKhD;AAED,wBAAgB,wBAAwB,CAAC,KAAK,EAAE,MAAM,EAAE,UAIvD;AAED,wBAAsB,kBAAkB,CACtC,YAAY,EAAE,MAAM,EACpB,UAAU,EAAE,MAAM,iBAgBnB;AAgED,wBAAsB,yBAAyB,CAC7C,YAAY,EAAE,MAAM,EACpB,OAAO,EAAE,MAAM,iBAwBhB;AAED,wBAAsB,cAAc,CAAC,EACnC,YAAY,EACZ,cAAc,EACd,QAAQ,GACT,EAAE;IACD,YAAY,EAAE,MAAM,CAAC;IACrB,cAAc,EAAE,MAAM,EAAE,CAAC;IACzB,QAAQ,EAAE,MAAM,CAAC;CAClB,iBAoBA;AAED,wBAAsB,yBAAyB,CAAC,EAC9C,QAAQ,EACR,QAAQ,EACR,KAAK,GACN,EAAE;IACD,QAAQ,EAAE,QAAQ,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB,GAAG,OAAO,CAAC,MAAM,CAAC,CAyDlB;AAyBD,wBAAsB,wBAAwB,CAAC,IAAI,EAAE,IAAI,iBA8HxD;AAED;;;GAGG;AACH,wBAAsB,oBAAoB,CACxC,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,oBAAoB,CAAC,CA2B/B;AAWD,wBAAsB,oBAAoB,CACxC,gBAAgB,EAAE,oBAAoB,GACrC,OAAO,CAAC,MAAM,EAAE,CAAC,CAQnB;AAED;;;;;GAKG;AACH,wBAAsB,iBAAiB,CACrC,YAAY,EAAE,MAAM,EACpB,gBAAgB,EAAE,oBAAoB,EACtC,gBAAgB,GAAE,MAAM,EAAU,GACjC,OAAO,CAAC,MAAM,CAAC,CA+CjB"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/agent/browsing/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAI3D,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAClC,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAsBvD,wBAAgB,QAAQ,CAAC,GAAG,EAAE,GAAG,GAAG,GAAG,IAAI,MAAM,CAKhD;AAED,wBAAgB,wBAAwB,CAAC,KAAK,EAAE,MAAM,EAAE,UAIvD;AAED,wBAAsB,kBAAkB,CACtC,YAAY,EAAE,MAAM,EACpB,UAAU,EAAE,MAAM,iBAgBnB;AAgED,wBAAsB,yBAAyB,CAC7C,YAAY,EAAE,MAAM,EACpB,OAAO,EAAE,MAAM,iBAwBhB;AAED,wBAAsB,cAAc,CAAC,EACnC,YAAY,EACZ,cAAc,EACd,QAAQ,GACT,EAAE;IACD,YAAY,EAAE,MAAM,CAAC;IACrB,cAAc,EAAE,MAAM,EAAE,CAAC;IACzB,QAAQ,EAAE,MAAM,CAAC;CAClB,iBAoBA;AAED,wBAAsB,yBAAyB,CAAC,EAC9C,QAAQ,EACR,QAAQ,EACR,KAAK,GACN,EAAE;IACD,QAAQ,EAAE,QAAQ,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB,GAAG,OAAO,CAAC,MAAM,CAAC,CAyDlB;AAyBD,wBAAsB,wBAAwB,CAAC,IAAI,EAAE,IAAI,iBA6HxD;AAED;;;GAGG;AACH,wBAAsB,oBAAoB,CACxC,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,oBAAoB,CAAC,CA2B/B;AAWD,wBAAsB,oBAAoB,CACxC,gBAAgB,EAAE,oBAAoB,GACrC,OAAO,CAAC,MAAM,EAAE,CAAC,CAQnB;AAED;;;;;GAKG;AACH,wBAAsB,iBAAiB,CACrC,YAAY,EAAE,MAAM,EACpB,gBAAgB,EAAE,oBAAoB,EACtC,gBAAgB,GAAE,MAAM,EAAU,GACjC,OAAO,CAAC,MAAM,CAAC,CA+CjB"}

package/dist/agent/browsing/utils.js

@@ -183,7 +183,6 @@ function newContentsWithTestOnly(existingContents, originalTestBlock, updatedTes
 }
 async function injectPwLocatorGenerator(page) {
     let pathToInstalledTestGen = require.resolve(".").split("dist")[0];
-    // eslint-disable-next-line turbo/no-undeclared-env-vars
     if (process.env.RUNNING_BROWSER_TESTS_FOR_TEST_GEN) {
         pathToInstalledTestGen = process.cwd();
     }
@@ -220,7 +219,7 @@ async function injectPwLocatorGenerator(page) {
                             iframeDoc.head.appendChild(script);
                         });
                     }
-                    catch (e) {
+                    catch {
                         console.warn("Error injecting script in iframe.");
                     }
                 };
@@ -264,7 +263,7 @@ async function injectPwLocatorGenerator(page) {
                         iframeDoc.head.appendChild(scr);
                     });
                 }
-                catch (e) {
+                catch {
                     console.warn("Error injecting script in iframe.");
                 }
             };
@@ -284,7 +283,7 @@ async function injectPwLocatorGenerator(page) {
             }
         });
     }
-    catch (e) {
+    catch {
         console.warn("Error injecting script in iframe.");
     }
 }

package/dist/agent/chat/agent-loop.d.ts

@@ -1,19 +1,18 @@
 import { TraceClient } from "@empiricalrun/llm";
 import { IChatModel } from "@empiricalrun/llm/chat";
-import { FileInfo, PendingToolCall, ReporterFunction, StreamingMessageReporterFunc, SupportedChatModels, ToolDefinition } from "@empiricalrun/shared-types";
-import { ToolExecutor } from "../../tools/executor";
-export declare function chatAgentLoop({ chatModel, selectedModel, reporter, streamingMessageReporter, trace, isToolExecutionRemote, toolExecutor, repoInfo, tools, onPendingToolCall, signal, featureFlags, }: {
-    chatModel: IChatModel<any>;
+import { CanonicalMessage, FileInfo, PendingToolCall, ReporterFunction, StreamingMessageReporterFunc, SupportedChatModels, ToolDefinition } from "@empiricalrun/shared-types";
+export declare function chatAgentLoop<T = CanonicalMessage>({ chatModel, selectedModel, reporter, streamingMessageReporter, trace, repoInfo, signal, featureFlags, tools, onPendingToolCall, onLLMResponse, mode, }: {
+    chatModel: IChatModel<T>;
     selectedModel: SupportedChatModels;
-    isToolExecutionRemote: boolean;
-    toolExecutor: ToolExecutor | undefined;
     reporter: ReporterFunction;
     streamingMessageReporter?: StreamingMessageReporterFunc;
     trace?: TraceClient;
     repoInfo: FileInfo;
-    tools: ToolDefinition[];
-    onPendingToolCall: (toolCalls: PendingToolCall[]) => Promise<void>;
     signal?: AbortSignal;
     featureFlags: string[];
+    tools?: ToolDefinition[];
+    onPendingToolCall: (toolCalls: PendingToolCall[]) => Promise<void>;
+    onLLMResponse?: (response: T, selectedModel: SupportedChatModels) => Promise<void>;
+    mode: string;
 }): Promise<void>;
 //# sourceMappingURL=agent-loop.d.ts.map

package/dist/agent/chat/agent-loop.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent-loop.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/agent-loop.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,UAAU,EAAmB,MAAM,wBAAwB,CAAC;AACrE,OAAO,EACL,QAAQ,EACR,eAAe,EACf,gBAAgB,EAChB,4BAA4B,EAC5B,mBAAmB,EACnB,cAAc,EACf,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AAKpD,wBAAsB,aAAa,CAAC,EAClC,SAAS,EACT,aAAa,EACb,QAAQ,EACR,wBAAwB,EACxB,KAAK,EACL,qBAAqB,EACrB,YAAY,EACZ,QAAQ,EACR,KAAK,EACL,iBAAiB,EACjB,MAAM,EACN,YAAY,GACb,EAAE;IACD,SAAS,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;IAC3B,aAAa,EAAE,mBAAmB,CAAC;IACnC,qBAAqB,EAAE,OAAO,CAAC;IAC/B,YAAY,EAAE,YAAY,GAAG,SAAS,CAAC;IACvC,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,wBAAwB,CAAC,EAAE,4BAA4B,CAAC;IACxD,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,QAAQ,EAAE,QAAQ,CAAC;IACnB,KAAK,EAAE,cAAc,EAAE,CAAC;IACxB,iBAAiB,EAAE,CAAC,SAAS,EAAE,eAAe,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACnE,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,YAAY,EAAE,MAAM,EAAE,CAAC;CACxB,GAAG,OAAO,CAAC,IAAI,CAAC,CAoEhB"}
+{"version":3,"file":"agent-loop.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/agent-loop.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EACL,gBAAgB,EAChB,QAAQ,EACR,eAAe,EACf,gBAAgB,EAChB,4BAA4B,EAC5B,mBAAmB,EACnB,cAAc,EACf,MAAM,4BAA4B,CAAC;AAOpC,wBAAsB,aAAa,CAAC,CAAC,GAAG,gBAAgB,EAAE,EACxD,SAAS,EACT,aAAa,EACb,QAAQ,EACR,wBAAwB,EACxB,KAAK,EACL,QAAQ,EACR,MAAM,EACN,YAAY,EACZ,KAAK,EACL,iBAAiB,EACjB,aAAa,EACb,IAAI,GACL,EAAE;IACD,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC;IACzB,aAAa,EAAE,mBAAmB,CAAC;IACnC,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,wBAAwB,CAAC,EAAE,4BAA4B,CAAC;IACxD,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,QAAQ,EAAE,QAAQ,CAAC;IACnB,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,KAAK,CAAC,EAAE,cAAc,EAAE,CAAC;IACzB,iBAAiB,EAAE,CAAC,SAAS,EAAE,eAAe,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACnE,aAAa,CAAC,EAAE,CACd,QAAQ,EAAE,CAAC,EACX,aAAa,EAAE,mBAAmB,KAC/B,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,IAAI,EAAE,MAAM,CAAC;CACd,GAAG,OAAO,CAAC,IAAI,CAAC,CAmEhB"}

package/dist/agent/chat/agent-loop.js

@@ -1,15 +1,18 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.chatAgentLoop = chatAgentLoop;
-const chat_1 = require("@empiricalrun/llm/chat");
+const tools_1 = require("../../tools");
 const prompt_1 = require("./prompt");
 const state_1 = require("./state");
 const utils_1 = require("./utils");
-async function chatAgentLoop({ chatModel, selectedModel, reporter, streamingMessageReporter, trace, isToolExecutionRemote, toolExecutor, repoInfo, tools, onPendingToolCall, signal, featureFlags, }) {
-    const systemPrompt = await (0, prompt_1.buildSystemPrompt)(repoInfo);
+async function chatAgentLoop({ chatModel, selectedModel, reporter, streamingMessageReporter, trace, repoInfo, signal, featureFlags, tools, onPendingToolCall, onLLMResponse, mode, }) {
+    const systemPrompt = await (0, prompt_1.getSystemPrompt)(repoInfo, mode);
     trace?.update({ input: { systemPrompt } });
+    if (!mode && !tools) {
+        throw new Error("Either mode or tools must be provided");
+    }
+    const availableToolsData = tools || (0, tools_1.availableTools)(selectedModel, mode === "triage");
     while (!chatModel.askUserForInput) {
-        // Abort if signal is set
         if (signal?.aborted) {
             (0, utils_1.log)("[chatAgentLoop] Aborted by signal");
             break;
@@ -18,32 +21,25 @@ async function chatAgentLoop({ chatModel, selectedModel, reporter, streamingMess
             const toolCalls = chatModel.getPendingToolCalls();
             if (toolCalls.length > 0) {
                 await onPendingToolCall(toolCalls);
-                if (isToolExecutionRemote) {
-                    (0, utils_1.log)(`Tool call remote execution in progress`);
-                    break;
-                }
-                else {
-                    if (!toolExecutor) {
-                        throw new Error("ToolExecutor is required for local execution");
-                    }
-                    const toolResults = await toolExecutor.execute(toolCalls);
-                    chatModel.pushToolResultsMessage(toolCalls, toolResults);
-                }
+                break;
             }
             (0, utils_1.log)(`${(0, utils_1.getModelName)(selectedModel)} is working...`);
             const response = await chatModel.getLLMResponse({
                 systemPrompt,
-                tools: tools.map((tool) => (0, chat_1.zodToOpenAITool)(tool.schema)),
+                tools: availableToolsData.map((t) => t.schema),
                 selectedModel,
                 trace,
                 streamingMessageReporter,
-                hasThinkingEnabled: featureFlags.includes("extendedThinkingClaude"),
+                hasThinkingEnabled: true,
                 hasInterleavedThinkingEnabled: featureFlags.includes("interleavedThinkingClaude"),
             });
             if (!response) {
                 throw new Error("Error getting response from LLM");
             }
             chatModel.pushMessage(response);
+            if (onLLMResponse) {
+                await onLLMResponse(response, selectedModel);
+            }
             const latest = chatModel.getHumanReadableLatestMessage();
             if (signal?.aborted) {
                 (0, utils_1.log)("[chatAgentLoop] Aborted by signal");

package/dist/agent/chat/exports.d.ts

@@ -1,8 +1,9 @@
-export { getFileInfoFromGitHub } from "../../file-info/github";
+export { getFileInfoFromGitHub, viewFileUsingGitHub, } from "../../file-info/adapters/github";
+export { allToolsDefinitions } from "../../tools";
 export { chatAgentLoop } from "./agent-loop";
-export { defaultModel } from "./models";
-export { CHAT_STATE_VERSIONS_MIGRATIONS_MAP, chatStateFromModel, createChatState, createChatStateForMessages, fetchToolCallAvailability, getLatestDownloadBuildUrl, LATEST_CHAT_STATE_VERSION, migrateChatState, processAttachments, } from "./state";
-export { extractAttachments } from "./utils";
+export { getDefaultChatModelId, resolveChatModelBasedOnInput } from "./models";
+export { CHAT_STATE_VERSIONS_MIGRATIONS_MAP, chatStateFromModel, createChatState, createChatStateForMessages, fetchToolCallFromId, getLatestDownloadBuildUrl, LATEST_CHAT_STATE_VERSION, migrateChatState, processAttachments, } from "./state";
+export { extractAttachments, getChatAgentMode } from "./utils";
 export { processToolCallsAndUpdateChatState, rejectPendingToolCalls, } from "./utils/tool-calls";
 export type { IChatModel } from "@empiricalrun/llm/chat";
 export { createChatModel, SUPPORTED_CHAT_MODELS } from "@empiricalrun/llm/chat";

package/dist/agent/chat/exports.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"exports.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/exports.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,EAAE,MAAM,wBAAwB,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AACxC,OAAO,EACL,kCAAkC,EAClC,kBAAkB,EAClB,eAAe,EACf,0BAA0B,EAC1B,yBAAyB,EACzB,yBAAyB,EACzB,yBAAyB,EACzB,gBAAgB,EAChB,kBAAkB,GACnB,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,kBAAkB,EAAE,MAAM,SAAS,CAAC;AAC7C,OAAO,EACL,kCAAkC,EAClC,sBAAsB,GACvB,MAAM,oBAAoB,CAAC;AAC5B,YAAY,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,wBAAwB,CAAC"}
+{"version":3,"file":"exports.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/exports.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,qBAAqB,EACrB,mBAAmB,GACpB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,EAAE,qBAAqB,EAAE,4BAA4B,EAAE,MAAM,UAAU,CAAC;AAC/E,OAAO,EACL,kCAAkC,EAClC,kBAAkB,EAClB,eAAe,EACf,0BAA0B,EAC1B,mBAAmB,EACnB,yBAAyB,EACzB,yBAAyB,EACzB,gBAAgB,EAChB,kBAAkB,GACnB,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAC/D,OAAO,EACL,kCAAkC,EAClC,sBAAsB,GACvB,MAAM,oBAAoB,CAAC;AAC5B,YAAY,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,wBAAwB,CAAC"}

package/dist/agent/chat/exports.js

@@ -1,24 +1,29 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.SUPPORTED_CHAT_MODELS = exports.createChatModel = exports.rejectPendingToolCalls = exports.processToolCallsAndUpdateChatState = exports.extractAttachments = exports.processAttachments = exports.migrateChatState = exports.LATEST_CHAT_STATE_VERSION = exports.getLatestDownloadBuildUrl = exports.fetchToolCallAvailability = exports.createChatStateForMessages = exports.createChatState = exports.chatStateFromModel = exports.CHAT_STATE_VERSIONS_MIGRATIONS_MAP = exports.defaultModel = exports.chatAgentLoop = exports.getFileInfoFromGitHub = void 0;
-var github_1 = require("../../file-info/github");
+exports.SUPPORTED_CHAT_MODELS = exports.createChatModel = exports.rejectPendingToolCalls = exports.processToolCallsAndUpdateChatState = exports.getChatAgentMode = exports.extractAttachments = exports.processAttachments = exports.migrateChatState = exports.LATEST_CHAT_STATE_VERSION = exports.getLatestDownloadBuildUrl = exports.fetchToolCallFromId = exports.createChatStateForMessages = exports.createChatState = exports.chatStateFromModel = exports.CHAT_STATE_VERSIONS_MIGRATIONS_MAP = exports.resolveChatModelBasedOnInput = exports.getDefaultChatModelId = exports.chatAgentLoop = exports.allToolsDefinitions = exports.viewFileUsingGitHub = exports.getFileInfoFromGitHub = void 0;
+var github_1 = require("../../file-info/adapters/github");
 Object.defineProperty(exports, "getFileInfoFromGitHub", { enumerable: true, get: function () { return github_1.getFileInfoFromGitHub; } });
+Object.defineProperty(exports, "viewFileUsingGitHub", { enumerable: true, get: function () { return github_1.viewFileUsingGitHub; } });
+var tools_1 = require("../../tools");
+Object.defineProperty(exports, "allToolsDefinitions", { enumerable: true, get: function () { return tools_1.allToolsDefinitions; } });
 var agent_loop_1 = require("./agent-loop");
 Object.defineProperty(exports, "chatAgentLoop", { enumerable: true, get: function () { return agent_loop_1.chatAgentLoop; } });
 var models_1 = require("./models");
-Object.defineProperty(exports, "defaultModel", { enumerable: true, get: function () { return models_1.defaultModel; } });
+Object.defineProperty(exports, "getDefaultChatModelId", { enumerable: true, get: function () { return models_1.getDefaultChatModelId; } });
+Object.defineProperty(exports, "resolveChatModelBasedOnInput", { enumerable: true, get: function () { return models_1.resolveChatModelBasedOnInput; } });
 var state_1 = require("./state");
 Object.defineProperty(exports, "CHAT_STATE_VERSIONS_MIGRATIONS_MAP", { enumerable: true, get: function () { return state_1.CHAT_STATE_VERSIONS_MIGRATIONS_MAP; } });
 Object.defineProperty(exports, "chatStateFromModel", { enumerable: true, get: function () { return state_1.chatStateFromModel; } });
 Object.defineProperty(exports, "createChatState", { enumerable: true, get: function () { return state_1.createChatState; } });
 Object.defineProperty(exports, "createChatStateForMessages", { enumerable: true, get: function () { return state_1.createChatStateForMessages; } });
-Object.defineProperty(exports, "fetchToolCallAvailability", { enumerable: true, get: function () { return state_1.fetchToolCallAvailability; } });
+Object.defineProperty(exports, "fetchToolCallFromId", { enumerable: true, get: function () { return state_1.fetchToolCallFromId; } });
 Object.defineProperty(exports, "getLatestDownloadBuildUrl", { enumerable: true, get: function () { return state_1.getLatestDownloadBuildUrl; } });
 Object.defineProperty(exports, "LATEST_CHAT_STATE_VERSION", { enumerable: true, get: function () { return state_1.LATEST_CHAT_STATE_VERSION; } });
 Object.defineProperty(exports, "migrateChatState", { enumerable: true, get: function () { return state_1.migrateChatState; } });
 Object.defineProperty(exports, "processAttachments", { enumerable: true, get: function () { return state_1.processAttachments; } });
 var utils_1 = require("./utils");
 Object.defineProperty(exports, "extractAttachments", { enumerable: true, get: function () { return utils_1.extractAttachments; } });
+Object.defineProperty(exports, "getChatAgentMode", { enumerable: true, get: function () { return utils_1.getChatAgentMode; } });
 var tool_calls_1 = require("./utils/tool-calls");
 Object.defineProperty(exports, "processToolCallsAndUpdateChatState", { enumerable: true, get: function () { return tool_calls_1.processToolCallsAndUpdateChatState; } });
 Object.defineProperty(exports, "rejectPendingToolCalls", { enumerable: true, get: function () { return tool_calls_1.rejectPendingToolCalls; } });

package/dist/agent/chat/filesystem-cache.d.ts

@@ -0,0 +1,12 @@
+import { LLMCache } from "@empiricalrun/llm/chat";
+export declare class FilesystemLLMCache implements LLMCache {
+    private cacheDir;
+    constructor(cacheDir?: string);
+    private ensureCacheDir;
+    private getCacheFilePath;
+    get(key: string): Promise<string | null>;
+    set(key: string, value: string): Promise<void>;
+    clear(): Promise<void>;
+    size(): Promise<number>;
+}
+//# sourceMappingURL=filesystem-cache.d.ts.map

package/dist/agent/chat/filesystem-cache.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"filesystem-cache.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/filesystem-cache.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAKlD,qBAAa,kBAAmB,YAAW,QAAQ;IACjD,OAAO,CAAC,QAAQ,CAAS;gBAEb,QAAQ,CAAC,EAAE,MAAM;YAIf,cAAc;IAQ5B,OAAO,CAAC,gBAAgB;IAKlB,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAWxC,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAU9C,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IActB,IAAI,IAAI,OAAO,CAAC,MAAM,CAAC;CAS9B"}

package/dist/agent/chat/filesystem-cache.js

@@ -0,0 +1,101 @@
+"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 __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FilesystemLLMCache = void 0;
+const crypto = __importStar(require("crypto"));
+const fs = __importStar(require("fs/promises"));
+const path = __importStar(require("path"));
+class FilesystemLLMCache {
+    cacheDir;
+    constructor(cacheDir) {
+        this.cacheDir = cacheDir || path.join(process.cwd(), ".llm-cache");
+    }
+    async ensureCacheDir() {
+        try {
+            await fs.access(this.cacheDir);
+        }
+        catch {
+            await fs.mkdir(this.cacheDir, { recursive: true });
+        }
+    }
+    getCacheFilePath(key) {
+        const hash = crypto.createHash("sha256").update(key).digest("hex");
+        return path.join(this.cacheDir, `${hash}.json`);
+    }
+    async get(key) {
+        try {
+            await this.ensureCacheDir();
+            const filePath = this.getCacheFilePath(key);
+            const data = await fs.readFile(filePath, "utf-8");
+            return data;
+        }
+        catch {
+            return null;
+        }
+    }
+    async set(key, value) {
+        try {
+            await this.ensureCacheDir();
+            const filePath = this.getCacheFilePath(key);
+            await fs.writeFile(filePath, value, "utf-8");
+        }
+        catch (error) {
+            console.warn("Failed to write to LLM cache:", error);
+        }
+    }
+    async clear() {
+        try {
+            await this.ensureCacheDir();
+            const files = await fs.readdir(this.cacheDir);
+            await Promise.all(files
+                .filter((file) => file.endsWith(".json"))
+                .map((file) => fs.unlink(path.join(this.cacheDir, file))));
+        }
+        catch (error) {
+            console.warn("Failed to clear LLM cache:", error);
+        }
+    }
+    async size() {
+        try {
+            await this.ensureCacheDir();
+            const files = await fs.readdir(this.cacheDir);
+            return files.filter((file) => file.endsWith(".json")).length;
+        }
+        catch {
+            return 0;
+        }
+    }
+}
+exports.FilesystemLLMCache = FilesystemLLMCache;

package/dist/agent/chat/index.d.ts

@@ -1,8 +1,11 @@
 import { SupportedChatModels } from "@empiricalrun/shared-types";
 export declare function fetchEnvironmentVariables(): Promise<Record<string, string>>;
-export declare function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialPromptContent, }: {
+export declare function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialPromptContent, useTriage, resetChat, useFSCache, }: {
     selectedModel: SupportedChatModels;
     useDiskForChatState: boolean;
     initialPromptContent: string | undefined;
+    useTriage: boolean;
+    useFSCache: boolean;
+    resetChat: boolean;
 }): Promise<void>;
 //# sourceMappingURL=index.d.ts.map

package/dist/agent/chat/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAKL,mBAAmB,EACpB,MAAM,4BAA4B,CAAC;AAkCpC,wBAAsB,yBAAyB,IAAI,OAAO,CACxD,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CACvB,CAqBA;AAED,wBAAsB,kBAAkB,CAAC,EACvC,mBAAmB,EACnB,aAAa,EACb,oBAAoB,GACrB,EAAE;IACD,aAAa,EAAE,mBAAmB,CAAC;IACnC,mBAAmB,EAAE,OAAO,CAAC;IAC7B,oBAAoB,EAAE,MAAM,GAAG,SAAS,CAAC;CAC1C,iBA4JA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAML,mBAAmB,EACpB,MAAM,4BAA4B,CAAC;AAwCpC,wBAAsB,yBAAyB,IAAI,OAAO,CACxD,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CACvB,CAkCA;AAED,wBAAsB,kBAAkB,CAAC,EACvC,mBAAmB,EACnB,aAAa,EACb,oBAAoB,EACpB,SAAS,EACT,SAAS,EACT,UAAU,GACX,EAAE;IACD,aAAa,EAAE,mBAAmB,CAAC;IACnC,mBAAmB,EAAE,OAAO,CAAC;IAC7B,oBAAoB,EAAE,MAAM,GAAG,SAAS,CAAC;IACzC,SAAS,EAAE,OAAO,CAAC;IACnB,UAAU,EAAE,OAAO,CAAC;IACpB,SAAS,EAAE,OAAO,CAAC;CACpB,iBA2KA"}

package/dist/agent/chat/index.js

@@ -6,29 +6,42 @@ const llm_1 = require("@empiricalrun/llm");
 const chat_1 = require("@empiricalrun/llm/chat");
 const picocolors_1 = require("picocolors");
 const client_1 = require("../../dashboard/client");
-const file_system_1 = require("../../file-info/file-system");
+const reader_1 = require("../../file-info/adapters/file-system/reader");
 const human_in_the_loop_1 = require("../../human-in-the-loop");
-const tools_1 = require("../../tools");
+const validation_1 = require("../../recorder/validation");
 const executor_1 = require("../../tools/executor");
 const git_1 = require("../../tools/executor/utils/git");
 const agent_loop_1 = require("./agent-loop");
+const filesystem_cache_1 = require("./filesystem-cache");
 const state_1 = require("./state");
 const utils_1 = require("./utils");
 function stopCriteria(userPrompt) {
     return userPrompt?.toLowerCase() === "stop";
 }
 function concludeAgent(chatModel, useDiskForChatState, selectedModel, error) {
-    console.log(`\n${(0, picocolors_1.gray)("Usage summary -> " + chatModel.getUsageSummary())}`);
+    console.log(`\n${(0, picocolors_1.gray)("Usage summary -> " + (0, state_1.getUsageSummary)(chatModel))}`);
     if (useDiskForChatState) {
         (0, state_1.saveToDisk)(chatModel.messages, selectedModel, chatModel.askUserForInput, error);
     }
 }
 async function fetchEnvironmentVariables() {
-    const apiClient = new client_1.DashboardAPIClient({
-        authType: "project-api-key",
-        projectApiKey: process.env.EMPIRICALRUN_API_KEY,
-    });
-    const data = await apiClient.request("/api/environment-variables", { method: "GET" });
+    const projectApiKey = process.env.EMPIRICALRUN_API_KEY;
+    let apiClient;
+    let queryParams = {};
+    if (projectApiKey) {
+        apiClient = new client_1.DashboardAPIClient({
+            authType: "project-api-key",
+            projectApiKey: process.env.EMPIRICALRUN_API_KEY,
+        });
+    }
+    else {
+        apiClient = new client_1.DashboardAPIClient({
+            authType: "user-access-token",
+        });
+        const repoName = await (0, validation_1.validatePackageJson)(process.cwd());
+        queryParams = { project_repo_name: repoName };
+    }
+    const data = await apiClient.request("/api/environment-variables", { method: "GET", params: queryParams });
     if (!data.data || !data.data.environment_variables) {
         console.error("Failed to fetch environment variables:", data);
         throw new Error("Failed to fetch environment variables");
@@ -39,9 +52,13 @@ async function fetchEnvironmentVariables() {
     }, {});
     return envVars;
 }
-async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialPromptContent, }) {
+async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialPromptContent, useTriage, resetChat, useFSCache, }) {
     let chatState;
-    let enableStreaming = true;
+    const enableStreaming = !useFSCache;
+    const cache = useFSCache ? new filesystem_cache_1.FilesystemLLMCache() : undefined;
+    if (resetChat) {
+        (0, state_1.clearChatState)();
+    }
     if (useDiskForChatState) {
         chatState = (0, state_1.loadChatState)();
     }
@@ -50,11 +67,10 @@ async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialP
     const branchName = `branch-${randomId}`;
     await (0, git_1.checkoutBranch)(branchName, process.cwd());
     let messagesLoadedFromDisk = chatState?.messages || [];
-    let chatModel = (0, chat_1.createChatModel)(messagesLoadedFromDisk, selectedModel);
+    let chatModel = (0, chat_1.createChatModel)(messagesLoadedFromDisk, selectedModel, undefined, cache);
     chatModel.validateEnvVarsForAuth();
     if (initialPromptContent && chatModel.messages.length === 0) {
         chatModel.pushUserMessage(initialPromptContent, []);
-        chatModel.askUserForInput = false;
     }
     else if (initialPromptContent && chatModel.messages.length > 0) {
         console.warn(`Ignoring initial prompt because we have existing messages.`);
@@ -99,6 +115,13 @@ async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialP
         const traceUrl = trace.getTraceUrl();
         console.log(`Starting ${selectedModel}: ${traceUrl}`);
     }
+    let authType = "user-access-token";
+    if (process.env.EMPIRICALRUN_API_KEY) {
+        authType = "project-api-key";
+    }
+    const apiClient = new client_1.DashboardAPIClient({
+        authType,
+    });
     while (!stopCriteria(userPrompt)) {
         if (chatModel.askUserForInput) {
             try {
@@ -132,14 +155,12 @@ async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialP
                 chatSession: null,
                 branchName,
                 repoPath: process.cwd(),
-                apiClient: new client_1.DashboardAPIClient({
-                    authType: "user-access-token",
-                }),
+                apiClient,
                 trace,
                 featureFlags: [],
                 environmentOverrides: await fetchEnvironmentVariables(),
             });
-            const fileInfo = await (0, file_system_1.getFileInfoFromFS)(process.cwd());
+            const fileInfo = await (0, reader_1.getFileInfoFromFS)(process.cwd());
             await (0, agent_loop_1.chatAgentLoop)({
                 chatModel,
                 selectedModel,
@@ -170,16 +191,17 @@ async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialP
                 })(),
                 trace,
                 repoInfo: fileInfo,
-                isToolExecutionRemote: false,
-                toolExecutor,
-                tools: (0, tools_1.availableTools)(selectedModel),
-                onPendingToolCall: async () => { },
-                featureFlags: ["extendedThinkingClaude"],
+                onPendingToolCall: async (toolCalls) => {
+                    const toolResults = await toolExecutor.execute(toolCalls);
+                    chatModel.pushToolResultsMessage(toolCalls, toolResults);
+                },
+                featureFlags: [],
+                mode: (0, utils_1.getChatAgentMode)(useTriage ? "triage" : "test_generation"),
             });
         }
     }
     trace?.update({ output: { messages: chatModel.messages } });
     await llm_1.langfuseInstance?.flushAsync();
-    const usageSummary = chatModel.getUsageSummary();
+    const usageSummary = (0, state_1.getUsageSummary)(chatModel);
     console.log(`\n${(0, picocolors_1.gray)("Usage summary -> " + usageSummary)}`);
 }

package/dist/agent/chat/models.d.ts

@@ -1,4 +1,5 @@
 import { SupportedChatModels } from "@empiricalrun/shared-types";
-export declare const defaultModel: SupportedChatModels;
+export declare function getDefaultChatModelId(): SupportedChatModels;
+export declare const resolveChatModelBasedOnInput: (requestedModel?: string) => SupportedChatModels;
 export declare const modelLabels: Record<SupportedChatModels, string>;
 //# sourceMappingURL=models.d.ts.map