이 기사에서는 JetBrains AI Assistant의 AI Chat에서 작성된 프롬프트를 저장하기를 다룹니다.

이 기능을 사용하면 프롬프트를 AI 액션 (AI Action)으로 저장할 수 있습니다.

예시로, Spring Boot의 데모 프로젝트 spring-demo-project에서 선택한 Java 메서드에 대한 JUnit5 테스트 생성 프롬프트를 저장하는 흐름을 소개합니다.

예를 들어, PaymentService.processPayment()와 같은 메서드가 있습니다.

OrderService에 의존함 -
UserService에 의존함 -
JUnit5와 Mockito로 테스트를 작성하고 싶음

@DisplayName이나 given / when / then 작성 방식도 맞추고 싶음

@Transactional
public void processPayment(Long orderId) {
Order order = orderService.findOrder(orderId);
...

AI Assistant에게 테스트를 작성하게 하기 위해, AI Chat에 다음과 같은 프롬프트를 입력합니다.

이 메서드의 테스트를 작성해 주세요.
- JUnit5
- Mockito로 외부 의존성을 모킹 (Mocking)
...

이 프롬프트를 저장해 두면, 다음부터 AI 액션 (AI Action)으로 호출할 수 있게 됩니다.

JetBrains AI Assistant에는 프롬프트를 저장할 수 있는 **프롬프트 라이브러리 (Prompt Library)**가 있습니다.

공식 문서에서는 AI Chat의 입력창에 작성한 프롬프트를 저장하는 절차를 소개하고 있습니다. https://pleiades.io/help/ai-assistant/prompt-library.html

AI Chat에서 작성된 프롬프트를 저장하는 방법

• 프롬프트 위에 마우스 커서를 올리고, 세로 점(⋮)을 클릭
• 메뉴에서 「현재 프롬프트 저장 (Save current prompt)」 옵션을 선택
• 프롬프트 라이브러리 (Prompt Library) 다이얼로그에서 프롬프트 설정을 구성
• OK를 클릭

저장된 프롬프트는 우클릭 메뉴나 Alt + Enter를 통해 열 수 있는 **AI 액션 (AI Action)**에서 호출할 수 있게 됩니다.

여기서는 선택한 메서드에 대해 JUnit5 테스트를 작성하게 하는 프롬프트를 예로 듭니다.

이 메서드의 테스트를 작성해 주세요.
요구사항:
- JUnit5를 사용
...

$SELECTION은 현재 에디터에서 선택하고 있는 코드를 프롬프트에 삽입하기 위한 변수입니다.

따라서 다음부터는,

• 테스트를 작성하고 싶은 메서드를 선택한다
• 우클릭하거나 Alt + Enter를 누른다
• AI 액션 (AI Action)에서 저장된 프롬프트를 선택한다

처음부터 프롬프트 라이브러리 (Prompt Library)에 등록하기보다, 먼저 AI Chat에서 실행해 보는 것이 좋을 것 같습니다.

spring-demo-project/src/main/java/com/example/order/PaymentService.java의 processPayment()를 선택한 상태에서 프롬프트를 실행하면, 다음과 같은 테스트가 생성되었습니다.

package com.example.order;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
...

지정한 내용이 반영되어 있음을 확인할 수 있습니다.

• JUnit5를 사용하고 있음
• Mockito의 @Mock과 @InjectMocks를 사용하고 있음
• @DisplayName이 void 메서드에 맞는 일본어로 되어 있음
• given / when / then 주석이 들어가 있음
• AssertJ의 assertThat을 사용하고 있음
• 예외 케이스에서 never()를 사용하여 사용자 잔액 업데이트가 호출되지 않았음을 확인하고 있음

이와 같이 기대한 형식으로 출력된 프롬프트는 저장하여 재사용할 후보가 됩니다.

기대한 대로 출력이 되는 것을 확인했다면, AI Chat의 입력창에 있는 프롬프트를 저장합니다.

프롬프트 이름: JUnit5 테스트 생성 규약

• 호출 후 추가 사용자 입력 대기 (Wait for additional user input after call): On/Off -
  「AI Action」 팝업에 프롬프트 표시 (Show prompt in "AI Action" popup): On/Off

우클릭 메뉴나 Alt + Enter를 통해 호출하고 싶다면, AI Action 표시를 On으로 설정합니다.

저장 후에는 다른 메서드에서도 동일한 프롬프트를 즉시 사용할 수 있습니다.

• 대상 메서드를 선택한다
• 우클릭하거나 Alt + Enter를 누른다 -
• AI Action을 연다 -
• JUnit5 테스트 생성 규약을 선택한다

이렇게 하면 저장된 프롬프트가 실행됩니다.

$SELECTION 위치에는 현재 선택된 메서드가 들어갑니다.

대상 메서드를 선택한 상태에서 Alt + Enter를 누르고, 표시된 메뉴에서 AI Action을 선택해도 동일하게 실행할 수 있습니다.

마우스 조작을 거치지 않아도 되므로, 코드를 작성하는 흐름을 유지하며 저장된 프롬프트를 쉽게 호출할 수 있습니다.

프롬프트 라이브러리는 개인의 IDE 설정으로 사용하는 기능이라고 생각하는 것이 좋습니다.

팀 전체가 동일한 규칙을 공유하고 싶다면, 프롬프트 라이브러리보다는 Project Rules가 더 적합합니다.

Project Rules는 .aiassistant/rules/*.md 형태로 Git 관리가 가능합니다.

설정 화면에서 새로운 프롬프트를 만들 수도 있습니다.

다만, 처음부터 완벽한 템플릿을 설계하려고 하기보다, 채팅에서 시도해 본 뒤 잘 작동하는 프롬프트를 저장하는 방식이 더 사용하기 편리할 것입니다.

'미리 설계해서 저장하기'보다는, 효과가 있었던 프롬프트를 그 자리에서 저장하는 것이 이 기능의 취지에 더 부합합니다.

AI Chat에 동일한 지시 사항을 반복해서 붙여넣고 있다면, **현재 프롬프트 저장 (Save current prompt)**을 사용하여 번거로움을 줄일 수 있습니다.

선택한 코드를 사용하는 작업에서는 $SELECTION이 매우 유용합니다.

• 메서드를 선택한다
• 저장된 프롬프트를 AI Action에서 호출한다
• 선택된 코드가 자동으로 프롬프트에 포함된다

이러한 흐름으로 진행할 수 있어, 테스트 생성이나 리뷰 요청과 같은 반복 작업에 적합합니다.

'채팅에서 효과적이었던 지시 사항을 저장한다'는 방식이 이 기능에 가장 잘 맞는 활용법이라고 생각합니다.

저희는 JetBrains 제품에 관한 질문이나 상담 등을 받고 있습니다. 저희 X 또는 이메일로 연락해 주세요.