From fbf303c321a0f7bf1d19d5aaeb3c05233293258e Mon Sep 17 00:00:00 2001 From: marcocapozzoli Date: Wed, 27 Mar 2024 14:23:04 -0300 Subject: [PATCH] add kwargs in fetch method --- hyperon_das/query_engines.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/hyperon_das/query_engines.py b/hyperon_das/query_engines.py index 51816144..bcdae8db 100644 --- a/hyperon_das/query_engines.py +++ b/hyperon_das/query_engines.py @@ -533,8 +533,11 @@ def fetch( if not host and not port: host = self.query_engine.host port = self.query_engine.port - kwargs['running_on_server'] = True - documents = self.cache_manager.fetch_data( - query=query, host=host, port=port, server=self.remote_das + kwargs.update( + { + 'running_on_server': True, + 'server': self.remote_das, + } ) + documents = self.cache_manager.fetch_data(query=query, host=host, port=port, **kwargs) self.cache_manager.bulk_insert(documents)