The burly left-arm spinner claimed seven wickets in the fourth innings to derail India's 176-run chase in the last test they played at Galle in 2015.
India went on to win the next two matches to clinch the series 2-1 and Kohli said his team have learnt their lesson from that series.
"He's a quality bowler. Last time he really dented us in that small chase," Kohli said of Herath, who will lead Sri Lanka in the series opener in absence of ailing regular skipper Dinesh Chandimal.
"It was actually a great eye-opener for us. We worked on those things in the next two games and were able to win the series. He played both the games and the ball was still doing a bit but our batsmen had counter-plans in place."
News that makes sense
Your trusted source for staying up-to-date with the world around you. Get free daily news updates and analysis, straight to your inbox.
Herath managed eight wickets in the remaining four innings as the Indians appeared to have found a way to deal with Sri Lanka's spin spearhead.
"We understand what he brings to the table and we understand what needs to be done," Kohli said.
"I'm sure everyone is taking personal responsibility to plan their innings... We've come a long way from that tour. We believe our batsmen will deliver the goods when it's required."
While Sri Lanka will miss Chandimal, who is recovering from pneumonia, India will be without regular opener Lokesh Rahul, who is down with high fever.
Kohli ruled out any experiment and suggested Abhinav Mukund will join Shikhar Dhawan at the top of the innings and said all-rounder Hardik Pandya could make his test debut in the first match of the three-test series.
"We got great balance in the side. We got a guy like Hardik Pandya who is a wicket-taking bowler. Every game he plays, on any surface, he has a knack of picking up wickets. He has a great chance of playing."
Counterpart Herath was worried how his young team would fare against the mighty Indians who have been beaten only once since the 2015 Galle test.
"They are the number one team at the moment and are playing very good cricket," said the 39-year-old.
"This is an interesting and challenging series. We need to do something special."
(Reporting by Amlan Chakraborty in New Delhi, editing by Pritha Sarkar)
