优化自动注入，可以创建多个 client

yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/AiPlatformEnum.java

@@ -0,0 +1,38 @@
+package cn.iocoder.yudao.framework.ai;
+
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+
+import java.util.Arrays;
+import java.util.Map;
+import java.util.stream.Collectors;
+
+/**
+ * 讯飞星火 模型
+ *
+ * 文档地址：https://www.xfyun.cn/doc/spark/Web.html#_1-%E6%8E%A5%E5%8F%A3%E8%AF%B4%E6%98%8E
+ *
+ * 1tokens 约等于1.5个中文汉字 或者 0.8个英文单词
+ * 星火V1.5支持[搜索]内置插件；星火V2.0、V3.0和V3.5支持[搜索]、[天气]、[日期]、[诗词]、[字词]、[股票]六个内置插件
+ * 星火V3.5 现已支持system、Function Call 功能。
+ *
+ * author: fansili
+ * time: 2024/3/11 10:12
+ */
+@Getter
+@AllArgsConstructor
+public enum AiPlatformEnum {
+
+
+    YI_YAN("yiyan"),
+    QIAN_WEN("qianwen"),
+    XING_HUO("xinghuo"),
+
+    ;
+
+    public static final Map<String, AiPlatformEnum> mapValues
+            = Arrays.stream(values()).collect(Collectors.toMap(AiPlatformEnum::name, o -> o));
+
+    private String value;
+
+}

yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/config/YudaoAiAutoConfiguration.java

@@ -1,11 +1,29 @@
 package cn.iocoder.yudao.framework.ai.config;
 
+import cn.hutool.core.bean.BeanUtil;
+import cn.iocoder.yudao.framework.ai.AiPlatformEnum;
+import cn.iocoder.yudao.framework.ai.chatqianwen.QianWenChatClient;
+import cn.iocoder.yudao.framework.ai.chatqianwen.QianWenOptions;
+import cn.iocoder.yudao.framework.ai.chatqianwen.api.QianWenApi;
 import cn.iocoder.yudao.framework.ai.chatxinghuo.XingHuoChatClient;
 import cn.iocoder.yudao.framework.ai.chatxinghuo.XingHuoOptions;
 import cn.iocoder.yudao.framework.ai.chatxinghuo.api.XingHuoApi;
+import cn.iocoder.yudao.framework.ai.chatyiyan.YiYanChatClient;
+import cn.iocoder.yudao.framework.ai.chatyiyan.YiYanOptions;
+import cn.iocoder.yudao.framework.ai.chatyiyan.api.YiYanApi;
+import cn.iocoder.yudao.framework.ai.exception.AiException;
+import org.springframework.beans.BeansException;
+import org.springframework.beans.factory.InitializingBean;
+import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
 import org.springframework.boot.autoconfigure.AutoConfiguration;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
 import org.springframework.boot.context.properties.EnableConfigurationProperties;
+import org.springframework.context.ApplicationContext;
+import org.springframework.context.ApplicationContextAware;
 import org.springframework.context.annotation.Bean;
