Bläddra i källkod

expand rerank range (#2746)

### What problem does this PR solve?


### Type of change

- [x] Performance Improvement
tags/v0.13.0
Kevin Hu 1 år sedan
förälder
incheckning
04ff9cda7c
Inget konto är kopplat till bidragsgivarens mejladress
1 ändrade filer med 1 tillägg och 1 borttagningar
  1. 1
    1
      rag/nlp/search.py

+ 1
- 1
rag/nlp/search.py Visa fil

@@ -364,7 +364,7 @@ class Dealer:
if not question:
return ranks
RERANK_PAGE_LIMIT = 3
req = {"kb_ids": kb_ids, "doc_ids": doc_ids, "size": page_size*RERANK_PAGE_LIMIT,
req = {"kb_ids": kb_ids, "doc_ids": doc_ids, "size": max(page_size*RERANK_PAGE_LIMIT, 128),
"question": question, "vector": True, "topk": top,
"similarity": similarity_threshold,
"available_int": 1}

Laddar…
Avbryt
Spara