ai_query

Appelle un point de terminaison Databricks Model Serving et retourne la réponse.

Pour obtenir plus de détails sur la fonction SQL de Databricks correspondante, consultez ai_query.

Syntax

from pyspark.sql import functions as dbf

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

Parameters

Parameter Type Description
endpoint str Nom du point de terminaison Databricks Model Serving à appeler.
request pyspark.sql.Column ou str Colonne contenant les données d’entrée à envoyer au point de terminaison.
return_type str ou pyspark.sql.types.DataType, facultatif Type de retour attendu, sous la forme d’une chaîne DDL (par exemple, "STRING", "ARRAY<FLOAT>") ou d’une instance PySpark pyspark.sql.types.DataType (par exemple, StringType(), ArrayType(FloatType())). Lorsqu’il est omis, le type est déduit du point de terminaison.

Returns

pyspark.sql.Column: nouvelle colonne contenant la réponse du point de terminaison.