@@ -482,7 +482,7 @@ async def test_hybrid_search_weighted_sum_vector_bias(
482482 result_ids = [doc .metadata ["doc_id_key" ] for doc in results ]
483483
484484 assert len (result_ids ) > 0
485- assert result_ids [0 ] == "hs_doc_orange_fruit "
485+ assert result_ids [0 ] == "hs_doc_generic_tech "
486486
487487 async def test_hybrid_search_weighted_sum_fts_bias (
488488 self , vs_hybrid_search_with_tsv_column
@@ -611,7 +611,7 @@ async def test_hybrid_search_fts_empty_results(
611611 assert len (result_ids ) > 0
612612 assert "hs_doc_apple_fruit" in result_ids or "hs_doc_apple_tech" in result_ids
613613 # The top result should be one of the apple documents based on vector search
614- assert results [0 ].metadata ["doc_id_key" ].startswith ("hs_doc_unrelated_cat " )
614+ assert results [0 ].metadata ["doc_id_key" ].startswith ("hs_doc_apple_fruit " )
615615
616616 async def test_hybrid_search_vector_empty_results_effectively (
617617 self , vs_hybrid_search_with_tsv_column
@@ -638,7 +638,7 @@ async def test_hybrid_search_vector_empty_results_effectively(
638638
639639 # Expect results based purely on FTS search for "orange fruit"
640640 assert len (result_ids ) == 1
641- assert result_ids [0 ] == "hs_doc_generic_tech "
641+ assert result_ids [0 ] == "hs_doc_orange_fruit "
642642
643643 async def test_hybrid_search_without_tsv_column (self , engine ):
644644 """Test hybrid search without a TSV column."""
0 commit comments