Skip to content

Commit d4cca33

Browse files
committed
cr fixes
Signed-off-by: chrislai <[email protected]>
1 parent a185042 commit d4cca33

File tree

1 file changed

+3
-1
lines changed
  • ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/agent/tracing

1 file changed

+3
-1
lines changed

ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/agent/tracing/MLAgentTracer.java

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
import org.opensearch.telemetry.tracing.Tracer;
1515
import org.opensearch.telemetry.tracing.attributes.Attributes;
1616
import org.opensearch.telemetry.tracing.noop.NoopTracer;
17+
import org.opensearch.ml.repackage.com.google.common.annotations.VisibleForTesting;
1718

1819
import lombok.extern.log4j.Log4j2;
1920

@@ -36,7 +37,7 @@ public static synchronized void initialize(Tracer tracer, MLFeatureEnabledSettin
3637

3738
public static synchronized MLAgentTracer getInstance() {
3839
if (instance == null) {
39-
throw new IllegalStateException("MLAgentTracer is not initialized. Call initialize() first or check feature flag.");
40+
throw new IllegalStateException("MLAgentTracer is not initialized. Call initialize() first or enable plugins.ml_commons.agent_tracing_feature_enabled setting.");
4041
}
4142
return instance;
4243
}
@@ -107,6 +108,7 @@ public Tracer getTracer() {
107108
return tracer;
108109
}
109110

111+
@VisibleForTesting
110112
static void resetForTest() {
111113
instance = null;
112114
}

0 commit comments

Comments
 (0)