* add token_count_util * remove token_count from retrieval util * format * update dependency * update test