iOS에서 ML Kit를 사용한 스마트 답장 생성

ML Kit는 기기별 모델을 사용하여 메시지에 대한 짧은 답장을 생성할 수 있습니다.

스마트 답장을 생성하려면 ML Kit에 대화의 최근 메시지 로그를 전달합니다. ML Kit에서 대화가 영어로 작성되었으며 대화에 민감한 주제가 포함되어 있지 않다고 판단하면 ML Kit는 사용자에게 추천할 수 있는 답장을 최대 3개까지 생성합니다.

사용해 보기

  • 샘플 앱을 살펴보고 이 API의 사용 예를 확인하세요.

시작하기 전에

  1. Podfile에 다음 ML Kit 포드를 추가합니다.
    pod 'GoogleMLKit/SmartReply', '3.2.0'
    
  2. 프로젝트의 포드를 설치하거나 업데이트한 후 .xcworkspace를 사용하여 Xcode 프로젝트를 엽니다. ML Kit는 Xcode 버전 12.4 이상에서 지원됩니다.

1. 대화 기록 객체 만들기

스마트 답장을 생성하려면 ML Kit에 시간순으로 정렬된 TextMessage 객체의 배열을 전달합니다. 가장 오래된 타임스탬프가 먼저 나와야 합니다. 사용자가 메시지를 보내거나 받을 때마다 메시지, 타임스탬프, 메시지 발신자의 사용자 ID를 대화 기록에 추가합니다.

사용자 ID는 대화 내에서 발신자를 고유하게 식별하는 모든 문자열일 수 있습니다. 사용자 ID는 사용자 데이터와 일치할 필요가 없으며, 스마트 답장 생성기의 호출 또는 대화 간에 사용자 ID가 일관되지 않아도 됩니다.

답장을 추천하려는 사용자가 보낸 메시지인 경우 isLocalUser를 true로 설정합니다.

Swift

var conversation: [TextMessage] = []

// Then, for each message sent and received:
let message = TextMessage(
    text: "How are you?",
    timestamp: Date().timeIntervalSince1970,
    userID: "userId",
    isLocalUser: false)
conversation.append(message)

Objective-C

NSMutableArray *conversation = [NSMutableArray array];

// Then, for each message sent and received:
MLKTextMessage *message = [[MLKTextMessage alloc]
        initWithText:@"How are you?"
        timestamp:[NSDate date].timeIntervalSince1970
        userID:userId
        isLocalUser:NO];
[conversation addObject:message];

대화 기록 객체는 다음 예시와 같습니다.

타임스탬프 userID isLocalUser 메시지
2019년 2월 21일 목요일 13:13:39 PST true 가는 중인가요?
2019년 2월 21일 목요일 13:15:03 PST FRIEND0 false 늦어서 죄송합니다.

ML Kit는 대화 기록에서 마지막 메시지에 대한 답장을 제안합니다. 마지막 메시지는 로컬이 아닌 사용자가 보낸 것이어야 합니다. 위의 예에서 대화의 마지막 메시지는 로컬이 아닌 사용자인 friend0이 보낸 메시지입니다. 이 로그를 통과하는 ML Kit를 사용하면 FriendsO의 메시지에 대한 회신을 제안합니다. "늦습니다. 죄송합니다!"

2. 메시지 답장 받기

메시지에 대한 스마트 답장을 생성하려면 SmartReply의 인스턴스를 가져와서 suggestReplies(for:completion:) 메서드에 대화 기록을 전달합니다.

Swift

SmartReply.smartReply().suggestReplies(for: conversation) { result, error in
    guard error == nil, let result = result else {
        return
    }
    if (result.status == .notSupportedLanguage) {
        // The conversation's language isn't supported, so
        // the result doesn't contain any suggestions.
    } else if (result.status == .success) {
        // Successfully suggested smart replies.
        // ...
    }
}

Objective-C

MLKSmartReply *smartReply = [MLKSmartReply smartReply];
[smartReply suggestRepliesForMessages:inputText
                           completion:^(MLKSmartReplySuggestionResult * _Nullable result,
                                        NSError * _Nullable error) {
  if (error || !result) {
    return;
  }
  if (result.status == MLKSmartReplyResultStatusNotSupportedLanguage) {
      // The conversation's language isn't supported, so
      // the result doesn't contain any suggestions.
  } else if (result.status == MLKSmartReplyResultStatusSuccess) {
      // Successfully suggested smart replies.
      // ...
  }
}];

작업이 성공하면 SmartReplySuggestionResult 객체가 완료 핸들러에 전달됩니다. 이 객체에는 사용자에게 표시할 수 있는 최대 3개의 추천 답장 목록이 포함됩니다.

Swift

for suggestion in result.suggestions {
  print("Suggested reply: \(suggestion.text)")
}

Objective-C

for (MLKSmartReplySuggestion *suggestion in result.suggestions) {
  NSLog(@"Suggested reply: %@", suggestion.text);
}

모델이 추천 답장의 관련성을 확신하지 못하거나, 입력 대화가 영어가 아니거나, 모델이 민감한 주제를 감지하는 경우 ML Kit가 결과를 반환하지 않을 수 있습니다.