Update chat model output type (#11833)

Commit

2 years ago

Update chat model output type (#11833) --------- Co-authored-by: Bagatur <baskaryan@gmail.com>