From 62bee249243e4a0024109dd4c94bba47ec62e94a Mon Sep 17 00:00:00 2001
From: timothycarambat <rambat1010@gmail.com>
Date: Thu, 5 Jun 2025 07:34:31 -0700
Subject: [PATCH] remove LLM Instruction `inputVariable` requirement

---
 .../Admin/AgentBuilder/BlockList/index.jsx    |  1 -
 .../nodes/LLMInstructionNode/index.jsx        | 11 -----------
 .../agentFlows/executors/llm-instruction.js   | 19 ++++---------------
 server/utils/agentFlows/flowTypes.js          |  4 ----
 4 files changed, 4 insertions(+), 31 deletions(-)

diff --git a/frontend/src/pages/Admin/AgentBuilder/BlockList/index.jsx b/frontend/src/pages/Admin/AgentBuilder/BlockList/index.jsx
index 2bdf18c3ddc..eaaf05b89eb 100644
--- a/frontend/src/pages/Admin/AgentBuilder/BlockList/index.jsx
+++ b/frontend/src/pages/Admin/AgentBuilder/BlockList/index.jsx
@@ -115,7 +115,6 @@ const BLOCK_INFO = {
     description: "Process data using LLM instructions",
     defaultConfig: {
       instruction: "",
-      inputVariable: "",
       resultVariable: "",
     },
     getSummary: (config) => config.instruction || "No instruction",
diff --git a/frontend/src/pages/Admin/AgentBuilder/nodes/LLMInstructionNode/index.jsx b/frontend/src/pages/Admin/AgentBuilder/nodes/LLMInstructionNode/index.jsx
index 24b7821b5de..bd52f81db5f 100644
--- a/frontend/src/pages/Admin/AgentBuilder/nodes/LLMInstructionNode/index.jsx
+++ b/frontend/src/pages/Admin/AgentBuilder/nodes/LLMInstructionNode/index.jsx
@@ -7,17 +7,6 @@ export default function LLMInstructionNode({
 }) {
   return (
     <div className="space-y-4">
-      <div>
-        <label className="block text-sm font-medium text-theme-text-primary mb-2">
-          Input Variable
-        </label>
-        {renderVariableSelect(
-          config.inputVariable,
-          (value) => onConfigChange({ ...config, inputVariable: value }),
-          "Select input variable"
-        )}
-      </div>
-
       <div>
         <label className="block text-sm font-medium text-theme-text-primary mb-2">
           Instruction
diff --git a/server/utils/agentFlows/executors/llm-instruction.js b/server/utils/agentFlows/executors/llm-instruction.js
index b5000a89c10..4c4b0d3b6fa 100644
--- a/server/utils/agentFlows/executors/llm-instruction.js
+++ b/server/utils/agentFlows/executors/llm-instruction.js
@@ -1,24 +1,17 @@
 /**
  * Execute an LLM instruction flow step
  * @param {Object} config Flow step configuration
- * @param {{introspect: Function, variables: Object, logger: Function}} context Execution context with introspect function
+ * @param {{introspect: Function, logger: Function}} context Execution context with introspect function
  * @returns {Promise<string>} Processed result
  */
 async function executeLLMInstruction(config, context) {
-  const { instruction, inputVariable, resultVariable } = config;
-  const { introspect, variables, logger, aibitat } = context;
+  const { instruction, resultVariable } = config;
+  const { introspect, logger, aibitat } = context;
   logger(
     `\x1b[43m[AgentFlowToolExecutor]\x1b[0m - executing LLM Instruction block`
   );
   introspect(`Processing data with LLM instruction...`);
 
-  if (!variables[inputVariable]) {
-    logger(`Input variable ${inputVariable} (${inputVariable}) not found`);
-    throw new Error(
-      `Input variable ${inputVariable} (${inputVariable}) not found`
-    );
-  }
-
   try {
     logger(
       `Sending request to LLM (${aibitat.defaultProvider.provider}::${aibitat.defaultProvider.model})`
@@ -26,16 +19,12 @@ async function executeLLMInstruction(config, context) {
     introspect(`Sending request to LLM...`);
 
     // Ensure the input is a string since we are sending it to the LLM direct as a message
-    let input = variables[inputVariable];
+    let input = instruction;
     if (typeof input === "object") input = JSON.stringify(input);
     if (typeof input !== "string") input = String(input);
 
     const provider = aibitat.getProviderForConfig(aibitat.defaultProvider);
     const completion = await provider.complete([
-      {
-        role: "system",
-        content: `Follow these instructions carefully: ${instruction}`,
-      },
       {
         role: "user",
         content: input,
diff --git a/server/utils/agentFlows/flowTypes.js b/server/utils/agentFlows/flowTypes.js
index bde8d3073ed..f4bd2a6db79 100644
--- a/server/utils/agentFlows/flowTypes.js
+++ b/server/utils/agentFlows/flowTypes.js
@@ -104,10 +104,6 @@ const FLOW_TYPES = {
         type: "string",
         description: "The instruction for the LLM to follow",
       },
-      inputVariable: {
-        type: "string",
-        description: "Variable containing the input data to process",
-      },
       resultVariable: {
         type: "string",
         description: "Variable to store the processed result",