Utiliser l'IA générative pour obtenir des recommandations personnalisées dans une application d'e-commerce

Remarque sur l'interface PostgreSQL : L'intégration de Vertex AI dans Spanner n'est pas compatible avec les bases de données de dialecte PostgreSQL.

Objectif

Dans ce tutoriel, vous allez apprendre à effectuer les opérations suivantes :

• Utilisez les modèles d'IA générative Vertex AI fournis par Google dans une base de données Spanner.
• Utilisez l'IA générative pour fournir des recommandations de produits personnalisées dans un exemple d'application d'e-commerce.

Coûts

Ce tutoriel utilise des composants facturables de Google Cloud, dont :

• Spanner
• Vertex AI

Pour en savoir plus sur les coûts de Spanner, consultez la Tarifs de Spanner.

Pour en savoir plus sur les coûts de Vertex AI, consultez la page Tarifs de Vertex AI.

Créer le schéma du site Web d'e-commerce

Pour ce tutoriel, nous utilisons le schéma et les données suivants:

CREATE TABLE Products (
  id INT64,
  name STRING(MAX),
  description STRING(MAX),
  category_id INT64,
) PRIMARY KEY(id);

CREATE TABLE Categories (
  id INT64,
  name STRING(MAX)
) PRIMARY KEY(id);

CREATE TABLE Users (
  id INT64,
  age INT64,
  likes STRING(MAX)
) PRIMARY KEY(id);

INSERT INTO Categories (id, name) VALUES
    (1, "Toys"),
    (2, "Tools");

INSERT INTO Products (id, name, description, category_id) VALUES
    (1, "Plush Bear", "Really fluffy. Safe for infants.", 1),
    (2, "Bike", "Bike for teenagers.", 1),
    (3, "Drill", "Cordless.", 2);

INSERT INTO Users (id, age, likes) VALUES
    (1, 30, "DIY"),
    (2, 14, "Toys");

Enregistrer un modèle d'IA générative dans un schéma Spanner

Dans ce tutoriel, nous utilisons le modèle text-bison de Vertex AI pour fournir des recommandations de produits personnalisées aux clients finaux. Pour enregistrer ce modèle dans une base de données Spanner, Exécutez l'instruction LDD suivante:

CREATE MODEL TextBison
INPUT (prompt STRING(MAX))
OUTPUT (content STRING(MAX))
REMOTE
OPTIONS (
  endpoint = '//aiplatform.googleapis.com/projects/PROJECT/locations/LOCATION/publishers/google/models/text-bison'
);

Remplacez les éléments suivants :

• PROJECT : ID du projet
• LOCATION: région dans laquelle vous utilisez Vertex AI

La découverte et la validation de schémas ne sont pas disponibles pour l'IA générative des modèles de ML. Par conséquent, vous devez fournir des clauses INPUT et OUTPUT qui correspondent au schéma du modèle. Vous trouverez le schéma complet de text-bison sur la plate-forme Vertex AI Documentation de référence de l'API de modèle

Tant que la base de données et les points de terminaison se trouvent dans le même projet, Spanner doit accorder automatiquement les autorisations appropriées. Sinon, consultez la section Contrôle des accès aux points de terminaison du modèle de la page de référence CREATE MODEL.

Pour vérifier que le modèle a été correctement enregistré, interrogez-le avec la fonction ML.PREDICT. Le modèle n'attend qu'un seul Colonne STRING nommée prompt. Vous pouvez utiliser une sous-requête Spanner pour générer la colonne prompt. Le modèle TextBison nécessite de spécifier un paramètre de modèle maxOutputTokens. Les autres paramètres sont facultatifs. Le modèle text-bison de Vertex AI n'est pas compatible avec le traitement par lot. Vous devez donc utiliser le paramètre @{remote_udf_max_rows_per_rpc=1} pour définir la taille de lot sur 1.

SELECT content
FROM ML.PREDICT(
  MODEL TextBison,
  (SELECT "Is 13 prime?" AS prompt),
  STRUCT(256 AS maxOutputTokens, 0.2 AS temperature, 40 as topK, 0.95 AS topP)
) @{remote_udf_max_rows_per_rpc=1};

+--------------------+
| content            |
+--------------------+
| "Yes, 13 is prime" |
+--------------------+

Répondre aux questions des clients à l'aide du modèle TextBison

Les modèles de texte basés sur l'IA générative peuvent résoudre un large éventail de problèmes. Par exemple, un utilisateur sur un site Web d'e-commerce peut rechercher des produits sûrs pour les nourrissons. Avec une seule requête, nous pouvons transmettre sa question au modèle TextBison. Tout ce que nous avons à faire est de Fournissez un contexte pertinent pour répondre à votre question en récupérant des informations détaillées sur le produit de la base de données.

REMARQUE : Certaines réponses du modèle ont été modifiées pour plus de concision.

