fix: use only cot/reasoning id when consuming it (#86)

* fix: use only cot/reasoning ids with assistant messages

* fix: update tests

* fix: tests