Handle belief state values longer than 1 token

scripts/UBAR_code/interaction/UBAR_interact.py

@@ -127,11 +127,26 @@ class UbarSystemModel:  # may inherit convlab or not, just like andy's
                     # in the belief state and then if there is a match adding the next token.
                     # This is not perfect as some are more than one word but its probably good enough.
                     if not db_success:
+                        # The DB doesn't contain a postcode for the police station so fill it here
+                        if token == "[value_postcode]" and self.turn_domain == ["police"]:
+                            token = "CB11QD"
+                            continue
                         decoded_belief_states = decoded_belief_state_subseq.split()
                         for idx, belief_state_slot in enumerate(decoded_belief_states):
                             if token in slots_to_db_keys_map.keys():
                                 if slots_to_db_keys_map[token] == belief_state_slot:
-                                    token = decoded_belief_states[idx + 1]
+                                    curr_slot_resp = ""
+                                    # We dont know the length of the value we need to extract from the belief state
+                                    for belief_state_token in decoded_belief_states[idx + 1 :]:
+                                        if (
+                                            belief_state_token not in slots_to_db_keys_map.values()
+                                            and belief_state_token != "<eos_b>"
+                                        ):
+                                            curr_slot_resp += belief_state_token + " "
+                                        else:
+                                            break
+                                    token = curr_slot_resp[:-1]
+                                    continue
 
                     # Otherwise just leave the slot as it is as we have failed to fill it