added span exist check

Author: wrisa

instrumentation-genai/opentelemetry-instrumentation-langchain/src/opentelemetry/instrumentation/langchain/callback_handler.py

@@ -115,6 +115,10 @@ def on_llm_end(
     ) -> None:
         span = self.span_manager.get_span(run_id)
 
+        if span is None:
+            # If the span does not exist, we cannot set attributes or end it
+            return
+
         finish_reasons: List[str] = []
         for generation in getattr(response, "generations", []):  # type: ignore
             for chat_generation in generation:
@@ -175,6 +179,9 @@ def on_llm_error(
 
     def _handle_error(self, error: BaseException, run_id: UUID):
         span = self.span_manager.get_span(run_id)
+        if span is None:
+            # If the span does not exist, we cannot set the error status
+            return
         span.set_status(Status(StatusCode.ERROR, str(error)))
         span.set_attribute(
             ErrorAttributes.ERROR_TYPE, type(error).__qualname__

instrumentation-genai/opentelemetry-instrumentation-langchain/src/opentelemetry/instrumentation/langchain/span_manager.py

@@ -98,4 +98,5 @@ def end_span(self, run_id: UUID) -> None:
         state.span.end()
 
     def get_span(self, run_id: UUID) -> Span:
-        return self.spans[run_id].span
+        state = self.spans.get(run_id)
+        return state.span if state else None