harnessed_agent/wwwroot/api/hermes_executions_task_update.dspy

35 lines
1.7 KiB
Python

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""Execution update API"""
import json, time
result = {'widgettype': 'Message', 'options': {'title': 'Error', 'message': 'Invalid request', 'type': 'error'}}
try:
dbname = get_module_dbname('harnessed_agent')
user_id = await get_user()
now = time.strftime('%Y-%m-%d %H:%M:%S')
record_id = params_kw.get('id', '').strip()
if not record_id:
result['options'] = {'title': 'Error', 'message': '记录ID不能为空', 'type': 'error'}
else:
async with DBPools().sqlorContext(dbname) as sor:
await sor.sqlExe("""UPDATE hermes_executions_task SET workflow_id = ${workflow_id}$, task_id = ${task_id}$, execution_status = ${execution_status}$, start_time = ${start_time}$, end_time = ${end_time}$, duration_seconds = ${duration_seconds}$, result_json = ${result_json}$, error_message = ${error_message}$, retry_count = ${retry_count}$, updated_at = ${updated_at}$ WHERE id = ${id}$ AND user_id = ${user_id}$""", {
'id': record_id, 'user_id': user_id, 'updated_at': now,
'workflow_id': workflow_id,
'task_id': task_id,
'execution_status': execution_status,
'start_time': start_time,
'end_time': end_time,
'duration_seconds': duration_seconds,
'result_json': result_json,
'error_message': error_message,
'retry_count': retry_count,
})
result = {'widgettype': 'Message', 'options': {'title': 'Success', 'message': '更新成功', 'type': 'success'}}
except Exception as e:
result['options'] = {'title': 'Error', 'message': '更新失败: ' + str(e), 'type': 'error'}
return json.dumps(result, ensure_ascii=False)