ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter2/prompt-engineering/tau_bench/model_utils/func_tools/map.py

@@ -0,0 +1,20 @@
+from concurrent.futures import ThreadPoolExecutor
+from typing import Callable, Iterable, TypeVar
+
+T = TypeVar("T")
+U = TypeVar("U")
+
+
+def map(
+    func: Callable[[T], U],
+    iterable: Iterable[T],
+    max_concurrency: int | None = None,
+    use_tqdm: bool = False,
+) -> Iterable[U]:
+    assert max_concurrency is None or max_concurrency > 0
+    with ThreadPoolExecutor(max_workers=max_concurrency) as executor:
+        if use_tqdm:
+            from tqdm import tqdm
+
+            return list(tqdm(executor.map(func, iterable), total=len(iterable)))
+        return executor.map(func, iterable)