Merge branch 'main' into azure-open-ai-structured-outputs

langchain4j · Dec 20, 2024 · 0335070 · 0335070
2 parents 2cefbfa + 3bdf8f2
commit 0335070
Show file tree

Hide file tree

Showing 2 changed files with 2 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
@@ -122,6 +122,7 @@ OpenAiEmbeddingModel openAiEmbeddingModel(Properties properties) {
                 .organizationId(embeddingModelProperties.organizationId())
                 .modelName(embeddingModelProperties.modelName())
                 .dimensions(embeddingModelProperties.dimensions())
+                .maxSegmentsPerBatch(embeddingModelProperties.maxSegmentsPerBatch())
                 .user(embeddingModelProperties.user())
                 .timeout(embeddingModelProperties.timeout())
                 .maxRetries(embeddingModelProperties.maxRetries())

diff --git a/...ng-boot-starter/src/main/java/dev/langchain4j/openai/spring/EmbeddingModelProperties.java b/...ng-boot-starter/src/main/java/dev/langchain4j/openai/spring/EmbeddingModelProperties.java
@@ -13,6 +13,7 @@ record EmbeddingModelProperties(
     String organizationId,
     String modelName,
     Integer dimensions,
+    Integer maxSegmentsPerBatch,
     String user,
     Duration timeout,
     Integer maxRetries,