Embeddings
Texteinbettungen sind numerische Darstellungen von Text, die das Messen semantischer Ähnlichkeit ermöglichen. Dieser Leitfaden stellt Embeddings, ihre Anwendungen und die Verwendung von Embedding-Modellen für Aufgaben wie Suche, Empfehlungen und Anomalieerkennung vor.
Vor der Implementierung von Embeddings
Bei der Auswahl eines Embedding-Anbieters gibt es je nach Ihren Anforderungen und Präferenzen mehrere Faktoren zu berücksichtigen:
- Datensatzgröße & Domänenspezifität: Größe des Modeltrainingsdatensatzes und seine Relevanz für die Domäne, die Sie einbetten möchten. Größere oder domänenspezifischere Daten erzeugen im Allgemeinen bessere domäneninterne Embeddings
- Inferenzleistung: Geschwindigkeit des Embedding-Lookups und End-to-End-Latenz. Dies ist eine besonders wichtige Überlegung für groß angelegte Produktionsbereitstellungen
- Anpassung: Optionen für die Weiterbildung mit privaten Daten oder die Spezialisierung von Modellen für sehr spezifische Domänen. Dies kann die Leistung bei einzigartigen Vokabularen verbessern
So erhalten Sie Embeddings mit Anthropic
Anthropic bietet kein eigenes Embedding-Modell an. Ein Embedding-Anbieter, der eine Vielzahl von Optionen und Funktionen bietet, die alle oben genannten Überlegungen umfassen, ist Voyage AI.
Voyage AI erstellt modernste Embedding-Modelle und bietet maßgeschneiderte Modelle für bestimmte Branchenbereiche wie Finanzen und Gesundheitswesen oder maßgeschneiderte feinabgestimmte Modelle für einzelne Kunden.
Der Rest dieses Leitfadens ist für Voyage AI, aber wir empfehlen Ihnen, eine Vielzahl von Embedding-Anbietern zu bewerten, um die beste Lösung für Ihren spezifischen Anwendungsfall zu finden.
Erste Schritte mit Voyage AI
So greifen Sie auf Voyage-Embeddings zu:
- Melden Sie sich auf der Voyage AI-Website an
- Holen Sie sich einen API-Schlüssel
- Setzen Sie den API-Schlüssel der Einfachheit halber als Umgebungsvariable:
export VOYAGE_API_KEY="<your secret key>"
Sie können die Embeddings entweder mit dem offiziellen voyageai Python-Paket oder mit HTTP-Anfragen ausführen, wie unten beschrieben.
Voyage Python-Paket
Das voyageai
-Paket kann mit dem folgenden Befehl installiert werden:
pip install -U voyageai
Dann können Sie ein Client-Objekt erstellen und es verwenden, um Ihre Texte einzubetten:
import voyageai
vo = voyageai.Client()
# Dies verwendet automatisch die Umgebungsvariable VOYAGE_API_KEY.
# Alternativ können Sie vo = voyageai.Client(api_key="<your secret key>") verwenden
texts = ["Beispieltext 1", "Beispieltext 2"]
result = vo.embed(texts, model="voyage-2", input_type="document")
print(result.embeddings[0])
print(result.embeddings[1])
result.embeddings
wird eine Liste von zwei Embedding-Vektoren sein, die jeweils 1024 Gleitkommazahlen enthalten.
Nach dem Ausführen des obigen Codes werden die beiden Embeddings auf dem Bildschirm ausgegeben:
[0.02012746, 0.01957859, ...] # Embedding für "Beispieltext 1"
[0.01429677, 0.03077182, ...] # Embedding für "Beispieltext 2"
Beim Erstellen der Embeddings können Sie der Funktion embed()
einige weitere Argumente übergeben. Hier ist die Spezifikation:
voyageai.Client.embed(texts : List[str], model : str, input_type : Optional[str] = None, truncation : Optional[bool] = None)
- texts (List[str]) - Eine Liste von Texten als Liste von Strings, wie z.B.
["Ich mag Katzen", "Ich mag auch Hunde"]
. Derzeit beträgt die maximale Länge der Liste 128 und die Gesamtzahl der Token in der Liste höchstens 320K fürvoyage-2
und 120K fürvoyage-large-2
/voyage-code-2
. - model (str) - Name des Modells. Empfohlene Optionen:
voyage-2
,voyage-large-2
,voyage-code-2
. - input_type (str, optional, Standardwert
None
) - Typ des Eingabetextes. Standardwert istNone
. Andere Optionen:query
,document
- Wenn der input_type auf
None
gesetzt ist, wird der Eingabetext direkt von Voyage’s Embedding-Modell kodiert. Alternativ können die Benutzer, wenn die Eingaben Dokumente oder Abfragen sind,input_type
aufquery
bzw.document
setzen. In solchen Fällen wird Voyage dem Eingabetext eine spezielle Eingabeaufforderung voranstellen und die erweiterten Eingaben an das Embedding-Modell senden - Für Retrieval-/Suchanwendungsfälle empfehlen wir, dieses Argument beim Kodieren von Abfragen oder Dokumenten anzugeben, um die Retrieval-Qualität zu verbessern. Embeddings, die mit und ohne das Argument
input_type
generiert wurden, sind kompatibel
- Wenn der input_type auf
- truncation (bool, optional, Standardwert
None
) - Ob die Eingabetexte abgeschnitten werden sollen, um in die Kontextlänge zu passen.- Wenn
True
, werden zu lange Eingabetexte abgeschnitten, um in die Kontextlänge zu passen, bevor sie vom Embedding-Modell vektorisiert werden - Wenn
False
, wird ein Fehler ausgelöst, wenn ein gegebener Text die Kontextlänge überschreitet - Wenn nicht angegeben (Standardwert
None
), wird Voyage den Eingabetext abschneiden, bevor er an das Embedding-Modell gesendet wird, wenn er die Kontextfensterlänge leicht überschreitet. Wenn er die Kontextfensterlänge deutlich überschreitet, wird ein Fehler ausgelöst
- Wenn
Voyage HTTP API
Sie können Embeddings auch erhalten, indem Sie die Voyage HTTP API anfordern. Zum Beispiel können Sie eine HTTP-Anfrage über den Befehl curl
in einem Terminal senden:
curl https://api.voyageai.com/v1/embeddings \
-H "Content-Type: application/json" \
-H "Authorization: Bearer $VOYAGE_API_KEY" \
-d '{
"input": ["Beispieltext 1", "Beispieltext 2"],
"model": "voyage-2"
}'
Die Antwort, die Sie erhalten würden, ist ein JSON-Objekt, das die Embeddings und die Token-Nutzung enthält:
{
"object": "list",
"data": [
{
"embedding": [0.02012746, 0.01957859, ...],
"index": 0
},
{
"embedding": [0.01429677, 0.03077182, ...],
"index": 1
}
],
"model": "voyage-2",
"usage": {
"total_tokens": 10
}
}
Der Embedding-Endpunkt von Voyage AI ist https://api.voyageai.com/v1/embeddings
(POST). Der Anfrage-Header muss den API-Schlüssel enthalten. Der Anfrage-Body ist ein JSON-Objekt, das die folgenden Argumente enthält:
- input (str, List[str]) - Ein einzelner Textstring oder eine Liste von Texten als Liste von Strings. Derzeit beträgt die maximale Länge der Liste 128 und die Gesamtzahl der Token in der Liste höchstens 320K für
voyage-2
und 120K fürvoyage-large-2
/voyage-code-2
. - model (str) - Name des Modells. Empfohlene Optionen:
voyage-2
,voyage-large-2
,voyage-code-2
. - input_type (str, optional, Standardwert
None
) - Typ des Eingabetextes. Standardwert istNone
. Andere Optionen:query
,document
- truncation (bool, optional, Standardwert
None
) - Ob die Eingabetexte abgeschnitten werden sollen, um in die Kontextlänge zu passen- Wenn
True
, werden zu lange Eingabetexte abgeschnitten, um in die Kontextlänge zu passen, bevor sie vom Embedding-Modell vektorisiert werden - Wenn
False
, wird ein Fehler ausgelöst, wenn ein gegebener Text die Kontextlänge überschreitet - Wenn nicht angegeben (Standardwert
None
), wird Voyage den Eingabetext abschneiden, bevor er an das Embedding-Modell gesendet wird, wenn er die Kontextfensterlänge leicht überschreitet. Wenn er die Kontextfensterlänge deutlich überschreitet, wird ein Fehler ausgelöst
- Wenn
- encoding_format (str, optional, Standardwert
None
) - Format, in dem die Embeddings kodiert sind. Voyage unterstützt derzeit zwei Optionen:- Wenn nicht angegeben (Standardwert
None
): Die Embeddings werden als Listen von Gleitkommazahlen dargestellt "base64"
: Die Embeddings werden in Base64-Kodierungen komprimiert
- Wenn nicht angegeben (Standardwert
Voyage Embedding-Beispiel
Jetzt, da wir wissen, wie man Embeddings mit Voyage erhält, lassen Sie uns das mit einem kurzen Beispiel in Aktion sehen.
Angenommen, wir haben einen kleinen Korpus von sechs Dokumenten, aus denen wir abrufen können
documents = [
"Die mediterrane Ernährung betont Fisch, Olivenöl und Gemüse, von denen angenommen wird, dass sie chronische Krankheiten reduzieren.",
"Die Photosynthese in Pflanzen wandelt Lichtenergie in Glukose um und produziert lebenswichtigen Sauerstoff.",
"Innovationen des 20. Jahrhunderts, vom Radio bis zum Smartphone, konzentrierten sich auf elektronische Fortschritte.",
"Flüsse liefern Wasser, Bewässerung und Lebensraum für Wasserlebewesen, die für Ökosysteme lebenswichtig sind.",
"Apples Telefonkonferenz zur Erörterung der Ergebnisse des vierten Geschäftsquartals und der Geschäftsupdates ist für Donnerstag, den 2. November 2023 um 14:00 Uhr PT / 17:00 Uhr ET geplant.",
"Shakespeares Werke wie 'Hamlet' und 'Ein Sommernachtstraum' überdauern in der Literatur."
]
Wir werden zuerst Voyage verwenden, um jedes von ihnen in einen Embedding-Vektor umzuwandeln
import voyageai
vo = voyageai.Client()
# Dokumente einbetten
doc_embds = vo.embed(
documents, model="voyage-2", input_type="document"
).embeddings
Die Embeddings ermöglichen es uns, semantische Suche/Retrieval im Vektorraum durchzuführen. Wir können dann eine Beispielabfrage umwandeln,
query = "Wann ist die Telefonkonferenz von Apple geplant?"
in ein Embedding und dann eine Nearest-Neighbor-Suche durchführen, um das relevanteste Dokument basierend auf der Distanz im Embedding-Raum zu finden.
import numpy as np
# Abfrage einbetten
query_embd = vo.embed(
[query], model="voyage-2", input_type="query"
).embeddings[0]
# Ähnlichkeit berechnen
# Voyage-Embeddings sind auf Länge 1 normiert, daher sind Skalarprodukt
# und Kosinus-Ähnlichkeit gleich.
similarities = np.dot(doc_embds, query_embd)
retrieved_id = np.argmax(similarities)
print(documents[retrieved_id])
Beachten Sie, dass wir input_type="document"
und input_type="query"
für das Einbetten des Dokuments bzw. der Abfrage verwenden. Weitere Spezifikationen finden Sie hier.
Die Ausgabe wäre das 5. Dokument, das tatsächlich am relevantesten für die Abfrage ist:
Apples Telefonkonferenz zur Erörterung der Ergebnisse des vierten Geschäftsquartals und der Geschäftsupdates ist für Donnerstag, den 2. November 2023 um 14:00 Uhr PT / 17:00 Uhr ET geplant.
Verfügbare Voyage-Modelle
Voyage empfiehlt die Verwendung der folgenden Embedding-Modelle:
Modell | Kontextlänge | Embedding-Dimension | Beschreibung |
---|---|---|---|
voyage-large-2 | 16000 | 1536 | Voyage AIs leistungsstärkstes generalistisches Embedding-Modell. |
voyage-code-2 | 16000 | 1536 | Optimiert für Code-Retrieval (17% besser als Alternativen) und auch SoTA auf allgemeinen Korpora. Siehe diesen Voyage Blog-Beitrag für Details. |
voyage-2 | 4000 | 1024 | Basis-generalistisches Embedding-Modell, optimiert sowohl für Latenz als auch für Qualität. |
voyage-lite-02-instruct | 4000 | 1024 | Anweisungsoptimiert für Klassifizierungs-, Clustering- und Satz-Textähnlichkeitsaufgaben, die die einzigen empfohlenen Anwendungsfälle für dieses Modell sind. |
voyage-2
und voyage-large-2
sind generalistische Embedding-Modelle, die domänenübergreifend modernste Leistung erzielen und eine hohe Effizienz beibehalten. voyage-code-2
ist für den Code-Bereich optimiert und bietet die 4-fache Kontextlänge für eine flexiblere Nutzung, allerdings mit einer relativ höheren Latenz.
Voyage entwickelt aktiv fortschrittlichere und spezialisierte Modelle und bietet auch Feinabstimmungsdienste an, um maßgeschneiderte Modelle für einzelne Kunden anzupassen. Senden Sie eine E-Mail an Ihren Anthropic-Account-Manager oder wenden Sie sich an den Anthropic-Support für weitere Informationen zu maßgeschneiderten Modellen.
voyage-finance-2
: demnächst verfügbarvoyage-law-2
: demnächst verfügbarvoyage-multilingual-2
: demnächst verfügbarvoyage-healthcare-2
: demnächst verfügbar
Voyage auf dem AWS Marketplace
Voyage-Embeddings sind auch auf dem AWS Marketplace verfügbar. Hier sind die Anweisungen für den Zugriff auf Voyage auf AWS:
- Abonnieren Sie das Modellpaket
- Navigieren Sie zur Modellpaket-Listenseite und wählen Sie das zu implementierende Modell aus
- Klicken Sie auf die Schaltfläche
Weiter zum Abonnieren
- Überprüfen Sie sorgfältig die Details auf der Seite
Dieses Software-Abonnement
. Wenn Sie mit der Standard-Endbenutzer-Lizenzvereinbarung (EULA), den Preisen und den Support-Bedingungen einverstanden sind, klicken Sie auf “Angebot annehmen” - Nachdem Sie
Weiter zur Konfiguration
ausgewählt und eine Region ausgewählt haben, wird Ihnen eine Produkt-Arn angezeigt. Dies ist die Modellpaket-ARN, die zum Erstellen eines implementierbaren Modells mit Boto3 erforderlich ist- Kopieren Sie die ARN, die Ihrer ausgewählten Region entspricht, und verwenden Sie sie in der nächsten Zelle
- Implementieren Sie das Modellpaket
Erstellen Sie von hier aus einen JupyterLab-Bereich in Sagemaker Studio, laden Sie das Notebook von Voyage hoch und folgen Sie den darin enthaltenen Anweisungen.
FAQ
Preise
Besuchen Sie die Preisseite von Voyage für die aktuellsten Preisinformationen.