def func_1(message):
if message.text.isdigit():
i = int(message.text)
b = i * i
bot.send_message(message.chat.id, str(i)+' + '+str(i)+' = '+str(b))
else:
bot.send_message(message.chat.id, "число")
Ну и в else нужно добавить next step handler на эту же функцию, или как там реализовано это все