ai_query

Anropar en Databricks Model Serving-slutpunkt och returnerar svaret.

För motsvarande Databricks SQL-funktion, se ai_query funktion.

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 Namnet på databricksmodellens serverdelsslutpunkt som ska anropas.
request pyspark.sql.Column eller str En kolumn som innehåller indata som ska skickas till slutpunkten.
return_type str eller pyspark.sql.types.DataType, valfritt Den förväntade returtypen, antingen som en DDL-sträng (till exempel "STRING", "ARRAY<FLOAT>") eller en PySpark-instans pyspark.sql.types.DataType (till exempel StringType(), ArrayType(FloatType())). När den utelämnas härleds typen från slutpunkten.

Returns

pyspark.sql.Column: En ny kolumn som innehåller slutpunktssvaret.