hhyykk/ipms-sjy
1
1
Fork 0
mirror of https://gitee.com/hhyykk/ipms-sjy.git synced 2025-07-17 04:25:06 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

【代码优化】AI：完善 LlamaChatModelTests 单测，方便大家快速体验

Browse Source
This commit is contained in:
YunaiV
2024-07-06 14:47:56 +08:00
parent 1feff2b12b
commit 0139317ac4
2 changed files with 66 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/factory/AiModelFactoryImpl.java
View File

@ -55,8 +55,6 @@ public class AiModelFactoryImpl implements AiModelFactory {
return Singleton.get(cacheKey, (Func0<ChatModel>) () -> {
//noinspection EnhancedSwitchMigration
switch (platform) {
case OLLAMA:
return buildOllamaChatClient(url);
case YI_YAN:
return buildYiYanChatClient(apiKey);
case XING_HUO:
@ -67,6 +65,8 @@ public class AiModelFactoryImpl implements AiModelFactory {
return buildDeepSeekChatClient(apiKey);
case OPENAI:
return buildOpenAiChatModel(apiKey, url);
case OLLAMA:
return buildOllamaChatModel(url);
default:
throw new IllegalArgumentException(StrUtil.format("未知平台({})", platform));
}
@ -163,7 +163,7 @@ public class AiModelFactoryImpl implements AiModelFactory {
/**
* 可参考 {@link OllamaAutoConfiguration}
*/
private static OllamaChatModel buildOllamaChatClient(String url) {
private static OllamaChatModel buildOllamaChatModel(String url) {
OllamaApi ollamaApi = new OllamaApi(url);
return new OllamaChatModel(ollamaApi);
}