add kwargs in fetch method

singnet · Mar 27, 2024 · fbf303c · fbf303c
1 parent ec77412
commit fbf303c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git 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)