diff --git a/service-ai/service-ai-web/src/main/java/com/lanyuanxiaoyao/service/ai/web/WebApplication.java b/service-ai/service-ai-web/src/main/java/com/lanyuanxiaoyao/service/ai/web/WebApplication.java index 90348c8..b3298c4 100644 --- a/service-ai/service-ai-web/src/main/java/com/lanyuanxiaoyao/service/ai/web/WebApplication.java +++ b/service-ai/service-ai-web/src/main/java/com/lanyuanxiaoyao/service/ai/web/WebApplication.java @@ -34,6 +34,10 @@ public class WebApplication implements ApplicationRunner, ApplicationContextAwar return context.getBean(clazz); } + public static T getBean(String name, Class clazz) { + return context.getBean(name, clazz); + } + @Override public void run(ApplicationArguments args) { } diff --git a/service-ai/service-ai-web/src/main/java/com/lanyuanxiaoyao/service/ai/web/tools/ChartTool.java b/service-ai/service-ai-web/src/main/java/com/lanyuanxiaoyao/service/ai/web/tools/ChartTool.java index ae5bad8..3393066 100644 --- a/service-ai/service-ai-web/src/main/java/com/lanyuanxiaoyao/service/ai/web/tools/ChartTool.java +++ b/service-ai/service-ai-web/src/main/java/com/lanyuanxiaoyao/service/ai/web/tools/ChartTool.java @@ -44,7 +44,7 @@ public class ChartTool { """) String request ) { log.info("Enter method: mermaid[request]. request:{}", request); - ChatClient.Builder builder = WebApplication.getBean(ChatClient.Builder.class); + ChatClient.Builder builder = WebApplication.getBean("chat", ChatClient.Builder.class); ChatClient client = builder.build(); return client.prompt() // language=TEXT