diff --git a/metagpt/actions/minecraft/control_primitives/__init__.py b/metagpt/actions/minecraft/control_primitives/__init__.py new file mode 100644 index 000000000..369e587d7 --- /dev/null +++ b/metagpt/actions/minecraft/control_primitives/__init__.py @@ -0,0 +1,18 @@ +import pkg_resources +import os +import voyager.utils as U + + +def load_control_primitives(primitive_names=None): + package_path = pkg_resources.resource_filename("voyager", "") + if primitive_names is None: + primitive_names = [ + primitives[:-3] + for primitives in os.listdir(f"{package_path}/control_primitives") + if primitives.endswith(".js") + ] + primitives = [ + U.load_text(f"{package_path}/control_primitives/{primitive_name}.js") + for primitive_name in primitive_names + ] + return primitives