diff --git a/.codegen.json b/.codegen.json
index 41241aeb2..49ca5c829 100644
--- a/.codegen.json
+++ b/.codegen.json
@@ -1 +1 @@
-{ "engineHash": "f9e2519", "specHash": "ccdb456", "version": "10.4.0" }
+{ "engineHash": "b181eba", "specHash": "ccdb456", "version": "10.4.0" }
diff --git a/docs/ai.md b/docs/ai.md
index cb2746275..157ed87c8 100644
--- a/docs/ai.md
+++ b/docs/ai.md
@@ -149,7 +149,7 @@ See the endpoint docs at
 
 <!-- sample post_ai_extract_structured -->
 ```
-client.getAi().createAiExtractStructured(new AiExtractStructured.Builder(Arrays.asList(new AiItemBase(file.getId()))).fields(Arrays.asList(new AiExtractStructuredFieldsField.Builder("firstName").description("Person first name").displayName("First name").prompt("What is the your first name?").type("string").build(), new AiExtractStructuredFieldsField.Builder("lastName").description("Person last name").displayName("Last name").prompt("What is the your last name?").type("string").build(), new AiExtractStructuredFieldsField.Builder("dateOfBirth").description("Person date of birth").displayName("Birth date").prompt("What is the date of your birth?").type("date").build(), new AiExtractStructuredFieldsField.Builder("age").description("Person age").displayName("Age").prompt("How old are you?").type("float").build(), new AiExtractStructuredFieldsField.Builder("hobby").description("Person hobby").displayName("Hobby").prompt("What is your hobby?").type("multiSelect").options(Arrays.asList(new AiExtractStructuredFieldsOptionsField("guitar"), new AiExtractStructuredFieldsOptionsField("books"))).build())).aiAgent(aiExtractStructuredAgentBasicTextConfig).build())
+client.getAi().createAiExtractStructured(new AiExtractStructured.Builder(Arrays.asList(new AiItemBase(file.getId()))).fields(Arrays.asList(new AiExtractStructuredFieldsField.Builder("firstName").description("Person first name").displayName("First name").prompt("What is the your first name?").type("string").build(), new AiExtractStructuredFieldsField.Builder("lastName").description("Person last name").displayName("Last name").prompt("What is the your last name?").type("string").build(), new AiExtractStructuredFieldsField.Builder("dateOfBirth").description("Person date of birth").displayName("Birth date").prompt("What is the date of your birth?").type("date").build(), new AiExtractStructuredFieldsField.Builder("age").description("Person age").displayName("Age").prompt("How old are you?").type("float").build(), new AiExtractStructuredFieldsField.Builder("hobby").description("Person hobby").displayName("Hobby").prompt("What is your hobby?").type("multiSelect").options(Arrays.asList(new AiExtractStructuredFieldsOptionsField("guitar"), new AiExtractStructuredFieldsOptionsField("books"))).build())).includeConfidenceScore(true).aiAgent(aiExtractStructuredAgentBasicTextConfig).build())
 ```
 
 ### Arguments
diff --git a/src/test/java/com/box/sdkgen/ai/AiITest.java b/src/test/java/com/box/sdkgen/ai/AiITest.java
index dbfbe8533..892e21d9e 100644
--- a/src/test/java/com/box/sdkgen/ai/AiITest.java
+++ b/src/test/java/com/box/sdkgen/ai/AiITest.java
@@ -261,7 +261,10 @@ public void testAiExtractStructuredWithFields() {
                         String.join("", getUuid(), ".txt"),
                         new UploadFileRequestBodyAttributesParentField("0")),
                     stringToByteStream(
-                        "My name is John Doe. I was born in 4th July 1990. I am 34 years old. My hobby is guitar.")));
+                        String.join(
+                            "",
+                            "My name is John Doe. I was born in 4th July 1990. I am 34 years old. My hobby is guitar. My UUID is ",
+                            getUuid()))));
     FileFull file = uploadedFiles.getEntries().get(0);
     delayInSeconds(5);
     AiExtractStructuredResponse response =
@@ -305,8 +308,10 @@ public void testAiExtractStructuredWithFields() {
                                         new AiExtractStructuredFieldsOptionsField("guitar"),
                                         new AiExtractStructuredFieldsOptionsField("books")))
                                 .build()))
+                    .includeConfidenceScore(true)
                     .aiAgent(aiExtractStructuredAgentBasicTextConfig)
                     .build());
+    assert !(response.getConfidenceScore() == null);
     assert convertToString(response.getAnswer().get("hobby"))
         .equals(convertToString(Arrays.asList("guitar")));
     assert convertToString(response.getAnswer().get("firstName")).equals("John");
@@ -328,7 +333,10 @@ public void testAiExtractStructuredWithMetadataTemplate() {
                         String.join("", getUuid(), ".txt"),
                         new UploadFileRequestBodyAttributesParentField("0")),
                     stringToByteStream(
-                        "My name is John Doe. I was born in 4th July 1990. I am 34 years old. My hobby is guitar.")));
+                        String.join(
+                            "",
+                            "My name is John Doe. I was born in 4th July 1990. I am 34 years old. My hobby is guitar. My UUID is ",
+                            getUuid()))));
     FileFull file = uploadedFiles.getEntries().get(0);
     delayInSeconds(5);
     String templateKey = String.join("", "key", getUuid());