重命名 tool-calls to tools.enabled
This commit is contained in:
@@ -1,15 +1,22 @@
|
|||||||
package com.jeesite.modules.cms.ai.properties;
|
package com.jeesite.modules.ai.cms.properties;
|
||||||
|
|
||||||
import org.springframework.boot.context.properties.ConfigurationProperties;
|
import org.springframework.boot.context.properties.ConfigurationProperties;
|
||||||
import org.springframework.boot.context.properties.NestedConfigurationProperty;
|
import org.springframework.boot.context.properties.NestedConfigurationProperty;
|
||||||
|
|
||||||
@ConfigurationProperties("spring.ai")
|
@ConfigurationProperties("spring.ai")
|
||||||
public class CmsAiProperties {
|
public class AiCmsProperties {
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 是否启用 Tool calling 工具调用
|
* 向量数据库设置
|
||||||
*/
|
*/
|
||||||
private Boolean toolCalls = false;
|
@NestedConfigurationProperty
|
||||||
|
private final Vectorstore vectorstore = new Vectorstore();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 是否启用 Tool calling 工具调用【例子详见 TestAiTools.java、UserAiTools.java 】
|
||||||
|
*/
|
||||||
|
@NestedConfigurationProperty
|
||||||
|
private final Tools tools = new Tools();
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 默认系统提示词
|
* 默认系统提示词
|
||||||
@@ -21,18 +28,12 @@ public class CmsAiProperties {
|
|||||||
*/
|
*/
|
||||||
private String defaultPromptTemplate = "";
|
private String defaultPromptTemplate = "";
|
||||||
|
|
||||||
/**
|
public Vectorstore getVectorstore() {
|
||||||
* 向量数据库设置
|
return vectorstore;
|
||||||
*/
|
|
||||||
@NestedConfigurationProperty
|
|
||||||
private final Vectorstore vectorstore = new Vectorstore();
|
|
||||||
|
|
||||||
public Boolean getToolCalls() {
|
|
||||||
return toolCalls;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public void setToolCalls(Boolean toolCalls) {
|
public Tools getTools() {
|
||||||
this.toolCalls = toolCalls;
|
return tools;
|
||||||
}
|
}
|
||||||
|
|
||||||
public String getDefaultSystem() {
|
public String getDefaultSystem() {
|
||||||
@@ -51,10 +52,6 @@ public class CmsAiProperties {
|
|||||||
this.defaultPromptTemplate = defaultPromptTemplate;
|
this.defaultPromptTemplate = defaultPromptTemplate;
|
||||||
}
|
}
|
||||||
|
|
||||||
public Vectorstore getVectorstore() {
|
|
||||||
return vectorstore;
|
|
||||||
}
|
|
||||||
|
|
||||||
public static class Vectorstore {
|
public static class Vectorstore {
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -70,4 +67,20 @@ public class CmsAiProperties {
|
|||||||
this.type = type;
|
this.type = type;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public static class Tools {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 是否启用 Tool calling 工具调用【例子详见 TestAiTools.java、UserAiTools.java 】
|
||||||
|
*/
|
||||||
|
private Boolean enabled = false;
|
||||||
|
|
||||||
|
public Boolean getEnabled() {
|
||||||
|
return enabled;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setEnabled(Boolean enabled) {
|
||||||
|
this.enabled = enabled;
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
@@ -6,7 +6,7 @@ package com.jeesite.test;
|
|||||||
|
|
||||||
import com.jeesite.common.mapper.JsonMapper;
|
import com.jeesite.common.mapper.JsonMapper;
|
||||||
import com.jeesite.common.tests.BaseSpringContextTests;
|
import com.jeesite.common.tests.BaseSpringContextTests;
|
||||||
import com.jeesite.modules.cms.ai.service.CmsAiChatService;
|
import com.jeesite.modules.ai.cms.service.AiCmsChatService;
|
||||||
import com.jeesite.modules.sys.entity.Area;
|
import com.jeesite.modules.sys.entity.Area;
|
||||||
import org.junit.FixMethodOrder;
|
import org.junit.FixMethodOrder;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
@@ -26,20 +26,20 @@ import java.util.Map;
|
|||||||
@ActiveProfiles("test")
|
@ActiveProfiles("test")
|
||||||
@SpringBootApplication
|
@SpringBootApplication
|
||||||
@FixMethodOrder(MethodSorters.NAME_ASCENDING)
|
@FixMethodOrder(MethodSorters.NAME_ASCENDING)
|
||||||
@SpringBootTest(properties = {"spring.ai.tool-calls=true"})
|
@SpringBootTest(properties = {"spring.ai.tools.enabled=true"})
|
||||||
public class AiChatServiceTest extends BaseSpringContextTests {
|
public class AiChatServiceTest extends BaseSpringContextTests {
|
||||||
|
|
||||||
private final CmsAiChatService cmsAiChatService;
|
private final AiCmsChatService aiCmsChatService;
|
||||||
|
|
||||||
public AiChatServiceTest(CmsAiChatService cmsAiChatService) {
|
public AiChatServiceTest(AiCmsChatService aiCmsChatService) {
|
||||||
this.cmsAiChatService = cmsAiChatService;
|
this.aiCmsChatService = aiCmsChatService;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void test01Text() {
|
public void test01Text() {
|
||||||
logger.info("===== 聊天对话,文本输出");
|
logger.info("===== 聊天对话,文本输出");
|
||||||
String message = "你好";
|
String message = "你好";
|
||||||
String text = cmsAiChatService.chatText(message);
|
String text = aiCmsChatService.chatText(message);
|
||||||
System.out.println(text);
|
System.out.println(text);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -47,7 +47,7 @@ public class AiChatServiceTest extends BaseSpringContextTests {
|
|||||||
public void test02Json() {
|
public void test02Json() {
|
||||||
logger.info("===== 聊天对话,结构化输出 JSON");
|
logger.info("===== 聊天对话,结构化输出 JSON");
|
||||||
String message = "张三";
|
String message = "张三";
|
||||||
Map<String, Object> map = cmsAiChatService.chatJson(message);
|
Map<String, Object> map = aiCmsChatService.chatJson(message);
|
||||||
System.out.println(JsonMapper.toJson(map));
|
System.out.println(JsonMapper.toJson(map));
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -55,10 +55,10 @@ public class AiChatServiceTest extends BaseSpringContextTests {
|
|||||||
public void test03Tool() {
|
public void test03Tool() {
|
||||||
logger.info("===== 聊天对话,结构化输出 Tool Calling");
|
logger.info("===== 聊天对话,结构化输出 Tool Calling");
|
||||||
String message = "打开客厅的灯";
|
String message = "打开客厅的灯";
|
||||||
Map<String, Object> map = cmsAiChatService.chatJson(message);
|
Map<String, Object> map = aiCmsChatService.chatJson(message);
|
||||||
System.out.println(JsonMapper.toJson(map));
|
System.out.println(JsonMapper.toJson(map));
|
||||||
message = "关闭客厅的灯";
|
message = "关闭客厅的灯";
|
||||||
map = cmsAiChatService.chatJson(message);
|
map = aiCmsChatService.chatJson(message);
|
||||||
System.out.println(JsonMapper.toJson(map));
|
System.out.println(JsonMapper.toJson(map));
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -66,7 +66,7 @@ public class AiChatServiceTest extends BaseSpringContextTests {
|
|||||||
public void test04Entity() {
|
public void test04Entity() {
|
||||||
logger.info("===== 聊天对话,结构化输出 Entity");
|
logger.info("===== 聊天对话,结构化输出 Entity");
|
||||||
String message = "北京";
|
String message = "北京";
|
||||||
List<Area> list = cmsAiChatService.chatArea(message);
|
List<Area> list = aiCmsChatService.chatArea(message);
|
||||||
System.out.println(JsonMapper.toJson(list));
|
System.out.println(JsonMapper.toJson(list));
|
||||||
}
|
}
|
||||||
|
|
||||||
Reference in New Issue
Block a user