Function search_dataset_vectors

tests/test_multiple_datasets.py:220–222

(query_id, query_emb)

    print(colored("Vectors upserted successfully!", "green"))

    def search_dataset_vectors(query_id, query_emb):
        result = ann_vector(query_id, constants["COLLECTION_NAME"], query_emb)
        return result

    query_vecs = pl.read_csv(bf_csv_path)
    df = pl.read_parquet(parquet_path, low_memory=True)