From 8034305e0ef98cfc00e7448f566254677e3ee004 Mon Sep 17 00:00:00 2001 From: dvora-h Date: Thu, 31 Aug 2023 13:51:41 +0300 Subject: [PATCH] Fix protocol checking for search commands --- redis/commands/search/commands.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/redis/commands/search/commands.py b/redis/commands/search/commands.py index 87b572195c..83dea106d2 100644 --- a/redis/commands/search/commands.py +++ b/redis/commands/search/commands.py @@ -5,7 +5,7 @@ from redis.client import Pipeline from redis.utils import deprecated_function -from ..helpers import parse_to_dict +from ..helpers import get_protocol_version, parse_to_dict from ._util import to_string from .aggregation import AggregateRequest, AggregateResult, Cursor from .document import Document @@ -64,7 +64,7 @@ class SearchCommands: """Search commands.""" def _parse_results(self, cmd, res, **kwargs): - if self.client.connection_pool.connection_kwargs.get("protocol") in ["3", 3]: + if get_protocol_version(self.client) in ["3", 3]: return res else: return self._RESP2_MODULE_CALLBACKS[cmd](res, **kwargs)