Pārlūkot izejas kodu

fix: Ensure the order of execution steps are correct when logging with Weave by W&B (#25183)

tags/2.0.0-beta.2^2
Anubhav Singh pirms 1 mēnesi
vecāks
revīzija
f721c778ad
Revīzijas autora e-pasta adrese nav piesaistīta nevienam kontam
1 mainītis faili ar 4 papildinājumiem un 1 dzēšanām
  1. 4
    1
      api/core/ops/weave_trace/weave_trace.py

+ 4
- 1
api/core/ops/weave_trace/weave_trace.py Parādīt failu

@@ -120,7 +120,7 @@ class WeaveDataTrace(BaseTraceInstance):
workflow_attributes["trace_id"] = trace_id
workflow_attributes["start_time"] = trace_info.start_time
workflow_attributes["end_time"] = trace_info.end_time
workflow_attributes["tags"] = ["workflow"]
workflow_attributes["tags"] = ["dify_workflow"]

workflow_run = WeaveTraceModel(
file_list=trace_info.file_list,
@@ -156,6 +156,9 @@ class WeaveDataTrace(BaseTraceInstance):
workflow_run_id=trace_info.workflow_run_id
)

# rearrange workflow_node_executions by starting time
workflow_node_executions = sorted(workflow_node_executions, key=lambda x: x.created_at)

for node_execution in workflow_node_executions:
node_execution_id = node_execution.id
tenant_id = trace_info.tenant_id # Use from trace_info instead

Notiek ielāde…
Atcelt
Saglabāt