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

chapter2/prompt-engineering/tau_bench/envs/__init__.py

@@ -0,0 +1,40 @@
+# Copyright Sierra
+
+from typing import Optional, Union
+from tau_bench.envs.base import Env
+from tau_bench.envs.user import UserStrategy
+
+
+def get_env(
+    env_name: str,
+    user_strategy: Union[str, UserStrategy],
+    user_model: str,
+    task_split: str,
+    user_provider: Optional[str] = None,
+    task_index: Optional[int] = None,
+    user_seed: Optional[int] = None,
+) -> Env:
+    if env_name == "retail":
+        from tau_bench.envs.retail import MockRetailDomainEnv
+
+        return MockRetailDomainEnv(
+            user_strategy=user_strategy,
+            user_model=user_model,
+            task_split=task_split,
+            user_provider=user_provider,
+            task_index=task_index,
+            user_seed=user_seed,
+        )
+    elif env_name == "airline":
+        from tau_bench.envs.airline import MockAirlineDomainEnv
+
+        return MockAirlineDomainEnv(
+            user_strategy=user_strategy,
+            user_model=user_model,
+            task_split=task_split,
+            user_provider=user_provider,
+            task_index=task_index,
+            user_seed=user_seed,
+        )
+    else:
+        raise ValueError(f"Unknown environment: {env_name}")