estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

feat: economical index support retrieval testing (#3563)

Browse Source
This commit is contained in:
Joel
2024-04-17 17:40:28 +08:00
committed by GitHub
parent be3b37114c
commit 40b48510f4
2 changed files with 22 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
web/app/components/datasets/hit-testing/textarea.tsx
View File

@@ -49,7 +49,14 @@ const TextAreaWithButton = ({
const onSubmit = async () => {
setLoading(true)
const [e, res] = await asyncRunSafe<HitTestingResponse>(
hitTesting({ datasetId, queryText: text, retrieval_model: retrievalConfig }) as Promise<HitTestingResponse>,
hitTesting({
datasetId,
queryText: text,
retrieval_model: {
...retrievalConfig,
search_method: isEconomy ? RETRIEVE_METHOD.keywordSearch : retrievalConfig.search_method,
},
}) as Promise<HitTestingResponse>,
)
if (!e) {
setHitResult(res)
@@ -102,7 +109,7 @@ const TextAreaWithButton = ({
<Tag color="red" className="!text-red-600">
{text?.length}
<span className="text-red-300 mx-0.5">/</span>
200
200
</Tag>
</div>
</Tooltip>
@@ -114,25 +121,20 @@ const TextAreaWithButton = ({
>
{text?.length}
<span className="text-gray-300 mx-0.5">/</span>
200
200
</Tag>
)}
<Tooltip
selector="hit-testing-submit"
disabled={indexingTechnique === 'high_quality'}
content={t('datasetHitTesting.input.indexWarning') as string}
>
<div>
<Button
onClick={onSubmit}
type="primary"
loading={loading}
disabled={indexingTechnique !== 'high_quality' ? true : (!text?.length || text?.length > 200)}
>
{t('datasetHitTesting.input.testing')}
</Button>
</div>
</Tooltip>
<div>
<Button
onClick={onSubmit}
type="primary"
loading={loading}
disabled={(!text?.length || text?.length > 200)}
>
{t('datasetHitTesting.input.testing')}
</Button>
</div>
</div>
</div>