@@ -199,7 +199,7 @@ def _do_work(self):
199199 self ._upload_queue .put ((span_to_export , evaluation_context ))
200200 self ._upload_queue .task_done ()
201201
202- def _complete_flow_log (self , span_id : int ) -> None :
202+ def _mark_span_completed (self , span_id : int ) -> None :
203203 for flow_log_span_id , flow_children_span_ids in self ._flow_log_prerequisites .items ():
204204 if span_id in flow_children_span_ids :
205205 flow_children_span_ids .remove (span_id )
@@ -282,7 +282,7 @@ def _export_prompt(self, span: ReadableSpan) -> None:
282282 self ._span_id_to_uploaded_log_id [span .context .span_id ] = log_response .id
283283 except HumanloopApiError :
284284 self ._span_id_to_uploaded_log_id [span .context .span_id ] = None
285- self ._complete_flow_log (span_id = span .context .span_id )
285+ self ._mark_span_completed (span_id = span .context .span_id )
286286
287287 def _export_tool (self , span : ReadableSpan ) -> None :
288288 file_object : dict [str , Any ] = read_from_opentelemetry_span (
@@ -318,7 +318,7 @@ def _export_tool(self, span: ReadableSpan) -> None:
318318 self ._span_id_to_uploaded_log_id [span .context .span_id ] = log_response .id
319319 except HumanloopApiError :
320320 self ._span_id_to_uploaded_log_id [span .context .span_id ] = None
321- self ._complete_flow_log (span_id = span .context .span_id )
321+ self ._mark_span_completed (span_id = span .context .span_id )
322322
323323 def _export_flow (self , span : ReadableSpan ) -> None :
324324 file_object : dict [str , Any ] = read_from_opentelemetry_span (
@@ -362,4 +362,4 @@ def _export_flow(self, span: ReadableSpan) -> None:
362362 except HumanloopApiError as e :
363363 logger .error (str (e ))
364364 self ._span_id_to_uploaded_log_id [span .context .span_id ] = None
365- self ._complete_flow_log (span_id = span .context .span_id )
365+ self ._mark_span_completed (span_id = span .context .span_id )
0 commit comments