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

【代码新增】AI：开启 AI 租户的能力

Browse Source
This commit is contained in:
YunaiV
2024-07-05 09:42:36 +08:00
parent c98495dfcb
commit 2e73959a2c
9 changed files with 38 additions and 25 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/AiClientFactoryImpl.java
View File

@ -163,7 +163,7 @@ public class AiClientFactoryImpl implements AiClientFactory {
* 可参考 {@link QianFanAutoConfiguration#qianFanChatModel(QianFanConnectionProperties, QianFanChatProperties, RestClient.Builder, RetryTemplate, ResponseErrorHandler)}
*/
private static QianFanChatModel buildYiYanChatClient(String key) {
// TODO 芋艿：貌似目前设置，request 势必会报错
// TODO @xin：貌似目前设置，request 势必会报错；看看能不能有办法，参考 buildQianWenChatClient，调用 QianFanAutoConfiguration#qianFanChatModel初始化，当然 key 要用自己的哈
List<String> keys = StrUtil.split(key, '|');
Assert.equals(keys.size(), 2, "YiYanChatClient 的密钥需要 (appKey|secretKey) 格式");
String appKey = keys.get(0);
@ -191,8 +191,8 @@ public class AiClientFactoryImpl implements AiClientFactory {
private static TongYiChatModel buildQianWenChatClient(String key) {
com.alibaba.dashscope.aigc.generation.Generation generation = SpringUtil.getBean(Generation.class);
TongYiChatProperties chatOptions = SpringUtil.getBean(TongYiChatProperties.class);
// TODO @芋艿：貌似 apiKey 是全局唯一的？？？得测试下
// TODO @芋艿：貌似阿里云不是增量返回的
// TODO @xin：貌似 apiKey 是全局唯一的？？？得测试下
// TODO @xin：貌似阿里云不是增量返回的
TongYiConnectionProperties connectionProperties = new TongYiConnectionProperties();
connectionProperties.setApiKey(key);
return new TongYiAutoConfiguration().tongYiChatClient(generation, chatOptions, connectionProperties);