ai_query

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

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

Sintaxe

Python
from pyspark.sql import functions as dbf

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

Parâmetro	Tipo	Descrição
`endpoint`	`str`	O nome do endpoint de Model Serving do Databricks a ser invocado.
`request`	`pyspark.sql.Column` ou `str`	Uma coluna contendo os dados de entrada para enviar ao endpoint.
`return_type`	`str` ou `pyspark.sql.types.DataType`, opcional	O tipo de retorno esperado, seja como strings 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 endpoint.

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