estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

chore(api/tasks): apply ruff reformatting (#7594)

Browse Source
This commit is contained in:
Bowen Liang
2024-08-26 13:38:37 +08:00
committed by GitHub
parent 3be756eaed
commit 979422cdc6
29 changed files with 546 additions and 508 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
api/tasks/ops_trace_task.py
View File

@@ -10,7 +10,7 @@ from models.model import Message
from models.workflow import WorkflowRun
@shared_task(queue='ops_trace')
@shared_task(queue="ops_trace")
def process_trace_tasks(tasks_data):
"""
Async process trace tasks
@@ -20,17 +20,17 @@ def process_trace_tasks(tasks_data):
"""
from core.ops.ops_trace_manager import OpsTraceManager
trace_info = tasks_data.get('trace_info')
app_id = tasks_data.get('app_id')
trace_info_type = tasks_data.get('trace_info_type')
trace_info = tasks_data.get("trace_info")
app_id = tasks_data.get("app_id")
trace_info_type = tasks_data.get("trace_info_type")
trace_instance = OpsTraceManager.get_ops_trace_instance(app_id)
if trace_info.get('message_data'):
trace_info['message_data'] = Message.from_dict(data=trace_info['message_data'])
if trace_info.get('workflow_data'):
trace_info['workflow_data'] = WorkflowRun.from_dict(data=trace_info['workflow_data'])
if trace_info.get('documents'):
trace_info['documents'] = [Document(**doc) for doc in trace_info['documents']]
if trace_info.get("message_data"):
trace_info["message_data"] = Message.from_dict(data=trace_info["message_data"])
if trace_info.get("workflow_data"):
trace_info["workflow_data"] = WorkflowRun.from_dict(data=trace_info["workflow_data"])
if trace_info.get("documents"):
trace_info["documents"] = [Document(**doc) for doc in trace_info["documents"]]
try:
if trace_instance: