ai_extract

Estrae dati strutturati da una colonna di documento usando AI/LLM.

Per la funzione SQL di Databricks corrispondente, vedere ai_extract funzione.

Sintassi

from pyspark.sql import functions as dbf

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

Parameters

Parametro Tipo Description
col pyspark.sql.Column oppure str Colonna contenente il contenuto del documento da estrarre.
schema dict oppure list Un Python dict (nome campo a {"type": ..., "description": ...}) o un elenco di stringhe nome campo. Serializzato automaticamente in un valore letterale JSON.
options dictopzionale Dizionario di opzioni per controllare il comportamento di estrazione.

Returns

pyspark.sql.Column: nuova colonna di VariantType contenente i campi estratti.

Esempi

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