[BUG] Episode search drops candidates before the cross-encoder can score them

import os
os.environ['GRAPHITI_TELEMETRY_ENABLED'] = 'false'

import asyncio
from datetime import datetime, timezone
from types import SimpleNamespace
from unittest.mock import patch

from graphiti_core.nodes import EpisodeType, EpisodicNode
from graphiti_core.search.search import episode_search
from graphiti_core.search.search_config import (
    EpisodeReranker, EpisodeSearchConfig, EpisodeSearchMethod,
)
from graphiti_core.search.search_filters import SearchFilters

episodes = [
    EpisodicNode(
        uuid=uuid, name=uuid, group_id='alice', content=content,
        source=EpisodeType.message, source_description='conversation',
        valid_at=datetime(2026, 1, 1, tzinfo=timezone.utc),
    )
    for uuid, content in [
        ('first', 'Alice discussed dinner plans.'),
        ('second', 'Alice prefers vegetarian dinners.'),
    ]
]

async def fulltext(*args):
    return episodes

class Reranker:
    async def rank(self, query, passages):
        scores = {episodes[0].content: 0.1, episodes[1].content: 0.9}
        return sorted(
            [(text, scores[text]) for text in passages],
            key=lambda item: item[1], reverse=True,
        )

async def main():
    with patch('graphiti_core.search.search.episode_fulltext_search', fulltext):
        result, scores = await episode_search(
            driver=SimpleNamespace(), cross_encoder=Reranker(),
            query='What dinner should I recommend to Alice?',
            _query_vector=[], group_ids=['alice'], limit=1,
            config=EpisodeSearchConfig(
                search_methods=[EpisodeSearchMethod.bm25],
                reranker=EpisodeReranker.cross_encoder,
            ),
            search_filter=SearchFilters(),
        )
    print([episode.uuid for episode in result], scores)

asyncio.run(main())