ai_query

Invoca um endpoint Databricks Model Serving e devolve a resposta.

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

Sintaxe

from pyspark.sql import functions as dbf

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

Parameters

Parâmetro Tipo Description
endpoint str O nome do endpoint Databricks Model Serving a invocar.
request pyspark.sql.Column ou str Uma coluna contendo os dados de entrada a enviar para o endpoint.
return_type str ou pyspark.sql.types.DataType, opcional O tipo de retorno esperado, seja como uma cadeia DDL (por exemplo, "STRING", "ARRAY<FLOAT>") ou uma instância PySpark pyspark.sql.types.DataType (por exemplo, StringType(), ArrayType(FloatType())). Quando omitido, o tipo é inferido a partir do ponto final.

Returns

pyspark.sql.Column: Uma nova coluna contendo a resposta do endpoint.