Use end_on_exit

Author: drewby

instrumentation-genai/opentelemetry-instrumentation-openai-v2/src/opentelemetry/instrumentation/openai_v2/patch.py

@@ -184,7 +184,7 @@ def traced_method(wrapped, instance, args, kwargs):
             name=span_name,
             kind=SpanKind.CLIENT,
             attributes=span_attributes,
-            end_on_exit=False,
+            end_on_exit=True,
         ) as span:
             # Store the input for later use in the response attributes
             input_text = kwargs.get("input", "")
@@ -200,7 +200,6 @@ def traced_method(wrapped, instance, args, kwargs):
                         span, result, event_logger, capture_content, input_text
                     )
 
-                span.end()
                 return result
 
             except Exception as error:
@@ -249,7 +248,7 @@ async def traced_method(wrapped, instance, args, kwargs):
             name=span_name,
             kind=SpanKind.CLIENT,
             attributes=span_attributes,
-            end_on_exit=False,
+            end_on_exit=True,
         ) as span:
             # Store the input for later use in the response attributes
             input_text = kwargs.get("input", "")
@@ -265,7 +264,6 @@ async def traced_method(wrapped, instance, args, kwargs):
                         span, result, event_logger, capture_content, input_text
                     )
 
-                span.end()
                 return result
 
             except Exception as error: