Преглед на файлове

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

Jyong преди 5 месеца
родител
ревизия
dad041c49f
променени са 1 файла, в които са добавени 2 реда и са изтрити 0 реда
  1. 2 0
      api/core/rag/datasource/retrieval_service.py

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

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