wanjia9506
/
langchain4j
mirror of https://github.com/langchain4j/langchain4j.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add support for custom base url for OpenAI model in Spring Boot starter (#228)

This commit is contained in:
marlukcz 2023-10-17 21:26:45 +02:00 committed by GitHub
parent f43cc8f4ca
commit 9719d8233f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
langchain4j-spring-boot-starter/src/main/java/dev/langchain4j/LangChain4jAutoConfiguration.java
View File

@ -46,6 +46,7 @@ public class LangChain4jAutoConfiguration {
throw illegalConfiguration("\n\nPlease define 'langchain4j.chat-model.openai.api-key' property"); throw illegalConfiguration("\n\nPlease define 'langchain4j.chat-model.openai.api-key' property");
} }
return OpenAiChatModel.builder() return OpenAiChatModel.builder()
.baseUrl(openAi.getBaseUrl())
.apiKey(openAi.getApiKey()) .apiKey(openAi.getApiKey())
.modelName(openAi.getModelName()) .modelName(openAi.getModelName())
.temperature(openAi.getTemperature()) .temperature(openAi.getTemperature())

9
langchain4j-spring-boot-starter/src/main/java/dev/langchain4j/OpenAi.java
View File

@ -4,6 +4,7 @@ import java.time.Duration;
class OpenAi { class OpenAi {
private String baseUrl;
private String apiKey; private String apiKey;
private String modelName; private String modelName;
private Double temperature; private Double temperature;
@ -16,6 +17,14 @@ class OpenAi {
private Boolean logRequests; private Boolean logRequests;
private Boolean logResponses; private Boolean logResponses;
public String getBaseUrl() {
return baseUrl;
}
public void setBaseUrl(String baseUrl) {
this.baseUrl = baseUrl;
}
public String getApiKey() { public String getApiKey() {
return apiKey; return apiKey;
} }