AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-16 10:05:56 +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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 28 additions and 2 deletions
Show Stats 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()
) )