Skip to content

Commit d8c06f8

Browse files
defensive check
1 parent 56ec48f commit d8c06f8

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

sentry_sdk/integrations/langchain.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -426,9 +426,11 @@ def on_chat_model_start(
426426
if ai_system:
427427
span.set_data(SPANDATA.GEN_AI_SYSTEM, ai_system)
428428

429-
agent_name = kwargs.get("metadata", {}).get("lc_agent_name")
430-
if agent_name is not None:
431-
span.set_data(SPANDATA.GEN_AI_AGENT_NAME, agent_name)
429+
agent_metadata = kwargs.get("metadata")
430+
if isinstance(agent_metadata, dict) and "lc_agent_name" in agent_metadata:
431+
span.set_data(
432+
SPANDATA.GEN_AI_AGENT_NAME, agent_metadata["lc_agent_name"]
433+
)
432434

433435
for key, attribute in DATA_FIELDS.items():
434436
if key in all_params and all_params[key] is not None:
@@ -626,9 +628,11 @@ def on_tool_start(
626628
if tool_description is not None:
627629
span.set_data(SPANDATA.GEN_AI_TOOL_DESCRIPTION, tool_description)
628630

629-
agent_name = kwargs.get("metadata", {}).get("lc_agent_name")
630-
if agent_name is not None:
631-
span.set_data(SPANDATA.GEN_AI_AGENT_NAME, agent_name)
631+
agent_metadata = kwargs.get("metadata")
632+
if isinstance(agent_metadata, dict) and "lc_agent_name" in agent_metadata:
633+
span.set_data(
634+
SPANDATA.GEN_AI_AGENT_NAME, agent_metadata["lc_agent_name"]
635+
)
632636

633637
if should_send_default_pii() and self.include_prompts:
634638
set_data_normalized(

0 commit comments

Comments
 (0)