INNER CODE UNIT · Python

search

yurijmikhalevich/rclip · rclip/main.py:300

  def search(
    self,
    query: str,
    directory: str,
    top_k: int | None = 10,
    positive_queries: List[str] = [],
    negative_queries: List[str] = [],
    *,
    cancel_event: threading.Event | None = None,
  ) -> List[SearchResult]:
    positive_queries = [query] + positive_queries
    helpers.raise_if_cancelled(cancel_event)
    positive_features = self._model.compute_features_for_queries(positive_queries)
    helpers.raise_if_cancelled(cancel_event)
    negative_features = self._model.compute_features_for_queries(negative_queries)
    helpers.raise_if_cancelled(cancel_event)
    query_features = positive_features - negative_features

View source record →