harnessed_agent/wwwroot/api/hermes_workflows_update.dspy

33 lines
1.6 KiB
Python

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""Workflow 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_workflows SET name = ${name}$, description = ${description}$, workflow_type = ${workflow_type}$, max_concurrent_tasks = ${max_concurrent_tasks}$, timeout_seconds = ${timeout_seconds}$, retry_count = ${retry_count}$, status = ${status}$, updated_at = ${updated_at}$ WHERE id = ${id}$ AND user_id = ${user_id}$""", {
'id': record_id, 'user_id': user_id, 'updated_at': now,
'name': name,
'description': description,
'workflow_type': workflow_type,
'max_concurrent_tasks': max_concurrent_tasks,
'timeout_seconds': timeout_seconds,
'retry_count': retry_count,
'status': status,
})
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)