youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-04 01:18:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: expose version to MemoryBlock

Browse Source
This commit is contained in:
Stream
2025-09-23 23:09:45 +08:00
parent 3d7d4182a6
commit 5bf642c3f9
4 changed files with 59 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
api/controllers/service_api/app/chatflow_memory.py
View File

@ -61,20 +61,32 @@ class MemoryEditApi(Resource):
memory_spec = next((it for it in workflow.memory_blocks if it.id == args['id']), None)
if not memory_spec:
return {'error': 'Memory not found'}, 404
ChatflowMemoryService.save_memory(
MemoryBlock(
spec=memory_spec,
tenant_id=app_model.tenant_id,
value=update,
conversation_id=conversation_id,
node_id=node_id,
app_id=app_model.id,
edited_by_user=True,
created_by=MemoryCreatedBy(end_user_id=end_user.id),
),
variable_pool=VariablePool(),
# First get existing memory
existing_memory = ChatflowMemoryService.get_memory_by_spec(
spec=memory_spec,
tenant_id=app_model.tenant_id,
app_id=app_model.id,
created_by=MemoryCreatedBy(end_user_id=end_user.id),
conversation_id=conversation_id,
node_id=node_id,
is_draft=False
)
# Create updated memory instance
updated_memory = MemoryBlock(
spec=existing_memory.spec,
tenant_id=existing_memory.tenant_id,
app_id=existing_memory.app_id,
conversation_id=existing_memory.conversation_id,
node_id=existing_memory.node_id,
value=update, # New value
version=existing_memory.version, # Keep current version (save_memory will handle version increment)
edited_by_user=True,
created_by=existing_memory.created_by,
)
ChatflowMemoryService.save_memory(updated_memory, VariablePool(), False)
return '', 204