Genera respuestas inteligentes con ML Kit en iOS

Puedes usar el Kit de AA para generar respuestas de mensajes con un modelo en el dispositivo.

Para generar respuestas inteligentes, pasa al Kit de AA un registro de mensajes recientes de una conversación. Si el Kit de AA determina que la conversación está en inglés y que no contiene cuestiones potencialmente delicadas, generará hasta tres respuestas que puedes sugerir al usuario.

Antes de comenzar

  1. Si aún no agregaste Firebase a tu app, sigue los pasos en la guía de introducción para hacerlo.
  2. Incluye las bibliotecas del ML Kit en tu Podfile:
    pod 'Firebase/MLCommon', '6.25.0'
    pod 'Firebase/MLNLSmartReply', '6.25.0'
    
    Después de instalar o actualizar los Pods de tu proyecto, asegúrate de abrir el proyecto de Xcode con su .xcworkspace.
  3. En tu app, importa Firebase:

    Swift

    import Firebase

    Objective-C

    @import Firebase;

1. Crea un objeto de historial de conversaciones

Para generar respuestas inteligentes, debes pasar al Kit de AA un arreglo de objetos TextMessage ordenado cronológicamente, con la marca de tiempo más antigua. Cuando los usuarios envían o reciben un mensaje, puedes agregarlo junto con su marca de tiempo y el ID de usuario del remitente al historial de conversaciones.

El ID de usuario puede ser cualquier string que identifique de forma única al remitente en la conversación. El ID no tiene que corresponder a los datos de ningún usuario ni ser coherente con las conversaciones o invocaciones del generador de respuestas inteligentes.

Si el mensaje lo envió el usuario al que quieres sugerirle respuestas, configura isLocalUser en verdadero.

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:
FIRTextMessage *message = [[FIRTextMessage alloc]
        initWithText:@"How are you?"
        timestamp:[NSDate date].timeIntervalSince1970
        userID:userId
        isLocalUser:NO];
[conversation addObject:message];

Un objeto de historial de conversaciones tiene un aspecto similar al del siguiente ejemplo:

Marca de tiempo ID de usuario ¿Es un usuario local? Mensaje
Jue 21 de feb 13:13:39 PST 2019 verdadero ¿vienes en camino?
Jue 21 de feb 13:15:03 PST 2019 FRIEND0 falso Llegaré tarde, lo siento.

Ten en cuenta que el mensaje más reciente en el ejemplo anterior es de un usuario no local. Esto es importante, ya que el Kit de AA sugiere respuestas cuyo envío está destinado a realizarlo el usuario de tu app, es decir, el usuario local. Debes asegurarte de que pasas al Kit de AA un registro de conversación que termine con un mensaje al que el usuario podría querer responder.

2. Recibe respuestas de mensajes

Obtén una instancia de SmartReply y pasa el historial de la conversación a su método suggestReplies(for:completion:) a fin de generar respuestas inteligentes para un mensaje:

Swift

let naturalLanguage = NaturalLanguage.naturalLanguage()
naturalLanguage.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
        // the result doesn't contain any suggestions.
    } else if (result.status == .success) {
        // Successfully suggested smart replies.
        // ...
    }
}

Objective-C

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

Si la operación se realiza correctamente, se pasará un objeto SmartReplySuggestionResult al controlador de finalización. Este objeto contiene una lista de hasta 3 respuestas sugeridas que puedes presentar a un usuario:

Swift

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

Objective-C

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

Ten en cuenta que el Kit de AA podría no mostrar resultados si el modelo no está seguro de la relevancia de las respuestas sugeridas, la conversación que se ingresa no está en inglés o si el modelo detecta cuestiones sensibles.