(feature)(webapp) change agent tool enum value (#454)

This commit is contained in:
williamhliu
2023-11-30 21:14:07 +08:00
committed by GitHub
parent 507c02a8fd
commit 39a85dc4ed
7 changed files with 48 additions and 68 deletions

View File

@@ -113,7 +113,8 @@ const ToolModal: React.FC<Props> = ({ editTool, onSaveTool, onCancel }) => {
<FormItem name="name" label="名称">
<Input placeholder="请输入工具名称" />
</FormItem>
{(toolType === AgentToolTypeEnum.RULE || toolType === AgentToolTypeEnum.LLM_S2SQL) && (
{(toolType === AgentToolTypeEnum.NL2SQL_RULE ||
toolType === AgentToolTypeEnum.NL2SQL_LLM) && (
<FormItem name="modelIds" label="主题域">
<Select
options={modelList.map((model) => ({ label: model.name, value: model.id }))}
@@ -122,7 +123,7 @@ const ToolModal: React.FC<Props> = ({ editTool, onSaveTool, onCancel }) => {
/>
</FormItem>
)}
{toolType === AgentToolTypeEnum.LLM_S2SQL && (
{toolType === AgentToolTypeEnum.NL2SQL_LLM && (
<FormItem name="exampleQuestions" label="示例问题">
<div className={styles.paramsSection}>
{examples.map((example) => {
@@ -158,7 +159,7 @@ const ToolModal: React.FC<Props> = ({ editTool, onSaveTool, onCancel }) => {
</div>
</FormItem>
)}
{toolType === AgentToolTypeEnum.INTERPRET && (
{toolType === AgentToolTypeEnum.ANALYTICS && (
<>
<FormItem name="modelId" label="主题域">
<Select
@@ -240,7 +241,7 @@ const ToolModal: React.FC<Props> = ({ editTool, onSaveTool, onCancel }) => {
/>
</FormItem>
)}
{toolType === AgentToolTypeEnum.RULE && (
{toolType === AgentToolTypeEnum.NL2SQL_RULE && (
<FormItem name="queryTypes" label="查询模式">
<Select
placeholder="请选择查询模式"

View File

@@ -2,48 +2,44 @@ export type MetricOptionType = {
id: string;
metricId?: number;
modelId?: number;
}
};
export enum AgentToolTypeEnum {
RULE = 'RULE',
LLM_S2SQL = 'LLM_S2SQL',
NL2SQL_RULE = 'NL2SQL_RULE',
NL2SQL_LLM = 'NL2SQL_LLM',
PLUGIN = 'PLUGIN',
INTERPRET = 'INTERPRET'
ANALYTICS = 'ANALYTICS',
}
export const AGENT_TOOL_TYPE_LIST = [
{
label: '规则语义解析',
value: AgentToolTypeEnum.RULE
value: AgentToolTypeEnum.NL2SQL_RULE,
},
{
label: '大模型语义解析',
value: AgentToolTypeEnum.LLM_S2SQL
},
{
label: '大模型指标解读',
value: AgentToolTypeEnum.INTERPRET
value: AgentToolTypeEnum.NL2SQL_LLM,
},
{
label: '第三方插件',
value: AgentToolTypeEnum.PLUGIN
value: AgentToolTypeEnum.PLUGIN,
},
]
];
export enum QueryModeEnum {
METRIC = 'METRIC',
TAG = 'TAG'
TAG = 'TAG',
}
export const QUERY_MODE_LIST = [
{
label: '指标模式',
value: QueryModeEnum.METRIC
value: QueryModeEnum.METRIC,
},
{
label: '标签模式',
value: QueryModeEnum.TAG
}
value: QueryModeEnum.TAG,
},
];
export type AgentToolType = {
@@ -55,11 +51,11 @@ export type AgentToolType = {
metricOptions?: MetricOptionType[];
exampleQuestions?: string[];
modelIds?: number[];
}
};
export type AgentConfigType = {
tools: AgentToolType[];
}
};
export type AgentType = {
id?: number;
@@ -73,7 +69,7 @@ export type AgentType = {
status?: 0 | 1;
enableSearch?: 0 | 1;
agentConfig?: AgentConfigType;
}
};
export type ModelType = {
id: number | string;