@@ -84,7 +84,6 @@ def new_generate_content_stream(
8484 chat_span .set_data (SPANDATA .GEN_AI_REQUEST_MODEL , model_name )
8585 set_span_data_for_request (chat_span , integration , model_name , contents , kwargs )
8686 chat_span .set_data (SPANDATA .GEN_AI_RESPONSE_STREAMING , True )
87- chat_span .set_data (SPANDATA .GEN_AI_AGENT_NAME , model_name )
8887
8988 try :
9089 stream = f (self , * args , ** kwargs )
@@ -143,7 +142,6 @@ async def new_async_generate_content_stream(
143142 chat_span .set_data (SPANDATA .GEN_AI_REQUEST_MODEL , model_name )
144143 set_span_data_for_request (chat_span , integration , model_name , contents , kwargs )
145144 chat_span .set_data (SPANDATA .GEN_AI_RESPONSE_STREAMING , True )
146- chat_span .set_data (SPANDATA .GEN_AI_AGENT_NAME , model_name )
147145
148146 try :
149147 stream = await f (self , * args , ** kwargs )
@@ -195,7 +193,6 @@ def new_generate_content(self: "Any", *args: "Any", **kwargs: "Any") -> "Any":
195193 chat_span .set_data (SPANDATA .GEN_AI_OPERATION_NAME , "chat" )
196194 chat_span .set_data (SPANDATA .GEN_AI_SYSTEM , GEN_AI_SYSTEM )
197195 chat_span .set_data (SPANDATA .GEN_AI_REQUEST_MODEL , model_name )
198- chat_span .set_data (SPANDATA .GEN_AI_AGENT_NAME , model_name )
199196 set_span_data_for_request (
200197 chat_span , integration , model_name , contents , kwargs
201198 )
0 commit comments