diff --git a/core/src/main/java/com/datastax/oss/sga/impl/agents/ai/GenAIToolKitFunctionAgentProvider.java b/core/src/main/java/com/datastax/oss/sga/impl/agents/ai/GenAIToolKitFunctionAgentProvider.java
index 4ea8aff3d..c2f30964b 100644
--- a/core/src/main/java/com/datastax/oss/sga/impl/agents/ai/GenAIToolKitFunctionAgentProvider.java
+++ b/core/src/main/java/com/datastax/oss/sga/impl/agents/ai/GenAIToolKitFunctionAgentProvider.java
@@ -111,6 +111,7 @@ public void generateSteps(Map<String, Object> step, Map<String, Object> original
                     requiredField(step, agentConfiguration, originalConfiguration, "fields");
                     requiredField(step, agentConfiguration, originalConfiguration, "query");
                     requiredField(step, agentConfiguration, originalConfiguration, "output-field");
+                    optionalField(step, agentConfiguration, originalConfiguration, "only-first", null);
                 }
             },
             "ai-chat-completions", new StepConfigurationInitializer() {
diff --git a/pom.xml b/pom.xml
index 186d0a746..18909b168 100644
--- a/pom.xml
+++ b/pom.xml
@@ -64,7 +64,7 @@
     <bouncy-castle.version>1.70</bouncy-castle.version>
     <kindcontainer.version>1.4.1</kindcontainer.version>
     <wiremock.version>3.0.0-beta-10</wiremock.version>
-    <streaming-ai.version>3.1.4</streaming-ai.version>
+    <streaming-ai.version>3.1.5</streaming-ai.version>
     <jandex-maven-plugin.version>1.2.3</jandex-maven-plugin.version>
     <burningwave.version>0.25.5</burningwave.version>
     <jjwt.version>0.11.5</jjwt.version>