Skip to content

Commit

Permalink
added timelog after openai completion
Browse files Browse the repository at this point in the history
  • Loading branch information
star-nox committed Dec 12, 2023
1 parent bb70a0b commit 152af14
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions ai_ta_backend/filtering_contexts.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit 152af14

Please sign in to comment.