opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-28 20:17:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add MemoryBlock to VariablePool

Browse Source
This commit is contained in:
Stream 2025-08-20 19:45:18 +08:00
parent 0c97bbf137
commit bbb640c9a2
No known key found for this signature in database
GPG Key ID: 9475891C9507B4F3
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/core/workflow/entities/variable_pool.py
View File

@ -46,6 +46,10 @@ class VariablePool(BaseModel):
description="Conversation variables.", description="Conversation variables.",
default_factory=list, default_factory=list,
) )
memory_blocks: Mapping[str, str] = Field(
description="Memory blocks.",
default_factory=dict,
)
def model_post_init(self, context: Any, /) -> None: def model_post_init(self, context: Any, /) -> None:
# Create a mapping from field names to SystemVariableKey enum values # Create a mapping from field names to SystemVariableKey enum values
@ -56,6 +60,9 @@ class VariablePool(BaseModel):
# Add conversation variables to the variable pool # Add conversation variables to the variable pool
for var in self.conversation_variables: for var in self.conversation_variables:
self.add((CONVERSATION_VARIABLE_NODE_ID, var.name), var) self.add((CONVERSATION_VARIABLE_NODE_ID, var.name), var)
# Add memory blocks to the variable pool
for memory_id, memory_value in self.memory_blocks.items():
self.add(['memory_block', memory_id], memory_value)
def add(self, selector: Sequence[str], value: Any, /) -> None: def add(self, selector: Sequence[str], value: Any, /) -> None:
""" """