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

[fix][chat]Model provider is case sensitive.#1739

Browse Source
This commit is contained in:
jerryjzhang
2024-10-09 17:48:42 +08:00
parent d1902e2704
commit cd57d2344e
4 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
launchers/standalone/src/test/java/com/tencent/supersonic/util/LLMConfigUtils.java
View File

@@ -65,11 +65,11 @@ public class LLMConfigUtils {
ChatModelConfig chatModelConfig;
if (type.isOllam) {
chatModelConfig = ChatModelConfig.builder().provider("ollama").baseUrl(baseUrl)
chatModelConfig = ChatModelConfig.builder().provider("OLLAMA").baseUrl(baseUrl)
.modelName(modelName).temperature(temperature).timeOut(60000L).build();
} else {
chatModelConfig =
ChatModelConfig.builder().provider("open_ai").baseUrl(baseUrl).apiKey(apiKey)
ChatModelConfig.builder().provider("OPEN_AI").baseUrl(baseUrl).apiKey(apiKey)
.modelName(modelName).temperature(temperature).timeOut(60000L).build();
}