diff --git a/blocks/blocks/memory.ts b/blocks/blocks/input.ts
similarity index 54%
rename from blocks/blocks/memory.ts
rename to blocks/blocks/input.ts
index 6e6618ca4c..70b29605f9 100644
--- a/blocks/blocks/memory.ts
+++ b/blocks/blocks/input.ts
@@ -1,13 +1,13 @@
-import { BrainIcon } from '@/components/icons'
+import { InputIcon } from '@/components/icons'
 import { BlockConfig } from '../types'
 
-export const MemoryBlock: BlockConfig = {
-  type: 'memory',
+export const InputBlock: BlockConfig = {
+  type: 'input',
   toolbar: {
-    title: 'Memory',
-    description: 'Add memory store',
-    bgColor: '#FF65BF',
-    icon: BrainIcon,
+    title: 'Input',
+    description: 'Add workflow input',
+    bgColor: '#2FB3FF',
+    icon: InputIcon,
     category: 'blocks',
   },
   tools: {
@@ -16,8 +16,6 @@ export const MemoryBlock: BlockConfig = {
   workflow: {
     inputs: {
       code: { type: 'string', required: true },
-      timeout: { type: 'number', required: false },
-      memoryLimit: { type: 'number', required: false },
     },
     outputs: {
       response: {
@@ -30,8 +28,9 @@ export const MemoryBlock: BlockConfig = {
     },
     subBlocks: [
       {
-        id: 'code',
-        type: 'code',
+        title: 'Input',
+        id: 'input',
+        type: 'short-input',
         layout: 'full',
       },
     ],
diff --git a/blocks/index.ts b/blocks/index.ts
index 791ab0e4fd..7808d47e68 100644
--- a/blocks/index.ts
+++ b/blocks/index.ts
@@ -9,7 +9,6 @@ import { FunctionBlock } from './blocks/function'
 import { GitHubBlock } from './blocks/github'
 import { GmailBlock } from './blocks/gmail'
 import { JinaBlock } from './blocks/jina'
-import { MemoryBlock } from './blocks/memory'
 import { NotionBlock } from './blocks/notion'
 import { RouterBlock } from './blocks/router'
 import { SerperBlock } from './blocks/serper'
@@ -40,7 +39,6 @@ export {
   NotionBlock,
   GmailBlock,
   XBlock,
-  MemoryBlock,
 }
 
 // Registry of all block configurations
@@ -51,7 +49,6 @@ const blocks: Record<string, BlockConfig> = {
   function: FunctionBlock,
   router: RouterBlock,
   evaluator: EvaluatorBlock,
-  memory: MemoryBlock,
   crewai_vision: CrewAIVisionBlock,
   firecrawl_scrape: FirecrawlScrapeBlock,
   jina_reader: JinaBlock,
diff --git a/components/icons.tsx b/components/icons.tsx
index 92f41c96be..f6a72cc7a1 100644
--- a/components/icons.tsx
+++ b/components/icons.tsx
@@ -1352,3 +1352,21 @@ export function BrainIcon(props: SVGProps<SVGSVGElement>) {
     </svg>
   )
 }
+
+export function InputIcon(props: SVGProps<SVGSVGElement>) {
+  return (
+    <svg
+      {...props}
+      width="26"
+      height="21"
+      viewBox="0 0 26 21"
+      fill="none"
+      xmlns="http://www.w3.org/2000/svg"
+    >
+      <path
+        d="M2.6 20.8C1.885 20.8 1.27314 20.5456 0.764404 20.0369C0.255671 19.5282 0.000871149 18.9159 4.48275e-06 18.2V15.6C4.48275e-06 15.2317 0.124805 14.9231 0.374405 14.6744C0.624004 14.4257 0.932538 14.3009 1.3 14.3C1.66747 14.2991 1.97644 14.4239 2.2269 14.6744C2.47737 14.9249 2.60174 15.2334 2.6 15.6V18.2H23.4V2.6H2.6V5.2C2.6 5.56833 2.4752 5.8773 2.2256 6.1269C1.976 6.3765 1.66747 6.50087 1.3 6.5C0.932538 6.49913 0.624004 6.37433 0.374405 6.1256C0.124805 5.87687 4.48275e-06 5.56833 4.48275e-06 5.2V2.6C4.48275e-06 1.885 0.254805 1.27313 0.764404 0.7644C1.274 0.255666 1.88587 0.000866666 2.6 0H23.4C24.115 0 24.7273 0.2548 25.2369 0.7644C25.7465 1.274 26.0009 1.88587 26 2.6V18.2C26 18.915 25.7456 19.5273 25.2369 20.0369C24.7282 20.5465 24.1159 20.8009 23.4 20.8H2.6ZM13.2275 11.7H1.3C0.931671 11.7 0.623138 11.5752 0.374405 11.3256C0.125671 11.076 0.000871149 10.7675 4.48275e-06 10.4C-0.000862184 10.0325 0.123938 9.724 0.374405 9.4744C0.624871 9.2248 0.933404 9.1 1.3 9.1H13.2275L11.44 7.345C11.18 7.10666 11.0552 6.80897 11.0656 6.4519C11.076 6.09483 11.2008 5.78587 11.44 5.525C11.7 5.265 12.009 5.1298 12.3669 5.1194C12.7248 5.109 13.0334 5.23337 13.2925 5.4925L17.29 9.49C17.55 9.75 17.68 10.0533 17.68 10.4C17.68 10.7467 17.55 11.05 17.29 11.31L13.2925 15.3075C13.0325 15.5675 12.724 15.6975 12.3669 15.6975C12.0098 15.6975 11.7009 15.5675 11.44 15.3075C11.2017 15.0475 11.0769 14.739 11.0656 14.3819C11.0543 14.0248 11.1791 13.7159 11.44 13.455L13.2275 11.7Z"
+        fill="currentColor"
+      />
+    </svg>
+  )
+}