wave-agent-sdk 0.16.12 → 0.17.0

package/src/utils/containerSetup.ts

@@ -81,6 +81,12 @@ export function setupAgentContainer(
   container.register("AgentOptions", options);
   container.register("Workdir", workdir);
 
+  const mergedEnv: Record<string, string> = {
+    ...(process.env as Record<string, string>),
+    ...(options.env || {}),
+  };
+  container.register("MergedEnv", mergedEnv);
+
   if (options.worktreeName) {
     container.register("WorktreeName", options.worktreeName);
     container.register("MainRepoRoot", getGitMainRepoRoot(workdir));
@@ -218,9 +224,7 @@ export function setupAgentContainer(
             cwd: workdir,
             toolName: context.toolName,
             toolInput: context.toolInput,
-            env: Object.fromEntries(
-              Object.entries(process.env).filter((e) => e[1] !== undefined),
-            ) as Record<string, string>,
+            env: mergedEnv,
           });
 
           if (results.length > 0) {

package/src/utils/editUtils.ts

@@ -10,8 +10,11 @@ export function escapeRegExp(string: string): string {
 }
 
 /**
- * Returns a generic error message when old_string is not found.
+ * Returns an error message when old_string is not found, including
+ * the attempted string to help the model self-correct on retry.
  */
-export function analyzeEditMismatch(): string {
-  return "old_string not found in file";
+export function analyzeEditMismatch(oldString: string): string {
+  const displayString =
+    oldString.length > 200 ? oldString.substring(0, 200) + "..." : oldString;
+  return `String to replace not found in file.\nString: ${displayString}`;
 }