From 612112c9192548c5b08b80561466ff1384149fe6 Mon Sep 17 00:00:00 2001
From: JzoNg <jzongcode@gmail.com>
Date: Fri, 12 Sep 2025 16:27:20 +0800
Subject: [PATCH] add memory variable in llm node

---
 .../llm/components/memory-system/index.tsx    |  6 ++
 .../memory-system/memory-create-button.tsx    | 68 +++++++++++++++++++
 2 files changed, 74 insertions(+)
 create mode 100644 web/app/components/workflow/nodes/llm/components/memory-system/memory-create-button.tsx

diff --git a/web/app/components/workflow/nodes/llm/components/memory-system/index.tsx b/web/app/components/workflow/nodes/llm/components/memory-system/index.tsx
index e7bead9ff8..3ab14758a5 100644
--- a/web/app/components/workflow/nodes/llm/components/memory-system/index.tsx
+++ b/web/app/components/workflow/nodes/llm/components/memory-system/index.tsx
@@ -6,7 +6,9 @@ import Collapse from '@/app/components/workflow/nodes/_base/components/collapse'
 import type {
   Node,
 } from '@/app/components/workflow/types'
+import Divider from '@/app/components/base/divider'
 import Tooltip from '@/app/components/base/tooltip'
+import MemoryCreateButton from './memory-create-button'
 import MemorySelector from './memory-selector'
 import LinearMemory from './linear-memory'
 import type { Memory } from '@/app/components/workflow/types'
@@ -54,6 +56,10 @@ const MemorySystem = ({
                     triggerClassName='w-4 h-4'
                   />
                   {collapseIcon}
+                  <Divider type='vertical' className='!ml-1.5 !mr-1 h-3 !w-px bg-divider-regular' />
+                  <div onClick={e => e.stopPropagation()}>
+                    <MemoryCreateButton />
+                  </div>
                 </div>
                 <MemorySelector
                   value={memoryType}
diff --git a/web/app/components/workflow/nodes/llm/components/memory-system/memory-create-button.tsx b/web/app/components/workflow/nodes/llm/components/memory-system/memory-create-button.tsx
new file mode 100644
index 0000000000..4236791710
--- /dev/null
+++ b/web/app/components/workflow/nodes/llm/components/memory-system/memory-create-button.tsx
@@ -0,0 +1,68 @@
+import { useCallback, useState } from 'react'
+import { RiAddLine } from '@remixicon/react'
+import VariableModal from '@/app/components/workflow/panel/chat-variable-panel/components/variable-modal'
+import {
+  PortalToFollowElem,
+  PortalToFollowElemContent,
+  PortalToFollowElemTrigger,
+} from '@/app/components/base/portal-to-follow-elem'
+import ActionButton from '@/app/components/base/action-button'
+import type { ConversationVariable } from '@/app/components/workflow/types'
+import { useStore } from '@/app/components/workflow/store'
+import { useNodesSyncDraft } from '@/app/components/workflow/hooks/use-nodes-sync-draft'
+import useInspectVarsCrud from '@/app/components/workflow/hooks/use-inspect-vars-crud'
+
+const MemoryCreateButton = () => {
+  const [open, setOpen] = useState(false)
+  const varList = useStore(s => s.conversationVariables) as ConversationVariable[]
+  const updateChatVarList = useStore(s => s.setConversationVariables)
+  const { doSyncWorkflowDraft } = useNodesSyncDraft()
+  const {
+    invalidateConversationVarValues,
+  } = useInspectVarsCrud()
+
+  const handleVarChanged = useCallback(() => {
+    doSyncWorkflowDraft(false, {
+      onSuccess() {
+        invalidateConversationVarValues()
+      },
+    })
+  }, [doSyncWorkflowDraft, invalidateConversationVarValues])
+
+  const handleSave = useCallback(async (newChatVar: ConversationVariable) => {
+    const newList = [newChatVar, ...varList]
+    updateChatVarList(newList)
+    handleVarChanged()
+    setOpen(false)
+  }, [varList, updateChatVarList, handleVarChanged, setOpen])
+
+  return (
+    <>
+      <PortalToFollowElem
+        open={open}
+        onOpenChange={setOpen}
+        placement='left-start'
+        offset={{
+          mainAxis: 128,
+          alignmentAxis: -48,
+        }}
+      >
+        <PortalToFollowElemTrigger onClick={() => setOpen(v => !v)}>
+          <ActionButton className='shrink-0'>
+            <RiAddLine className='h-4 w-4' />
+          </ActionButton>
+        </PortalToFollowElemTrigger>
+        <PortalToFollowElemContent className='z-[11]'>
+          <VariableModal
+            onSave={handleSave}
+            onClose={() => {
+              setOpen(false)
+            }}
+          />
+        </PortalToFollowElemContent>
+      </PortalToFollowElem>
+    </>
+  )
+}
+
+export default MemoryCreateButton