Databricks Model Serving エンドポイントを呼び出し、応答を返します。
対応する Databricks SQL 関数については、 ai_query 関数を参照してください。
構文
from pyspark.sql import functions as dbf
dbf.ai_query(endpoint=<endpoint>, request=<request>, return_type=<return_type>)
Parameters
| パラメーター | タイプ | 説明 |
|---|---|---|
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: エンドポイント応答を含む新しい列。