The role of Artificial Intelligence (AI) is becoming more and more important in the area of production and manufacturing. While experiencing the rapid changes in the Industry 4.0 era, companies are striving to remain competitive by exploiting the potential of digitization. In the road towards Industry 4.0, companies have to evaluate their capabilities and determine goals for the future: these are supported by maturity assessment models, such as the Company CoMpaSs (CCMS) model. As technologies evolve, these models have to be improved as well: the aim of the paper is to update the CCMS model with AI aspects that were not detailed enough previously. In the paper, the authors first perform a literature review on the topic of application of AI in the production area, and also interview experts from the industry and academia in order to create a basis for the AI aspect in the CCMS model. Second, the model is extended with one additional dimension called AI applications, based on the above-mentioned results and taking the pathways and development levels determined in the Connected Factories project into consideration.