From 655de86e69890875dfd96ecd7523616b4548569f Mon Sep 17 00:00:00 2001
From: huangge1199 <huangge1199@hotmail.com>
Date: Sat, 17 May 2025 12:55:09 +0800
Subject: [PATCH] =?UTF-8?q?=E8=A7=A3=E5=86=B3=E9=97=AE=E9=A2=98=EF=BC=9ACh?=
 =?UTF-8?q?atClientAutoConfiguration=E6=A3=80=E6=B5=8B=E5=88=B0=E4=B8=A4?=
 =?UTF-8?q?=E4=B8=AAChatModel=E7=B1=BB=E5=9E=8B=E7=9A=84bean=EF=BC=88dashs?=
 =?UTF-8?q?copeChatModel=E5=92=8CollamaChatModel=EF=BC=89=EF=BC=8C?=
 =?UTF-8?q?=E5=AF=BC=E8=87=B4=E6=97=A0=E6=B3=95=E8=87=AA=E5=8A=A8=E6=B3=A8?=
 =?UTF-8?q?=E5=85=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pom.xml                                       |  4 +++
 .../Service/impl/InvokeServiceImpl.java       |  3 ++
 .../aiagent/config/ChatModelConfig.java       | 31 +++++++++++++++++++
 3 files changed, 38 insertions(+)
 create mode 100644 src/main/java/com/huangge1199/aiagent/config/ChatModelConfig.java

diff --git a/pom.xml b/pom.xml
index 52a79b4..288ba11 100644
--- a/pom.xml
+++ b/pom.xml
@@ -132,6 +132,10 @@
                             <version>1.18.38</version>
                         </path>
                     </annotationProcessorPaths>
+                    <parameters>true</parameters>
+                    <compilerArgs>
+                        <arg>-parameters</arg>
+                    </compilerArgs>
                 </configuration>
             </plugin>
             <plugin>
diff --git a/src/main/java/com/huangge1199/aiagent/Service/impl/InvokeServiceImpl.java b/src/main/java/com/huangge1199/aiagent/Service/impl/InvokeServiceImpl.java
index 8929e36..b6cf7a1 100644
--- a/src/main/java/com/huangge1199/aiagent/Service/impl/InvokeServiceImpl.java
+++ b/src/main/java/com/huangge1199/aiagent/Service/impl/InvokeServiceImpl.java
@@ -18,6 +18,7 @@ import jakarta.annotation.Resource;
 import org.springframework.ai.chat.messages.AssistantMessage;
 import org.springframework.ai.chat.prompt.Prompt;
 import org.springframework.ai.ollama.OllamaChatModel;
+import org.springframework.beans.factory.annotation.Qualifier;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 
@@ -34,9 +35,11 @@ public class InvokeServiceImpl implements InvokeService {
     private String baiLianKey;
 
     @Resource
+    @Qualifier("dashscopeChatModel")
     private DashScopeChatModel dashscopeChatModel;
 
     @Resource
+    @Qualifier("ollamaChatModel")
     private OllamaChatModel ollamaChatModel;
 
     @Override
diff --git a/src/main/java/com/huangge1199/aiagent/config/ChatModelConfig.java b/src/main/java/com/huangge1199/aiagent/config/ChatModelConfig.java
new file mode 100644
index 0000000..1ac5d5a
--- /dev/null
+++ b/src/main/java/com/huangge1199/aiagent/config/ChatModelConfig.java
@@ -0,0 +1,31 @@
+package com.huangge1199.aiagent.config;
+
+import org.springframework.ai.chat.model.ChatModel;
+import org.springframework.ai.ollama.OllamaChatModel;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.Primary;
+
+/**
+ * ChatModelConfig
+ *
+ * @author huangge1199
+ * @since 2025/5/17 12:48:28
+ */
+@Configuration
+public class ChatModelConfig {
+
+    // 选择DashScope作为默认模型
+//    @Bean
+//    @Primary
+//    public ChatModel primaryChatModel(DashScopeChatModel dashscopeChatModel) {
+//        return dashscopeChatModel;
+//    }
+
+    // 或者选择Ollama：
+     @Bean
+     @Primary
+     public ChatModel primaryChatModel(OllamaChatModel ollamaChatModel) {
+         return ollamaChatModel;
+     }
+}