Fix context

Author: anuraaga

instrumentation/openai/openai-java-1.1/library/src/main/java/io/opentelemetry/instrumentation/openai/v1_1/InstrumentedChatCompletionServiceAsync.java

@@ -92,7 +92,7 @@ private CompletableFuture<ChatCompletion> create(
     future =
         future.whenComplete(
             (res, t) -> instrumenter.end(context, chatCompletionCreateParams, res, t));
-    return CompletableFutureWrapper.wrap(future, context);
+    return CompletableFutureWrapper.wrap(future, parentContext);
   }
 
   private CompletableFuture<ChatCompletion> createWithLogs(

instrumentation/openai/openai-java-1.1/testing/src/main/java/io/opentelemetry/instrumentation/openai/v1_1/AbstractChatTest.java

@@ -59,7 +59,9 @@
 import io.opentelemetry.api.common.AttributeKey;
 import io.opentelemetry.api.common.KeyValue;
 import io.opentelemetry.api.common.Value;
+import io.opentelemetry.api.trace.Span;
 import io.opentelemetry.api.trace.SpanContext;
+import io.opentelemetry.context.Context;
 import io.opentelemetry.instrumentation.testing.junit.InstrumentationExtension;
 import io.opentelemetry.instrumentation.testing.recording.RecordingExtension;
 import io.opentelemetry.sdk.testing.assertj.SpanDataAssert;
@@ -158,7 +160,15 @@ protected final ChatCompletion doCompletions(
       case ASYNC_FROM_SYNC:
         OpenAIClientAsync cl = testType == TestType.ASYNC ? clientAsync : client.async();
         try {
-          return cl.chat().completions().create(params).join();
+          return cl.chat()
+              .completions()
+              .create(params)
+              .thenApply(
+                  res -> {
+                    assertThat(Span.fromContextOrNull(Context.current())).isNull();
+                    return res;
+                  })
+              .join();
         } catch (CompletionException e) {
           if (e.getCause() instanceof OpenAIIoException) {
             throw ((OpenAIIoException) e.getCause());