fix: pass instance logger to createGraphTracker for resumed runs

Author: mattrmc1

lib/sdk/server-ai/src/main/java/com/launchdarkly/sdk/server/ai/AIGraphTracker.java

@@ -92,6 +92,22 @@ public final class AIGraphTracker {
    */
   public static AIGraphTracker fromResumptionToken(
       String token, LDClientInterface client, LDContext context) {
+    return fromResumptionToken(token, client, context, defaultLogger());
+  }
+
+  /**
+   * Reconstructs a graph tracker from a resumption token, preserving the original run identity,
+   * and logging through the supplied logger.
+   *
+   * @param token the resumption token produced by {@link #getResumptionToken()}
+   * @param client the LaunchDarkly client; must not be {@code null}
+   * @param context the evaluation context; must not be {@code null}
+   * @param logger the logger to use for at-most-once warnings; must not be {@code null}
+   * @return a new tracker with the decoded run identity
+   * @throws IllegalArgumentException if the token is malformed
+   */
+  public static AIGraphTracker fromResumptionToken(
+      String token, LDClientInterface client, LDContext context, LDLogger logger) {
     ResumptionTokens.DecodedGraph d = ResumptionTokens.decodeGraph(token);
     int version = Math.max(1, d.getVersion());
     return new AIGraphTracker(
@@ -101,7 +117,7 @@ public static AIGraphTracker fromResumptionToken(
         d.getVariationKey(),
         version,
         context,
-        defaultLogger());
+        logger);
   }
 
   /**

lib/sdk/server-ai/src/main/java/com/launchdarkly/sdk/server/ai/LDAIClientImpl.java

@@ -444,7 +444,7 @@ private static Set<String> collectReachableKeys(AgentGraphFlagValue flagValue) {
 
   @Override
   public AIGraphTracker createGraphTracker(String resumptionToken, LDContext context) {
-    return AIGraphTracker.fromResumptionToken(resumptionToken, client, context);
+    return AIGraphTracker.fromResumptionToken(resumptionToken, client, context, logger);
   }
 
   @Override

lib/sdk/server-ai/src/test/java/com/launchdarkly/sdk/server/ai/AIGraphTrackerTest.java

@@ -382,6 +382,24 @@ public void fromResumptionTokenPreservesRunId() {
     assertThat(captor.getValue().get("graphKey").stringValue(), is(GRAPH_KEY));
   }
 
+  @Test
+  public void fromResumptionTokenWithLoggerRoutesWarningsThroughIt() {
+    LogCapture captureForResumed = Logs.capture();
+    LDLogger resumedLogger = LDLogger.withAdapter(captureForResumed, "test");
+    String token = tracker.getResumptionToken();
+    AIGraphTracker reconstructed = AIGraphTracker.fromResumptionToken(token, client, CONTEXT, resumedLogger);
+
+    reconstructed.trackInvocationSuccess();
+    reconstructed.trackInvocationSuccess(); // duplicate — should warn on resumedLogger, not the base logCapture
+
+    List<String> resumedWarnings = captureForResumed.getMessages().stream()
+        .filter(m -> m.getLevel() == LDLogLevel.WARN)
+        .map(LogCapture.Message::getText)
+        .collect(Collectors.toList());
+    assertThat(resumedWarnings.stream().anyMatch(w -> w.contains("invocation already recorded")), is(true));
+    assertThat(logCapture.getMessages(), is(org.hamcrest.Matchers.empty()));
+  }
+
   @Test
   public void fromResumptionTokenClampsVersionLessThanOne() {
     AIGraphTracker t = new AIGraphTracker(client, RUN_ID, GRAPH_KEY, null, 0, CONTEXT, logger);