diff --git a/robowaiter/llm_client/single_round.py b/robowaiter/llm_client/single_round.py index 49622c5..21e9694 100644 --- a/robowaiter/llm_client/single_round.py +++ b/robowaiter/llm_client/single_round.py @@ -47,165 +47,11 @@ def single_round(question, prefix=""): if __name__ == '__main__': question = ''' -from actions import MoveTo (/), PickUp (), PutDown (,), PutDown(Anything,Anywhere), Make (), Clean (), Turn (,) -from states import At (Robor,/), On (,), Holding (), Is (,), Exist () - -obj=['Coffee', 'Water', 'Dessert', 'Softdrink', 'BottledDrink', 'Yogurt', 'ADMilk', 'MilkDrink', 'Milk','VacuumCup','Chips', 'NFCJuice', 'Bernachon', 'ADMilk', 'SpringWater'] -place=['Bar', 'Bar2', 'WaterTable', 'CoffeeTable', 'Table1', 'Table2', 'Table3','BrightTable6'] -obj2=['Coffee', 'Water', 'Dessert'] -obj3=['Table1','Floor','Chairs'] -obj4=['AC','TubeLight','HallLight','Curtain','ACTemperature'] -obj4state=['On','Off','Up','Down'] - -currents_state={'At(Robot,Bar)', 'Is(AC,Off)', - 'Exist(Yogurt)', 'Exist(BottledDrink)','Exist(Softdrink)','Exist(VacuumCup)', - 'Holding(Coffee)', - 'On(VacuumCup,Table2)', 'On(Yogurt,Bar)', 'On(BottledDrink,Bar)', 'On(Softdrink,Table1)', - 'Is(HallLight,Off)', 'Is(TubeLight,On)', 'Is(Curtain,On)', - 'Is(Table1,Dirty)', 'Is(Floor,Dirty)', 'Is(Chairs,Dirty)'} -def Put_SoftDrink_on_WaterTable(currents_state): - selector - cond On(Softdrink,WaterTable) - sequence - cond Holding(Softdrink) - selector - sequence - cond At(Robot,WaterTable) - act PutDown(Softdrink,WaterTable) - act MoveTo(WaterTable) - sequence - cond At(Robot,Softdrink) - selector - sequence - cond Holding(Nothing) - act PickUp(Softdrink) - act PutDown(Anything,Anywhere) - sequence - cond Exist(Softdrink) - act MoveTo(Softdrink) - - -currents_state={'At(Robot,Bar)', 'Is(AC,On)', - 'Exist(Yogurt)', 'Exist(BottledDrink)', - 'Exist(Chips)', 'Exist(VacuumCup)', 'Exist(ADMilk)', - 'Holding(Nothing)', - 'On(VacuumCup,Table2)', 'On(Chips,Bar)','On(ADMilk,Bar)', - 'Is(HallLight,Off)', 'Is(TubeLight,On)', 'Is(Curtain,On)', - 'Is(Table1,Dirty)', 'Is(Floor,Dirty)', 'Is(Chairs,Dirty)'} -def Put_Dessert_on_Bar(currents_state): - selector - cond On(Dessert,Bar) - sequence - cond Holding(Nothing) - act Make(Dessert) - - -currents_state={'At(Robot,Bar)', 'Is(AC,Off)', - 'Exist(VacuumCup)','Exist(Coffee)', - 'On(VacuumCup,Table2)', 'On(Coffee,CoffeeTable)', - 'Holding(Nothing)', - 'Is(HallLight,Off)', 'Is(TubeLight,On)', 'Is(Curtain,On)', - 'Is(Table1,Dirty)', 'Is(Floor,Dirty)', 'Is(Chairs,Dirty)'} -def Put_Coffee_on_Bar(currents_state): - selector - cond On(Coffee,Bar) - sequence - cond Holding(Coffee) - selector - sequence - cond At(Robot,Bar) - act PutDown(Coffee,Bar) - act MoveTo(Bar) - sequence - cond At(Robot,Coffee) - selector - sequence - cond Holding(Nothing) - act PickUp(Coffee) - act PutDown(Anything,Anywhere) - sequence - cond Exist(Coffee) - cond Holding(Nothing) - act MoveTo(Coffee) - -currents_state={'At(Robot,Bar)', 'Is(AC,Off)', - 'Exist(VacuumCup)', 'Exist(Yogurt)', - 'On(VacuumCup,Table2)', - 'Holding(Yogurt)', - 'Is(HallLight,Off)', 'Is(TubeLight,On)', 'Is(Curtain,On)', - 'Is(Table1,Dirty)', 'Is(Floor,Dirty)', 'Is(Chairs,Dirty)'} -def Put_Coffee_on_Bar(currents_state): - selector - cond On(Coffee,Bar) - sequence - cond Holding(Coffee) - selector - sequence - cond At(Robot,Bar) - act PutDown(Coffee,Bar) - act MoveTo(Bar) - sequence - cond At(Robot,Coffee) - selector - sequence - cond Holding(Nothing) - act PickUp(Coffee) - act PutDown(Anything,Anywhere) - sequence - cond Exist(Coffee) - act MoveTo(Coffee) - sequence - cond Holding(Nothing) - act Make(Coffee) - sequence - cond At(Robot,Bar) - cond Holding(Yogurt) - act PutDown(Yogurt,Bar) - - -currents_state={'At(Robot,Bar)', 'Is(AC,Off)', - 'Exist(Yogurt)', 'Exist(BottledDrink)','Exist(Softdrink)', - 'Exist(Chips)', 'Exist(NFCJuice)', 'Exist(Bernachon)', 'Exist(ADMilk)', 'Exist(SpringWater)', - 'Holding(Softdrink)', - 'Exist(VacuumCup)', 'On(VacuumCup,Table2)', - 'Is(HallLight,Off)', 'Is(TubeLight,On)', 'Is(Curtain,On)', - 'Is(Table1,Clean)', 'Is(Floor,Clean)', 'Is(Chairs,Clean)'} -def Turn_AC_Temperature_Down(currents_state): - selector - cond Is(ACTemperature,Down) - sequence - cond Is(AC,On) - selector - sequence - cond Holding(Nothing) - act Turn(ACTemperature,Down) - act PutDown(Anything,Anywhere) - sequence - cond Is(AC,Off) - selector - sequence - cond Holding(Nothing) - act Turn(AC,On) - act PutDown(Anything,Anywhere) - - -currents_state={'At(Robot,Bar)', 'Is(AC,Off)', - 'Exist(Yogurt)','Exist(VacuumCup)','Exist(Coffee)', - 'Holding(Nothing)', - 'On(Yogurt,Bar)','On(VacuumCup,Table2)','On(Coffee,CoffeeTable)', - 'Is(HallLight,Off)', 'Is(TubeLight,On)', 'Is(Curtain,On)', - 'Is(Table1,Dirty)', 'Is(Floor,Dirty)', 'Is(Chairs,Dirty)'} -def Put_one_Coffee_on_WaterTable_and_Put_another_Coffee_on_BrightTable6 -(currents_state): - - -Please write the content of this function based on the content of currents_state - -Please write the content of this function based on the content of currents_state + 给我来杯咖啡 ''' import timeit cur_time = time.time() - print(single_round(question)) - # print(single_round(question, prefix='现在给出符合这句话要求的目标状态: ')) + # print(single_round(question)) + print(single_round(question, prefix='现在给出符合这句话要求的目标状态: ')) print(f"单次生成耗时:{time.time() - cur_time} s \n")