Gerar respostas inteligentes com o Kit de ML no iOS

O Kit de ML pode gerar respostas curtas a mensagens usando um modelo no dispositivo.

Para gerar respostas inteligentes, você transmite ao Kit de ML um registro de mensagens recentes em um conversa. Se o Kit de ML determinar que a conversa está em inglês e que a conversa não pode ter um assunto potencialmente sensível, o kit de ML gera até três respostas, que você pode sugerir ao usuário.

Faça um teste

Antes de começar

  1. Inclua os seguintes pods do kit de ML no seu Podfile:
    pod 'GoogleMLKit/SmartReply', '3.2.0'
    
  2. Depois de instalar ou atualizar os pods do seu projeto, abra o projeto Xcode usando o .xcworkspace: O Kit de ML é compatível com a versão 12.4 ou mais recente do Xcode.

1. Criar um objeto de histórico de conversas

Para gerar respostas inteligentes, você transmite ao Kit de ML uma matriz ordenada cronologicamente de Objetos TextMessage, com o carimbo de data/hora mais antigo primeiro. Sempre que o usuário envia ou recebe uma mensagem, adiciona a mensagem, o carimbo de data/hora dela ID do usuário do remetente ao histórico de conversas.

O ID do usuário pode ser qualquer string que identifique exclusivamente o remetente no campo conversa. O ID do usuário não precisa corresponder a nenhum dado do usuário. e o User-ID não precisa ser consistente entre conversas ou invocações do gerador de respostas inteligentes.

Se a mensagem foi enviada pelo usuário para o qual você quer sugerir respostas, defina isLocalUser como verdadeiro.

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];

Um objeto de histórico de conversas é parecido com este exemplo:

Carimbo de data/hora userID isLocalUser Mensagem
Quinta-feira, 21 de fevereiro, 13:13:39 PST 2019 verdadeiro você está a caminho?
Quinta-feira, 21 de fevereiro, 13:15:03 PST 2019 FRIEND0 falso Sinto muito!

O Kit de ML sugere respostas para a última mensagem no histórico de conversas. A última mensagem deve ser de um usuário não local. No exemplo acima, a última mensagem da conversa pertence ao usuário não local FRIEND0. Quando você usa esse registro, ele sugere responde à mensagem de FRIENDO: "Está atrasado, sinto muito!".

2. Receber respostas de mensagens

Para gerar respostas inteligentes a uma mensagem, receba uma instância de SmartReply e transmita o histórico de conversas para o método 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.
      // ...
  }
}];

Se a operação for bem-sucedida, um objeto SmartReplySuggestionResult será transmitido para o gerenciador de conclusão. Esse objeto contém uma lista de até três sugestões de resposta, que você pode apresentar ao usuário:

Swift

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

Objective-C

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

O kit de ML poderá não retornar resultados se o modelo não confiar na a relevância das respostas sugeridas, a conversa de entrada não está inglês ou se o modelo detectar um assunto sensível.