@@ -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