Freigeben über

ai_extract-Funktion

  • Artikel

Gilt für: durch Häkchen mit „Ja“ markiert Databricks SQL durch Häkchen mit „Ja“ markiert Databricks Runtime

Wichtig

Dieses Feature befindet sich in der Public Preview.

In der Preview ist Folgendes enthalten:

  • Das zugrunde liegende Sprachmodell kann mehrere Sprachen verarbeiten. Diese Funktionen sind jedoch für Englisch optimiert.
  • Für die zugrunde liegenden Foundation-Modell-APIs gibt es Ratenbegrenzung. Weitere Informationen finden Sie unter Foundation Model-APIs, um diese Grenzwerte zu aktualisieren.

Mit der ai_extract()-Funktion können Sie ein hochmodernes Modell für generative KI aufrufen, um Entitäten zu extrahieren, die mithilfe von SQL durch Bezeichnungen aus einem bestimmten Text angegeben wurden. Diese Funktion verwendet einen Bereitstellungsendpunkt für Chatmodelle, der durch die Basismodell-APIs von Databricks verfügbar gemacht wird.

Anforderungen

Wichtig

Die zugrunde liegenden Modelle, die derzeit verwendet werden können, sind unter der Apache 2.0-Lizenz, Copyright The Apache Software Foundation oder der LLAMA 3.1 Community License Copyright © © Meta Platforms, Inc. lizenziert. Alle Rechte vorbehalten. Kunden sind dafür verantwortlich, die Einhaltung anwendbarer Modelllizenzen sicherzustellen.

Databricks empfiehlt, diese Lizenzen zu überprüfen, um die Einhaltung der geltenden Bedingungen zu gewährleisten. Wenn in Zukunft Modelle entstehen, die gemäß den internen Benchmarks von Databricks besser funktionieren, kann Databricks das Modell (und die Liste der anwendbaren Lizenzen auf dieser Seite) ändern.

Derzeit ist Meta-Llama-3.1-70B-Instruct das zugrunde liegende Modell, das diese KI-Funktionen unterstützt.

Hinweis

In Databricks Runtime 15.1 und höher wird diese Funktion in Databricks-Notebooks unterstützt, einschließlich in Notebooks, die als Task in einem Databricks-Workflow ausgeführt werden.

Syntax

ai_extract(content, labels)

Argumente

  • content: Ein STRING-Ausdruck.
  • labels: Ein ARRAY<STRING>-Literal. Jedes Element ist ein Entitätstyp, der extrahiert werden soll.

Gibt zurück

Ein STRUCT-Ausdruck, bei dem jedes Feld einem in labels angegebenen Entitätstyp entspricht. Jedes Feld enthält eine Zeichenfolge, die die extrahierte Entität darstellt. Wenn mehrere potenzielle Entitätstypen gefunden werden, wird lediglich einer zurückgegeben.

Wenn contentNULL ist, ist das Ergebnis NULL.

Beispiele

> SELECT ai_extract(
    'John Doe lives in New York and works for Acme Corp.',
    array('person', 'location', 'organization')
  );
 {"person": "John Doe", "location": "New York", "organization": "Acme Corp."}

> SELECT ai_extract(
    'Send an email to jane.doe@example.com about the meeting at 10am.',
    array('email', 'time')
  );
 {"email": "jane.doe@example.com", "time": "10am"}