Spaces:
Sleeping
Sleeping
Update src/utils/search_docs_utils.py
Browse files
src/utils/search_docs_utils.py
CHANGED
|
@@ -104,7 +104,7 @@ def filter_by_language(metadatas: List[Dict], lang: str) -> List[int]:
|
|
| 104 |
Returns:
|
| 105 |
List[int]: Indices of filtered documents
|
| 106 |
"""
|
| 107 |
-
return [i for i, doc in enumerate(metadatas) if doc['Langue'].strip().lower()
|
| 108 |
|
| 109 |
def select_documents(query: str, embeddings: np.ndarray, metadatas: List[Dict], lang: str = None, top_k: int = 5) -> List[Dict]:
|
| 110 |
"""
|
|
|
|
| 104 |
Returns:
|
| 105 |
List[int]: Indices of filtered documents
|
| 106 |
"""
|
| 107 |
+
return [i for i, doc in enumerate(metadatas) if doc['Langue'].strip().lower() != 'unknown' ]
|
| 108 |
|
| 109 |
def select_documents(query: str, embeddings: np.ndarray, metadatas: List[Dict], lang: str = None, top_k: int = 5) -> List[Dict]:
|
| 110 |
"""
|