fix: fix bugs check by Claude Code

This commit is contained in:
Stream
2025-08-22 19:59:17 +08:00
parent 48f3c69c69
commit 05d231ad33
4 changed files with 46 additions and 28 deletions

View File

@ -27,7 +27,7 @@ class MemoryEditApi(WebApiResource):
def put(self, app_model):
parser = reqparse.RequestParser()
parser.add_argument('id', type=str, required=True)
parser.add_argument('node_id', type=str, required=False)
parser.add_argument('node_id', type=str, required=False, default=None)
parser.add_argument('update', type=str, required=True)
args = parser.parse_args()
workflow = WorkflowService().get_published_workflow(app_model)
@ -39,18 +39,27 @@ class MemoryEditApi(WebApiResource):
if not memory_spec.end_user_editable:
return {'error': 'Memory not editable'}, 403
with Session(db.engine) as session:
session.merge(
ChatflowMemoryVariable(
tenant_id=app_model.tenant_id,
app_id=app_model.id,
node_id=args['node_id'],
memory_id=args['id'],
name=memory_spec.name,
value=args['update'],
scope=memory_spec.scope,
term=memory_spec.term,
existing = session.query(ChatflowMemoryVariable).filter_by(
memory_id=args['id'],
tenant_id=app_model.tenant_id,
app_id=app_model.id,
node_id=args['node_id']
).first()
if existing:
existing.value = args['update']
else:
session.add(
ChatflowMemoryVariable(
tenant_id=app_model.tenant_id,
app_id=app_model.id,
node_id=args['node_id'],
memory_id=args['id'],
name=memory_spec.name,
value=args['update'],
scope=memory_spec.scope,
term=memory_spec.term,
)
)
)
session.commit()
return '', 204