feat: add context handling to sampling prompts in executors

Author: yaonyan

packages/core/src/executors/sampling/agentic-sampling-executor.ts

@@ -99,6 +99,9 @@ export class SamplingExecutor extends BaseSamplingExecutor {
     const systemPrompt = this.buildSystemPrompt(
       args.userRequest as string,
       agenticSchema,
+      (args.context && typeof args.context === "object")
+        ? args.context as Record<string, unknown>
+        : undefined,
     );
     return this.runSamplingLoop(() => systemPrompt, agenticSchema);
   }
@@ -151,6 +154,7 @@ export class SamplingExecutor extends BaseSamplingExecutor {
   private buildSystemPrompt(
     userRequest: string,
     agenticSchema: Record<string, unknown>,
+    context?: Record<string, unknown>,
   ): string {
     const toolList = this.allToolNames
       .map((name) => {
@@ -168,7 +172,12 @@ export class SamplingExecutor extends BaseSamplingExecutor {
       })
       .join("\n");
 
-    // Build the complete schema using forAgentic
+    let contextInfo = "";
+    if (
+      context && typeof context === "object" && Object.keys(context).length > 0
+    ) {
+      contextInfo = `\n\nContext:\n${JSON.stringify(context, null, 2)}`;
+    }
 
     // Use compiled sampling prompt
     const basePrompt = CompiledPrompts.samplingExecution({
@@ -180,7 +189,7 @@ export class SamplingExecutor extends BaseSamplingExecutor {
     const taskPrompt = `
 
 ## Current Task
-I will now use agentic sampling to complete the following task: "${userRequest}"
+I will now use agentic sampling to complete the following task: "${userRequest}"${contextInfo}
 
 When I need to use a tool, I should specify the tool name in 'action' and provide tool-specific parameters as additional properties.
 When the task is complete, I should use "action": "complete".`;

packages/core/src/executors/sampling/workflow-sampling-executor.ts

@@ -132,8 +132,15 @@ export class WorkflowSamplingExecutor extends BaseSamplingExecutor {
     });
 
     // Create workflow-specific sampling prompt using existing patterns
+    let contextInfo = "";
+    if (
+      args.context && typeof args.context === "object" &&
+      Object.keys(args.context).length > 0
+    ) {
+      contextInfo = `\n\nContext:\n${JSON.stringify(args.context, null, 2)}`;
+    }
     const workflowPrompt =
-      `\n\nCurrent Task: <user_request>${args.userRequest}</user_request>`;
+      `\n\nCurrent Task: <user_request>${args.userRequest}</user_request>${contextInfo}`;
 
     // Use JSON instruction injection pattern
     return this.injectJsonInstruction({

packages/core/src/factories/args-def-factory.ts

@@ -154,8 +154,14 @@ Workflow step definitions - provide ONLY on initial call.
             description:
               "The task or request that should be completed autonomously by the agentic system using available tools",
           },
+          context: {
+            type: "object",
+            description:
+              "Necessary context for the request, e.g., the absolute path of the current working directory. This is just an example; any relevant context fields are allowed.",
+            additionalProperties: true,
+          },
         },
-        required: ["userRequest"],
+        required: ["userRequest", "context"],
       };
     },