소스 검색

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 []