Selaa lähdekoodia

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

tags/0.11.0
Jyong 1 vuosi sitten
vanhempi
commit
dad041c49f
No account linked to committer's email address
1 muutettua tiedostoa jossa 2 lisäystä ja 0 poistoa
  1. 2
    0
      api/core/rag/datasource/retrieval_service.py

+ 2
- 0
api/core/rag/datasource/retrieval_service.py Näytä tiedosto

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

Loading…
Peruuta
Tallenna