RoboWaiter/robowaiter/scene/tasks/Open_tasks_test.py

49 lines
1.5 KiB
Python
Raw Normal View History

2023-11-15 14:30:57 +08:00
"""
人提出请求机器人完成任务
1. 做咖啡固定动画接收到做咖啡指令走到咖啡机拿杯子操作咖啡机取杯子送到客人桌子上
2. 倒水
3. 夹点心
具体描述设计一套点单规则如菜单包含咖啡点心等按照规则拟造随机的订单在收到订单后通过大模型让机器人输出合理的备餐计划并尝试在模拟环境中按照这个规划实现任务
"""
# todo: 接收点单信息,大模型生成任务规划
from robowaiter.scene.scene import Scene
class SceneOT(Scene):
def __init__(self, robot):
super().__init__(robot)
# 在这里加入场景中发生的事件
self.event_list = [
2023-11-17 18:56:58 +08:00
# (5, self.set_goal("At(Robot,WaterTable)"))
# (5, self.set_goal("On(Yogurt,Table4)"))
(5, self.set_goal("At(Robot,BrightTable4)"))
2023-11-15 14:30:57 +08:00
]
def _reset(self):
2023-11-17 18:56:58 +08:00
self.gen_obj()
self.state["condition_set"] = {'At(Robot,Bar)', 'Is(AC,Off)',
'Holding(Nothing)','Exist(Yogurt)','Exist(BottledDrink)','On(Yogurt,Bar)','On(BottledDrink,Table1)',
'Is(HallLight,Off)', 'Is(TubeLight,On)', 'Is(Curtain,On)',
'Is(Table1,Dirty)', 'Is(Floor,Dirty)', 'Is(Chairs,Dirty)'}
pass
2023-11-15 14:30:57 +08:00
def _run(self):
pass
if __name__ == '__main__':
import os
from robowaiter.robot.robot import Robot
robot = Robot()
# create task
task = SceneOT(robot)
task.reset()
task.run()