SELECT product_id, product_name, content
FROM ML.PREDICT(
  MODEL TextBison,
  (SELECT
    product.id as product_id,
    product.name as product_name,
    CONCAT("Is this product safe for infants?", "\n",
        "Product Name: ", product.name, "\n",
        "Category Name: ", category.name, "\n",
        "Product Description:", product.description) AS prompt
   FROM
     Products AS product JOIN Categories AS category
       ON product.category_id = category.id),
  STRUCT(100 AS maxOutputTokens)
) @{remote_udf_max_rows_per_rpc=1};

-- The model correctly recommends a Plush Bear as safe for infants.
-- Other products are not safe and the model provides justification why.
+------------+-----------------+--------------------------------------------------------------------------------------------------+
| product_id | product_name    | content                                                                                          |
+------------+-----------------+--------------------------------------------------------------------------------------------------+
|          1 | "Plush Bear"    | "Yes, this product is infant safe. [...] "                                                       |
|            |                 | "The product description says that the product is safe for infants. [...]"                       |
+------------+-----------------+--------------------------------------------------------------------------------------------------+
|          2 | "Bike"          | "No, this product is not infant safe. [...] "                                                    |
|            |                 | "It is not safe for infants because it is too big and heavy for them to use. [...]"              |
+------------+-----------------+--------------------------------------------------------------------------------------------------+
|          3 | "Drill"         | "No, this product is not infant safe. [...]"                                                     |
|            |                 | " If an infant were to grab the drill, they could pull it on themselves and cause injury. [...]" |
+------------+-----------------+--------------------------------------------------------------------------------------------------+

Vous pouvez remplacer le littéral de la question par un paramètre de requête, tel que @UserQuestion, si vous souhaitez renseigner directement le paramètre avec une question client. Le client bénéficie ainsi d'une expérience d'achat en ligne optimisée par l'IA.

Proposer des recommandations de produits personnalisées aux clients

En plus des informations détaillées sur le produit, nous pouvons ajouter des informations le client à l'prompt. Le modèle peut ainsi prendre en compte afin de proposer des recommandations de produits entièrement personnalisées.

SELECT product_id, product_name, content
FROM ML.PREDICT(
  MODEL TextBison,
  (SELECT
    product.id as product_id,
    product.name as product_name,
    CONCAT(
        "Answer with YES or NO only: Is this a good fit for me?",
        "My age:", CAST(user.age AS STRING), "\n",
        "I like:", user.likes,  "\n",
        "Product name: ", product.name, "\n",
        "Category mame: ", category.name, "\n",
        "Product description:", product.description) AS prompt,
   FROM
     Products AS product
       JOIN Categories AS category ON product.category_id = category.id
       JOIN Users AS user ON user.id = 1),
  STRUCT(256 AS maxOutputTokens)
) @{remote_udf_max_rows_per_rpc=1};

-- The model correctly guessed that the user might be interested in a Drill
-- as they are interested in DIY.
+------------+-----------------+-------------+
| product_id | product_name    | content     |
+------------+-----------------+-------------+
|          1 | "Plush Bear"    | "NO"        |
+------------+-----------------+-------------+
|          2 | "Bike"          | "NO"        |
+------------+-----------------+-------------+
|          3 | "Drill"         | "YES"       |
+------------+-----------------+-------------+

Pour chercher un cadeau pour son enfant, l'utilisateur peut créer un profil pour son adolescent et voir une autre liste de recommandations:

SELECT product_id, product_name, content
FROM ML.PREDICT(
  MODEL TextBison,
  (SELECT
    product.id as product_id,
    product.name as product_name,
    CONCAT(
        "Answer with YES or NO only: Is this a good fit for me?",
        "\nMy's age:", CAST(user.age AS STRING),
        "\nI like:", user.likes,
        "\nProduct Name: ", product.name,
        "\nCategory Name: ", category.name,
        "\nProduct Description:", product.description) AS prompt,
   FROM
     Products AS product
       JOIN Categories AS category ON product.category_id = category.id
       JOIN Users AS user ON user.id = 2),
  STRUCT(40 AS maxOutputTokens)
) @{remote_udf_max_rows_per_rpc=1};

-- The model correctly guesses that a teenager is interested in a Bike,
-- but not a plush bear for infants or spicy peppers.
+------------+-----------------+---------+
| product_id | product_name    | content |
+------------+-----------------+---------+
|          1 | "Plush Bear"    | "NO"    |
+------------+-----------------+---------+
|          2 | "Bike"          | "YES"   |
+------------+-----------------+---------+
|          3 | "Spicy peppers" | "NO"    |
+------------+-----------------+---------+

Vous pouvez ajouter l'historique des achats ou d'autres informations pertinentes à l'invite pour offrir au client une expérience plus personnalisée.

L'intégration de Vertex AI dans Spanner vous aide à assembler des requêtes complexes contenant des données actives pour créer des applications basées sur l'IA.