ai_extract

Extrahiert strukturierte Daten aus einer Dokumentspalte mithilfe von AI/LLM.

Die entsprechende Databricks SQL-Funktion finden Sie unter ai_extract Funktion.

Syntax

from pyspark.sql import functions as dbf

dbf.ai_extract(col=<col>, schema=<schema>, options=<options>)

Parameter

Parameter Typ Beschreibung
col pyspark.sql.Column oder str Eine Spalte, die den Dokumentinhalt enthält, aus der extrahiert werden soll.
schema dict oder list Ein Python Diktieren (Feldname in {"type": ..., "description": ...}) oder eine Liste von Feldnamenzeichenfolgen. Serialisiert in ein JSON-Literal automatisch.
options dictwahlfrei Ein Wörterbuch mit Optionen zum Steuern des Extraktionsverhaltens.

Returns

pyspark.sql.Column: Eine neue Spalte von VariantType, die die extrahierten Felder enthält.

Beispiele

df.select(ai_extract("text", {"name": {"type": "string", "description": "Name"}}))
df.select(ai_extract("text", ["name", "age"]))