Procházet zdrojové kódy

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

tags/0.11.0
Jyong před 1 rokem
rodič
revize
dad041c49f
Žádný účet není propojen s e-mailovou adresou tvůrce revize
1 změnil soubory, kde provedl 2 přidání a 0 odebrání
  1. 2
    0
      api/core/rag/datasource/retrieval_service.py

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

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

Načítá se…
Zrušit
Uložit