File size: 3,214 Bytes
8c4ec99 0a903c7 1a156e6 0a903c7 1a156e6 8c4ec99 0a903c7 1a156e6 0a903c7 1a156e6 0a903c7 1a156e6 8c4ec99 1a156e6 0a903c7 8c4ec99 1a156e6 0a903c7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 |
import SimilaritySlider from '@/components/similarity-slider';
import { DeleteOutlined, HistoryOutlined } from '@ant-design/icons';
import {
Button,
Card,
Divider,
Flex,
Form,
Input,
Slider,
Space,
Tag,
} from 'antd';
import { FormInstance } from 'antd/lib';
import styles from './index.less';
const list = [1, 2, 3];
type FieldType = {
similarity_threshold?: number;
vector_similarity_weight?: number;
top_k?: number;
question: string;
};
interface IProps {
form: FormInstance;
handleTesting: () => Promise<any>;
}
const TestingControl = ({ form, handleTesting }: IProps) => {
const question = Form.useWatch('question', { form, preserve: true });
const buttonDisabled =
!question || (typeof question === 'string' && question.trim() === '');
return (
<section className={styles.testingControlWrapper}>
<p>
<b>Retrieval testing</b>
</p>
<p>Final step! After success, leave the rest to Infiniflow AI.</p>
<Divider></Divider>
<section>
<Form
name="testing"
layout="vertical"
form={form}
initialValues={{
similarity_threshold: 0.2,
vector_similarity_weight: 0.3,
top_k: 1024,
}}
>
<SimilaritySlider></SimilaritySlider>
<Form.Item<FieldType> label="Top k" name={'top_k'}>
<Slider marks={{ 0: 0, 2048: 2048 }} defaultValue={0} max={2048} />
</Form.Item>
<Card size="small" title="Test text">
<Form.Item<FieldType>
name={'question'}
rules={[
{ required: true, message: 'Please input your question!' },
]}
>
<Input.TextArea autoSize={{ minRows: 8 }}></Input.TextArea>
</Form.Item>
<Flex justify={'space-between'}>
<Tag>10/200</Tag>
<Button
type="primary"
size="small"
onClick={handleTesting}
disabled={buttonDisabled}
>
Testing
</Button>
</Flex>
</Card>
</Form>
</section>
<section>
<p className={styles.historyTitle}>
<Space size={'middle'}>
<HistoryOutlined className={styles.historyIcon} />
<b>Test history</b>
</Space>
</p>
<Space
direction="vertical"
size={'middle'}
className={styles.historyCardWrapper}
>
{list.map((x) => (
<Card className={styles.historyCard} key={x}>
<Flex justify={'space-between'} gap={'small'}>
<span>{x}</span>
<div className={styles.historyText}>
content dcjsjl snldsh svnodvn svnodrfn svjdoghdtbnhdo
sdvhodhbuid sldghdrlh
</div>
<Flex gap={'small'}>
<span>time</span>
<DeleteOutlined></DeleteOutlined>
</Flex>
</Flex>
</Card>
))}
</Space>
</section>
</section>
);
};
export default TestingControl;
|