Unverified Commit 117a209a authored by yezhwi's avatar yezhwi Committed by GitHub

Fix:condition for dataset availability check (#973)

parent 071e7800
......@@ -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,
......
......@@ -15,6 +15,8 @@
# production
/build
/.history
# misc
.DS_Store
*.pem
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment