balibabu
commited on
Commit
·
70b1cb8
1
Parent(s):
622e082
feat: add LLMSelect (#1270)
Browse files### What problem does this PR solve?
feat: add LLMSelect #918
### Type of change
- [x] New Feature (non-breaking change which adds functionality)
web/src/components/llm-select/index.tsx
ADDED
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import { Popover, Select } from 'antd';
|
2 |
+
import LlmSettingItems from '../llm-setting-items';
|
3 |
+
|
4 |
+
const LLMSelect = () => {
|
5 |
+
const content = (
|
6 |
+
<div>
|
7 |
+
<LlmSettingItems handleParametersChange={() => {}}></LlmSettingItems>
|
8 |
+
</div>
|
9 |
+
);
|
10 |
+
|
11 |
+
return (
|
12 |
+
<Popover content={content} trigger="click" placement="left" arrow={false}>
|
13 |
+
{/* <Button>Click me</Button> */}
|
14 |
+
<Select
|
15 |
+
defaultValue="lucy"
|
16 |
+
style={{ width: '100%' }}
|
17 |
+
dropdownStyle={{ display: 'none' }}
|
18 |
+
/>
|
19 |
+
</Popover>
|
20 |
+
);
|
21 |
+
};
|
22 |
+
|
23 |
+
export default LLMSelect;
|
web/src/pages/flow/categorize-form/index.tsx
ADDED
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import LLMSelect from '@/components/llm-select';
|
2 |
+
|
3 |
+
const CategorizeForm = () => {
|
4 |
+
return (
|
5 |
+
<section>
|
6 |
+
<LLMSelect></LLMSelect>
|
7 |
+
</section>
|
8 |
+
);
|
9 |
+
};
|
10 |
+
|
11 |
+
export default CategorizeForm;
|
web/src/pages/flow/constant.tsx
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
import {
|
|
|
2 |
MergeCellsOutlined,
|
3 |
RocketOutlined,
|
4 |
SendOutlined,
|
@@ -10,6 +11,7 @@ export enum Operator {
|
|
10 |
Retrieval = 'Retrieval',
|
11 |
Generate = 'Generate',
|
12 |
Answer = 'Answer',
|
|
|
13 |
}
|
14 |
|
15 |
export const operatorIconMap = {
|
@@ -17,6 +19,7 @@ export const operatorIconMap = {
|
|
17 |
[Operator.Generate]: MergeCellsOutlined,
|
18 |
[Operator.Answer]: SendOutlined,
|
19 |
[Operator.Begin]: SlidersOutlined,
|
|
|
20 |
};
|
21 |
|
22 |
export const operatorMap = {
|
@@ -26,6 +29,7 @@ export const operatorMap = {
|
|
26 |
[Operator.Generate]: { description: 'Generate description' },
|
27 |
[Operator.Answer]: { description: 'Answer description' },
|
28 |
[Operator.Begin]: { description: 'Begin description' },
|
|
|
29 |
};
|
30 |
|
31 |
export const componentMenuList = [
|
@@ -41,6 +45,10 @@ export const componentMenuList = [
|
|
41 |
name: Operator.Answer,
|
42 |
description: operatorMap[Operator.Answer].description,
|
43 |
},
|
|
|
|
|
|
|
|
|
44 |
];
|
45 |
|
46 |
export const initialRetrievalValues = {
|
|
|
1 |
import {
|
2 |
+
DatabaseOutlined,
|
3 |
MergeCellsOutlined,
|
4 |
RocketOutlined,
|
5 |
SendOutlined,
|
|
|
11 |
Retrieval = 'Retrieval',
|
12 |
Generate = 'Generate',
|
13 |
Answer = 'Answer',
|
14 |
+
Categorize = 'Categorize',
|
15 |
}
|
16 |
|
17 |
export const operatorIconMap = {
|
|
|
19 |
[Operator.Generate]: MergeCellsOutlined,
|
20 |
[Operator.Answer]: SendOutlined,
|
21 |
[Operator.Begin]: SlidersOutlined,
|
22 |
+
[Operator.Categorize]: DatabaseOutlined,
|
23 |
};
|
24 |
|
25 |
export const operatorMap = {
|
|
|
29 |
[Operator.Generate]: { description: 'Generate description' },
|
30 |
[Operator.Answer]: { description: 'Answer description' },
|
31 |
[Operator.Begin]: { description: 'Begin description' },
|
32 |
+
[Operator.Categorize]: { description: 'Categorize description' },
|
33 |
};
|
34 |
|
35 |
export const componentMenuList = [
|
|
|
45 |
name: Operator.Answer,
|
46 |
description: operatorMap[Operator.Answer].description,
|
47 |
},
|
48 |
+
{
|
49 |
+
name: Operator.Categorize,
|
50 |
+
description: operatorMap[Operator.Categorize].description,
|
51 |
+
},
|
52 |
];
|
53 |
|
54 |
export const initialRetrievalValues = {
|
web/src/pages/flow/flow-drawer/index.tsx
CHANGED
@@ -4,6 +4,7 @@ import { useEffect } from 'react';
|
|
4 |
import { Node } from 'reactflow';
|
5 |
import AnswerForm from '../answer-form';
|
6 |
import BeginForm from '../begin-form';
|
|
|
7 |
import { Operator } from '../constant';
|
8 |
import GenerateForm from '../generate-form';
|
9 |
import { useHandleFormValuesChange } from '../hooks';
|
@@ -18,6 +19,7 @@ const FormMap = {
|
|
18 |
[Operator.Retrieval]: RetrievalForm,
|
19 |
[Operator.Generate]: GenerateForm,
|
20 |
[Operator.Answer]: AnswerForm,
|
|
|
21 |
};
|
22 |
|
23 |
const FlowDrawer = ({
|
|
|
4 |
import { Node } from 'reactflow';
|
5 |
import AnswerForm from '../answer-form';
|
6 |
import BeginForm from '../begin-form';
|
7 |
+
import CategorizeForm from '../categorize-form';
|
8 |
import { Operator } from '../constant';
|
9 |
import GenerateForm from '../generate-form';
|
10 |
import { useHandleFormValuesChange } from '../hooks';
|
|
|
19 |
[Operator.Retrieval]: RetrievalForm,
|
20 |
[Operator.Generate]: GenerateForm,
|
21 |
[Operator.Answer]: AnswerForm,
|
22 |
+
[Operator.Categorize]: CategorizeForm,
|
23 |
};
|
24 |
|
25 |
const FlowDrawer = ({
|