fix: fix bugs check by Claude Code

This commit is contained in:
Stream 2025-08-22 17:54:18 +08:00
parent 8b68020453
commit 48f3c69c69
No known key found for this signature in database
GPG Key ID: 033728094B100D70
3 changed files with 24 additions and 22 deletions

View File

@ -39,17 +39,18 @@ class MemoryEditApi(Resource):
if not memory_spec: if not memory_spec:
return {'error': 'Memory not found'}, 404 return {'error': 'Memory not found'}, 404
with Session(db.engine) as session: with Session(db.engine) as session:
ChatflowMemoryVariable( session.merge(
tenant_id=app_model.tenant_id, ChatflowMemoryVariable(
app_id=app_model.id, tenant_id=app_model.tenant_id,
node_id=args['node_id'], app_id=app_model.id,
memory_id=args['id'], node_id=args['node_id'],
name=memory_spec.name, memory_id=args['id'],
value=args['update'], name=memory_spec.name,
scope=memory_spec.scope, value=args['update'],
term=memory_spec.term, scope=memory_spec.scope,
term=memory_spec.term,
)
) )
session.add(memory_spec)
session.commit() session.commit()
return '', 204 return '', 204

View File

@ -39,17 +39,18 @@ class MemoryEditApi(WebApiResource):
if not memory_spec.end_user_editable: if not memory_spec.end_user_editable:
return {'error': 'Memory not editable'}, 403 return {'error': 'Memory not editable'}, 403
with Session(db.engine) as session: with Session(db.engine) as session:
ChatflowMemoryVariable( session.merge(
tenant_id=app_model.tenant_id, ChatflowMemoryVariable(
app_id=app_model.id, tenant_id=app_model.tenant_id,
node_id=args['node_id'], app_id=app_model.id,
memory_id=args['id'], node_id=args['node_id'],
name=memory_spec.name, memory_id=args['id'],
value=args['update'], name=memory_spec.name,
scope=memory_spec.scope, value=args['update'],
term=memory_spec.term, scope=memory_spec.scope,
term=memory_spec.term,
)
) )
session.add(memory_spec)
session.commit() session.commit()
return '', 204 return '', 204

View File

@ -156,7 +156,7 @@ class ChatflowMemoryService:
ChatflowMemoryVariable.tenant_id == tenant_id, ChatflowMemoryVariable.tenant_id == tenant_id,
ChatflowMemoryVariable.app_id == app_id, ChatflowMemoryVariable.app_id == app_id,
ChatflowMemoryVariable.node_id == \ ChatflowMemoryVariable.node_id == \
(node_id if spec.term == MemoryScope.NODE else None), (node_id if spec.scope == MemoryScope.NODE else None),
ChatflowMemoryVariable.conversation_id == \ ChatflowMemoryVariable.conversation_id == \
(conversation_id if spec.term == MemoryTerm.SESSION else None), (conversation_id if spec.term == MemoryTerm.SESSION else None),
) )
@ -344,7 +344,7 @@ class ChatflowMemoryService:
memory_block: MemoryBlock, memory_block: MemoryBlock,
visible_history: Sequence[PromptMessage] visible_history: Sequence[PromptMessage]
) -> bool: ) -> bool:
return len(visible_history) > memory_block.spec.update_turns return len(visible_history) >= memory_block.spec.update_turns
@staticmethod @staticmethod
def _app_submit_async_memory_update( def _app_submit_async_memory_update(