Move test func & add comment

This commit is contained in:
yuymf 2023-09-28 01:17:02 +08:00
parent 7f763b57d0
commit 3a705a0e89
25 changed files with 489 additions and 333 deletions

View file

@ -1,18 +1,20 @@
import pkg_resources
import os
import voyager.utils as U
import metagpt.utils.minecraft as utils
from metagpt.logs import logger
def load_control_primitives_context(primitive_names=None):
package_path = pkg_resources.resource_filename("metagpt", "")
if primitive_names is None:
primitive_names = [
primitive[:-3]
for primitive in os.listdir(f"{package_path}/actions/minecraft/control_primitives_context")
if primitive.endswith(".js")
def load_skills_code_context(skill_names=None):
skills_dir = os.path.dirname(os.path.abspath(__file__))
if skill_names is None:
skill_names = [
skill[:-3] for skill in os.listdir(f"{skills_dir}") if skill.endswith(".js")
]
primitives = [
U.load_text(f"{package_path}/actions/minecraft/control_primitives_context/{primitive_name}.js")
for primitive_name in primitive_names
skills = [
utils.load_text(os.path.join(skills_dir, f"{skill_name}.js"))
for skill_name in skill_names
]
return primitives
return skills
if __name__ == "__main__":
logger.info(load_skills_code_context(["craftItem", "exploreUntil"]))