diff --git a/core/src/main/java/cucumber/runtime/formatter/JSONFormatter.java b/core/src/main/java/cucumber/runtime/formatter/JSONFormatter.java index 49f1f6ff6b..8a3ce356a3 100644 --- a/core/src/main/java/cucumber/runtime/formatter/JSONFormatter.java +++ b/core/src/main/java/cucumber/runtime/formatter/JSONFormatter.java @@ -290,11 +290,11 @@ private void addOutputToHookMap(String text) { } private void addEmbeddingToHookMap(byte[] data, String mimeType) { - if (!currentStepOrHookMap.containsKey("embedding")) { - currentStepOrHookMap.put("embedding", new ArrayList>()); + if (!currentStepOrHookMap.containsKey("embeddings")) { + currentStepOrHookMap.put("embeddings", new ArrayList>()); } Map embedMap = createEmbeddingMap(data, mimeType); - ((List>)currentStepOrHookMap.get("embedding")).add(embedMap); + ((List>)currentStepOrHookMap.get("embeddings")).add(embedMap); } private Map createEmbeddingMap(byte[] data, String mimeType) { diff --git a/core/src/test/java/cucumber/runtime/formatter/JSONFormatterTest.java b/core/src/test/java/cucumber/runtime/formatter/JSONFormatterTest.java index 7ccbc23d21..7b743675d4 100755 --- a/core/src/test/java/cucumber/runtime/formatter/JSONFormatterTest.java +++ b/core/src/test/java/cucumber/runtime/formatter/JSONFormatterTest.java @@ -584,7 +584,7 @@ public void should_handle_embed_from_a_hooks() throws Throwable { " \"match\": {\n" + " \"location\": \"Hooks.before_hook_1()\"\n" + " },\n" + - " \"embedding\": [\n" + + " \"embeddings\": [\n" + " {\n" + " \"mime_type\": \"mime-type;base64\",\n" + " \"data\": \"AQID\"\n" +