plancraft 0.4.7__py3-none-any.whl → 0.4.9__py3-none-any.whl

plancraft/environment/actions.py

@@ -216,6 +216,16 @@ class StopAction(BaseModel):
         return f"impossible: {self.reason}"
 
 
+class ClearAction(BaseModel):
+    """
+    Fake action that represents clearing the crafting grid by moving all items to inventory slots
+    Used for a more semantically aware planner step
+    """
+
+    def __str__(self):
+        return "clearing crafting grid"
+
+
 # when symbolic action is true, can either move objects around or smelt
 SymbolicAction = MoveAction | SmeltAction
 

plancraft/environment/planner.py

@@ -5,6 +5,7 @@ from collections import Counter
 import networkx as nx
 
 from plancraft.environment.actions import (
+    ClearAction,
     MoveAction,
     SmeltAction,
     StopAction,
@@ -500,7 +501,9 @@ def decompose_subgoal(
 
 
 def get_subplans(
-    observation: dict, return_items=False
+    observation: dict,
+    return_items=False,
+    clear_first=False,
 ) -> tuple[list[list[str]], list, list[str]] | tuple[list[list[str]], list]:
     current_inventory = copy.deepcopy(observation["inventory"])
     plan = get_plan(observation)
@@ -513,6 +516,31 @@ def get_subplans(
     subplans = []
     action_items_used = []
 
+    if clear_first:
+        # move all the current stuff in the crafting grid to free inventory slots
+        # While this makes a worse planner, this makes a clearer distinction between crafting steps
+        subplan = []
+        action_items = []
+        for slot in range(1, 10):  # crafting grid slots 1-9
+            if slot in current_inventory and current_inventory[slot]["quantity"] > 0:
+                free_slot = find_free_inventory_slot(current_inventory, from_slot=slot)
+                action = MoveAction(
+                    slot_from=slot,
+                    slot_to=free_slot,
+                    quantity=current_inventory[slot]["quantity"],
+                )
+                subplan.append(str(action))
+                action_items.append(current_inventory[slot]["type"])
+                current_inventory = update_inventory(
+                    current_inventory,
+                    slot,
+                    free_slot,
+                    current_inventory[slot]["quantity"],
+                )
+        if subplan:  # only add clear subplan if there were items to clear
+            subplans.append(subplan)
+            action_items_used.append([action_items])
+
     # Calculate the subplans for each step in the plan
     for plan_recipe, new_inventory in plan:
         subplan, current_inventory, action_items = decompose_subgoal(
@@ -520,6 +548,17 @@ def get_subplans(
         )
         subplans.append(subplan)
         action_items_used.append(action_items)
+
+    # If we added a clear action, we need to modify the plan to include it
+    if clear_first and len(subplans) > len(plan):
+        # Insert a fake clear action at the beginning of the plan
+        clear_action = ClearAction()
+        plan_inventory_counter = copy.deepcopy(plan[0][1])
+        plan = [(clear_action, plan_inventory_counter)] + plan
+
+    # If we are returning items, we need to return the action items used
+    # this is a list of the item (from_slot) used in each action
     if return_items:
         return subplans, plan, action_items_used
+
     return subplans, plan

{plancraft-0.4.7.dist-info → plancraft-0.4.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plancraft
-Version: 0.4.7
+Version: 0.4.9
 Summary: Plancraft: an evaluation dataset for planning with LLM agents
 License: MIT License
         

{plancraft-0.4.7.dist-info → plancraft-0.4.9.dist-info}/RECORD

@@ -15,10 +15,10 @@ plancraft/data/val.repeated.json,sha256=gLeCg94oB0B_e8QyH8-FBygcJctGA4bkXYldoOHt
 plancraft/data/val.small.easy.json,sha256=9zEmqepjXG2NIp88xnFqOCkwsUsku3HEwHoQGxgTr6U,190252
 plancraft/data/val.small.json,sha256=76E9EFaljDQyAokg97e-IblvcOe6KbrdKkXvRxhhkgo,237653
 plancraft/environment/__init__.py,sha256=XFsFny4lH195AwAmL-WeCaF9ZCMgc7IgXIwhQ8FTdgE,505
-plancraft/environment/actions.py,sha256=Pub21caxM5iZ9IaX-ny1-xxr_peJIwwV_QAx3BVSry0,11551
+plancraft/environment/actions.py,sha256=bMHR8bmgE_DD10K4Bii_QRgTs-GPkcgw9NKwHDoDmyQ,11813
 plancraft/environment/env.py,sha256=B7VpIMcQKITyDmkHgBoR4KbmxmM1b4A6Y-1_b90EMXo,16428
 plancraft/environment/items.py,sha256=Z9rhSyVDEoHF1pxRvhyiT94tyQJaWHi3wUHVcamz82o,221
-plancraft/environment/planner.py,sha256=b8p4tH-EyC8v9Rc2wBQr9YJ8AS1dmG0LUBOsoba8Qd8,20695
+plancraft/environment/planner.py,sha256=TKOMMqdBtk0dSc8X-ZgMnFeQnG_gzp99i3PHP-WI_hE,22415
 plancraft/environment/prompts.py,sha256=NU9YHAz3id-IgaukQvEi5uLlpEstpE5_Hccvvq1At2Y,6950
 plancraft/environment/recipes.py,sha256=0vwzOU86eZmGN2EpZVSIvzxpx0AOBWNPxTtAOFBN2A0,19570
 plancraft/environment/sampler.py,sha256=BworSMWQ-TLbV9068tkNOdo4ZLP-UDox6Laeb4Weu9k,7723
@@ -1924,7 +1924,7 @@ plancraft/models/generators.py,sha256=7COMLjjx_HbTWJqINNLqqExQv7gLikfLTViacAdSt5
 plancraft/models/oracle.py,sha256=f-0KWlBuHy6wcxmDsxM3MQ_QwfBstzfbA26mlk1MgLA,1657
 plancraft/models/utils.py,sha256=xgkP5jqCeFfkKe3Xd4ZYfTqiEJ-dA-qgFAC-J35ub3E,4029
 plancraft/train/dataset.py,sha256=oFqEd4LG9oEQ-71teh0Wf7-jJbtybT2ZibfM2bBdBkM,5474
-plancraft-0.4.7.dist-info/METADATA,sha256=hSaf9GuTMHGsBRW8wE4svMMIpfmL7yUVCtEFQKgwJw0,12391
-plancraft-0.4.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-plancraft-0.4.7.dist-info/licenses/LICENSE,sha256=YGR8ehDB4t-T-lOQKMfKNR-2zsOU7E3E5NA8t25HKE0,1070
-plancraft-0.4.7.dist-info/RECORD,,
+plancraft-0.4.9.dist-info/METADATA,sha256=jS7Fsd8WyiqI9w7joE9onIe8QJ339Lvo7vAxi3ihV6s,12391
+plancraft-0.4.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+plancraft-0.4.9.dist-info/licenses/LICENSE,sha256=YGR8ehDB4t-T-lOQKMfKNR-2zsOU7E3E5NA8t25HKE0,1070
+plancraft-0.4.9.dist-info/RECORD,,