ai_query

Invoca um ponto de extremidade do Databricks Model Serving e retorna a resposta.

Para a função SQL do Databricks correspondente, consulte a função ai_query.

Sintaxe

from pyspark.sql import functions as dbf

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

Parameters

Parâmetro Tipo Descrição
endpoint str O nome do ponto de extremidade do Databricks Model Serving a ser invocado.
request pyspark.sql.Column ou str Uma coluna que contém os dados de entrada a serem enviados para o ponto de extremidade.
return_type str ou pyspark.sql.types.DataType, opcional O tipo de retorno esperado, como uma cadeia de caracteres DDL (por exemplo, "STRING", "ARRAY<FLOAT>") ou uma instância do PySpark pyspark.sql.types.DataType (por exemplo, StringType(), ). ArrayType(FloatType()) Quando omitido, o tipo é inferido do ponto de extremidade.

Returns

pyspark.sql.Column: uma nova coluna que contém a resposta do ponto de extremidade.