diff --git a/blocks/configs/agent.ts b/blocks/configs/agent.ts
index 1ab93b5899..f14ebe8a65 100644
--- a/blocks/configs/agent.ts
+++ b/blocks/configs/agent.ts
@@ -25,6 +25,11 @@ export const AgentBlock: BlockConfig = {
         type: 'long-input',
         layout: 'full',
       },
+      {
+        title: 'Context',
+        type: 'long-input',
+        layout: 'full',
+      },
       {
         title: 'Model',
         type: 'dropdown',
@@ -44,7 +49,12 @@ export const AgentBlock: BlockConfig = {
         layout: "full",
         placeholder: "Enter your API key",
         password: true
-      }
+      },
+      {
+        title: 'Response Format',
+        type: 'code',
+        layout: 'full',
+      },
     ],
   },
 }
\ No newline at end of file