Vektoreinbettungen mit der Modellendpunktverwaltung generieren

Auf dieser Seite wird eine Vorabversion beschrieben, mit der Sie einen Endpunkt für ein KI-Modell registrieren und Vorhersagen mit der Modellendpunktverwaltung aufrufen können. Informationen zur Verwendung von KI-Modellen in Produktionsumgebungen finden Sie unter Generative KI-Anwendungen mit AlloyDB AI erstellen und Mit Vektoreinbettungen arbeiten.

Nachdem die Modellendpunkte hinzugefügt und in der Modellendpunktverwaltung registriert wurden, können Sie sie mithilfe der Modell-ID referenzieren, um Einbettungen zu generieren.

Hinweise

Sie müssen Ihren Modellendpunkt in der Modellendpunktverwaltung registriert haben. Weitere Informationen finden Sie unter Modellendpunkt bei der Modellendpunktverwaltung registrieren.

Einbettungen generieren

Verwenden Sie die SQL-Funktion google_ml.embedding(), um den registrierten Modellendpunkt mit dem Modelltyp für Texteinbettungen aufzurufen und Einbettungen zu generieren.

Verwenden Sie die folgende SQL-Abfrage, um das Modell aufzurufen und Einbettungen zu generieren:

SELECT
  google_ml.embedding(
    model_id => 'MODEL_ID',
    content => 'CONTENT');

Ersetzen Sie Folgendes:

  • MODEL_ID: die Modell-ID, die Sie beim Registrieren des Modellendpunkts festgelegt haben.
  • CONTENT: Der Text, der in eine Vektoreinbettung umgewandelt werden soll.

Beispiele

In diesem Abschnitt finden Sie einige Beispiele für das Generieren von Einbettungen mit einem registrierten Modellendpunkt.

Texteinbettungsmodelle mit integrierter Unterstützung

Wenn Sie Embeds für einen registrierten textembedding-gecko@002-Modellendpunkt generieren möchten, führen Sie den folgenden Befehl aus:

    SELECT
      google_ml.embedding(
        model_id => 'textembedding-gecko@002',
        content => 'AlloyDB is a managed, cloud-hosted SQL database service');

Wenn Sie embeddings für einen registrierten text-embedding-ada-002-Modellendpunkt von OpenAI generieren möchten, führen Sie den folgenden Befehl aus:

    SELECT
      google_ml.embedding(
        model_id => 'text-embedding-ada-002',
        content => 'e-mail spam');

Andere Modelle für die Texteinbettung

Wenn Sie Einbettungen für registrierte text-embedding-3-small- oder text-embedding-3-large-Modellendpunkte von OpenAI generieren möchten, führen Sie die folgende Anweisung aus:

  SELECT
    google_ml.embedding(
      model_id => 'text-embedding-3-small',
      content => 'Vector embeddings in AI');