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

Job	Run time
Collect changes / collect-changes	7s
Python tests / types-check	1m 2s
Python tests / lint-and-format-check	35s
Python tests / unit-tests (py313)	3m 0s
Python tests / unit-tests (py312)	6m 28s
Python tests / unit-tests (py311)	2m 19s
Python tests / unit-tests (py310)	2m 16s
Python tests / unit-tests (py39)	2m 41s
status-check	1s
	18m 29s

Provide feedback