From c56732022bdb95b3129ac51f83ba0755c72103db Mon Sep 17 00:00:00 2001 From: ramashishmaurya Date: Fri, 25 Sep 2026 13:05:14 +0530 Subject: [PATCH] fix(client): resolve mypy type narrowing issue for cross encoder --- src/dynavec/client.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/dynavec/client.py b/src/dynavec/client.py index 67c159c..4bc592f 100644 --- a/src/dynavec/client.py +++ b/src/dynavec/client.py @@ -722,11 +722,13 @@ def _cross_encoder_rerank( "rerank", ) from exc - if self._cross_encoder is None: - self._cross_encoder = CrossEncoder(self.config.cross_encoder_model) + encoder = self._cross_encoder + if encoder is None: + encoder = CrossEncoder(self.config.cross_encoder_model) + self._cross_encoder = encoder pairs = [(query, result.text) for result in results] - scores = self._cross_encoder.predict(pairs) + scores = encoder.predict(pairs) reranked = sorted( zip(results, scores),