Spaces:
Sleeping
Sleeping
Update doc_searcher_v2.py
Browse files- 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 =
|
| 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:
|
| 25 |
-
top_k: Broj najboljih rezultata nakon rerankinga (default:
|
| 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()
|