+import org.springframework.context.support.GenericApplicationContext;
+
+import java.util.Map;
 
 /**
  * ai 自动配置
@@ -18,15 +36,118 @@ import org.springframework.context.annotation.Bean;
 @EnableConfigurationProperties(YudaoAiProperties.class)
 public class YudaoAiAutoConfiguration {
 
+    // TODO @芋艿：我看sharding jdbc 差不多这么玩的
     @Bean
-    public XingHuoChatClient xingHuoChatClient(YudaoAiProperties yudaoAiProperties) {
-        return new XingHuoChatClient(
-                new XingHuoApi(
-                        yudaoAiProperties.getXingHuo().getAppId(),
-                        yudaoAiProperties.getXingHuo().getAppKey(),
-                        yudaoAiProperties.getXingHuo().getSecretKey()
-                ),
-                new XingHuoOptions().setChatModel(yudaoAiProperties.getXingHuo().getChatModel())
-        );
+    @ConditionalOnMissingBean(value = InitChatClient.class)
+    public InitChatClient initChatClient(YudaoAiProperties yudaoAiProperties) {
+        return new InitChatClient(yudaoAiProperties);
     }
-}
+
+    public static class InitChatClient implements InitializingBean, ApplicationContextAware {
+
+        private GenericApplicationContext applicationContext;
+        private YudaoAiProperties yudaoAiProperties;
+
+        public InitChatClient(YudaoAiProperties yudaoAiProperties) {
+            this.yudaoAiProperties = yudaoAiProperties;
+        }
+
+        @Override
+        public void afterPropertiesSet() throws Exception {
+            for (Map.Entry<String, Map<String, Object>> properties : yudaoAiProperties.entrySet()) {
+                String beanName = properties.getKey();
+                Map<String, Object> aiPlatformMap = properties.getValue();
+
+                // 检查平台类型是否正确
+                String aiPlatform = String.valueOf(aiPlatformMap.get("aiPlatform"));
+                if (!AiPlatformEnum.mapValues.containsKey(aiPlatform)) {
+                    throw new AiException("AI平台名称错误! 可以参考 AiPlatformEnum 类!");
+                }
+                // 获取平台类型
+                AiPlatformEnum aiPlatformEnum = AiPlatformEnum.mapValues.get(aiPlatform);
+                // 获取 chat properties
+                YudaoAiProperties.ChatProperties chatProperties = getChatProperties(aiPlatformEnum, aiPlatformMap);
+                // 创建客户端
+                registerChatClient(applicationContext, chatProperties, beanName);
+//                applicationContext.refresh();
+
+
+            }
+
+            System.err.println(applicationContext.getBean("qianWen"));
+            System.err.println(applicationContext.getBean("yiYan"));
+        }
+
+        @Override
+        public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
+            this.applicationContext = (GenericApplicationContext) applicationContext;
+        }
+    }
+
+    private static void registerChatClient(GenericApplicationContext applicationContext, YudaoAiProperties.ChatProperties chatProperties, String beanName) {
+        ConfigurableListableBeanFactory beanFactory = applicationContext.getBeanFactory();
+        Object wrapperBean = null;
+        if (AiPlatformEnum.XING_HUO == chatProperties.getAiPlatform()) {
+            YudaoAiProperties.XingHuoProperties xingHuoProperties = (YudaoAiProperties.XingHuoProperties) chatProperties;
+            wrapperBean = beanFactory.initializeBean(
+                    new XingHuoChatClient(
+                            new XingHuoApi(
+                                    xingHuoProperties.getAppId(),
+                                    xingHuoProperties.getAppKey(),
+                                    xingHuoProperties.getSecretKey()
+                            ),
+                            new XingHuoOptions().setChatModel(xingHuoProperties.getChatModel())
+                    ),
+                    beanName
+            );
+        } else if (AiPlatformEnum.QIAN_WEN == chatProperties.getAiPlatform()) {
+            YudaoAiProperties.QianWenProperties qianWenProperties = (YudaoAiProperties.QianWenProperties) chatProperties;
+            wrapperBean = beanFactory.initializeBean(new QianWenChatClient(
+                            new QianWenApi(
+                                    qianWenProperties.getAccessKeyId(),
+                                    qianWenProperties.getAccessKeySecret(),
+                                    qianWenProperties.getAgentKey(),
+                                    qianWenProperties.getEndpoint()
+                            ),
+                            new QianWenOptions()
+                                    .setAppId(qianWenProperties.getAppId())
+                    ),
+                    beanName
+            );
+        } else if (AiPlatformEnum.YI_YAN == chatProperties.getAiPlatform()) {
+            YudaoAiProperties.YiYanProperties yiYanProperties = (YudaoAiProperties.YiYanProperties) chatProperties;
+
+            wrapperBean = beanFactory.initializeBean(new YiYanChatClient(
+                            new YiYanApi(
+                                    yiYanProperties.getAppKey(),
+                                    yiYanProperties.getSecretKey(),
+                                    yiYanProperties.getChatModel(),
+                                    yiYanProperties.getRefreshTokenSecondTime()
+                            ),
+                            new YiYanOptions().setMax_output_tokens(2048)),
+                    beanName
+            );
+        }
+        if (wrapperBean != null) {
+            beanFactory.registerSingleton(beanName, wrapperBean);
+        }
+    }
+
+
+    private static YudaoAiProperties.ChatProperties getChatProperties(AiPlatformEnum aiPlatformEnum, Map<String, Object> aiPlatformMap) {
+        if (AiPlatformEnum.XING_HUO == aiPlatformEnum) {
+            YudaoAiProperties.XingHuoProperties xingHuoProperties = new YudaoAiProperties.XingHuoProperties();
+            BeanUtil.fillBeanWithMap(aiPlatformMap, xingHuoProperties, true);
+            return xingHuoProperties;
+        } else if (AiPlatformEnum.YI_YAN == aiPlatformEnum) {
+            YudaoAiProperties.YiYanProperties yiYanProperties = new YudaoAiProperties.YiYanProperties();
+            BeanUtil.fillBeanWithMap(aiPlatformMap, yiYanProperties, true);
+            return yiYanProperties;
+        } else if (AiPlatformEnum.QIAN_WEN == aiPlatformEnum) {
+            YudaoAiProperties.QianWenProperties qianWenProperties = new YudaoAiProperties.QianWenProperties();
+            BeanUtil.fillBeanWithMap(aiPlatformMap, qianWenProperties, true);
+            return qianWenProperties;
+        }
+        throw new AiException("不支持的Ai类型!");
+    }
+}

yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/config/YudaoAiProperties.java

@@ -1,11 +1,15 @@
 package cn.iocoder.yudao.framework.ai.config;
 
+import cn.iocoder.yudao.framework.ai.AiPlatformEnum;
 import cn.iocoder.yudao.framework.ai.chatxinghuo.XingHuoChatModel;
 import cn.iocoder.yudao.framework.ai.chatyiyan.YiYanChatModel;
 import lombok.Data;
 import lombok.experimental.Accessors;
 import org.springframework.boot.context.properties.ConfigurationProperties;
 
+import java.util.LinkedHashMap;
+import java.util.Map;
+
 /**
  * ai 自动配置
  *
@@ -15,16 +19,18 @@ import org.springframework.boot.context.properties.ConfigurationProperties;
  */
 @Data
 @ConfigurationProperties(prefix = "yudao.ai")
