optimize test cases

Jashinck · Sep 30, 2023 · bc5e5cd · bc5e5cd
1 parent ab11233
commit bc5e5cd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...src/test/java/com/hw/langchain/agents/toolkits/flink/sql/toolkit/FlinkSqlToolkitTest.java b/...src/test/java/com/hw/langchain/agents/toolkits/flink/sql/toolkit/FlinkSqlToolkitTest.java
@@ -96,6 +96,7 @@ void testRunQuery() {
                 .init();
 
         var llm = ChatOpenAI.builder()
+                .model("gpt-4")
                 .temperature(0)
                 .build().init();
 
@@ -105,7 +106,7 @@ void testRunQuery() {
         // SELECT SQRT(AVG(Age)) FROM titanic
         var actual = agentExecutor.run("whats the square root of the average age?");
         // sometimes it's 'The square root of the average age is approximately 5.07.'
-        assertEquals("5.07", actual);
+        assertEquals("The square root of the average age is approximately 5.07.", actual);
 
         // TODO: It should be DESC here, not ASC.
         // SELECT Name FROM titanic WHERE Survived = 1 ORDER BY Age ASC LIMIT 1