diff --git a/elm/wizard.py b/elm/wizard.py index 099d04b..b41efaa 100644 --- a/elm/wizard.py +++ b/elm/wizard.py @@ -62,7 +62,7 @@ def query_vector_db(self, query, limit=100): ranked strings/scores outputs. """ - def engineer_query(self, query, + async def engineer_query(self, query, token_budget=None, new_info_threshold=0.7, convo=False): @@ -104,7 +104,7 @@ def engineer_query(self, query, token_budget = token_budget or self.token_budget start_time = perf_counter() - strings, _, idx = self.query_vector_db(query) + strings, _, idx = await self.query_vector_db(query) end_time = perf_counter() vector_query_time = end_time - start_time message = copy.deepcopy(self.MODEL_INSTRUCTION)