ai_query

Databricks Model Servingエンドポイントを呼び出し、応答を返します。

対応するDatabricks SQL関数については、ai_query関数を参照してください。

構文

Python
from pyspark.sql import functions as dbf

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

パラメーター	Type	説明
`endpoint`	`str`	呼び出す Databricks Model Serving エンドポイントの名前。
`request`	`pyspark.sql.Column` または `str`	エンドポイントに送信する入力データを含む列。
`return_type`	`str` または `pyspark.sql.types.DataType`、オプション	予期される戻り値の型。DDL文字列（例: `"STRING"`、`"ARRAY<FLOAT>"`）またはPySpark `pyspark.sql.types.DataType`インスタンス（例: `StringType()`、`ArrayType(FloatType())`）のいずれかです。省略された場合、タイプはエンドポイントから推論されます。

pyspark.sql.Column：エンドポイント応答を含む新しい列です。