EmbodiedAI
/
RoboWaiter
mirror of https://github.com/HPCL-EI/RoboWaiter.git
3
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

添加serve节点

This commit is contained in:
ChenXL97 2023-11-19 16:55:22 +08:00
parent 576e312fec
commit 3e1b73f682
3 changed files with 24 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
robowaiter/behavior_lib/act/DealChat.py
View File

@ -34,8 +34,9 @@ class DealChat(Act):
history = self.scene.state["chat_history"][name]
self.scene.state["attention"]["customer"] = name
self.scene.state["serve_state"] = {
self.scene.state["serve_state"][name] = {
"last_chat_time": self.scene.time,
"served": False
}
function_call, response = ask_llm(sentence,history,func_map=self.func_map)
@ -128,7 +129,10 @@ class DealChat(Act):
# return near_location
def stop_serve(self,**args):
customer = self.scene.state["attention"]["customer"]
serve_state = self.scene.state["serve_state"][customer]
serve_state['served'] = True
return "好的"

12
robowaiter/behavior_lib/act/ServeCustomer.py
View File

@ -20,7 +20,13 @@ class ServeCustomer(Act):
# if self.scene.time - self.scene.state["serve_state"]["last_chat_time"] > 10:
# self.chat_bubble
goal = Act.place_xyz_dic['Bar']
self.scene.walk_to(goal[0]-5,goal[1], 180, 180, 0)
self.scene.chat_bubble("欢迎光临！请问有什么可以帮您？")
customer = self.scene.state["attention"]["customer"]
serve_state = self.scene.state["serve_state"][customer]
if self.scene.time - serve_state['last_chat_time'] > 3:
serve_state['served'] = True
# goal = Act.place_xyz_dic['Bar']
# self.scene.walk_to(goal[0]-5,goal[1], 180, 180, 0)
# self.scene.chat_bubble("欢迎光临！请问有什么可以帮您？")
return ptree.common.Status.RUNNING

11
robowaiter/behavior_lib/cond/CustomerServed.py
View File

@ -13,11 +13,20 @@ class CustomerServed(Cond):
def _update(self) -> ptree.common.Status:
# if self.scene.status?
if self.name in self.scene.state["condition_set"]:
customer = self.scene.state["attention"]["customer"]
serve_state = self.scene.state["serve_state"][customer]
if serve_state['served']:
return ptree.common.Status.SUCCESS
else:
return ptree.common.Status.FAILURE
# if self.scene.state["attention"]["customer"]
#
# if self.name in self.scene.state["condition_set"]:
# return ptree.common.Status.SUCCESS
# else:
# return ptree.common.Status.FAILURE
# if self.scene.state['chat_list'] == []:
# return ptree.common.Status.FAILURE
# else: