우리 팀 코드 스타일을 아는 AI 만들기: 테스트코드 작성, GitLab MR 리뷰 만들기

이 글은 팀의 코드 스타일을 이해하는 AI를 활용하여 테스트 코드 작성 자동화 및 GitLab Merge Request(MR) 리뷰 생성 기능을 구현한 사례를 공유합니다. 이전 글에서 소개한 RAG(Retrieval-Augmented Generation)와 Vector DB 기술을 기반으로, 실제 적용 과정을 상세히 설명합니다.

첫 번째로, 기존 레포지토리의 소스 코드를 임베딩하여 RAG 방식으로 테스트 코드를 자동으로 제안받는 `@pwb/testgen` 패키지를 개발하여 반복적인 테스트 코드 작성 부담을 줄였습니다. 두 번째로, GitLab MR에서 변경된 코드의 컨텍스트를 분석하여 RAG 기반으로 리뷰 코멘트를 생성하는 실험을 통해 리뷰어의 놓치기 쉬운 부분을 보완하고 코드 스타일/규칙 위반 사항을 빠르게 감지하는 데 활용했습니다.

이 두 기능은 일관된 아키텍처 구조를 기반으로 개발되었습니다.