From 117a209ad42200f83b7c5bafc3ee65fa585fa1b5 Mon Sep 17 00:00:00 2001
From: yezhwi <yzhw.2008@163.com>
Date: Wed, 23 Aug 2023 19:57:27 +0800
Subject: [PATCH] Fix:condition for dataset availability check (#973)

---
 api/services/hit_testing_service.py | 2 +-
 web/.gitignore                      | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/api/services/hit_testing_service.py b/api/services/hit_testing_service.py
index 4c2a8bf90..b6e622a90 100644
--- a/api/services/hit_testing_service.py
+++ b/api/services/hit_testing_service.py
@@ -19,7 +19,7 @@ from models.dataset import Dataset, DocumentSegment, DatasetQuery
 class HitTestingService:
     @classmethod
     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 {
                 "query": {
                     "content": query,
diff --git a/web/.gitignore b/web/.gitignore
index 35d1a1461..694294187 100644
--- a/web/.gitignore
+++ b/web/.gitignore
@@ -15,6 +15,8 @@
 # production
 /build
 
+/.history
+
 # misc
 .DS_Store
 *.pem