sxid003 commited on
Commit
5c94786
·
verified ·
1 Parent(s): b631eae

Update src/utils/search_docs_utils.py

Browse files
Files changed (1) hide show
  1. src/utils/search_docs_utils.py +1 -1
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() == lang]
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
  """