GenAI ToolKit: update to 3.1.5 to pick the 'only-first' option in the…

… query step (LangStream#171)
vectorize-io · Aug 3, 2023 · af98cc9 · af98cc9
1 parent 3027fa5
commit af98cc9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.../src/main/java/com/datastax/oss/sga/impl/agents/ai/GenAIToolKitFunctionAgentProvider.java b/.../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
@@ -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>