hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-11 20:06:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: update workflow trace query (#6010)

Browse Source
This commit is contained in:
Joe
2024-07-05 18:37:26 +08:00
committed by GitHub
parent cc63af8e72
commit 3f0da88ff7
2 changed files with 28 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
api/core/ops/langfuse_trace/langfuse_trace.py
View File

@@ -107,7 +107,20 @@ class LangFuseDataTrace(BaseTraceInstance):
# through workflow_run_id get all_nodes_execution # through workflow_run_id get all_nodes_execution
workflow_nodes_executions = ( workflow_nodes_executions = (
db.session.query(WorkflowNodeExecution) db.session.query(
WorkflowNodeExecution.id,
WorkflowNodeExecution.tenant_id,
WorkflowNodeExecution.app_id,
WorkflowNodeExecution.title,
WorkflowNodeExecution.node_type,
WorkflowNodeExecution.status,
WorkflowNodeExecution.inputs,
WorkflowNodeExecution.outputs,
WorkflowNodeExecution.created_at,
WorkflowNodeExecution.elapsed_time,
WorkflowNodeExecution.process_data,
WorkflowNodeExecution.execution_metadata,
)
.filter(WorkflowNodeExecution.workflow_run_id == trace_info.workflow_run_id) .filter(WorkflowNodeExecution.workflow_run_id == trace_info.workflow_run_id)
.all() .all()
) )

15
api/core/ops/langsmith_trace/langsmith_trace.py
View File

@@ -100,7 +100,20 @@ class LangSmithDataTrace(BaseTraceInstance):
# through workflow_run_id get all_nodes_execution # through workflow_run_id get all_nodes_execution
workflow_nodes_executions = ( workflow_nodes_executions = (
db.session.query(WorkflowNodeExecution) db.session.query(
WorkflowNodeExecution.id,
WorkflowNodeExecution.tenant_id,
WorkflowNodeExecution.app_id,
WorkflowNodeExecution.title,
WorkflowNodeExecution.node_type,
WorkflowNodeExecution.status,
WorkflowNodeExecution.inputs,
WorkflowNodeExecution.outputs,
WorkflowNodeExecution.created_at,
WorkflowNodeExecution.elapsed_time,
WorkflowNodeExecution.process_data,
WorkflowNodeExecution.execution_metadata,
)
.filter(WorkflowNodeExecution.workflow_run_id == trace_info.workflow_run_id) .filter(WorkflowNodeExecution.workflow_run_id == trace_info.workflow_run_id)
.all() .all()
) )