**Trace**: Block internal opentelemetry implementation

Author: JqRrt

CHANGELOG.md

@@ -6,7 +6,11 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
 
-## [0.1.1] - 2025-11-11
+## [0.1.2] - 2026-02-04
+### Changed
+- **Trace**: Block internal opentelemetry implementation
+
+## [0.1.1] - 2026-02-04
 ### Changed
 - **Cozeloop Client**: Support get workspace id and token from env
 

cozeloop-core/pom.xml

@@ -8,7 +8,7 @@
     <parent>
         <groupId>com.coze</groupId>
         <artifactId>cozeloop-java</artifactId>
-        <version>0.1.1</version>
+        <version>0.1.2</version>
         <relativePath>../pom.xml</relativePath>
     </parent>
 

cozeloop-core/src/main/java/com/coze/loop/client/CozeLoopClient.java

@@ -9,11 +9,9 @@
 import com.coze.loop.entity.Prompt;
 import com.coze.loop.prompt.GetPromptParam;
 import com.coze.loop.stream.StreamReader;
+import com.coze.loop.trace.CozeLoopContext;
 import com.coze.loop.trace.CozeLoopSpan;
 
-import io.opentelemetry.api.trace.Tracer;
-import io.opentelemetry.context.Context;
-
 /** Main interface for CozeLoop SDK client. Provides trace and prompt functionality. */
 public interface CozeLoopClient extends AutoCloseable {
 
@@ -58,24 +56,17 @@ public interface CozeLoopClient extends AutoCloseable {
    * @param parentContext the parent context
    * @return the span wrapper
    */
-  CozeLoopSpan startSpan(String name, String spanType, Context parentContext);
+  CozeLoopSpan startSpan(String name, String spanType, CozeLoopContext parentContext);
 
-  CozeLoopSpan startSpan(String name, String spanType, Context parentContext, String scene);
+  CozeLoopSpan startSpan(String name, String spanType, CozeLoopContext parentContext, String scene);
 
   /**
    * Extract context from headers for cross-service propagation.
    *
    * @param headers the map of headers (e.g. from an incoming HTTP request)
    * @return the extracted context
    */
-  Context extractContext(Map<String, String> headers);
-
-  /**
-   * Get the underlying OpenTelemetry Tracer.
-   *
-   * @return tracer instance
-   */
-  Tracer getTracer();
+  CozeLoopContext extractContext(Map<String, String> headers);
 
   // ========== Prompt Operations ==========
 

cozeloop-core/src/main/java/com/coze/loop/client/CozeLoopClientImpl.java

@@ -19,6 +19,7 @@
 import com.coze.loop.prompt.PromptProvider;
 import com.coze.loop.spec.tracespec.SpanKeys;
 import com.coze.loop.stream.StreamReader;
+import com.coze.loop.trace.CozeLoopContext;
 import com.coze.loop.trace.CozeLoopSpan;
 import com.coze.loop.trace.CozeLoopTracerProvider;
 
@@ -84,10 +85,7 @@ public CozeLoopSpan startSpan(String name, String spanType, String scene) {
   public CozeLoopSpan startSpan(String name, String spanType, CozeLoopSpan parentSpan) {
     checkNotClosed();
 
-    Context parentContext =
-        parentSpan != null && parentSpan.getSpan() != null
-            ? Context.current().with(parentSpan.getSpan())
-            : null;
+    CozeLoopContext parentContext = parentSpan != null ? parentSpan.getContext() : null;
 
     return startSpan(name, spanType, parentContext);
   }
@@ -97,23 +95,21 @@ public CozeLoopSpan startSpan(
       String name, String spanType, CozeLoopSpan parentSpan, String scene) {
     checkNotClosed();
 
-    Context parentContext =
-        parentSpan != null && parentSpan.getSpan() != null
-            ? Context.current().with(parentSpan.getSpan())
-            : null;
+    CozeLoopContext parentContext = parentSpan != null ? parentSpan.getContext() : null;
 
     return startSpan(name, spanType, parentContext, scene);
   }
 
   @Override
-  public CozeLoopSpan startSpan(String name, String spanType, Context parentContext) {
+  public CozeLoopSpan startSpan(String name, String spanType, CozeLoopContext parentContext) {
     checkNotClosed();
 
     return startSpan(name, spanType, parentContext, "");
   }
 
   @Override
-  public CozeLoopSpan startSpan(String name, String spanType, Context parentContext, String scene) {
+  public CozeLoopSpan startSpan(
+      String name, String spanType, CozeLoopContext parentContext, String scene) {
     checkNotClosed();
 
     SpanBuilder spanBuilder = tracer.spanBuilder(name);
@@ -140,39 +136,35 @@ public CozeLoopSpan startSpan(String name, String spanType, Context parentContex
     Context finalContext = contextToUse.with(span);
     Scope scope = finalContext.makeCurrent();
 
-    return new CozeLoopSpan(span, scope, tracerProvider.getPropagators(), finalContext, scene);
+    return new CozeLoopSpan(
+        span, scope, tracerProvider.getPropagators(), new CozeLoopContext(finalContext), scene);
   }
 
   @Override
-  public Context extractContext(Map<String, String> headers) {
+  public CozeLoopContext extractContext(Map<String, String> headers) {
     checkNotClosed();
     if (headers == null || headers.isEmpty()) {
-      return Context.current();
+      return CozeLoopContext.current();
     }
 
-    return tracerProvider
-        .getPropagators()
-        .getTextMapPropagator()
-        .extract(
-            Context.current(),
-            headers,
-            new TextMapGetter<Map<String, String>>() {
-              @Override
-              public Iterable<String> keys(Map<String, String> carrier) {
-                return carrier.keySet();
-              }
-
-              @Override
-              public String get(Map<String, String> carrier, String key) {
-                return carrier.get(key);
-              }
-            });
-  }
-
-  @Override
-  public Tracer getTracer() {
-    checkNotClosed();
-    return tracer;
+    return new CozeLoopContext(
+        tracerProvider
+            .getPropagators()
+            .getTextMapPropagator()
+            .extract(
+                Context.current(),
+                headers,
+                new TextMapGetter<Map<String, String>>() {
+                  @Override
+                  public Iterable<String> keys(Map<String, String> carrier) {
+                    return carrier.keySet();
+                  }
+
+                  @Override
+                  public String get(Map<String, String> carrier, String key) {
+                    return carrier.get(key);
+                  }
+                }));
   }
 
   // ========== Prompt Operations ==========

cozeloop-core/src/main/java/com/coze/loop/entity/Message.java

@@ -3,9 +3,11 @@
 import java.util.ArrayList;
 import java.util.List;
 
+import com.fasterxml.jackson.annotation.JsonInclude;
 import com.fasterxml.jackson.annotation.JsonProperty;
 
 /** Message entity. */
+@JsonInclude(JsonInclude.Include.NON_NULL)
 public class Message {
   @JsonProperty("role")
   private Role role;

cozeloop-core/src/main/java/com/coze/loop/internal/Constants.java

@@ -1,7 +1,7 @@
 package com.coze.loop.internal;
 
 public class Constants {
-  public static final String Version = "0.1.0";
+  public static final String Version = "0.1.2";
   public static final String TraceContextHeaderParent = "X-Cozeloop-Traceparent";
   public static final String TraceContextHeaderBaggage = "X-Cozeloop-Tracestate";
 }

cozeloop-core/src/main/java/com/coze/loop/trace/CozeLoopContext.java

@@ -0,0 +1,45 @@
+package com.coze.loop.trace;
+
+import org.jetbrains.annotations.Nullable;
+
+import io.opentelemetry.context.Context;
+import io.opentelemetry.context.ContextKey;
+
+/**
+ * Wrapper for OpenTelemetry Context to shield internal implementation.
+ *
+ * <p>This class provides a way to work with trace context without directly exposing OpenTelemetry
+ * APIs.
+ */
+public class CozeLoopContext implements Context {
+  private final Context context;
+
+  /**
+   * Create a new CozeLoopContext wrapping an OpenTelemetry Context.
+   *
+   * @param context the underlying OpenTelemetry context
+   */
+  public CozeLoopContext(Context context) {
+    this.context = context;
+  }
+
+  /**
+   * Get the current context from thread-local storage.
+   *
+   * @return the current context
+   */
+  public static CozeLoopContext current() {
+    return new CozeLoopContext(Context.current());
+  }
+
+  @Nullable
+  @Override
+  public <V> V get(ContextKey<V> contextKey) {
+    return context.get(contextKey);
+  }
+
+  @Override
+  public <V> Context with(ContextKey<V> contextKey, V v) {
+    return context.with(contextKey, v);
+  }
+}