feat(open-ai): add support to configure custom query params via spring boot config

maff · maff · commit 78b0217caa3b · 2025-10-15T20:07:03.000+02:00
diff --git a/langchain4j-open-ai-spring-boot-starter/src/main/java/dev/langchain4j/openai/spring/AutoConfig.java b/langchain4j-open-ai-spring-boot-starter/src/main/java/dev/langchain4j/openai/spring/AutoConfig.java
@@ -86,6 +86,7 @@ OpenAiChatModel openAiChatModel(
                 .logRequests(chatModelProperties.logRequests())
                 .logResponses(chatModelProperties.logResponses())
                 .customHeaders(chatModelProperties.customHeaders())
+                .customQueryParams(chatModelProperties.customQueryParams())
                 .listeners(listeners.orderedStream().toList())
                 .build();
     }
@@ -140,6 +141,7 @@ OpenAiStreamingChatModel openAiStreamingChatModel(
                 .logRequests(chatModelProperties.logRequests())
                 .logResponses(chatModelProperties.logResponses())
                 .customHeaders(chatModelProperties.customHeaders())
+                .customQueryParams(chatModelProperties.customQueryParams())
                 .listeners(listeners.orderedStream().toList())
                 .build();
     }
@@ -198,6 +200,7 @@ OpenAiLanguageModel openAiLanguageModel(
                 .logRequests(languageModelProperties.logRequests())
                 .logResponses(languageModelProperties.logResponses())
                 .customHeaders(languageModelProperties.customHeaders())
+                .customQueryParams(languageModelProperties.customQueryParams())
                 .build();
     }
 
@@ -230,6 +233,7 @@ OpenAiStreamingLanguageModel openAiStreamingLanguageModel(
                 .logRequests(languageModelProperties.logRequests())
                 .logResponses(languageModelProperties.logResponses())
                 .customHeaders(languageModelProperties.customHeaders())
+                .customQueryParams(languageModelProperties.customQueryParams())
                 .build();
     }
 
@@ -290,6 +294,7 @@ OpenAiEmbeddingModel openAiEmbeddingModel(
                 .logRequests(embeddingModelProperties.logRequests())
                 .logResponses(embeddingModelProperties.logResponses())
                 .customHeaders(embeddingModelProperties.customHeaders())
+                .customQueryParams(embeddingModelProperties.customQueryParams())
                 .build();
     }
 
@@ -322,6 +327,7 @@ OpenAiModerationModel openAiModerationModel(
                 .logRequests(moderationModelProperties.logRequests())
                 .logResponses(moderationModelProperties.logResponses())
                 .customHeaders(moderationModelProperties.customHeaders())
+                .customQueryParams(moderationModelProperties.customQueryParams())
                 .build();
     }
 
@@ -359,6 +365,7 @@ OpenAiImageModel openAiImageModel(
                 .logRequests(imageModelProperties.logRequests())
                 .logResponses(imageModelProperties.logResponses())
                 .customHeaders(imageModelProperties.customHeaders())
+                .customQueryParams(imageModelProperties.customQueryParams())
                 .build();
     }
 
diff --git a/langchain4j-open-ai-spring-boot-starter/src/main/java/dev/langchain4j/openai/spring/ChatModelProperties.java b/langchain4j-open-ai-spring-boot-starter/src/main/java/dev/langchain4j/openai/spring/ChatModelProperties.java
@@ -38,6 +38,7 @@ record ChatModelProperties(
         Boolean logRequests,
         Boolean logResponses,
         Map<String, String> customHeaders,
+        Map<String, String> customQueryParams,
         Map<String, Object> customParameters
 ) {
 }
diff --git a/langchain4j-open-ai-spring-boot-starter/src/main/java/dev/langchain4j/openai/spring/EmbeddingModelProperties.java b/langchain4j-open-ai-spring-boot-starter/src/main/java/dev/langchain4j/openai/spring/EmbeddingModelProperties.java
@@ -16,7 +16,8 @@ record EmbeddingModelProperties(
         Integer maxRetries,
         Boolean logRequests,
         Boolean logResponses,
-        Map<String, String> customHeaders
+        Map<String, String> customHeaders,
+        Map<String, String> customQueryParams
 ) {
 
 }
