OpenAI: added maxCompletionTokens

langchain4j · Sep 16, 2024 · dfd6c11 · dfd6c11
1 parent bb1556f
commit dfd6c11
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/...j-open-ai-spring-boot-starter/src/main/java/dev/langchain4j/openai/spring/AutoConfig.java b/...j-open-ai-spring-boot-starter/src/main/java/dev/langchain4j/openai/spring/AutoConfig.java
@@ -26,6 +26,7 @@ OpenAiChatModel openAiChatModel(Properties properties) {
                 .topP(chatModelProperties.getTopP())
                 .stop(chatModelProperties.getStop())
                 .maxTokens(chatModelProperties.getMaxTokens())
+                .maxCompletionTokens(chatModelProperties.getMaxCompletionTokens())
                 .presencePenalty(chatModelProperties.getPresencePenalty())
                 .frequencyPenalty(chatModelProperties.getFrequencyPenalty())
                 .logitBias(chatModelProperties.getLogitBias())
@@ -57,6 +58,7 @@ OpenAiStreamingChatModel openAiStreamingChatModel(Properties properties) {
                 .topP(chatModelProperties.getTopP())
                 .stop(chatModelProperties.getStop())
                 .maxTokens(chatModelProperties.getMaxTokens())
+                .maxCompletionTokens(chatModelProperties.getMaxCompletionTokens())
                 .presencePenalty(chatModelProperties.getPresencePenalty())
                 .frequencyPenalty(chatModelProperties.getFrequencyPenalty())
                 .logitBias(chatModelProperties.getLogitBias())

diff --git a/...-spring-boot-starter/src/main/java/dev/langchain4j/openai/spring/ChatModelProperties.java b/...-spring-boot-starter/src/main/java/dev/langchain4j/openai/spring/ChatModelProperties.java
@@ -20,6 +20,7 @@ class ChatModelProperties {
     Double topP;
     List<String> stop;
     Integer maxTokens;
+    Integer maxCompletionTokens;
     Double presencePenalty;
     Double frequencyPenalty;
     Map<String, Integer> logitBias;