[python] fix last token fetch logic

[python] fix last token fetch logic #5061