Fix:condition for dataset availability check (#973)

This commit is contained in:
yezhwi 2023-08-23 19:57:27 +08:00 • committed by GitHub
commit 117a209ad4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 1 deletions

View file

@ -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
web/.gitignore vendored
View file

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