Parcourir la source

feat: economical index support retrieval testing (#3563)

Joel il y a 1 an
Parent
commit
40b48510f4
2 fichiers modifiés avec 22 ajouts et 19 suppressions
  1. 21 19
      web/app/components/datasets/hit-testing/textarea.tsx
  2. 1 0
      web/types/app.ts

+ 21 - 19
web/app/components/datasets/hit-testing/textarea.tsx

@@ -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>
 

+ 1 - 0
web/types/app.ts

@@ -33,6 +33,7 @@ export enum RETRIEVE_METHOD {
   fullText = 'full_text_search',
   hybrid = 'hybrid_search',
   invertedIndex = 'invertedIndex',
+  keywordSearch = 'keyword_search',
 }
 
 export type VariableInput = {