feat: Add JSON cycle detection

This chanve enhances the JsonFormatter within the ADK core to detect and handle circular references in JSON structures during smart truncation, preventing infinite recursion.

PiperOrigin-RevId: 904554270

Author: google-genai-bot

core/src/main/java/com/google/adk/plugins/agentanalytics/JsonFormatter.java

@@ -30,6 +30,8 @@
 import com.google.genai.types.FunctionCall;
 import com.google.genai.types.Part;
 import java.util.ArrayList;
+import java.util.Collections;
+import java.util.IdentityHashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Optional;
@@ -304,8 +306,13 @@ static TruncationResult smartTruncate(Object obj, int maxLength) {
     if (obj == null) {
       return TruncationResult.create(mapper.nullNode(), false);
     }
+    if (obj instanceof JsonNode jsonNode) {
+      return recursiveSmartTruncate(
+          jsonNode, maxLength, Collections.newSetFromMap(new IdentityHashMap<>()));
+    }
     try {
-      return recursiveSmartTruncate(mapper.valueToTree(obj), maxLength);
+      return recursiveSmartTruncate(
+          mapper.valueToTree(obj), maxLength, Collections.newSetFromMap(new IdentityHashMap<>()));
     } catch (IllegalArgumentException e) {
       // Fallback for types that mapper can't handle directly as a tree
       return truncateWithStatus(String.valueOf(obj), maxLength);
@@ -324,33 +331,47 @@ static JsonNode convertToJsonNode(Object obj) {
     }
   }
 
-  private static TruncationResult recursiveSmartTruncate(JsonNode node, int maxLength) {
-    boolean isTruncated = false;
-    if (node.isTextual()) {
-      String text = node.asText();
-      if (text.length() > maxLength) {
-        return TruncationResult.create(mapper.valueToTree(truncate(text, maxLength)), true);
+  private static TruncationResult recursiveSmartTruncate(
+      JsonNode node, int maxLength, Set<JsonNode> visited) {
+    if (node.isContainerNode()) {
+      if (visited.contains(node)) {
+        return TruncationResult.create(mapper.valueToTree("[CYCLE DETECTED]"), true);
       }
-      return TruncationResult.create(node, false);
-    } else if (node.isObject()) {
-      ObjectNode newNode = mapper.createObjectNode();
-      Set<Map.Entry<String, JsonNode>> properties = node.properties();
-      for (Map.Entry<String, JsonNode> entry : properties) {
-        TruncationResult res = recursiveSmartTruncate(entry.getValue(), maxLength);
-        newNode.set(entry.getKey(), res.node());
-        isTruncated = isTruncated || res.isTruncated();
+      visited.add(node);
+    }
+
+    try {
+      boolean isTruncated = false;
+      if (node.isTextual()) {
+        String text = node.asText();
+        if (text.length() > maxLength) {
+          return TruncationResult.create(mapper.valueToTree(truncate(text, maxLength)), true);
+        }
+        return TruncationResult.create(node, false);
+      } else if (node.isObject()) {
+        ObjectNode newNode = mapper.createObjectNode();
+        Set<Map.Entry<String, JsonNode>> properties = node.properties();
+        for (Map.Entry<String, JsonNode> entry : properties) {
+          TruncationResult res = recursiveSmartTruncate(entry.getValue(), maxLength, visited);
+          newNode.set(entry.getKey(), res.node());
+          isTruncated = isTruncated || res.isTruncated();
+        }
+        return TruncationResult.create(newNode, isTruncated);
+      } else if (node.isArray()) {
+        ArrayNode newNode = mapper.createArrayNode();
+        for (JsonNode element : node) {
+          TruncationResult res = recursiveSmartTruncate(element, maxLength, visited);
+          newNode.add(res.node());
+          isTruncated = isTruncated || res.isTruncated();
+        }
+        return TruncationResult.create(newNode, isTruncated);
       }
-      return TruncationResult.create(newNode, isTruncated);
-    } else if (node.isArray()) {
-      ArrayNode newNode = mapper.createArrayNode();
-      for (JsonNode element : node) {
-        TruncationResult res = recursiveSmartTruncate(element, maxLength);
-        newNode.add(res.node());
-        isTruncated = isTruncated || res.isTruncated();
+      return TruncationResult.create(node, false);
+    } finally {
+      if (node.isContainerNode()) {
+        visited.remove(node);
       }
-      return TruncationResult.create(newNode, isTruncated);
     }
-    return TruncationResult.create(node, false);
   }
 
   private static TruncationResult truncateWithStatus(String s, int maxLength) {

core/src/test/java/com/google/adk/plugins/agentanalytics/JsonFormatterTest.java

@@ -21,7 +21,9 @@
 import static org.junit.Assert.assertTrue;
 
 import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
 import com.fasterxml.jackson.databind.node.ArrayNode;
+import com.fasterxml.jackson.databind.node.ObjectNode;
 import com.google.adk.models.LlmRequest;
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.ImmutableMap;
@@ -142,4 +144,17 @@ public void parse_list_truncatesElements() {
     assertEquals("short", arrayNode.get(0).asText());
     assertEquals("this is a ...[truncated]", arrayNode.get(1).asText());
   }
+
+  @Test
+  public void smartTruncate_withCycle_detectsCycle() {
+    ObjectMapper mapper = new ObjectMapper();
+    ObjectNode node = mapper.createObjectNode();
+    node.set("child", node);
+
+    // Verify that smartTruncate handles circular JsonNode structures by detecting the cycle.
+    JsonFormatter.TruncationResult result = JsonFormatter.smartTruncate(node, 100);
+
+    assertTrue(result.isTruncated());
+    assertEquals("[CYCLE DETECTED]", result.node().get("child").asText());
+  }
 }