senatus123 commited on
Commit
5f40d31
·
verified ·
1 Parent(s): 87ab461

Update doc_searcher_v2.py

Browse files
Files changed (1) hide show
  1. doc_searcher_v2.py +3 -3
doc_searcher_v2.py CHANGED
@@ -15,14 +15,14 @@ class DocSearcherV2:
15
  # self.sparse_model = SparseTextEmbedding(SPARSE_MODEL) # Disabled - not needed without sparse search
16
  self.qdrant_client = QdrantClient(QDRANT_URL,api_key=QDRANT_API_KEY,timeout=30)
17
 
18
- async def search_semantic(self, text: str, qdrant_limit: int = 20, top_k: int = 3):
19
  """
20
  Semantička pretraga sa reranking-om
21
 
22
  Args:
23
  text: Query tekst
24
- qdrant_limit: Broj chunk-ova iz Qdrant-a (default: 20)
25
- top_k: Broj najboljih rezultata nakon rerankinga (default: 3)
26
  """
27
  queries = [text]
28
  dense_query = self.model.encode(text).tolist()
 
15
  # self.sparse_model = SparseTextEmbedding(SPARSE_MODEL) # Disabled - not needed without sparse search
16
  self.qdrant_client = QdrantClient(QDRANT_URL,api_key=QDRANT_API_KEY,timeout=30)
17
 
18
+ async def search_semantic(self, text: str, qdrant_limit: int = 50, top_k: int = 5):
19
  """
20
  Semantička pretraga sa reranking-om
21
 
22
  Args:
23
  text: Query tekst
24
+ qdrant_limit: Broj chunk-ova iz Qdrant-a (default: 50)
25
+ top_k: Broj najboljih rezultata nakon rerankinga (default: 5)
26
  """
27
  queries = [text]
28
  dense_query = self.model.encode(text).tolist()