add few-shot examples

robohusky/conversation.py

@@ -252,13 +252,22 @@ register_conv_template(
             "Your task is to identify and list only the core, task-related robot actions clearly and accurately in English. "
             "Use simple imperative sentences, one action per line, with a numbered list. "
             "Only include meaningful, task-oriented actions where the robot directly interacts with objects or performs task steps. "
+            "Include the object involved and its precise relative location, mentioning colors or nearby landmarks. "
+            "Use placeholders like [color] and [object] only in examples, never in real output. "
+            "Never copy the examples verbatim."
             "Ignore supporting or movement-related actions like moving forward or turning. "
             "The video contains only a few actions. List no more than 3 actions, and do not repeat any action. "
             "Do not add any imagined or unrelated details. "
             "Keep the list concise, clear, and precise, suitable for automatic embodied intelligence data annotation."
         ),
-        roles=("Human", "Assistant"),
-        messages=(),
+        roles=["Human", "Assistant"],
+        messages=[
+            ["Human", "Analyze the video and list all core robot actions."],
+            ["Assistant",
+             "1. Pick up the [color] [object].\n"
+             "2. Place the [color] [object] next to the [color] [landmark]."
+             ],
+        ],
         offset=0,
         sep_style=SeparatorStyle.ADD_COLON_TWO,
         sep=" ",