Skip to content

Commit f5eac15

Browse files
committed
renamed search collection to hybrid in config
1 parent ac7a8c0 commit f5eac15

3 files changed

Lines changed: 8 additions & 8 deletions

File tree

bbconf/config_parser/bedbaseconfig.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -328,7 +328,7 @@ def _init_qdrant_hybrid(
328328
return None
329329

330330
dimensions = int(dense_encoder.get_embedding_size(self._config.path.text2vec))
331-
collection_name = self.config.qdrant.search_collection
331+
collection_name = self.config.qdrant.hybrid_collection
332332

333333
try:
334334
if not qdrant_cl.collection_exists(collection_name):

bbconf/config_parser/models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ class ConfigQdrant(BaseModel):
5656
api_key: Optional[str] = None
5757
file_collection: str = DEFAULT_QDRANT_FILE_COLLECTION_NAME
5858
text_collection: Optional[str] = DEFAULT_QDRANT_BIVEC_COLLECTION_NAME
59-
search_collection: Optional[str] = DEFAULT_QDRANT_HYBRID_COLLECTION_NAME
59+
hybrid_collection: Optional[str] = DEFAULT_QDRANT_HYBRID_COLLECTION_NAME
6060

6161

6262
class ConfigServer(BaseModel):

bbconf/modules/bedfiles.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2071,7 +2071,7 @@ def _update_sources(
20712071
# "Uploading points to qdrant using batch..."
20722072
# )
20732073
# operation_info = self.config._qdrant_advanced_engine.upsert(
2074-
# collection_name=self.config.config.qdrant.search_collection,
2074+
# collection_name=self.config.config.qdrant.hybrid_collection,
20752075
# points=points,
20762076
# )
20772077
# session.commit()
@@ -2083,7 +2083,7 @@ def _update_sources(
20832083
# pbar.update(1)
20842084
#
20852085
# operation_info = self.config._qdrant_advanced_engine.upsert(
2086-
# collection_name=self.config.config.qdrant.search_collection,
2086+
# collection_name=self.config.config.qdrant.hybrid_collection,
20872087
# points=points,
20882088
# )
20892089
# assert operation_info.status == "completed"
@@ -2186,7 +2186,7 @@ def reindex_hybrid_search(self, batch: int = 1000, purge: bool = False) -> None:
21862186
"Uploading points to qdrant using batch..."
21872187
)
21882188
operation_info = self.config.qdrant_client.upsert(
2189-
collection_name=self.config.config.qdrant.search_collection,
2189+
collection_name=self.config.config.qdrant.hybrid_collection,
21902190
points=points,
21912191
)
21922192
session.commit()
@@ -2199,7 +2199,7 @@ def reindex_hybrid_search(self, batch: int = 1000, purge: bool = False) -> None:
21992199

22002200
if points:
22012201
operation_info = self.config.qdrant_client.upsert(
2202-
collection_name=self.config.config.qdrant.search_collection,
2202+
collection_name=self.config.config.qdrant.hybrid_collection,
22032203
points=points,
22042204
)
22052205
assert operation_info.status == "completed"
@@ -2250,7 +2250,7 @@ def reindex_hybrid_search(self, batch: int = 1000, purge: bool = False) -> None:
22502250
# embeddings_list = list(self.config.dense_encoder.embed(query))[0]
22512251
#
22522252
# results: QueryResponse = self.config.qdrant_client.query_points(
2253-
# collection_name=self.config.config.qdrant.search_collection,
2253+
# collection_name=self.config.config.qdrant.hybrid_collection,
22542254
# query=list(embeddings_list),
22552255
# limit=limit,
22562256
# offset=offset,
@@ -2362,7 +2362,7 @@ def hybrid_search(
23622362
]
23632363

23642364
results = self.config.qdrant_client.query_points(
2365-
collection_name=self.config.config.qdrant.search_collection,
2365+
collection_name=self.config.config.qdrant.hybrid_collection,
23662366
limit=limit,
23672367
offset=offset,
23682368
prefetch=hybrid_query,

0 commit comments

Comments
 (0)