@@ -92,6 +92,7 @@ def new_generate_content_stream(self, *args, **kwargs):
9292 chat_span .set_data (SPANDATA .GEN_AI_REQUEST_MODEL , model_name )
9393 set_span_data_for_request (chat_span , integration , model_name , contents , kwargs )
9494 chat_span .set_data (SPANDATA .GEN_AI_RESPONSE_STREAMING , True )
95+ chat_span .set_data (SPANDATA .GEN_AI_AGENT_NAME , model_name )
9596
9697 try :
9798 stream = f (self , * args , ** kwargs )
@@ -165,6 +166,7 @@ async def new_async_generate_content_stream(self, *args, **kwargs):
165166 chat_span .set_data (SPANDATA .GEN_AI_REQUEST_MODEL , model_name )
166167 set_span_data_for_request (chat_span , integration , model_name , contents , kwargs )
167168 chat_span .set_data (SPANDATA .GEN_AI_RESPONSE_STREAMING , True )
169+ chat_span .set_data (SPANDATA .GEN_AI_AGENT_NAME , model_name )
168170
169171 try :
170172 stream = await f (self , * args , ** kwargs )
@@ -233,6 +235,7 @@ def new_generate_content(self, *args, **kwargs):
233235 chat_span .set_data (SPANDATA .GEN_AI_OPERATION_NAME , "chat" )
234236 chat_span .set_data (SPANDATA .GEN_AI_SYSTEM , GEN_AI_SYSTEM )
235237 chat_span .set_data (SPANDATA .GEN_AI_REQUEST_MODEL , model_name )
238+ chat_span .set_data (SPANDATA .GEN_AI_AGENT_NAME , model_name )
236239 set_span_data_for_request (
237240 chat_span , integration , model_name , contents , kwargs
238241 )
0 commit comments