ai_query

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

Pour la fonction Databricks SQL correspondante, consultez la fonctionai_query.

Syntaxe

Python

from pyspark.sql import functions as dbf

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

parameter

parameter	Type	Description
endpoint	str	Le nom de l'Endpoint Databricks Model Serving à invoquer.
request	pyspark.sql.Column OU str	Une colonne contenant les données d'entrée à envoyer à l'Endpoint.
return_type	str ou pyspark.sql.types.DataType, facultatif	Le type de retour attendu, soit sous forme de chaîne DDL (par exemple, "STRING", "ARRAY<FLOAT>") soit d'une instance PySpark pyspark.sql.types.DataType (par exemple, StringType(), ArrayType(FloatType())). Lorsqu'il est omis, le type est inféré à partir de l'Endpoint.

parameter	Type	Description
endpoint	str	Le nom de l'Endpoint Databricks Model Serving à invoquer.
request	pyspark.sql.Column OU str	Une colonne contenant les données d'entrée à envoyer à l'Endpoint.
return_type	str ou pyspark.sql.types.DataType, facultatif	Le type de retour attendu, soit sous forme de chaîne DDL (par exemple, "STRING", "ARRAY<FLOAT>") soit d'une instance PySpark pyspark.sql.types.DataType (par exemple, StringType(), ArrayType(FloatType())). Lorsqu'il est omis, le type est inféré à partir de l'Endpoint.

Renvoie

pyspark.sql.Column: Une nouvelle colonne contenant la réponse de l'Endpoint.