Commit 4a6b4e5f65d36d6af15ace9b022a0efbd1652fe2
1 parent
4db96fa3
【优化】优化 chat 初始化
Showing
1 changed file
with
0 additions
and
28 deletions
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; |