Commit 4a6b4e5f65d36d6af15ace9b022a0efbd1652fe2

Authored by cherishsince
1 parent 4db96fa3

【优化】优化 chat 初始化

yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/config/YudaoAiAutoConfiguration.java
... ... @@ -12,15 +12,10 @@ import cn.iocoder.yudao.framework.ai.chatyiyan.YiYanChatClient;
12 12 import cn.iocoder.yudao.framework.ai.chatyiyan.YiYanOptions;
13 13 import cn.iocoder.yudao.framework.ai.chatyiyan.api.YiYanApi;
14 14 import cn.iocoder.yudao.framework.ai.exception.AiException;
15   -import org.springframework.beans.BeansException;
16   -import org.springframework.beans.factory.InitializingBean;
17 15 import org.springframework.boot.autoconfigure.AutoConfiguration;
18 16 import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
19 17 import org.springframework.boot.context.properties.EnableConfigurationProperties;
20   -import org.springframework.context.ApplicationContext;
21   -import org.springframework.context.ApplicationContextAware;
22 18 import org.springframework.context.annotation.Bean;
23   -import org.springframework.context.support.GenericApplicationContext;
24 19  
25 20 import java.util.HashMap;
26 21 import java.util.Map;
... ... @@ -65,29 +60,6 @@ public class YudaoAiAutoConfiguration {
65 60 return chatMap;
66 61 }
67 62  
68   - public static class InitChatClient implements InitializingBean, ApplicationContextAware {
69   -
70   - private GenericApplicationContext applicationContext;
71   - private YudaoAiProperties yudaoAiProperties;
72   -
73   - public InitChatClient(YudaoAiProperties yudaoAiProperties) {
74   - this.yudaoAiProperties = yudaoAiProperties;
75   - }
76   -
77   - @Override
78   - public void afterPropertiesSet() {
79   -
80   -
81   - System.err.println(applicationContext.getBean("qianWen"));
82   - System.err.println(applicationContext.getBean("yiYan"));
83   - }
84   -
85   - @Override
86   - public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
87   - this.applicationContext = (GenericApplicationContext) applicationContext;
88   - }
89   - }
90   -
91 63 private static Object createChatClient(YudaoAiProperties.ChatProperties chatProperties) {
92 64 if (AiPlatformEnum.XING_HUO == chatProperties.getAiPlatform()) {
93 65 YudaoAiProperties.XingHuoProperties xingHuoProperties = (YudaoAiProperties.XingHuoProperties) chatProperties;
... ...