Skip to content

Commit a1a4f05

Browse files
committed
Add temperature and timeout
1 parent 8315cdc commit a1a4f05

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/main/java/schemacrawler/tools/command/utility/lanchain4j/Langchain4JChatAssistant.java

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
package schemacrawler.tools.command.utility.lanchain4j;
3030

3131
import java.sql.Connection;
32+
import java.time.Duration;
3233
import java.util.Collection;
3334
import java.util.Map;
3435
import java.util.logging.Level;
@@ -89,8 +90,10 @@ public Langchain4JChatAssistant(
8990
OpenAiChatModel.builder()
9091
.apiKey(commandOptions.getApiKey())
9192
.modelName(commandOptions.getModel())
92-
.strictTools(
93-
true) // https://docs.langchain4j.dev/integrations/language-models/open-ai#structured-outputs-for-tools
93+
.temperature(0.2)
94+
.timeout(Duration.ofSeconds(commandOptions.getTimeout()))
95+
// https://docs.langchain4j.dev/integrations/language-models/open-ai#structured-outputs-for-tools
96+
.strictTools(true)
9497
.build();
9598

9699
assistant =

0 commit comments

Comments
 (0)