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,15 +12,10 @@ import cn.iocoder.yudao.framework.ai.chatyiyan.YiYanChatClient;
12 import cn.iocoder.yudao.framework.ai.chatyiyan.YiYanOptions; 12 import cn.iocoder.yudao.framework.ai.chatyiyan.YiYanOptions;
13 import cn.iocoder.yudao.framework.ai.chatyiyan.api.YiYanApi; 13 import cn.iocoder.yudao.framework.ai.chatyiyan.api.YiYanApi;
14 import cn.iocoder.yudao.framework.ai.exception.AiException; 14 import cn.iocoder.yudao.framework.ai.exception.AiException;
15 -import org.springframework.beans.BeansException;  
16 -import org.springframework.beans.factory.InitializingBean;  
17 import org.springframework.boot.autoconfigure.AutoConfiguration; 15 import org.springframework.boot.autoconfigure.AutoConfiguration;
18 import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean; 16 import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
19 import org.springframework.boot.context.properties.EnableConfigurationProperties; 17 import org.springframework.boot.context.properties.EnableConfigurationProperties;
20 -import org.springframework.context.ApplicationContext;  
21 -import org.springframework.context.ApplicationContextAware;  
22 import org.springframework.context.annotation.Bean; 18 import org.springframework.context.annotation.Bean;
23 -import org.springframework.context.support.GenericApplicationContext;  
24 19
25 import java.util.HashMap; 20 import java.util.HashMap;
26 import java.util.Map; 21 import java.util.Map;
@@ -65,29 +60,6 @@ public class YudaoAiAutoConfiguration { @@ -65,29 +60,6 @@ public class YudaoAiAutoConfiguration {
65 return chatMap; 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 private static Object createChatClient(YudaoAiProperties.ChatProperties chatProperties) { 63 private static Object createChatClient(YudaoAiProperties.ChatProperties chatProperties) {
92 if (AiPlatformEnum.XING_HUO == chatProperties.getAiPlatform()) { 64 if (AiPlatformEnum.XING_HUO == chatProperties.getAiPlatform()) {
93 YudaoAiProperties.XingHuoProperties xingHuoProperties = (YudaoAiProperties.XingHuoProperties) chatProperties; 65 YudaoAiProperties.XingHuoProperties xingHuoProperties = (YudaoAiProperties.XingHuoProperties) chatProperties;