Skip to content

Commit de68cff

Browse files
lpi-tnCopilot
andauthored
Update welearn_datastack/modules/qdrant_handler.py
Co-authored-by: Copilot <[email protected]>
1 parent 71a66c2 commit de68cff

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

welearn_datastack/modules/qdrant_handler.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -35,22 +35,23 @@ def classify_documents_per_collection(
3535
lang = dslice.document.lang
3636
model = dslice.embedding_model.title
3737
collection_name = None
38+
multilingual_collection = f"collection_welearn_mul_{model}"
39+
mono_collection = f"collection_welearn_{lang}_{model}"
3840
# Check multilingual
3941
for cn in collections_names_in_qdrant:
40-
multilingual_collection = f"collection_welearn_mul_{model}"
4142
if cn == multilingual_collection:
4243
collection_name = multilingual_collection
4344

4445
# Check monolingual
4546
for cn in collections_names_in_qdrant:
46-
mono_collection = f"collection_welearn_{lang}_{model}"
4747
if cn == mono_collection:
4848
collection_name = mono_collection
4949

5050
if not collection_name:
5151
logger.error(
52-
"Collection %s not found in Qdrant, slice %s ignored",
53-
collection_name,
52+
"Collections %s or %s not found in Qdrant, slice %s ignored",
53+
multilingual_collection,
54+
mono_collection,
5455
dslice.id,
5556
)
5657
continue

0 commit comments

Comments
 (0)