revert langchain for now

Author: sentrivana

sentry_sdk/integrations/langchain.py

@@ -276,14 +276,6 @@ def on_chat_model_start(self, serialized, messages, *, run_id, **kwargs):
             elif "openai" in ai_type:
                 span.set_data(SPANDATA.GEN_AI_SYSTEM, "openai")
 
-            agent_name = (
-                sentry_sdk.get_current_scope()
-                ._contexts.get("langchain_agent", {})
-                .get("agent_name")
-            )
-            if agent_name:
-                span.set_data(SPANDATA.GEN_AI_AGENT_NAME, agent_name)
-
             for key, attribute in DATA_FIELDS.items():
                 if key in all_params and all_params[key] is not None:
                     set_data_normalized(span, attribute, all_params[key], unpack=False)
@@ -436,14 +428,6 @@ def on_tool_start(self, serialized, input_str, *, run_id, **kwargs):
             if tool_description is not None:
                 span.set_data(SPANDATA.GEN_AI_TOOL_DESCRIPTION, tool_description)
 
-            agent_name = (
-                sentry_sdk.get_current_scope()
-                ._contexts.get("langchain_agent", {})
-                .get("agent_name")
-            )
-            if agent_name:
-                span.set_data(SPANDATA.GEN_AI_AGENT_NAME, agent_name)
-
             if should_send_default_pii() and self.include_prompts:
                 set_data_normalized(
                     span,
@@ -772,9 +756,6 @@ def new_invoke(self, *args, **kwargs):
             name=f"invoke_agent {agent_name}" if agent_name else "invoke_agent",
             origin=LangchainIntegration.origin,
         ) as span:
-            sentry_sdk.get_current_scope().set_context(
-                "langchain_agent", {"agent_name": agent_name, "tools": tools}
-            )
             if agent_name:
                 span.set_data(SPANDATA.GEN_AI_AGENT_NAME, agent_name)
 
@@ -813,8 +794,6 @@ def new_invoke(self, *args, **kwargs):
             ):
                 set_data_normalized(span, SPANDATA.GEN_AI_RESPONSE_TEXT, output)
 
-            sentry_sdk.get_current_scope().remove_context("langchain_agent")
-
             return result
 
     return new_invoke
@@ -835,15 +814,11 @@ def new_stream(self, *args, **kwargs):
 
         span = start_span_function(
             op=OP.GEN_AI_INVOKE_AGENT,
-            name=f"invoke_agent {agent_name}" if agent_name else "invoke_agent",
+            name=f"invoke_agent {agent_name}".strip(),
             origin=LangchainIntegration.origin,
         )
         span.__enter__()
 
-        sentry_sdk.get_current_scope().set_context(
-            "langchain_agent", {"agent_name": agent_name, "tools": tools}
-        )
-
         if agent_name:
             span.set_data(SPANDATA.GEN_AI_AGENT_NAME, agent_name)
 
@@ -893,8 +868,6 @@ def new_iterator():
             ):
                 set_data_normalized(span, SPANDATA.GEN_AI_RESPONSE_TEXT, output)
 
-            sentry_sdk.get_current_scope().remove_context("langchain_agent")
-
             span.__exit__(None, None, None)
 
         async def new_iterator_async():
@@ -914,8 +887,6 @@ async def new_iterator_async():
             ):
                 set_data_normalized(span, SPANDATA.GEN_AI_RESPONSE_TEXT, output)
 
-            sentry_sdk.get_current_scope().remove_context("langchain_agent")
-
             span.__exit__(None, None, None)
 
         if str(type(result)) == "<class 'async_generator'>":