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 ...@@ -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,
......
...@@ -15,6 +15,8 @@ ...@@ -15,6 +15,8 @@
# production # production
/build /build
/.history
# misc # misc
.DS_Store .DS_Store
*.pem *.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