thanks for reply, yes, we are thinking of commenting out token counting logic if databricks side is not willing to change implementation. Just want to raise the point that the implementation like this should take into consideration of restricted network scenarios.