【代码优化】AI:完善 OpenAIChatModelTests、OpenAiImageModelTests 单测,方便大家快速体验

This commit is contained in:
YunaiV
2024-07-06 14:11:21 +08:00
parent e0f08a0f02
commit 1feff2b12b
6 changed files with 108 additions and 73 deletions

View File

@ -101,7 +101,7 @@ public class AiApiKeyServiceImpl implements AiApiKeyService {
public ChatModel getChatModel(Long id) {
AiApiKeyDO apiKey = validateApiKey(id);
AiPlatformEnum platform = AiPlatformEnum.validatePlatform(apiKey.getPlatform());
return modelFactory.getOrCreateChatClient(platform, apiKey.getApiKey(), apiKey.getUrl());
return modelFactory.getOrCreateChatModel(platform, apiKey.getApiKey(), apiKey.getUrl());
}
@Override