From 97dd3dec9559175a59c627048837c835fd607489 Mon Sep 17 00:00:00 2001
From: Michael Chou <michael.chou@scale.com>
Date: Tue, 5 May 2026 16:23:59 -0700
Subject: [PATCH 1/4] added ui to support input params for task creation in dev
 environment

---
 .../primary-content/prompt-input.tsx          | 36 +++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/agentex-ui/components/primary-content/prompt-input.tsx b/agentex-ui/components/primary-content/prompt-input.tsx
index 5d872c4..487f339 100644
--- a/agentex-ui/components/primary-content/prompt-input.tsx
+++ b/agentex-ui/components/primary-content/prompt-input.tsx
@@ -49,6 +49,9 @@ export function PromptInput({ prompt, setPrompt }: PromptInputProps) {
   const { taskID, agentName, updateParams } = useSafeSearchParams();
   const [isClient, setIsClient] = useState(false);
   const [isSendingJSON, setIsSendingJSON] = useState(false);
+  const [isTaskParamsOpen, setIsTaskParamsOpen] = useState(false);
+  const [taskParams, setTaskParams] = useState('{}');
+  const taskParamsViewRef = useRef<EditorView | null>(null);
 
   const { agentexClient } = useAgentexClient();
 
@@ -119,11 +122,22 @@ export function PromptInput({ prompt, setPrompt }: PromptInputProps) {
     setPrompt('');
 
     if (!currentTaskId) {
+      let extraTaskParams: Record<string, unknown> = {};
+      if (process.env.NODE_ENV === 'development' && taskParams.trim()) {
+        try {
+          extraTaskParams = JSON.parse(taskParams);
+        } catch {
+          toast.error('Invalid Task Parameters JSON');
+          return;
+        }
+      }
+
       const task = await createTaskMutation.mutateAsync({
         agentName: agentName,
         params: {
           description: prompt,
           content: currentPrompt,
+          ...extraTaskParams,
         },
       });
       currentTaskId = task.id;
@@ -159,10 +173,32 @@ export function PromptInput({ prompt, setPrompt }: PromptInputProps) {
     sendMessageMutation,
     setPrompt,
     isSendingJSON,
+    taskParams,
   ]);
 
   return (
     <div className="flex w-full max-w-3xl flex-col gap-2">
+      {process.env.NODE_ENV === 'development' && !taskID && (
+        <div className="flex flex-col gap-1">
+          <button
+            type="button"
+            className="text-muted-foreground hover:text-foreground ml-4 flex items-center gap-1 text-sm transition-colors"
+            onClick={() => setIsTaskParamsOpen(v => !v)}
+          >
+            <span>{isTaskParamsOpen ? '▾' : '▸'}</span>
+            Task Parameters
+          </button>
+          {isTaskParamsOpen && (
+            <DataInput
+              prompt={taskParams}
+              setPrompt={setTaskParams}
+              isDisabled={isDisabled}
+              handleSendPrompt={handleSendPrompt}
+              codeMirrorViewRef={taskParamsViewRef}
+            />
+          )}
+        </div>
+      )}
       <div
         className={`border-input dark:bg-input ${isDisabled ? 'bg-muted scale-90 cursor-not-allowed' : 'scale-100'} flex w-full items-center justify-between rounded-4xl border py-2 pr-2 pl-6 shadow-sm transition-transform duration-300 disabled:cursor-not-allowed`}
       >

From ea01ea0d70932752c423dc6a21f23833cdfb3619 Mon Sep 17 00:00:00 2001
From: Michael Chou <michael.chou@scale.com>
Date: Wed, 6 May 2026 09:11:54 -0700
Subject: [PATCH 2/4] added ui to support input params for task creation in dev
 environment

---
 agentex-ui/components/primary-content/prompt-input.tsx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/agentex-ui/components/primary-content/prompt-input.tsx b/agentex-ui/components/primary-content/prompt-input.tsx
index 487f339..b7d4a04 100644
--- a/agentex-ui/components/primary-content/prompt-input.tsx
+++ b/agentex-ui/components/primary-content/prompt-input.tsx
@@ -135,9 +135,9 @@ export function PromptInput({ prompt, setPrompt }: PromptInputProps) {
       const task = await createTaskMutation.mutateAsync({
         agentName: agentName,
         params: {
+          ...extraTaskParams,
           description: prompt,
           content: currentPrompt,
-          ...extraTaskParams,
         },
       });
       currentTaskId = task.id;

From 6ad266b7a214d2a63908297d4c7a58151390d663 Mon Sep 17 00:00:00 2001
From: Michael Chou <michael.chou@scale.com>
Date: Wed, 6 May 2026 13:02:01 -0700
Subject: [PATCH 3/4] added ui to support input params for task creation in dev
 environment

---
 agentex-ui/components/primary-content/prompt-input.tsx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/agentex-ui/components/primary-content/prompt-input.tsx b/agentex-ui/components/primary-content/prompt-input.tsx
index b7d4a04..7077d37 100644
--- a/agentex-ui/components/primary-content/prompt-input.tsx
+++ b/agentex-ui/components/primary-content/prompt-input.tsx
@@ -50,7 +50,7 @@ export function PromptInput({ prompt, setPrompt }: PromptInputProps) {
   const [isClient, setIsClient] = useState(false);
   const [isSendingJSON, setIsSendingJSON] = useState(false);
   const [isTaskParamsOpen, setIsTaskParamsOpen] = useState(false);
-  const [taskParams, setTaskParams] = useState('{}');
+  const [taskParams, setTaskParams] = useState('');
   const taskParamsViewRef = useRef<EditorView | null>(null);
 
   const { agentexClient } = useAgentexClient();

From b6739c50f3db59bd399759b8d532e2dda100a717 Mon Sep 17 00:00:00 2001
From: Michael Chou <michael.chou@scale.com>
Date: Wed, 6 May 2026 13:32:55 -0700
Subject: [PATCH 4/4] added ui to support input params for task creation in dev
 environment

---
 agentex-ui/components/primary-content/prompt-input.tsx | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/agentex-ui/components/primary-content/prompt-input.tsx b/agentex-ui/components/primary-content/prompt-input.tsx
index 7077d37..45f0f9d 100644
--- a/agentex-ui/components/primary-content/prompt-input.tsx
+++ b/agentex-ui/components/primary-content/prompt-input.tsx
@@ -123,7 +123,7 @@ export function PromptInput({ prompt, setPrompt }: PromptInputProps) {
 
     if (!currentTaskId) {
       let extraTaskParams: Record<string, unknown> = {};
-      if (process.env.NODE_ENV === 'development' && taskParams.trim()) {
+      if (taskParams.trim()) {
         try {
           extraTaskParams = JSON.parse(taskParams);
         } catch {
@@ -178,7 +178,7 @@ export function PromptInput({ prompt, setPrompt }: PromptInputProps) {
 
   return (
     <div className="flex w-full max-w-3xl flex-col gap-2">
-      {process.env.NODE_ENV === 'development' && !taskID && (
+      {!taskID && !isDisabled && (
         <div className="flex flex-col gap-1">
           <button
             type="button"