File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -774,7 +774,7 @@ def _get_log_func(
774774 "run_id" : run_id ,
775775 }
776776 if file_type == "flow" :
777- return partial (client .flows .log , ** log_request , trace_status = "complete" )
777+ return partial (client .flows .log , ** log_request , log_status = "complete" )
778778 elif file_type == "prompt" :
779779 return partial (client .prompts .log , ** log_request )
780780 elif file_type == "evaluator" :
@@ -912,7 +912,7 @@ def _run_local_evaluators(
912912
913913 # Wait for the Flow trace to complete before running evaluators
914914 while True :
915- if file_type != "flow" or log_dict ["trace_status " ] == "complete" :
915+ if file_type != "flow" or log_dict ["log_status " ] == "complete" :
916916 break
917917 log = client .logs .get (id = log_id )
918918 if not isinstance (log , dict ):
Original file line number Diff line number Diff line change @@ -371,7 +371,7 @@ def _mark_trace_complete_if_needed(self, trace_head_span_id: int):
371371 trace_head_span_id ,
372372 )
373373 else :
374- self ._client .flows .update_log (log_id = flow_log_id , trace_status = "complete" )
374+ self ._client .flows .update_log (log_id = flow_log_id , log_status = "complete" )
375375
376376 def _keep_track_of_trace (self , log_id : str , parent_log_id : str ):
377377 found = False
You can’t perform that action at this time.
0 commit comments