ai_query

Richiama un endpoint di gestione del modello di Databricks e restituisce la risposta.

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

Sintassi

from pyspark.sql import functions as dbf

dbf.ai_query(endpoint=<endpoint>, request=<request>, return_type=<return_type>)

Parameters

Parametro Tipo Description
endpoint str Nome dell'endpoint Databricks Model Serving da richiamare.
request pyspark.sql.Column oppure str Colonna contenente i dati di input da inviare all'endpoint.
return_type str o pyspark.sql.types.DataType, facoltativo Tipo restituito previsto, come stringa DDL (ad esempio, "STRING", "ARRAY<FLOAT>") o un'istanza di PySpark pyspark.sql.types.DataType (ad esempio, StringType(), ArrayType(FloatType())). Se omesso, il tipo viene dedotto dall'endpoint.

Returns

pyspark.sql.Column: nuova colonna contenente la risposta dell'endpoint.