Browse Source

Fix:condition for dataset availability check (#973)

yezhwi 1 year ago
parent
commit
117a209ad4
2 changed files with 3 additions and 1 deletions
  1. 1 1
      api/services/hit_testing_service.py
  2. 2 0
      web/.gitignore

+ 1 - 1
api/services/hit_testing_service.py

@@ -19,7 +19,7 @@ from models.dataset import Dataset, DocumentSegment, DatasetQuery
 class HitTestingService:
 class HitTestingService:
     @classmethod
     @classmethod
     def retrieve(cls, dataset: Dataset, query: str, account: Account, limit: int = 10) -> dict:
     def retrieve(cls, dataset: Dataset, query: str, account: Account, limit: int = 10) -> dict:
-        if dataset.available_document_count == 0 or dataset.available_document_count == 0:
+        if dataset.available_document_count == 0 or dataset.available_segment_count == 0:
             return {
             return {
                 "query": {
                 "query": {
                     "content": query,
                     "content": query,

+ 2 - 0
web/.gitignore

@@ -15,6 +15,8 @@
 # production
 # production
 /build
 /build
 
 
+/.history
+
 # misc
 # misc
 .DS_Store
 .DS_Store
 *.pem
 *.pem