-public class YudaoAiProperties {
+public class YudaoAiProperties extends LinkedHashMap<String, Map<String, Object>> {
 
-    private QianWenProperties qianWen;
-    private XingHuoProperties xingHuo;
-    private YiYanProperties yiYan;
+//    private QianWenProperties qianWen;
+//    private XingHuoProperties xingHuo;
+//    private YiYanProperties yiYan;
 
     @Data
     @Accessors(chain = true)
     public static class ChatProperties {
 
+        private AiPlatformEnum aiPlatform;
+
         private Float temperature;
 
         private Float topP;
@@ -48,9 +54,14 @@ public class YudaoAiProperties {
          */
         private String accessKeySecret;
         /**
-         * 阿里云：agentKey(相当于应用id)
+         * 阿里云：agentKey
          */
         private String agentKey;
+        /**
+         * 阿里云：agentKey(相当于应用id)
+         */
+        private String appId;
+
     }
 
     @Data

yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/exception/AiException.java

@@ -0,0 +1,15 @@
+package cn.iocoder.yudao.framework.ai.exception;
+
+/**
+ * ai 异常
+ *
+ * @author fansili
+ * @time 2024/4/13 17:05
+ * @since 1.0
+ */
+public class AiException extends RuntimeException {
+
+    public AiException(String message) {
+        super(message);
+    }
+}