From 152af1456c3956baff57bd5034f2fef152060488 Mon Sep 17 00:00:00 2001 From: star-nox Date: Tue, 12 Dec 2023 00:02:10 -0600 Subject: [PATCH] added timelog after openai completion --- ai_ta_backend/filtering_contexts.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ai_ta_backend/filtering_contexts.py b/ai_ta_backend/filtering_contexts.py index 5310c729..7c5985f2 100644 --- a/ai_ta_backend/filtering_contexts.py +++ b/ai_ta_backend/filtering_contexts.py @@ -44,6 +44,7 @@ def run_context_filtering(contexts, user_query, max_time_before_return=45, max_c with ProcessPoolExecutor(max_workers=100) as executor: print("max workers: ", executor._max_workers) anyscale_responses = list(executor.map(partial_func1, contexts)) + print("len of anyscale responses: ", len(anyscale_responses)) if len(anyscale_responses) > 0: executor.map(partial_func2, anyscale_responses) else: