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

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

@@ -0,0 +1,21 @@
+# Copyright Sierra
+
+import json
+import os
+from typing import Any
+
+FOLDER_PATH = os.path.dirname(__file__)
+
+
+def load_data() -> dict[str, Any]:
+    with open(os.path.join(FOLDER_PATH, "flights.json")) as f:
+        flight_data = json.load(f)
+    with open(os.path.join(FOLDER_PATH, "reservations.json")) as f:
+        reservation_data = json.load(f)
+    with open(os.path.join(FOLDER_PATH, "users.json")) as f:
+        user_data = json.load(f)
+    return {
+        "flights": flight_data,
+        "reservations": reservation_data,
+        "users": user_data,
+    }