2002 toy tacoma wont idle in drive

I just had my front brakes fixed at the dealershipand when I got home (1 hr away) the truck died at a stop light ( it is an automatic) it did this all the way home. at the lights I put it in N. ANY IDEAS?

It sounds like the torque converter is locking up like it should do at highway speeds.

Thanks!
I have 2000 miles left on my waranty. Sounds like I better take it in.