Prechádzať zdrojové kódy

fix issue: query is none when doing retrieval (#10129)

Jyong 5 mesiacov pred
rodič
commit
dad041c49f

+ 2 - 0
api/core/rag/datasource/retrieval_service.py

@@ -34,6 +34,8 @@ class RetrievalService:
         reranking_mode: Optional[str] = "reranking_model",
         reranking_mode: Optional[str] = "reranking_model",
         weights: Optional[dict] = None,
         weights: Optional[dict] = None,
     ):
     ):
+        if not query:
+            return []
         dataset = db.session.query(Dataset).filter(Dataset.id == dataset_id).first()
         dataset = db.session.query(Dataset).filter(Dataset.id == dataset_id).first()
         if not dataset:
         if not dataset:
             return []
             return []