going through tests

Author: Andrei Bratu

src/humanloop/otel/exporter.py

@@ -207,12 +207,12 @@ def _export_span_dispatch(self, span: ReadableSpan) -> None:
 
         if file_type == "prompt":
             self._export_prompt_span(span=span)
-        if file_type == "tool":
+        elif file_type == "tool":
             self._export_tool_span(span=span)
-        if file_type == "flow":
+        elif file_type == "flow":
             self._export_flow_span(span=span)
         else:
-            raise NotImplementedError(f"Unknown span type: {hl_file}")
+            raise NotImplementedError(f"Unknown span type: {file_type}")
 
     def _export_prompt_span(self, span: ReadableSpan) -> None:
         file_object: dict[str, Any] = read_from_opentelemetry_span(

src/humanloop/otel/processor/__init__.py

@@ -126,8 +126,8 @@ def _get_waiting_parent(self, span: ReadableSpan) -> Optional[ReadableSpan]:
 
     def _add_dependency_to_await(self, span: ReadableSpan):
         # We know this span has a parent, need to satisfy the type checker
-        parent_span_id = span.parent.span_id  # type: ignore
         if self._is_dependency(span):
+            parent_span_id = span.parent.span_id  # type: ignore
             self._dependencies[parent_span_id].append(DependantSpan(span=span, finished=False))
 
     def _track_flow_traces(self, span: ReadableSpan):