(improvement)(chat) Provide agent tool type list (#683)

Co-authored-by: jolunoluo
This commit is contained in:
LXW
2024-01-23 11:45:15 +08:00
committed by GitHub
parent 163e782f51
commit 42a6f61456
2 changed files with 31 additions and 7 deletions

View File

@@ -1,8 +1,25 @@
package com.tencent.supersonic.chat.core.agent;
import java.util.HashMap;
import java.util.Map;
public enum AgentToolType {
NL2SQL_RULE,
NL2SQL_LLM,
PLUGIN,
ANALYTICS
NL2SQL_RULE("基于规则Text-to-SQL"),
NL2SQL_LLM("基于大模型Text-to-SQL"),
PLUGIN("第三方插件");
private String title;
AgentToolType(String title) {
this.title = title;
}
public static Map<AgentToolType, String> getToolTypes() {
Map<AgentToolType, String> map = new HashMap<>();
map.put(NL2SQL_RULE, NL2SQL_RULE.title);
map.put(NL2SQL_LLM, NL2SQL_LLM.title);
map.put(PLUGIN, PLUGIN.title);
return map;
}
}