parallel_utils.py

import json
from concurrent.futures import ThreadPoolExecutor, as_completed


def assess_wrapper(mod_prompt, function_info, client):
    # Local import to avoid circular dependency
    from ast_utils import assess_function_relevance
    is_relevant = assess_function_relevance(mod_prompt, function_info, client)
    return function_info["name"], function_info.get("calls", []), is_relevant


def parallel_assess_functions(mod_prompt, functions, client):
    results = []
    with ThreadPoolExecutor() as executor:
        futures = [executor.submit(assess_wrapper, mod_prompt, func, client) for func in functions]
        for future in as_completed(futures):
            try:
                results.append(future.result())
            except Exception as e:
                print("[DEBUG] Exception in parallel assess:", e)
                exit()
    return results