langchain work

Author: antonpirker

sentry_sdk/integrations/langchain.py

@@ -5,11 +5,10 @@
 import sentry_sdk
 from sentry_sdk.ai.monitoring import set_ai_pipeline_name
 from sentry_sdk.ai.utils import set_data_normalized, get_start_span_function
-from sentry_sdk.consts import OP, SPANDATA, SPANSTATUS
+from sentry_sdk.consts import OP, SPANDATA
 from sentry_sdk.integrations import DidNotEnable, Integration
 from sentry_sdk.scope import should_send_default_pii
-from sentry_sdk.tracing import Span
-from sentry_sdk.tracing_utils import _get_value
+from sentry_sdk.tracing_utils import _get_value, set_span_errored
 from sentry_sdk.utils import logger, capture_internal_exceptions
 
 from typing import TYPE_CHECKING
@@ -26,6 +25,7 @@
         Union,
     )
     from uuid import UUID
+    from sentry_sdk.tracing import Span
 
 
 try:
@@ -116,7 +116,7 @@ def _handle_error(self, run_id, error):
 
             span_data = self.span_map[run_id]
             span = span_data.span
-            span.set_status(SPANSTATUS.ERROR)
+            set_span_errored(span)
 
             sentry_sdk.capture_exception(error, span.scope)
 

tests/integrations/langchain/test_langchain.py

@@ -267,6 +267,9 @@ def test_langchain_error(sentry_init, capture_events):
 
 
 def test_span_status_error(sentry_init, capture_events):
+    global llm_type
+    llm_type = "acme-llm"
+
     sentry_init(
         integrations=[LangchainIntegration(include_prompts=True)],
         traces_sample_rate=1.0,