topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-12 12:37:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[improvement](chat) fix InProcess config (#509)

Browse Source
This commit is contained in:
lexluo09
2023-12-14 17:10:27 +08:00
committed by GitHub
parent 3bf5b86535
commit 7db1cc270e
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
launchers/common/src/main/java/dev/langchain4j/S2LangChain4jAutoConfiguration.java
View File

@@ -20,6 +20,7 @@ import dev.langchain4j.model.openai.OpenAiChatModel;
import dev.langchain4j.model.openai.OpenAiEmbeddingModel; import dev.langchain4j.model.openai.OpenAiEmbeddingModel;
import dev.langchain4j.model.openai.OpenAiLanguageModel; import dev.langchain4j.model.openai.OpenAiLanguageModel;
import dev.langchain4j.model.openai.OpenAiModerationModel; import dev.langchain4j.model.openai.OpenAiModerationModel;
import java.util.Objects;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean; import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.context.properties.EnableConfigurationProperties; import org.springframework.boot.context.properties.EnableConfigurationProperties;
@@ -246,7 +247,7 @@ public class S2LangChain4jAutoConfiguration {
.build(); .build();
case IN_PROCESS: case IN_PROCESS:
InProcess inProcess = properties.getEmbeddingModel().getInProcess(); InProcess inProcess = properties.getEmbeddingModel().getInProcess();
if (isNullOrBlank(inProcess.getModelPath())) { if (Objects.isNull(inProcess) || isNullOrBlank(inProcess.getModelPath())) {
return new AllMiniLmL6V2EmbeddingModel(); return new AllMiniLmL6V2EmbeddingModel();
} }
return new S2OnnxEmbeddingModel(inProcess.getModelPath(), inProcess.getVocabularyPath()); return new S2OnnxEmbeddingModel(inProcess.getModelPath(), inProcess.getVocabularyPath());