Custom Componenet

Hello!

In the process method of the custom component, I am running into an issue as far as the message object goes. I am attempting to use the tokens from the message parameter but I am running into an error where message.get('tokens') is returning None.

This is the error I receive during training:

message_tokens = [t.get('text') for t in message.get('tokens')]

TypeError: 'NoneType' object is not iterable`

This is my process method where the issue is happening in the custom component py file:

def process(self, message: Message, **kwargs: Any) -> None:
    message_tokens = [t.get('text') for t in message.get('tokens')]

Let me know if I can provide any extra details!

Thanks! Abhi

Did you set a attribute named token ?