def execute_restaurant(self, entities):
api_manager = RestaurantAPIManager(entities)
ret_data = {}
ret_data["looking_for"] = "restaurant"
ret_data["enough_entities"], ret_data["next_entity"] = api_manager.are_enough_entities()
if ret_data["enough_entities"] == False:
return ret_data
ret_data["recv"] = api_manager.call_api()
if len(ret_data["recv"]["rest"]) > 0:
api_manager.print_data(ret_data["recv"])
return ret_data
def determine(self, parsed, last_state):
query_type = parsed["query_type"]
entities = parsed["entities"]
raw_tokens = parsed["raw_tokens"]
# Entity merging
if last_state is not None:
if query_type == "pardon":
if last_state["STATE"] == State.init or last_state["STATE"] == State.expect:
query_type = "question"
else:
last_state["query_type"] = parsed["query_type"]
merge_entity(last_state["entities"], parsed["entities"])
entities = last_state["entities"]
else:
last_state = parsed
last_state["STATE"] = State.init
last_state["STATE_HISTORY"] = [State.init]
# updating state
last_state["raw_tokens"] = raw_tokens
print("last_state:")
print(last_state)
if query_type == "question":
if entities["question"] == "is_there":
# reset conversation
# if different_location(last_state, parsed):
# Trying to query
if last_state["STATE"] != State.search:
api_manager = RestaurantAPIManager(entities)
api_satisfied, missing_entities = api_manager.are_enough_entities()
# If queries are OK, then execute
if api_satisfied:
last_state["STATE"] = State.search
last_state["STATE_HISTORY"] = [State.init, State.search]
action = Action(ActionType.exec_rest, entities)
else:
last_state["STATE"] = State.expect
last_state["STATE_HISTORY"] = [State.init, State.expect]
action = Action(ActionType.pardon, missing_entities)
else:
action = Action(ActionType.explain)
elif entities["question"] == "how" or entities["question"] == "where":
# Route
if State.search in last_state["STATE_HISTORY"]:
action = Action(ActionType.route)
else:
action = Action(ActionType.pardon, "location")
elif entities["question"] == "which":
if last_state["STATE"] == State.search:
action = Action(ActionType.explain)
else:
actoin = Action(ActionType.pardon, location)
else:
raise NotImplementedError()
elif query_type == "hello" or query_type == "bye":
if query_type == "bye":
last_state = None
action = Action(ActionType.dialogue, query_type)
else:
if last_state["STATE"] == State.search:
action = Action(ActionType.explain)
else:
action = Action(ActionType.pardon)
return action, last_state