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

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

@@ -0,0 +1,2 @@
+from tau_bench.model_utils.func_tools.filter import filter as filter
+from tau_bench.model_utils.func_tools.map import map as map

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

@@ -0,0 +1,17 @@
+from typing import Callable, Iterable, TypeVar
+
+from tau_bench.model_utils.func_tools.map import map
+
+T = TypeVar("T")
+
+builtin_filter = filter
+
+
+def filter(
+    func: Callable[[T], bool],
+    iterable: Iterable[T],
+    max_concurrency: int | None = None,
+) -> Iterable[T]:
+    assert max_concurrency is None or max_concurrency > 0
+    bits = map(func, iterable=iterable, max_concurrency=max_concurrency)
+    return [x for x, y in zip(iterable, bits) if y]

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)