change test to fit performance metrics

NREL · Dec 10, 2024 · 6d37186 · 6d37186
1 parent 96be266
commit 6d37186
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)