Create a vector of chunk indices to restrict retrieval to.
(
self, filter_languages: list[str] | None = None, filter_paths: list[str] | None = None
)
| 242 | return results |
| 243 | |
| 244 | def _get_selector_vector( |
| 245 | self, filter_languages: list[str] | None = None, filter_paths: list[str] | None = None |
| 246 | ) -> npt.NDArray[np.int_] | None: |
| 247 | """Create a vector of chunk indices to restrict retrieval to.""" |
| 248 | selector = [] |
| 249 | for language in filter_languages or []: |
| 250 | selector.extend(self._language_mapping.get(language, [])) |
| 251 | for filename in filter_paths or []: |
| 252 | selector.extend(self._file_mapping.get(filename, [])) |
| 253 | |
| 254 | return np.unique(selector) if selector else None |
| 255 | |
| 256 | def search( |
| 257 | self, |
no test coverage detected