ai_query

Wywołuje punkt końcowy obsługujący model usługi Databricks i zwraca odpowiedź.

Aby uzyskać odpowiednią funkcję SQL usługi Databricks, zobacz ai_query funkcja.

Składnia

from pyspark.sql import functions as dbf

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

Parameters

Parametr	Typ	Opis
endpoint	str	Nazwa punktu końcowego obsługującego model usługi Databricks, który ma być wywoływany.
request	pyspark.sql.Column lub str	Kolumna zawierająca dane wejściowe do wysłania do punktu końcowego.
return_type	str lub pyspark.sql.types.DataType, opcjonalnie	Oczekiwany typ zwracany jako ciąg DDL (na przykład "STRING", "ARRAY<FLOAT>") lub wystąpienie PySpark pyspark.sql.types.DataType (na przykład StringType(), ArrayType(FloatType())). Po pominięciu typ jest wnioskowany z punktu końcowego.

Returns

pyspark.sql.Column: nowa kolumna zawierająca odpowiedź punktu końcowego.