ai_query

Вызывает конечную точку обслуживания модели Databricks и возвращает ответ.

Для соответствующей функции Databricks SQL смотрите функцию ai_query.

Syntax

from pyspark.sql import functions as dbf

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

Parameters

Parameter Тип Описание
endpoint str Имя вызываемой конечной точки обслуживания модели Databricks.
request pyspark.sql.Column или str Столбец, содержащий входные данные для отправки в конечную точку.
return_type str или pyspark.sql.types.DataType, необязательно Ожидаемый тип возвращаемого значения либо в виде строки DDL (например, "STRING",), либо экземпляра PySpark "ARRAY<FLOAT>" (например, pyspark.sql.types.DataType). StringType()ArrayType(FloatType()) При опущении типа выводится из конечной точки.

Returns

pyspark.sql.Column: новый столбец, содержащий ответ конечной точки.