diff --git a/langchain4j-open-ai-spring-boot-starter/src/main/java/dev/langchain4j/openai/spring/ImageModelProperties.java b/langchain4j-open-ai-spring-boot-starter/src/main/java/dev/langchain4j/openai/spring/ImageModelProperties.java
@@ -18,6 +18,7 @@ record ImageModelProperties(
         Integer maxRetries,
         Boolean logRequests,
         Boolean logResponses,
-        Map<String, String> customHeaders
+        Map<String, String> customHeaders,
+        Map<String, String> customQueryParams
 ) {
 }
diff --git a/langchain4j-open-ai-spring-boot-starter/src/main/java/dev/langchain4j/openai/spring/LanguageModelProperties.java b/langchain4j-open-ai-spring-boot-starter/src/main/java/dev/langchain4j/openai/spring/LanguageModelProperties.java
@@ -14,6 +14,7 @@ record LanguageModelProperties(
         Integer maxRetries,
         Boolean logRequests,
         Boolean logResponses,
-        Map<String, String> customHeaders
+        Map<String, String> customHeaders,
+        Map<String, String> customQueryParams
 ) {
 }
diff --git a/langchain4j-open-ai-spring-boot-starter/src/main/java/dev/langchain4j/openai/spring/ModerationModelProperties.java b/langchain4j-open-ai-spring-boot-starter/src/main/java/dev/langchain4j/openai/spring/ModerationModelProperties.java
@@ -13,6 +13,7 @@ record ModerationModelProperties(
         Integer maxRetries,
         Boolean logRequests,
         Boolean logResponses,
-        Map<String, String> customHeaders
+        Map<String, String> customHeaders,
+        Map<String, String> customQueryParams
 ) {
 }

Original file line number	Diff line number	Diff line change
`@@ -38,6 +38,7 @@ record ChatModelProperties(`
`38`	`38`	`Boolean logRequests,`
`39`	`39`	`Boolean logResponses,`
`40`	`40`	`Map<String, String> customHeaders,`
	`41`	`+ Map<String, String> customQueryParams,`
`41`	`42`	`Map<String, Object> customParameters`
`42`	`43`	`) {`
`43`	`44`	`}`
Original file line number	Diff line number	Diff line change
`@@ -16,7 +16,8 @@ record EmbeddingModelProperties(`
`16`	`16`	`Integer maxRetries,`
`17`	`17`	`Boolean logRequests,`
`18`	`18`	`Boolean logResponses,`
`19`		`- Map<String, String> customHeaders`
	`19`	`+ Map<String, String> customHeaders,`
	`20`	`+ Map<String, String> customQueryParams`
`20`	`21`	`) {`
`21`	`22`
`22`	`23`	`}`
Original file line number	Diff line number	Diff line change
`@@ -18,6 +18,7 @@ record ImageModelProperties(`
`18`	`18`	`Integer maxRetries,`
`19`	`19`	`Boolean logRequests,`
`20`	`20`	`Boolean logResponses,`
`21`		`- Map<String, String> customHeaders`
	`21`	`+ Map<String, String> customHeaders,`
	`22`	`+ Map<String, String> customQueryParams`
`22`	`23`	`) {`
`23`	`24`	`}`
Original file line number	Diff line number	Diff line change
`@@ -14,6 +14,7 @@ record LanguageModelProperties(`
`14`	`14`	`Integer maxRetries,`
`15`	`15`	`Boolean logRequests,`
`16`	`16`	`Boolean logResponses,`
`17`		`- Map<String, String> customHeaders`
	`17`	`+ Map<String, String> customHeaders,`
	`18`	`+ Map<String, String> customQueryParams`
`18`	`19`	`) {`
`19`	`20`	`}`
Original file line number	Diff line number	Diff line change
`@@ -13,6 +13,7 @@ record ModerationModelProperties(`
`13`	`13`	`Integer maxRetries,`
`14`	`14`	`Boolean logRequests,`
`15`	`15`	`Boolean logResponses,`
`16`		`- Map<String, String> customHeaders`
	`16`	`+ Map<String, String> customHeaders,`
	`17`	`+ Map<String, String> customQueryParams`
`17`	`18`	`) {`
`18`	`19`	`}`