mirror of
https://github.com/tencentmusic/supersonic.git
synced 2025-12-12 20:51:48 +00:00
(improvement)(chat) Provide agent tool type list (#683)
Co-authored-by: jolunoluo
This commit is contained in:
@@ -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;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user