src/intent_extractor.py

class IntentExtractor:
    def extract(self, tokens):
        intents = []
        for token in tokens:
            if token in ['book','reserve','order','check']:
                intents.append(token)
        return list(set(intents))