fix: unify memory variable in VariablePool

This commit is contained in:
Stream
2025-10-15 14:39:05 +08:00
parent f4567fbf9e
commit 7ca06931ec
2 changed files with 15 additions and 5 deletions

View File

@ -139,9 +139,10 @@ class ChatflowMemoryService:
if is_draft:
with Session(bind=db.engine) as session:
draft_var_service = WorkflowDraftVariableService(session)
memory_selector = memory.spec.id if not memory.node_id else f"{memory.node_id}.{memory.spec.id}"
existing_vars = draft_var_service.get_draft_variables_by_selectors(
app_id=memory.app_id,
selectors=[['memory_block', memory.spec.id]]
selectors=[[MEMORY_BLOCK_VARIABLE_NODE_ID, memory_selector]]
)
if existing_vars:
draft_var = existing_vars[0]