Skip to content

Commit 05af12e

Browse files
committed
fix: update schedule_extract_long_term_memory function parameters
1 parent 37d9edb commit 05af12e

2 files changed

Lines changed: 3 additions & 11 deletions

File tree

apps/application/long_term_memory/__init__.py

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -451,7 +451,7 @@ def extract_long_term_memory(workspace_id, application_id, chat_user_id):
451451

452452

453453
@celery_app.task(name="celery:schedule_extract_long_term_memory")
454-
def schedule_extract_long_term_memory(workspace_id, application_id, trigger_setting):
454+
def schedule_extract_long_term_memory(workspace_id, application_id, enabled, trigger_type, trigger_setting):
455455
# 先清理旧的调度任务
456456
_remove_long_term_jobs(application_id)
457457

@@ -460,15 +460,6 @@ def schedule_extract_long_term_memory(workspace_id, application_id, trigger_sett
460460
return
461461

462462
# 应用关闭长期记忆或不再是定时触发,则只清理不再部署
463-
if application.type == 'WORK_FLOW':
464-
node_list = application.work_flow.get('nodes', []) if application.work_flow else []
465-
base_node = next((n for n in node_list if n.get('id') == 'base-node'), None)
466-
node_data = (base_node or {}).get('properties', {}).get('node_data', {})
467-
enabled = node_data.get('long_term_enable', False)
468-
trigger_type = node_data.get('long_term_trigger_type')
469-
else:
470-
enabled = application.long_term_enable
471-
trigger_type = application.long_term_trigger_type
472463
if not enabled or trigger_type != 'SCHEDULED':
473464
return
474465

apps/application/serializers/application.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1125,7 +1125,8 @@ def edit(self, instance: Dict, with_valid=True):
11251125
knowledge_id_list,
11261126
application_id))
11271127
schedule_extract_long_term_memory.delay(
1128-
application.workspace_id, application_id, application.long_term_trigger_setting
1128+
application.workspace_id, application_id,
1129+
application.long_term_enable, application.long_term_trigger_type, application.long_term_trigger_setting
11291130
)
11301131
return self.one(with_valid=False)
11311132

0 commit comments

Comments
 (0)