Gerar respostas inteligentes com o Kit de ML no iOS

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

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

Observação: as APIs do Kit de ML para iOS são executadas apenas em dispositivos de 64 bits. Se você criar seu app com suporte a 32 bits, verifique a arquitetura do dispositivo antes de usar essa API.

Faça um teste

• Teste o app de exemplo para ver um exemplo de uso dessa API.

Antes de começar

1. Inclua os seguintes pods de Kit de ML no seu Podfile:

   pod 'GoogleMLKit/SmartReply', '3.2.0'
   

2. Depois de instalar ou atualizar os pods do projeto, abra o projeto do Xcode usando o .xcworkspace. O Kit de ML é compatível com a versão 12.4 ou superior do Xcode.

1. Criar um objeto de histórico de conversas

Para gerar respostas inteligentes, você transmite para o Kit de ML uma matriz em ordem cronológica de objetos TextMessage, com o carimbo de data/hora mais antigo primeiro. Sempre que o usuário enviar ou receber uma mensagem, adicione a mensagem, o carimbo de data/hora e o ID do usuário do remetente da mensagem ao histórico de conversas.

O ID do usuário pode ser qualquer string que identifique exclusivamente o remetente na conversa. O ID do usuário não precisa corresponder a nenhum dado do usuário e não precisa ser consistente entre conversas ou invocações do gerador de resposta inteligente.

Se a mensagem foi enviada pelo usuário para o qual você quer sugerir respostas, defina isLocalUser como "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];

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

Carimbo de data/hora	ID do usuário	Usuário local	A mensagem
Quinta-feira, 21 de fevereiro, às 13:13:39 PST 2019		verdadeiro	Você está a caminho?
Quinta-feira, 21 de fevereiro, às 13:15:03 PST 2019	AMIGO0	false	Vou me atrasar!

O Kit de ML sugere respostas à última mensagem em um histórico de conversas. A última mensagem precisa ser de um usuário não local. No exemplo acima, a última mensagem na conversa é do usuário não local MIG0. Quando você usa esse registro do Kit de ML, ele sugere respostas à mensagem de Amigos: "Correndo tarde, desculpa!"

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 respostas sugeridas que podem ser apresentadas 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 pode não retornar resultados se o modelo não confiar na relevância das respostas sugeridas, se a conversa de entrada não estiver em inglês ou se o modelo detectar um assunto confidencial.