Function _batch

tests/test_hybrid.py:830–866

(batch_indices, batch_index)

    qps_tracker = QPSTracker()

    def _batch(batch_indices, batch_index):
        batch_start_time = time.time()
        failed_count = 0

        try:
            batch_dense = [dense_query_vectors[i].tolist() for i in batch_indices]
            batch_text = [text_queries[i] for i in batch_indices]

            # Use batch_hybrid for better performance
            batch_results = collection.search.batch_hybrid(
                dense_queries=batch_dense,
                sparse_queries=batch_text,
                top_k=top_k,
                dense_weight=0.5,
                sparse_weight=0.5,
            )

            # Format results consistently
            formatted_results = []
            for result in batch_results:
                if isinstance(result, dict) and "results" in result:
                    formatted = [
                        {"id": r["id"], "score": r["score"]}
                        for r in result["results"]
                    ]
                else:
                    formatted = [{"id": r["id"], "score": r["score"]} for r in result]
                formatted_results.append(formatted)

        except Exception as e:
            print(f"Batch hybrid search failed: {e}")
            failed_count = len(batch_indices)
            formatted_results = [[]] * len(batch_indices)

        batch_duration = time.time() - batch_start_time
        qps_tracker.record_query_batch(len(batch_indices), batch_duration, failed_count)
        return formatted_results

    # Prepare batches
    batches = []