В 1916 г. к двум орденам штабс-ротмистра (Св. Станислава 3-й степени от 18.08.1913 г. и Св. Анны 3-й степени от 02.04.1915 г. за окончание Военной авиационной школы) добавляются ещё три: Св. Анны 4-й степени с надписью «За храбрость» – от 27 января – «за боевые разведки на р. Равке в феврале и марте месяце 1915 г.»;
Св. Станислава 2-й степени с мечами – Приказом по 5-й армии № 659 от 4 июля и Св. Владимира 4-й степени с мечами и бантом – Приказом по армиям Северного фронта № 757 от 7 сентября.
В одной из служебных характеристик на Козакова, например, указывалось: «Выдающийся военный лётчик и офицер. Своей беззаветной храбростью и решительностью всегда служит блестящим примером своим подчинённым. Технически хорошо подготовлен и отряд в этом отношении держит высоко».
В 1-ю авиагруппу вошли три авиаотряда (19-й, 4-й и 2-й). После получения новой материальной части (аэропланы «ньюпор» и «Спад») лётчики авиагруппы в короткий срок осваивали основные элементы воздушного боя, фигуры высшего пилотажа, а также тактику группового боя. Козаков лично занимается обучением и передаёт свой богатый опыт подчинённым.
С 25 февраля 1917 г. он исполняет должность командира 1-й боевой авиационной группы Юго-Западного фронта. 14 марта его утверждают в этой должности, а 10 апреля производят в чин ротмистра.
«Под руководством Козакова, – пишет Д. В. Митюрин, – это соединение достаточно высокой степени эффективности, уничтожив только в течение сентября около десятка вражеских аэропланов и потеряв всего одну свою машину. Вероятно, под вдохновляющим влиянием этих успехов на фюзеляжах машин авиагрупп стала воинственная и устрашающая эмблема, позаимствованная с пиратского флага “Весёлый Роджер” – череп со скрещёнными костями. Сам Козаков к концу войны стал бесспорным рекордсменом среди русских истребителей, имея на счету 19 достоверных побед и 3 вероятно сбитых противников».
Не случайно, что из всех боевых групп, созданных в России, 1-я оставалась первой до конца войны и по заслугам, и по количеству побед.
В апреле – мае 1917-го ротмистр Козаков ведёт весьма активную боевую работу. Свои очередные победы он одерживает, как правило, в паре или в группе. Замечательные успехи не остаются не замеченными командованием. Ему предлагают возглавить школу воздушного боя в Евпатории. Но Козаков от заманчивого предложения отказывается:
«На должность начальника школы воздушного боя очень прошу не назначать. Хочу быть только в группе. Благодарю за предложение. Ротмистр Козаков».
14 (27) июня 1917 г. в двух вылетах в паре Александр Александрович одерживает ещё две победы. Однако впервые получает ранение четырьмя пулями в руку, с повреждением локтевого сустава. Командира группы срочно эвакуируют в тыл.
В том же 1917-м Козакова награждают ещё двумя боевыми орденами. 21 апреля – орденом Св. Анны 2-й степени с мечами «за воздушные разведки под г. Двинском в январе-феврале 1916 г.» и орденом Св. Георгия 4-й степени от 31 июля «за то, что, будучи в чине штабс-ротмистра, 8-го декабря 1916 г., вылетев на самолёте “ньюпор”,