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

Job	Run time
Collect changes / collect-changes	6s
Python tests / lint-and-format-check	36s
Python tests / unit-tests (py313)	3m 43s
Python tests / types-check	1m 4s
Python tests / unit-tests (py312)	6m 26s
Python tests / unit-tests (py311)	2m 15s
Python tests / unit-tests (py310)	2m 21s
Python tests / unit-tests (py39)	2m 47s
status-check	1s
	19m 19s

Provide feedback