feat(ai): Add support for AI features #642

Job	Run time
Collect changes / collect-changes	5s
Python tests / lint-and-format-check	41s
Python tests / types-check	1m 3s
Python tests / unit-tests (py313)	2m 55s
Python tests / unit-tests (py312)	6m 28s
Python tests / unit-tests (py311)	2m 17s
Python tests / unit-tests (py310)	2m 18s
Python tests / unit-tests (py39)	2m 44s
status-check	1s
	18m 32s

Provide feedback