REST API
MLflow REST API 允许您创建、列出和获取实验及运行,并记录参数、指标和制品。API 托管在 MLflow 跟踪服务器的 `/api` 路由下。例如,要搜索托管在 `https://:5000` 上的跟踪服务器中的实验,请向 `https://:5000/api/2.0/mlflow/experiments/search` 发送 POST 请求。
重要提示
MLflow REST API 要求所有 POST 请求的内容类型为 `application/json`。
目录
创建实验
端点 |
HTTP 方法 |
|---|---|
|
|
通过名称创建一个实验。返回新创建实验的 ID。会验证是否已存在同名实验,如果存在则失败。
如果存在同名实验,抛出 `RESOURCE_ALREADY_EXISTS` 异常。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
name |
|
实验名称。此字段为必需。 |
artifact_location |
|
存储实验所有制品的地点。如果不提供,远程服务器将选择一个合适的默认值。 |
tags |
ExperimentTag 的数组 |
要在实验上设置的一组标签。最大标签大小和单次请求中的标签数量取决于存储后端。所有存储后端都保证支持最大 250 字节的标签键和最大 5000 字节的标签值。所有存储后端也保证支持单次请求最多 20 个标签。 |
响应结构
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
实验的唯一标识符。 |
搜索实验
端点 |
HTTP 方法 |
|---|---|
|
|
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
max_results |
|
期望的最大实验数量。服务器可能会选择默认的 `max_results` 值。所有服务器保证支持至少 1,000 的 `max_results` 阈值,但也可能支持更多。建议调用此端点的人显式传递 max_results 并利用 page_token 来遍历实验。 |
page_token |
|
指示要获取的实验页面的令牌。 |
filter |
|
基于实验属性和标签的过滤表达式,允许返回实验的子集。语法是 SQL 的一个子集,支持将属性或标签与常量之间的二元操作进行 AND 连接。示例:`name LIKE 'test-%' AND tags.key = 'value'`。您可以使用双引号或反引号选择具有特殊字符(连字符、空格、句点等)的列。示例:`tags."extra-key" = 'value'` 或 `tags.`extra-key` = 'value'`。支持的运算符包括 `=`、`!=`、`LIKE` 和 `ILIKE`。 |
order_by |
STRING 的数组 |
用于对搜索结果排序的列列表,可以包括实验名称和 ID,并带有可选的“DESC”或“ASC”注解,默认为“ASC”。平局由实验 ID DESC 处理。 |
view_type |
要返回的实验类型的限定符。如果未指定,仅返回活动实验。 |
响应结构
字段名称 |
类型 |
描述 |
|---|---|---|
experiments |
Experiment 的数组 |
符合搜索条件的实验 |
next_page_token |
|
可用于检索下一页实验的令牌。空令牌表示没有更多实验可供检索。 |
获取实验
端点 |
HTTP 方法 |
|---|---|
|
|
获取实验的元数据。此方法适用于已删除的实验。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
关联实验的 ID。此字段为必需。 |
通过名称获取实验
端点 |
HTTP 方法 |
|---|---|
|
|
获取实验的元数据。
此端点将返回已删除的实验,但如果活动实验和已删除实验具有相同的名称,则优先返回活动实验。如果多个已删除实验具有相同的名称,API 将返回其中一个。
如果不存在指定名称的实验,抛出 `RESOURCE_DOES_NOT_EXIST` 异常。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_name |
|
关联实验的名称。此字段为必需。 |
删除实验
端点 |
HTTP 方法 |
|---|---|
|
|
将实验及其关联的元数据、运行、指标、参数和标签标记为删除。如果实验使用 FileStore,与实验关联的制品也会被删除。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
关联实验的 ID。此字段为必需。 |
恢复实验
端点 |
HTTP 方法 |
|---|---|
|
|
恢复已标记为删除的实验。这也将恢复关联的元数据、运行、指标、参数和标签。如果实验使用 FileStore,与实验关联的底层制品也会被恢复。
如果实验从未创建或已被永久删除,抛出 `RESOURCE_DOES_NOT_EXIST` 异常。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
关联实验的 ID。此字段为必需。 |
更新实验
端点 |
HTTP 方法 |
|---|---|
|
|
更新实验元数据。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
关联实验的 ID。此字段为必需。 |
new_name |
|
如果提供,实验名称将更改为新名称。新名称必须是唯一的。 |
设置实验标签
端点 |
HTTP 方法 |
|---|---|
|
|
在实验上设置标签。实验标签是可以更新的元数据。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
要记录标签的实验 ID。必须提供。此字段为必需。 |
key |
|
标签名称。最大大小取决于存储后端。所有存储后端都保证支持最大 250 字节的键值。此字段为必需。 |
value |
|
要记录的标签的字符串值。最大大小取决于存储后端。所有存储后端都保证支持最大 5000 字节的键值。此字段为必需。 |
删除实验标签
端点 |
HTTP 方法 |
|---|---|
|
|
删除实验上的标签。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
记录该标签的实验 ID。必须提供。此字段为必需。 |
key |
|
标签名称。最大大小为 255 字节。必须提供。此字段为必需。 |
创建运行
端点 |
HTTP 方法 |
|---|---|
|
|
在实验内创建一个新运行。运行通常是机器学习或数据 ETL 管道的单次执行。MLflow 使用运行来跟踪与单次执行关联的 Param、Metric 和 RunTag。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
关联实验的 ID。 |
user_id |
|
执行运行的用户的 ID。此字段自 MLflow 1.0 起已弃用,并将在未来的 MLflow 版本中删除。请改用“mlflow.user”标签。 |
run_name |
|
运行的名称。 |
start_time |
|
运行开始时的 Unix 时间戳(毫秒)。 |
tags |
RunTag 的数组 |
运行的其他元数据。 |
更新运行
端点 |
HTTP 方法 |
|---|---|
|
|
更新运行元数据。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
run_id |
|
要更新的运行 ID。必须提供。 |
run_uuid |
|
[已弃用,请改用 run_id] 要更新的运行 ID。此字段将在未来的 MLflow 版本中删除。 |
status |
运行的更新状态。 |
|
end_time |
|
运行结束时的 Unix 时间戳(毫秒)。 |
run_name |
|
运行的更新名称。 |
删除运行
端点 |
HTTP 方法 |
|---|---|
|
|
将运行标记为删除。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
run_id |
|
要删除的运行 ID。此字段为必需。 |
恢复运行
端点 |
HTTP 方法 |
|---|---|
|
|
恢复已删除的运行。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
run_id |
|
要恢复的运行 ID。此字段为必需。 |
获取运行
端点 |
HTTP 方法 |
|---|---|
|
|
获取运行的元数据、指标、参数和标签。如果为运行记录了多个具有相同键的指标,则仅返回带有最新时间戳的值。如果带有最新时间戳的值有多个,则返回这些值中的最大值。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
run_id |
|
要获取的运行 ID。必须提供。 |
run_uuid |
|
[已弃用,请改用 run_id] 要获取的运行 ID。此字段将在未来的 MLflow 版本中删除。 |
搜索运行
端点 |
HTTP 方法 |
|---|---|
|
|
搜索满足表达式的运行。搜索表达式可以使用 Metric 和 Param 键。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_ids |
STRING 的数组 |
要搜索的实验 ID 列表。 |
filter |
|
基于参数、指标和标签的过滤表达式,允许返回运行的子集。语法是 SQL 的一个子集,支持将参数、指标或标签与常量之间的二元操作进行 AND 连接。示例:`metrics.rmse < 1 and params.model_class = 'LogisticRegression'`。您可以使用双引号选择具有特殊字符(连字符、空格、句点等)的列:`metrics."model class" = 'LinearRegression' and tags."user- name" = 'Tomas'`。支持的运算符包括 `=`、`!=`、`>`、`>=`、`<` 和 `<=`。 |
run_view_type |
显示仅活动的、仅已删除的还是所有运行。默认为仅活动运行。 |
|
max_results |
|
期望的最大运行数量。如果未指定,默认为 1000。所有服务器保证支持至少 50,000 的 `max_results` 阈值,但也可能支持更多。建议调用此端点的人显式传递 max_results 并利用 page_token 来遍历实验。 |
order_by |
STRING 的数组 |
要排序的列列表,包括属性、参数、指标和标签,并带有可选的“DESC”或“ASC”注解,默认为“ASC”。示例:[“params.input DESC”, “metrics.alpha ASC”, “metrics.rmse”]。平局由 start_time DESC 处理,如果开始时间相同,则由 run_id 处理(如果未提供 order_by,这也是默认的排序标准)。 |
page_token |
|
响应结构
字段名称 |
类型 |
描述 |
|---|---|---|
runs |
Run 的数组 |
符合搜索条件的运行。 |
next_page_token |
|
记录指标
端点 |
HTTP 方法 |
|---|---|
|
|
为运行记录指标。指标是带有关联时间戳的键值对(字符串键,浮点值)。例如代表 ML 模型准确性的各种指标。指标可以被多次记录。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
run_id |
|
要记录指标的运行 ID。必须提供。 |
run_uuid |
|
[已弃用,请改用 run_id] 要记录指标的运行 ID。此字段将在未来的 MLflow 版本中删除。 |
key |
|
指标名称。此字段为必需。 |
value |
|
正在记录的指标的双精度值。此字段为必需。 |
timestamp |
|
记录指标时的 Unix 时间戳(毫秒)。此字段为必需。 |
step |
|
记录指标时的步数 |
model_id |
|
如果适用,与指标关联的记录模型 ID |
dataset_name |
|
与指标关联的数据集名称。例如“my.uc.table@2”、“nyc-taxi-dataset”、“fantastic-elk-3” |
dataset_digest |
|
与指标关联的数据集的摘要,例如数据集的 md5 哈希值,用于在同名数据集中唯一标识它。 |
记录参数
端点 |
HTTP 方法 |
|---|---|
|
|
记录运行中使用的参数。参数是键值对(字符串键,字符串值)。例如 ML 模型训练中使用的超参数,以及 ETL 管道中使用的常量日期和值。参数在一次运行中只能记录一次。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
run_id |
|
要记录参数的运行 ID。必须提供。 |
run_uuid |
|
[已弃用,请改用 run_id] 要记录参数的运行 ID。此字段将在未来的 MLflow 版本中删除。 |
key |
|
参数名称。最大大小为 255 字节。此字段为必需。 |
value |
|
正在记录的参数的字符串值。最大大小为 6000 字节。此字段为必需。 |
批量记录
端点 |
HTTP 方法 |
|---|---|
|
|
批量记录运行的指标、参数和标签。如果任何数据无法持久化,服务器将响应错误(非 200 状态码)。如果发生错误(由于内部服务器错误或无效请求),可能会写入部分数据。
您可以交替写入指标、参数和标签,但在特定实体类型内,保证遵循请求正文中指定的顺序。也就是说,对于像这样的 API 请求
{
"run_id": "2a14ed5c6a87499199e0106c3501eab8",
"metrics": [
{"key": "mae", "value": 2.5, "timestamp": 1552550804},
{"key": "rmse", "value": 2.7, "timestamp": 1552550804},
],
"params": [
{"key": "model_class", "value": "LogisticRegression"},
]
}
服务器保证在“mae”之后写入指标“rmse”,尽管它可能在两个指标之前、在“mae”之后或在两个指标之后写入参数“model_class”。
指标、参数和标签的覆盖行为如下
指标:指标值从不被覆盖。记录指标(键、值、时间戳)会追加到具有指定键的指标值集中。
标签:标签值可以被后续对同一标签键的写入覆盖。也就是说,如果同一个 API 请求中提供了多个具有相同键的标签值,则会写入最后提供的标签值。允许记录相同的标签(键、值)——也就是说,记录标签是幂等的。
参数:一旦写入,参数值就不能更改(尝试覆盖参数值将导致错误)。但是,允许记录相同的参数(键、值)——也就是说,记录参数是幂等的。
请求限制
单个 JSON 序列化的 API 请求大小最大可为 1 MB,并包含
总共不超过 1000 个指标、参数和标签
最多 1000 个指标
最多 100 个参数
最多 100 个标签
例如,一个有效的请求可能包含 900 个指标、50 个参数和 50 个标签,但记录 900 个指标、50 个参数和 51 个标签是无效的。以下限制也适用于指标、参数和标签的键与值
指标、参数和标签的键长度最多可为 250 个字符
参数和标签的值长度最多可为 250 个字符
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
run_id |
|
要记录的运行 ID |
metrics |
Metric 的数组 |
要记录的指标。单个请求最多可包含 1000 个指标,并且指标、参数和标签总数最多为 1000 个。 |
params |
Param 的数组 |
要记录的参数。单个请求最多可包含 100 个参数,并且指标、参数和标签总数最多为 1000 个。 |
tags |
RunTag 的数组 |
要记录的标签。单个请求最多可包含 100 个标签,并且指标、参数和标签总数最多为 1000 个。 |
记录模型
端点 |
HTTP 方法 |
|---|---|
|
|
注意
实验性功能:此 API 可能会在未来的版本中更改或删除,恕不另行通知。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
run_id |
|
要记录的运行 ID |
model_json |
|
JSON 格式的 MLmodel 文件。 |
记录输入
端点 |
HTTP 方法 |
|---|---|
|
|
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
run_id |
|
要记录的运行 ID。此字段为必需。 |
datasets |
DatasetInput 的数组 |
数据集输入 |
models |
ModelInput 的数组 |
模型输入(目前未记录在 LoggedModels 私有预览版中) |
设置标签
端点 |
HTTP 方法 |
|---|---|
|
|
在运行上设置标签。标签是运行的元数据,可以在运行期间和运行完成后进行更新。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
run_id |
|
要记录标签的运行 ID。必须提供。 |
run_uuid |
|
[已弃用,请改用 run_id] 要记录标签的运行 ID。此字段将在未来的 MLflow 版本中删除。 |
key |
|
标签名称。最大大小取决于存储后端。所有存储后端都保证支持最大 250 字节的键值。此字段为必需。 |
value |
|
要记录的标签的字符串值。最大大小取决于存储后端。所有存储后端都保证支持最大 5000 字节的键值。此字段为必需。 |
删除标签
端点 |
HTTP 方法 |
|---|---|
|
|
删除运行上的标签。标签是运行的元数据,可以在运行期间和运行完成后进行更新。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
run_id |
|
记录该标签的运行 ID。必须提供。此字段为必需。 |
key |
|
标签名称。最大大小为 255 字节。必须提供。此字段为必需。 |
获取指标历史
端点 |
HTTP 方法 |
|---|---|
|
|
获取指定运行中特定指标的所有值列表。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
run_id |
|
要获取指标值的运行 ID。必须提供。 |
run_uuid |
|
[已弃用,请改用 run_id] 要获取指标值的运行 ID。此字段将在未来的 MLflow 版本中删除。 |
metric_key |
|
指标名称。此字段为必需。 |
page_token |
|
指示要获取的指标历史页面的令牌 |
max_results |
|
单次调用中要返回的运行指标的记录实例的最大数量。后端服务器可能会根据性能要求限制 `max_results` 的值。未指定此值的请求将作为非分页查询处理,其中运行内给定指标的所有历史值将在单个响应中返回。 |
响应结构
字段名称 |
类型 |
描述 |
|---|---|---|
metrics |
Metric 的数组 |
该指标的所有记录值。 |
next_page_token |
|
可用于查询下一页指标历史值的令牌。缺少令牌表示没有额外的指标可供获取。 |
列出制品
端点 |
HTTP 方法 |
|---|---|
|
|
列出运行的制品。接受一个可选的 `artifact_path` 前缀,如果指定,响应仅包含具有指定前缀的制品。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
run_id |
|
要列出其制品的运行 ID。必须提供。 |
run_uuid |
|
[已弃用,请改用 run_id] 要列出其制品的运行 ID。此字段将在未来的 MLflow 版本中删除。 |
path |
|
过滤匹配此路径的制品(从根制品目录开始的相对路径)。 |
page_token |
|
指示要获取的制品结果页面的令牌 |
响应结构
字段名称 |
类型 |
描述 |
|---|---|---|
root_uri |
|
运行的根制品目录。 |
files |
FileInfo 的数组 |
制品的路径和元数据。 |
next_page_token |
|
可用于检索下一页制品结果的令牌 |
注册评分器
端点 |
HTTP 方法 |
|---|---|
|
|
为实验注册评分器。
请求结构
为实验注册评分器。
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
实验 ID。 |
name |
|
评分器名称。 |
serialized_scorer |
|
序列化的评分器字符串(JSON)。 |
响应结构
字段名称 |
类型 |
描述 |
|---|---|---|
version |
|
评分器的新版本号。 |
scorer_id |
|
评分器的唯一标识符。 |
experiment_id |
|
实验 ID(与请求相同)。 |
name |
|
评分器名称(与请求相同)。 |
serialized_scorer |
|
序列化的评分器字符串(与请求相同)。 |
creation_time |
|
评分器版本的创建时间(自纪元以来的毫秒数)。 |
列出评分器
端点 |
HTTP 方法 |
|---|---|
|
|
列出实验的所有评分器。
请求结构
列出所有评分器,可选地限定在一个实验内。
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
实验 ID。如果为空,则返回活动工作区中所有实验的评分器(由管理员 UI 评分器选择器使用)。 |
响应结构
字段名称 |
类型 |
描述 |
|---|---|---|
scorers |
Scorer 的数组 |
评分器实体列表(每个评分器名称的最新版本)。 |
列出评分器版本
端点 |
HTTP 方法 |
|---|---|
|
|
列出实验特定评分器的所有版本。
请求结构
列出实验特定评分器的所有版本。
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
实验 ID。 |
name |
|
评分器名称。 |
响应结构
字段名称 |
类型 |
描述 |
|---|---|---|
scorers |
Scorer 的数组 |
该评分器所有版本的评分器实体列表。 |
获取评分器
端点 |
HTTP 方法 |
|---|---|
|
|
获取实验的特定评分器。
请求结构
获取实验的特定评分器。
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
实验 ID。 |
name |
|
评分器名称。 |
version |
|
评分器版本。如果未指定,则返回最大版本的评分器。 |
删除评分器
端点 |
HTTP 方法 |
|---|---|
|
|
删除实验的评分器。
请求结构
删除实验的评分器。
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
实验 ID。 |
name |
|
评分器名称。 |
version |
|
要删除的评分器版本。如果未指定,则删除所有版本。 |
创建网关密钥
端点 |
HTTP 方法 |
|---|---|
|
|
为 LLM 提供程序认证创建一个新的加密密钥
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
secret_name |
|
密钥的用户友好名称(必须唯一) |
secret_value |
SecretValueEntry 的数组 |
要加密为键值对的密钥值。对于简单 API 密钥:{“api_key”: “sk-xxx”} 对于复合凭证:{“aws_access_key_id”: “…”, “aws_secret_access_key”: “…”} |
provider |
|
可选的 LLM 提供程序(例如,“openai”,“anthropic”) |
auth_config |
AuthConfigEntry 的数组 |
可选的提供程序特定认证配置。对于多认证提供程序,包含“auth_mode”键(例如 {“auth_mode”: “access_keys”, “aws_region_name”: “us-east-1”}) |
created_by |
|
创建者的用户名 |
获取网关密钥信息
端点 |
HTTP 方法 |
|---|---|
|
|
获取关于密钥的元数据(不包含加密值)
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
secret_id |
|
必须提供 secret_id 或 secret_name |
secret_name |
|
更新网关密钥
端点 |
HTTP 方法 |
|---|---|
|
|
更新现有密钥的值或认证配置
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
secret_id |
|
要更新的密钥 ID |
secret_value |
SecretValueEntry 的数组 |
用于密钥轮换的可选新密钥值(作为键值对,空映射 = 无更改)。对于简单 API 密钥:{“api_key”: “sk-xxx”} 对于复合凭证:{“aws_access_key_id”: “…”, “aws_secret_access_key”: “…”} |
auth_config |
AuthConfigEntry 的数组 |
可选的新认证配置。对于多认证提供程序,包含“auth_mode”键(例如 {“auth_mode”: “access_keys”, “aws_region_name”: “us-east-1”}) |
updated_by |
|
更新者的用户名 |
删除网关密钥
端点 |
HTTP 方法 |
|---|---|
|
|
删除密钥
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
secret_id |
|
要删除的密钥 ID |
列出网关密钥
端点 |
HTTP 方法 |
|---|---|
|
|
按提供程序进行可选过滤,列出所有密钥
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
provider |
|
按提供程序过滤(可选,例如“openai”,“anthropic”) |
响应结构
字段名称 |
类型 |
描述 |
|---|---|---|
secrets |
GatewaySecretInfo 的数组 |
密钥元数据列表(不包含加密值) |
创建网关模型定义
端点 |
HTTP 方法 |
|---|---|
|
|
创建可重用的模型定义
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
name |
|
模型定义的用户友好名称(必须唯一) |
secret_id |
|
包含认证凭证的密钥 ID |
provider |
|
LLM 提供程序(例如,“openai”,“anthropic”) |
model_name |
|
提供程序特定的模型标识符(例如,“gpt-4o”,“claude-3-5-sonnet”) |
created_by |
|
创建者的用户名 |
获取网关模型定义
端点 |
HTTP 方法 |
|---|---|
|
|
按 ID 获取模型定义
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
model_definition_id |
|
要检索的模型定义 ID |
列出网关模型定义
端点 |
HTTP 方法 |
|---|---|
|
|
列出带有可选过滤器的所有模型定义
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
provider |
|
按提供程序过滤(可选) |
secret_id |
|
按密钥 ID 过滤(可选) |
响应结构
字段名称 |
类型 |
描述 |
|---|---|---|
model_definitions |
GatewayModelDefinition 的数组 |
模型定义列表 |
更新网关模型定义
端点 |
HTTP 方法 |
|---|---|
|
|
更新模型定义
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
model_definition_id |
|
要更新的模型定义 ID |
name |
|
可选的新名称 |
secret_id |
|
可选的新密钥 ID |
model_name |
|
可选的新模型名称 |
updated_by |
|
更新者的用户名 |
provider |
|
可选的新提供程序 |
删除网关模型定义
端点 |
HTTP 方法 |
|---|---|
|
|
删除模型定义(如果任何端点正在使用,则失败)
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
model_definition_id |
|
要删除的模型定义 ID(如果任何端点正在使用,则失败) |
创建网关端点
端点 |
HTTP 方法 |
|---|---|
|
|
创建带有模型配置的新端点
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
name |
|
端点的可选用户友好名称 |
model_configs |
GatewayEndpointModelConfig 的数组 |
模型配置列表 |
created_by |
|
创建者的用户名 |
routing_strategy |
端点的可选路由策略 |
|
fallback_config |
可选的回退配置(包括策略、最大尝试次数) |
|
experiment_id |
|
用于追踪的可选实验 ID。仅在 usage_tracking 为 true 时使用。如果未提供且 usage_tracking 为 true,则会自动创建实验。 |
usage_tracking |
|
是否为此端点启用使用追踪。默认为 false。为 true 时,将记录端点调用的追踪。 |
获取网关端点
端点 |
HTTP 方法 |
|---|---|
|
|
获取包含所有模型配置的端点详情
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
endpoint_id |
|
必须提供 endpoint_id 或 name |
name |
|
更新网关端点
端点 |
HTTP 方法 |
|---|---|
|
|
更新端点名称
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
endpoint_id |
|
要更新的端点 ID |
name |
|
端点的可选新名称 |
updated_by |
|
更新者的用户名 |
model_configs |
GatewayEndpointModelConfig 的数组 |
模型配置的可选新列表(替换所有现有的模型链接) |
routing_strategy |
端点的可选新路由策略 |
|
fallback_config |
可选的回退配置(包括策略、最大尝试次数) |
|
experiment_id |
|
用于追踪的可选实验 ID。仅在 usage_tracking 为 true 时使用。 |
usage_tracking |
|
是否为此端点启用使用追踪。设为 true 时,将记录端点调用的追踪。设为 false 时,禁用使用追踪并清除 experiment_id。 |
删除网关端点
端点 |
HTTP 方法 |
|---|---|
|
|
删除端点及其所有模型配置
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
endpoint_id |
|
要删除的端点 ID |
列出网关端点
端点 |
HTTP 方法 |
|---|---|
|
|
按提供程序或密钥进行可选过滤,列出端点
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
provider |
|
按提供程序过滤(可选) |
secret_id |
|
按密钥 ID 过滤(可选) |
响应结构
字段名称 |
类型 |
描述 |
|---|---|---|
endpoints |
GatewayEndpoint 的数组 |
带有模型配置的端点列表 |
将模型附加到端点
端点 |
HTTP 方法 |
|---|---|
|
|
将现有模型定义附加到端点
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
endpoint_id |
|
要将模型附加到的端点 ID |
model_config |
要附加的模型配置 |
|
created_by |
|
创建者的用户名 |
从端点分离模型
端点 |
HTTP 方法 |
|---|---|
|
|
从端点分离模型定义(不删除模型定义)
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
endpoint_id |
|
端点 ID |
model_definition_id |
|
要分离的模型定义 ID |
创建端点绑定
端点 |
HTTP 方法 |
|---|---|
|
|
在端点和 MLflow 资源之间创建绑定
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
endpoint_id |
|
要绑定的端点 ID |
resource_type |
|
MLflow 资源类型 |
resource_id |
|
资源实例 ID |
created_by |
|
创建者的用户名 |
删除端点绑定
端点 |
HTTP 方法 |
|---|---|
|
|
删除端点和资源之间的绑定
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
endpoint_id |
|
端点 ID |
resource_type |
|
绑定到端点的资源类型 |
resource_id |
|
资源 ID |
列出端点绑定
端点 |
HTTP 方法 |
|---|---|
|
|
列出端点的所有绑定
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
endpoint_id |
|
要列出绑定的端点 ID |
resource_type |
|
要过滤绑定的资源类型(例如“scorer”) |
resource_id |
|
要过滤绑定的资源 ID |
响应结构
字段名称 |
类型 |
描述 |
|---|---|---|
bindings |
GatewayEndpointBinding 的数组 |
端点绑定列表 |
设置网关端点标签
端点 |
HTTP 方法 |
|---|---|
|
|
在端点上设置标签
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
endpoint_id |
|
要设置标签的端点 ID |
key |
|
要设置的标签键 |
value |
|
要设置的标签值 |
删除网关端点标签
端点 |
HTTP 方法 |
|---|---|
|
|
从端点删除标签
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
endpoint_id |
|
要删除标签的端点 ID |
key |
|
要删除的标签键 |
创建提示优化任务
端点 |
HTTP 方法 |
|---|---|
|
|
创建新的提示优化任务。此端点使用指定的配置启动优化运行。优化过程异步运行,可以通过 getPromptOptimizationJob 进行监控。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
用于跟踪优化任务的 MLflow 实验 ID。 |
source_prompt_uri |
|
要优化的源提示 URI(例如“prompts:/my-prompt/1”)。 |
config |
优化任务的配置。 |
|
tags |
PromptOptimizationJobTag 的数组 |
优化任务的可选标签。 |
获取提示优化任务
端点 |
HTTP 方法 |
|---|---|
|
|
获取提示优化任务的详情和状态。返回任务配置、当前状态、进度统计,如果优化已完成,则返回最佳提示。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
job_id |
|
优化任务的唯一标识符(与 run_id 相同)。 |
搜索提示优化任务
端点 |
HTTP 方法 |
|---|---|
|
|
搜索提示优化任务。返回匹配指定过滤器的优化任务列表。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
要搜索优化任务的 MLflow 实验 ID。 |
响应结构
字段名称 |
类型 |
描述 |
|---|---|---|
jobs |
PromptOptimizationJob 的数组 |
优化任务列表。 |
取消提示优化任务
端点 |
HTTP 方法 |
|---|---|
|
|
取消正在进行的提示优化任务。如果任务已完成或已取消,此操作无效。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
job_id |
|
要取消的优化任务的唯一标识符。 |
删除提示优化任务
端点 |
HTTP 方法 |
|---|---|
|
|
删除提示优化任务及其关联数据。这将永久移除任务及所有相关信息。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
job_id |
|
要删除的优化任务的唯一标识符。 |
创建预签名上传 URL
端点 |
HTTP 方法 |
|---|---|
|
|
生成一个预签名 URL,用于将制品直接上传到云存储。服务器使用其自己的凭证对 URL 进行签名,使客户端无需直接的云存储写入权限即可上传制品。
由外部制品存储库插件使用(例如 https://github.com/aws/sagemaker-mlflow)。
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
run_id |
|
拥有该制品的运行 ID。必须提供。 |
path |
|
运行制品目录内的相对路径(例如“models/model.pkl”)。必须提供。 |
expiration |
|
URL 过期时间(秒,默认:900)。 |
响应结构
字段名称 |
类型 |
描述 |
|---|---|---|
presigned_url |
|
用于直接制品上传的预签名 URL。 |
headers |
HeadersEntry 的数组 |
上传请求所需的请求头(例如 Content-Type)。 |
创建预算策略
端点 |
HTTP 方法 |
|---|---|
|
|
创建新的预算策略
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
budget_unit |
||
budget_amount |
|
|
duration |
||
target_scope |
||
budget_action |
||
created_by |
|
获取预算策略
端点 |
HTTP 方法 |
|---|---|
|
|
通过 ID 获取预算策略
更新预算策略
端点 |
HTTP 方法 |
|---|---|
|
|
更新预算策略
创建护栏
端点 |
HTTP 方法 |
|---|---|
|
|
创建一个由评分器支持的新护栏
更新端点护栏配置
端点 |
HTTP 方法 |
|---|---|
|
|
创建已注册模型 (RegisteredModel)
端点 |
HTTP 方法 |
|---|---|
|
|
如果给定名称的已注册模型已存在,则抛出 RESOURCE_ALREADY_EXISTS 错误。
获取已注册模型
端点 |
HTTP 方法 |
|---|---|
|
|
重命名已注册模型
端点 |
HTTP 方法 |
|---|---|
|
|
更新已注册模型
端点 |
HTTP 方法 |
|---|---|
|
|
删除已注册模型
端点 |
HTTP 方法 |
|---|---|
|
|
搜索已注册模型
端点 |
HTTP 方法 |
|---|---|
|
|
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
filter |
|
字符串过滤条件,如“name LIKE ‘my-model-name’”。在后端自动解释为“name LIKE ‘%my-model-name%’”。单个布尔条件,字符串值需用单引号包裹。 |
max_results |
|
期望返回的最大模型数量。默认为 100。最大阈值为 1000。 |
order_by |
STRING 的数组 |
用于对搜索结果排序的列列表,可包括模型名称和最后更新时间戳,带有可选的“DESC”或“ASC”注解(默认为“ASC”)。平局将按模型名称 ASC 排序。 |
page_token |
|
分页令牌,用于根据之前的搜索查询跳转到下一页。 |
获取最新模型版本
端点 |
HTTP 方法 |
|---|---|
|
|
响应结构
字段名称 |
类型 |
描述 |
|---|---|---|
model_versions |
ModelVersion 数组 |
每个请求阶段的最新版本模型。仅返回当前状态为 |
创建模型版本 (ModelVersion)
端点 |
HTTP 方法 |
|---|---|
|
|
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
name |
|
以此名称注册模型。此字段为必填项。 |
source |
|
指示模型工件位置的 URI。此字段为必填项。 |
run_id |
|
用于关联的 MLflow 运行 ID,如果 |
tags |
模型版本的附加元数据。 |
|
run_link |
|
MLflow 运行链接 - 这是生成此模型版本的运行的确切链接,可能托管在另一个 MLflow 实例上。 |
description |
|
模型版本的可选描述。 |
model_id |
|
用于将已注册模型链接到已记录源模型的模型版本可选 model_id |
获取模型版本
端点 |
HTTP 方法 |
|---|---|
|
|
更新模型版本
端点 |
HTTP 方法 |
|---|---|
|
|
删除模型版本
端点 |
HTTP 方法 |
|---|---|
|
|
搜索模型版本
端点 |
HTTP 方法 |
|---|---|
|
|
请求结构
字段名称 |
类型 |
描述 |
|---|---|---|
filter |
|
字符串过滤条件,如“name=’my-model-name’”。必须是单个布尔条件,字符串值需用单引号包裹。 |
max_results |
|
期望返回的最大模型数量。最大阈值为 20 万。后端可以选择更小的默认值和最大阈值。 |
order_by |
STRING 的数组 |
用于排序的列列表,包括模型名称、版本、阶段,带有可选的“DESC”或“ASC”注解(默认为“ASC”)。平局按最后阶段转换时间戳排序,其次是名称 ASC,最后是版本 DESC。 |
page_token |
|
分页令牌,用于根据之前的搜索查询跳转到下一页。 |
响应结构
字段名称 |
类型 |
描述 |
|---|---|---|
model_versions |
ModelVersion 数组 |
符合搜索条件的模型 |
next_page_token |
|
分页令牌,用于为相同的搜索查询请求下一页模型。 |
获取模型版本工件的下载 URI
端点 |
HTTP 方法 |
|---|---|
|
|
转换模型版本阶段
端点 |
HTTP 方法 |
|---|---|
|
|
设置已注册模型标签
端点 |
HTTP 方法 |
|---|---|
|
|
设置模型版本标签
端点 |
HTTP 方法 |
|---|---|
|
|
删除已注册模型标签
端点 |
HTTP 方法 |
|---|---|
|
|
删除模型版本标签
端点 |
HTTP 方法 |
|---|---|
|
|
设置已注册模型别名
端点 |
HTTP 方法 |
|---|---|
|
|
删除已注册模型别名
端点 |
HTTP 方法 |
|---|---|
|
|
按别名获取模型版本
端点 |
HTTP 方法 |
|---|---|
|
|
创建 Webhook
端点 |
HTTP 方法 |
|---|---|
|
|
请求结构
创建 Webhook 请求
字段名称 |
类型 |
描述 |
|---|---|---|
name |
|
Webhook 名称。此字段为必填项。 |
description |
|
Webhook 的可选描述 |
url |
|
发送 Webhook 事件的 URL。此字段为必填项。 |
events |
WebhookEvent 数组 |
要订阅的事件列表。此字段为必填项。 |
secret |
|
HMAC 签名验证的密钥 |
status |
初始状态(若未指定,默认为 ACTIVE) |
列出 Webhooks
端点 |
HTTP 方法 |
|---|---|
|
|
获取 Webhook
端点 |
HTTP 方法 |
|---|---|
|
|
更新 Webhook
端点 |
HTTP 方法 |
|---|---|
|
|
请求结构
更新 Webhook 请求
字段名称 |
类型 |
描述 |
|---|---|---|
webhook_id |
|
要更新的 Webhook ID。此字段为必填项。 |
name |
|
Webhook 的新名称 |
description |
|
Webhook 的新描述 |
url |
|
Webhook 的新 URL |
events |
WebhookEvent 数组 |
要订阅的新事件列表 |
secret |
|
用于 HMAC 签名的新密钥 |
status |
Webhook 的新状态 |
删除 Webhook
端点 |
HTTP 方法 |
|---|---|
|
|
测试 Webhook
端点 |
HTTP 方法 |
|---|---|
|
|
列出工件
端点 |
HTTP 方法 |
|---|---|
|
|
创建工件分段上传
端点 |
HTTP 方法 |
|---|---|
|
|
完成工件分段上传
端点 |
HTTP 方法 |
|---|---|
|
|
中止工件分段上传
端点 |
HTTP 方法 |
|---|---|
|
|
获取分段下载的预签名 URL
端点 |
HTTP 方法 |
|---|---|
|
|
响应结构
字段名称 |
类型 |
描述 |
|---|---|---|
url |
|
用于直接从云存储下载工件的预签名 URL。 |
headers |
HeadersEntry 数组 |
下载请求中必须包含的可选标头。 |
file_size |
|
可选的文件大小(字节)。 |
数据结构
AddDatasetToExperiments
字段名称 |
类型 |
描述 |
|---|---|---|
dataset_id |
|
要添加到实验的数据集 ID。此字段为必填项。 |
experiment_ids |
STRING 的数组 |
与数据集关联的实验 ID |
Assessment
轨迹评估的数据和元数据。
字段名称 |
类型 |
描述 |
|---|---|---|
assessment_id |
|
评估的唯一 ID。注意:这不是通过“validate_required”标记为必填的字段,因为该消息是在创建新评估的上下文中使用,此时 ID 未知。 |
assessment_name |
|
评估名称。名称不得包含“.”。此字段为必填项。 |
trace_id |
|
此评估关联的轨迹 ID。 |
span_id |
|
如果评估是针对特定范围 (span) 的,则为该范围的 ID(可选)。 |
source |
此评估的来源。 |
|
create_time |
|
此评估的创建时间。 |
last_update_time |
|
此评估的最后更新时间。 |
rationale(理由) |
|
评估的理由。 |
metadata |
描述评估并存储额外信息的附加元数据,例如块相关性 chunk_index。此元数据必须是 JSON 可序列化的。 |
|
overrides |
|
此评估所覆盖的评估 ID。 |
valid |
|
此评估是否有效(即尚未被取代)。默认为 true,若创建了新的替代评估,则设为 false。 |
|
如果是 |
AssessmentError
字段名称 |
类型 |
描述 |
|---|---|---|
error_code |
|
发生错误时评估的值。 |
error_message |
|
|
stack_trace |
|
错误的堆栈跟踪。截断为 1000 个字符,以避免 TraceInfo 过大。 |
AssessmentSource
字段名称 |
类型 |
描述 |
|---|---|---|
source_type |
源类型。此字段为必填项。 |
|
source_id |
|
源标识符。例如:人工 -> 用户名;LLM 判断 -> 判断来源(Databricks 或自定义);代码 -> 为空。此字段为必填项。 |
BudgetDuration
固定窗口持续时间:定义预算窗口时长的 (unit, value) 对
字段名称 |
类型 |
描述 |
|---|---|---|
unit |
时间单位 (MINUTES, HOURS, DAYS, WEEKS, MONTHS) |
|
value |
|
每个窗口的单位数 |
BudgetWindow
字段名称 |
类型 |
描述 |
|---|---|---|
budget_policy_id |
|
|
window_start_ms |
|
|
window_end_ms |
|
|
current_spend |
|
CalculateTraceFilterCorrelation
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_ids |
STRING 的数组 |
要搜索的实验 ID 列表。 |
filter_string1 |
|
第一个过滤条件(例如,“span.type = ‘LLM’”)。 |
filter_string2 |
|
第二个过滤条件(例如,“feedback.quality > 0.8”)。 |
base_filter |
|
可选的基础过滤器,filter1 和 filter2 都在其之上进行测试(例如,时间窗口使用 ‘request_time > … and request_time < …’)。 |
CreateDataset
字段名称 |
类型 |
描述 |
|---|---|---|
name |
|
数据集名称。此字段为必填项。 |
experiment_ids |
STRING 的数组 |
关联的实验 ID。若未提供,默认为当前活动的实验。 |
source_type |
源类型 |
|
source |
|
源信息 |
schema |
|
模式信息 (JSON) |
profile |
|
概况信息 (JSON) |
created_by |
|
创建数据集的用户 |
tags |
|
在数据集上设置的标签(键值对的 JSON 字符串) |
CreateLoggedModel
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
关联实验的 ID。此字段为必需。 |
name |
|
模型名称。可选。若未指定,后端将生成一个。 |
model_type |
|
模型类型,例如 “Agent”、“Classifier”、“LLM”。 |
source_run_id |
|
创建此模型的运行的 ID。 |
params |
已记录模型的参数。 |
|
tags |
已记录模型的标签。 |
CreateWorkspace
创建一个新的工作空间。
字段名称 |
类型 |
描述 |
|---|---|---|
name |
|
要创建的工作区名称。此字段为必填项。 |
description |
|
可选的工作区描述。 |
default_artifact_root |
|
可选的默认工件根覆盖,在创建时应用。 |
trace_archival_config |
可选的轨迹归档设置,在创建时应用。 |
Dataset
数据集。表示模型开发过程中用于训练、测试或评估的数据引用。
字段名称 |
类型 |
描述 |
|---|---|---|
name |
|
数据集名称。例如 “my.uc.table@2”、“nyc-taxi-dataset”、“fantastic-elk-3”。此字段为必填项。 |
digest |
|
数据集摘要,例如数据集的 md5 哈希值,在同名数据集中唯一标识它。此字段为必填项。 |
source_type |
|
数据集源类型,例如 ‘databricks-uc-table’、‘DBFS’、‘S3’ 等。此字段为必填项。 |
source |
|
数据集的源信息。注意,如果数据集在使用前经过转换/修改,源可能无法精确重现该数据集。此字段为必填项。 |
schema |
|
数据集的模式。例如,DataFrame 的 MLflow ColSpec JSON、ndarray 的 MLflow TensorSpec JSON 或其他模式格式。 |
profile |
|
数据集概况。数据集的汇总统计信息,例如表中的行数、表中每一列的平均值/标准差/众数或数组中的元素数。 |
Dataset
字段名称 |
类型 |
描述 |
|---|---|---|
dataset_id |
|
数据集的唯一标识符 |
name |
|
数据集名称(用户友好标识符) |
tags |
|
作为 JSON 字符串的标签(元数据的键值对) |
schema |
|
模式信息 (JSON) |
profile |
|
概况信息 (JSON) |
digest |
|
用于完整性检查的数据集摘要 |
created_time |
|
创建时间戳(毫秒) |
last_update_time |
|
最后更新时间戳(毫秒) |
created_by |
|
创建数据集的用户 |
last_updated_by |
|
最后更新数据集的用户 |
experiment_ids |
STRING 的数组 |
关联的实验 ID(从 entity_associations 表填充) |
DatasetInput
DatasetInput。表示数据集和输入标签。
字段名称 |
类型 |
描述 |
|---|---|---|
tags |
InputTag 数组 |
数据集输入的标签列表,例如值设为 “training” 的 “context” 标签 |
dataset |
作为运行输入使用的数据集。此字段为必填项。 |
DatasetRecord
字段名称 |
类型 |
描述 |
|---|---|---|
dataset_record_id |
|
记录的唯一标识符 |
dataset_id |
|
此记录所属的数据集 ID |
inputs |
|
作为 JSON 字符串的输入 |
期望值 (expectations) |
|
作为 JSON 字符串的预期 |
tags |
|
作为 JSON 字符串的标签 |
source |
|
作为 JSON 字符串的源信息 |
source_id |
|
用于快速查找的源 ID(例如 trace_id) |
source_type |
源类型 |
|
created_time |
|
创建时间戳(毫秒) |
last_update_time |
|
最后更新时间戳(毫秒) |
created_by |
|
创建该记录的用户 |
last_updated_by |
|
最后更新该记录的用户 |
输出 |
|
作为 JSON 字符串的输出 |
DatasetSummary
DatasetSummary。表示有关数据集信息的摘要。
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
实验的唯一标识符。此字段为必填项。 |
name |
|
数据集名称。例如 “my.uc.table@2”、“nyc-taxi-dataset”、“fantastic-elk-3”。此字段为必填项。 |
digest |
|
数据集摘要,例如数据集的 md5 哈希值,在同名数据集中唯一标识它。此字段为必填项。 |
context(上下文) |
|
如果为给定数据集设置,则为 “context” 标签的值。 |
DeleteAssessment
根据 trace_id 和 assessment_id 删除评估的请求。删除成功时响应为空。
字段名称 |
类型 |
描述 |
|---|---|---|
trace_id |
|
轨迹 ID。此字段为必填项。 |
assessment_id |
|
评估 ID。此字段为必填项。 |
DeleteDatasetRecords
字段名称 |
类型 |
描述 |
|---|---|---|
dataset_id |
|
要删除记录的数据集 ID。此字段为必填项。 |
dataset_record_ids |
STRING 的数组 |
要删除的数据集记录 ID 列表。 |
DeleteLoggedModelTag
字段名称 |
类型 |
描述 |
|---|---|---|
model_id |
|
要从中删除标签的 LoggedModel ID。此字段为必填项。 |
tag_key |
|
标签键。此字段为必填项。 |
DeleteTraces
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
关联实验的 ID。此字段为必需。 |
max_timestamp_millis |
|
情况 1:基于时间的删除必须指定 max_timestamp_millis 和 max_traces。自 UNIX 纪元以来删除轨迹的最大毫秒时间戳。 |
max_traces |
|
要删除的最大轨迹数。 |
request_ids |
STRING 的数组 |
情况 2:基于 ID 的删除必须指定 request_ids。要删除的一组请求 ID |
DeleteTracesV3
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
关联实验的 ID。此字段为必需。 |
max_timestamp_millis |
|
情况 1:基于时间的删除必须指定 max_timestamp_millis 和 max_traces。自 UNIX 纪元以来删除轨迹的最大毫秒时间戳。 |
max_traces |
|
要删除的最大轨迹数。 |
request_ids |
STRING 的数组 |
情况 2:基于 ID 的删除必须指定 request_ids。要删除的一组请求 ID |
EndTrace
字段名称 |
类型 |
描述 |
|---|---|---|
request_id |
|
要结束的轨迹 ID。 |
timestamp_ms |
|
轨迹结束时的 UNIX 时间戳(毫秒)。 |
status |
正在追踪操作的总体状态(OK、error 等)。 |
|
request_metadata |
关于正在追踪的操作的附加元数据。 |
|
tags |
TraceTag 数组 |
要添加到轨迹的附加标签。 |
Expectation
对模型或代理从轨迹包含的输入中应产生的输出值或指南的期望。
字段名称 |
类型 |
描述 |
|---|---|---|
value |
|
基于期望评估的值。其底层使用 |
serialized_value |
以指定格式序列化为字符串的基于期望的评估值。 |
Experiment
实验
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
实验的唯一标识符。 |
name |
|
用于标识实验的可读名称。 |
artifact_location |
|
存储实验工件的位置。 |
lifecycle_stage |
|
实验的当前生命周期阶段:“active”或“deleted”。已删除的实验不会由 API 返回。 |
last_update_time |
|
最后更新时间 |
creation_time |
|
创建时间 |
tags |
ExperimentTag 的数组 |
标签:附加的元数据键值对。 |
effective_trace_archival_retention |
|
应用更广泛范围和实验覆盖后的有效轨迹归档保留期。 |
workspace |
|
实验的工作区名称(若未启用工作区,则始终为 default)。 |
Feedback
对轨迹中包含的模型/代理输出提供的反馈
字段名称 |
类型 |
描述 |
|---|---|---|
value |
|
基于反馈的评估值。我们使用 google.protobuf.Value 来支持灵活的反馈值模式。支持的初始类型:- 数字值,如整数或浮点数 - 布尔值 - 文本值(可包含用户希望存储的 JSON 文本,但仅能作为文本搜索) - 仅包含字符串的非空列表值 - 目前拒绝其他值,如结构体、非字符串列表等 |
error |
生成反馈时遇到的错误。若值为 null,则必须提供。 |
FileInfo
单个工件文件或目录的元数据。
字段名称 |
类型 |
描述 |
|---|---|---|
path |
|
相对于根工件目录运行的路径。 |
is_dir |
|
该路径是否为目录。 |
file_size |
|
大小(字节)。目录此项未设置。 |
FileInfo
字段名称 |
类型 |
描述 |
|---|---|---|
path |
|
相对于根工件目录运行的路径。 |
is_dir |
|
该路径是否为目录。 |
file_size |
|
大小(字节)。目录此项未设置。 |
FinalizeLoggedModel
字段名称 |
类型 |
描述 |
|---|---|---|
model_id |
|
要最终确定的 LoggedModel ID。此字段为必填项。 |
status |
模型是否准备好使用。此消息中的有效值:ENUM<LOGGED_MODEL_READY, LOGGED_MODEL_UPLOAD_FAILED>(“LOGGED_MODEL_UPLOAD_FAILED”表示在记录模型权重/代理代码时出错)。此字段为必填项。 |
GatewayBudgetPolicy
表示 AI 网关的预算策略
字段名称 |
类型 |
描述 |
|---|---|---|
budget_policy_id |
|
此预算策略的唯一标识符 |
budget_unit |
预算计量单位(例如 USD) |
|
budget_amount |
|
预算限额金额 |
duration |
固定时间窗口(单位+长度对) |
|
target_scope |
预算范围(GLOBAL 或 WORKSPACE) |
|
budget_action |
超出预算时的操作 |
|
created_by |
|
创建策略的用户 ID |
created_at |
|
创建时间戳(毫秒) |
last_updated_by |
|
最后更新策略的用户 ID |
last_updated_at |
|
最后更新时间戳(毫秒) |
GatewayEndpoint
表示 LLM 网关端点的端点实体
字段名称 |
类型 |
描述 |
|---|---|---|
endpoint_id |
|
端点的唯一标识符 |
name |
|
端点的用户友好名称 |
created_at |
|
端点创建时间戳(纪元毫秒) |
last_updated_at |
|
端点最后更新时间戳(纪元毫秒) |
model_mappings |
绑定到此端点的模型映射列表 |
|
created_by |
|
创建端点的用户 ID |
last_updated_by |
|
最后更新端点的用户 ID |
tags |
与端点关联的标签 |
|
routing_strategy |
端点的路由策略 |
|
fallback_config |
故障转移配置(如果 routing_strategy 为 FALLBACK 则填充) |
|
experiment_id |
|
记录此端点轨迹的 MLflow 实验 ID |
usage_tracking |
|
是否为此端点启用使用情况跟踪。为 true 时,如果未提供实验,将自动创建,并记录端点调用的轨迹。 |
GatewayEndpointBinding
端点与 MLflow 资源之间的绑定。使用复合键(endpoint_id, resource_type, resource_id)进行标识。
字段名称 |
类型 |
描述 |
|---|---|---|
endpoint_id |
|
此绑定引用的端点 ID |
resource_type |
|
MLflow 资源类型(例如,“scorer”) |
resource_id |
|
特定资源实例的 ID |
created_at |
|
绑定创建时间戳(纪元毫秒) |
last_updated_at |
|
绑定最后更新时间戳(纪元毫秒) |
created_by |
|
创建绑定的用户 ID |
last_updated_by |
|
最后更新绑定的用户 ID |
display_name |
|
保留字段 8-9 - 已移除 endpoint_name 和 model_mappings(客户端连接)资源的易读显示名称(例如,评分器名称) |
GatewayEndpointModelConfig
附加到端点的模型配置
字段名称 |
类型 |
描述 |
|---|---|---|
model_definition_id |
|
模型定义的 ID |
linkage_type |
链接类型 |
|
weight |
|
流量分配的路由权重 |
fallback_order |
|
故障转移尝试顺序(仅适用于 FALLBACK 链接,PRIMARY 为 NULL) |
GatewayEndpointModelMapping
端点与模型定义之间的映射
字段名称 |
类型 |
描述 |
|---|---|---|
mapping_id |
|
此映射的唯一标识符 |
endpoint_id |
|
端点 ID |
model_definition_id |
|
模型定义的 ID |
model_definition |
完整的模型定义(通过 JOIN 填充) |
|
weight |
|
流量分配的路由权重 |
created_at |
|
映射创建时间戳(纪元毫秒) |
created_by |
|
创建映射的用户 ID |
linkage_type |
链接类型 |
|
fallback_order |
|
故障转移尝试顺序(仅适用于 FALLBACK 链接,PRIMARY 为 NULL) |
GatewayGuardrail
字段名称 |
类型 |
描述 |
|---|---|---|
guardrail_id |
|
|
name |
|
|
scorer |
||
stage |
||
action |
||
action_endpoint_id |
|
该操作所使用的 LLM 的可选网关端点 ID(例如消毒)。 |
created_by |
|
|
created_at |
|
|
last_updated_by |
|
|
last_updated_at |
|
GatewayGuardrailConfig
字段名称 |
类型 |
描述 |
|---|---|---|
endpoint_id |
|
|
guardrail_id |
|
|
execution_order |
|
|
created_by |
|
|
created_at |
|
|
guardrail |
完整的护栏实体,列出配置时填充。 |
GatewayModelDefinition
可在端点间共享的可重用模型定义
字段名称 |
类型 |
描述 |
|---|---|---|
model_definition_id |
|
此模型定义的唯一标识符 |
name |
|
用于标识和重用的用户友好名称 |
secret_id |
|
包含认证凭证的密钥 ID |
secret_name |
|
用于显示目的的密钥名称 |
provider |
|
LLM 提供商(例如 “openai”、“anthropic”、“cohere”、“bedrock”) |
model_name |
|
提供程序特定的模型标识符(例如,“gpt-4o”,“claude-3-5-sonnet”) |
created_at |
|
模型定义创建时间戳(纪元毫秒) |
last_updated_at |
|
模型定义最后更新时间戳(纪元毫秒) |
created_by |
|
创建模型定义的用户 ID |
last_updated_by |
|
最后更新模型定义的用户 ID |
GatewaySecretInfo
密钥元数据实体(不包括解密后的密钥值)
字段名称 |
类型 |
描述 |
|---|---|---|
secret_id |
|
密钥的唯一标识符 (UUID) |
secret_name |
|
密钥的用户友好名称(必须唯一) |
masked_values |
用于显示为键值对的密钥掩码版本。简单 API 密钥:{“api_key”: “sk-…xyz123”} 复合凭据: |
|
created_at |
|
密钥创建时间戳(纪元毫秒) |
last_updated_at |
|
密钥最后更新时间戳(纪元毫秒) |
provider |
|
LLM 提供商标识符(例如 “openai”、“anthropic”、“cohere”) |
created_by |
|
创建密钥的用户 ID |
last_updated_by |
|
最后更新密钥的用户 ID |
auth_config |
提供程序特定的身份验证配置(例如 auth_mode, region, project_id) |
GetAssessmentRequest
字段名称 |
类型 |
描述 |
|---|---|---|
trace_id |
|
该评估所属的轨迹 ID。此字段为必填项。 |
assessment_id |
|
评估 ID。此字段为必填项。 |
GetDatasetRecords
字段名称 |
类型 |
描述 |
|---|---|---|
dataset_id |
|
要获取记录的数据集 ID。此字段为必填项。 |
max_results |
|
可选分页 - 要返回的最大记录数 |
page_token |
|
用于获取下一页的可选分页令牌 |
GetMetricHistoryBulkInterval
字段名称 |
类型 |
描述 |
|---|---|---|
run_ids |
STRING 的数组 |
要获取指标值的运行 ID。必须提供。 |
metric_key |
|
指标名称。此字段为必需。 |
start_step |
|
可选的起始步,仅获取指定步之后的指标。如果定义了 end_step,则必须定义此项。 |
end_step |
|
可选的结束步,仅获取指定步之前的指标。如果定义了 start_step,则必须定义此项。 |
max_results |
|
每个运行获取的最大结果数。必须设置为正数。注意,实际上该 API 最多返回 (max_results + # of run IDs) x (# run IDs) 个指标数据点。 |
GetOnlineTraceDetails
字段名称 |
类型 |
描述 |
|---|---|---|
trace_id |
|
要检索的轨迹 ID。此字段为必填项。 |
sql_warehouse_id |
|
查询使用的 SQL 仓库。此字段为必填项。 |
source_inference_table |
|
查询使用的源推理表,例如 “ml.bbqiu.codegen_payload”。此字段为必填项。 |
source_databricks_request_id |
|
查询使用的源 Databricks 请求 ID,例如 “8d1992ce-ba3d-49e9-9701-e9b323c5cc8c”。此字段为必填项。 |
JobState
将状态与元数据相结合的通用作业状态消息。提供了一种跨不同作业类型表示作业状态的统一方式。
字段名称 |
类型 |
描述 |
|---|---|---|
status |
作业的当前状态。 |
|
error_message |
|
如果作业失败,则显示错误消息。仅在状态为 JOB_STATUS_FAILED 时设置。 |
metadata |
作为键值对的附加元数据。可用于存储作业特定的状态信息。 |
LinkTracesToRun
字段名称 |
类型 |
描述 |
|---|---|---|
trace_ids |
STRING 的数组 |
要链接到运行的轨迹 ID。单次请求中最多可链接 100 个轨迹 ID。 |
run_id |
|
要链接轨迹的运行 ID。此字段为必填项。 |
ListLoggedModelArtifacts
字段名称 |
类型 |
描述 |
|---|---|---|
model_id |
|
要列出工件的 LoggedModel ID。此字段为必填项。 |
artifact_directory_path |
|
过滤匹配此路径的制品(从根制品目录开始的相对路径)。 |
page_token |
|
指示要获取的制品结果页面的令牌 |
LogLoggedModelParamsRequest
字段名称 |
类型 |
描述 |
|---|---|---|
model_id |
|
要记录参数的 LoggedModel ID。此字段为必填项。 |
params |
附加到模型的参数。 |
LogOutputs
字段名称 |
类型 |
描述 |
|---|---|---|
run_id |
|
要从中记录输出的运行 ID。此字段为必填项。 |
models |
ModelOutput 数组 |
来自运行的模型输出。 |
LoggedModel
LoggedModel 消息包括已记录的模型属性、标签、注册信息、参数和链接的运行指标。
字段名称 |
类型 |
描述 |
|---|---|---|
info |
已记录模型的属性,如模型 ID、状态、标签等。 |
|
data |
已记录模型的参数和指标。 |
LoggedModelData
LoggedModelData 消息包括已记录的模型参数和链接的指标。
字段名称 |
类型 |
描述 |
|---|---|---|
params |
模型的不可变字符串键值对。 |
|
metrics |
Metric 的数组 |
链接到模型的性能指标。 |
LoggedModelInfo
LoggedModelInfo 包括已记录的模型属性、标签和注册信息。
字段名称 |
类型 |
描述 |
|---|---|---|
model_id |
|
模型的唯一标识符。 |
experiment_id |
|
拥有该模型的实验 ID。 |
name |
|
模型名称。 |
creation_timestamp_ms |
|
模型创建时间戳(自 UNIX 纪元以来的毫秒)。 |
last_updated_timestamp_ms |
|
模型最后更新时间戳(自 UNIX 纪元以来的毫秒) |
artifact_uri |
|
存储模型工件的目录 URI。 |
status |
模型是否准备好使用。 |
|
creator_id |
|
创建该模型用户或主体的 ID。 |
model_type |
|
模型类型,例如 “Agent”、“Classifier”、“LLM”。 |
source_run_id |
|
创建该模型的运行的运行 ID。 |
status_message |
|
当前状态的详细信息。 |
tags |
在模型上设置的可变字符串键值对。 |
|
registrations |
如果模型已提升至模型注册表(Model Registry),该字段将包含注册模型名称、模型版本号等信息。 |
LoggedModelRegistrationInfo
LoggedModel 的注册信息。
字段名称 |
类型 |
描述 |
|---|---|---|
name |
|
模型被提升至的注册模型名称。 |
version |
|
被提升模型的版本号。 |
Metric
与运行关联的指标,表示为键值对。
字段名称 |
类型 |
描述 |
|---|---|---|
key |
|
用于标识此指标的键。 |
value |
|
与此指标关联的值。 |
timestamp |
|
记录此指标的时间戳。 |
step |
|
记录指标的步数。 |
dataset_name |
|
与指标关联的数据集名称。例如“my.uc.table@2”、“nyc-taxi-dataset”、“fantastic-elk-3” |
dataset_digest |
|
与指标关联的数据集的摘要,例如数据集的 md5 哈希值,用于在同名数据集中唯一标识它。 |
model_id |
|
与该指标关联的 LoggedModel 或注册模型版本的 ID(如果适用)。 |
run_id |
|
包含该指标的运行 ID。 |
MetricAggregation
字段名称 |
类型 |
描述 |
|---|---|---|
aggregation_type |
要执行的聚合类型。 |
|
percentile_value |
|
要计算的百分位数(0-100),在 aggregation_type 为 PERCENTILE 时必需。示例:50(中位数)、75、90、95、99。该字段在其他聚合类型中将被忽略。 |
MetricDataPoint
包含维度值和指标值的单个数据点。
字段名称 |
类型 |
描述 |
|---|---|---|
metric_name |
|
指标名称,例如“latency”。 |
dimensions |
DimensionsEntry 的数组 |
此数据点的维度值。键对应维度,例如 {“status”: “OK”}。 |
values |
ValuesEntry 的数组 |
此数据点的指标值。键为聚合类型,例如 {“AVG”: 150, “P99”: 234.5}。 |
MetricWithRunId
字段名称 |
类型 |
描述 |
|---|---|---|
key |
|
用于标识此指标的键。 |
value |
|
与此指标关联的值。 |
timestamp |
|
记录此指标的时间戳。 |
step |
|
记录指标的步数。 |
run_id |
|
包含该指标的运行 ID。 |
ModelMetric
与模型关联的指标,表示为键值对。复制自 MLflow 指标。
字段名称 |
类型 |
描述 |
|---|---|---|
key |
|
用于标识此指标的键。 |
value |
|
与此指标关联的值。 |
timestamp |
|
记录此指标的时间戳。 |
step |
|
记录指标的步数。 |
dataset_name |
|
与指标关联的数据集名称。例如“my.uc.table@2”、“nyc-taxi-dataset”、“fantastic-elk-3” |
dataset_digest |
|
与指标关联的数据集的摘要,例如数据集的 md5 哈希值,用于在同名数据集中唯一标识它。 |
model_id |
|
与该指标关联的 LoggedModel 或注册模型版本的 ID。 |
run_id |
|
包含该指标的运行 ID。 |
ModelOutput
代表运行(Run)输出的 LoggedModel。
字段名称 |
类型 |
描述 |
|---|---|---|
model_id |
|
模型的唯一标识符。此字段为必填项。 |
step |
|
模型产生的步数。此字段为必填项。 |
ModelVersion
字段名称 |
类型 |
描述 |
|---|---|---|
name |
|
模型的唯一名称。 |
version |
|
模型的版本号。 |
creation_timestamp |
|
创建此 |
last_updated_timestamp |
|
此 |
user_id |
|
创建此 |
current_stage |
|
此 |
description |
|
此 |
source |
|
指示源模型制品位置的 URI,在创建 |
run_id |
|
创建 |
status |
|
|
status_message |
|
当前 |
tags |
标签:此 |
|
run_link |
|
运行链接:指向生成此版本的运行的直接链接。该字段仅在模型版本创建时为那些源运行来自与注册服务器不同的跟踪服务器的模型版本设置。 |
aliases |
STRING 的数组 |
指向此 |
model_id |
|
用于将已注册模型链接到已记录源模型的模型版本可选 model_id |
model_params |
ModelParam 的数组 |
模型的可选参数。 |
model_metrics |
ModelMetric 的数组 |
模型的可选指标。 |
deployment_job_state |
此模型版本的部署作业状态。 |
ModelVersionDeploymentJobState
字段名称 |
类型 |
描述 |
|---|---|---|
job_id |
|
|
run_id |
|
|
job_state |
||
run_state |
||
current_task_name |
|
MultipartUploadCredential
字段名称 |
类型 |
描述 |
|---|---|---|
url |
|
|
part_number |
|
|
headers |
HeadersEntry 的数组 |
OrderBy
字段名称 |
类型 |
描述 |
|---|---|---|
field_name |
|
排序依据的字段名称,例如“metrics.accuracy”。此字段为必填项。 |
ascending |
|
排序是否为升序。 |
dataset_name |
|
如果 |
dataset_digest |
|
如果 |
PromptOptimizationJob
代表提示词优化作业实体。
字段名称 |
类型 |
描述 |
|---|---|---|
job_id |
|
优化作业的唯一标识符。用于轮询作业执行状态(待处理/运行中/已完成/失败)。 |
run_id |
|
存储优化指标和结果的 MLflow 运行 ID。使用此 ID 在 MLflow UI 中查看结果。仅在作业开始运行后可用。 |
state |
作业的当前状态(状态 + 错误信息 + 元数据)。 |
|
experiment_id |
|
跟踪此优化作业的 MLflow 实验 ID。 |
source_prompt_uri |
|
优化开始的源提示词 URI(例如,“prompts:/my-prompt/1”)。 |
optimized_prompt_uri |
|
优化后的提示词 URI(例如,“prompts:/my-prompt/2”)。仅在优化成功完成后设置。 |
config |
优化任务的配置。 |
|
creation_timestamp_ms |
|
作业创建的时间戳(自纪元以来的毫秒数)。 |
completion_timestamp_ms |
|
作业完成的时间戳(自纪元以来的毫秒数)。仅在状态为 COMPLETED、FAILED 或 CANCELED 时设置。 |
tags |
PromptOptimizationJobTag 的数组 |
与此作业关联的标签。 |
initial_eval_scores |
优化前的初始评估分数,按评估器名称键入。示例:{“Correctness”: 0.65, “Safety”: 0.80} |
|
final_eval_scores |
优化后的最终评估分数,按评估器名称键入。示例:{“Correctness”: 0.89, “Safety”: 0.95} |
PromptOptimizationJobConfig
提示词优化作业的配置。作为运行参数存储在底层 MLflow 运行中。
字段名称 |
类型 |
描述 |
|---|---|---|
optimizer_type |
要使用的优化器类型。 |
|
dataset_id |
|
包含训练数据的 EvaluationDataset ID。 |
scorers |
STRING 的数组 |
评估器名称列表。可以是内置评估器类名(例如,“Correctness”、“Safety”)或注册的评估器名称。 |
optimizer_config_json |
|
JSON 序列化的特定于优化器的配置。不同的优化器接受不同的参数:- GEPA: {“reflection_model”: “openai:/gpt-5”, “max_metric_calls”: 300} - MetaPrompt: {“reflection_model”: “openai:/gpt-5”, “guidelines”: “…”, “lm_kwargs”: {…}} |
PromptVersionRef
用于链接到追踪的提示词版本引用。每个引用都包含提示词名称和版本。
字段名称 |
类型 |
描述 |
|---|---|---|
name |
|
此字段为必填项。 |
version |
|
此字段为必填项。 |
QueryTraceMetrics
查询追踪、跨度(span)或评估的聚合指标。
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_ids |
STRING 的数组 |
必填:用于搜索追踪的实验 ID。 |
view_type |
必填:聚合指标的级别。 |
|
metric_name |
|
必填:要查询的指标名称(例如“latency”)。 |
aggregations |
必填:要应用的聚合方式。 |
|
dimensions |
STRING 的数组 |
可选:按维度对指标进行分组(例如,“name”,“status”)。 |
filters |
STRING 的数组 |
可选:要应用的筛选表达式(例如 trace.status=”OK”)。 |
time_interval_seconds |
|
可选:分组的时间间隔(秒)。设置后,结果将自动包含按指定间隔分组的时间维度。示例:60(分)、3600(时)、86400(天)、604800(周)、2592000(月)。 |
start_time_ms |
|
可选:时间范围的起始点(自纪元以来的毫秒数)。如果设置了 time_interval_seconds,则为必填项。 |
end_time_ms |
|
可选:时间范围的结束点(自纪元以来的毫秒数)。如果设置了 time_interval_seconds,则为必填项。 |
max_results |
|
可选:返回的最大数据点数。默认值:1000。 |
page_token |
|
可选:用于获取下一页结果的分页令牌。 |
RegisteredModel
字段名称 |
类型 |
描述 |
|---|---|---|
name |
|
模型的唯一名称。 |
creation_timestamp |
|
创建此 |
last_updated_timestamp |
|
此 |
user_id |
|
创建此 |
description |
|
此 |
latest_versions |
ModelVersion 数组 |
每个阶段的最新模型版本集合。仅包含状态为 |
tags |
标签:此 |
|
aliases |
指向与此 |
|
deployment_job_id |
|
该模型的部署作业 ID。 |
deployment_job_state |
此模型的部署作业状态。 |
RemoveDatasetFromExperiments
字段名称 |
类型 |
描述 |
|---|---|---|
dataset_id |
|
要从实验中删除的数据集 ID。此字段为必填项。 |
experiment_ids |
STRING 的数组 |
要解除与数据集关联的实验 ID。 |
Response
字段名称 |
类型 |
描述 |
|---|---|---|
metrics |
MetricWithRunId 的数组 |
表示值和元数据历史记录的指标列表。 |
Response
字段名称 |
类型 |
描述 |
|---|---|---|
traces |
Trace 的数组 |
获取的追踪信息。 |
Response
字段名称 |
类型 |
描述 |
|---|---|---|
trace_infos |
TraceInfoV3 的数组 |
获取的追踪元数据。 |
Response
字段名称 |
类型 |
描述 |
|---|---|---|
traces |
TraceInfo 的数组 |
有关匹配搜索条件的追踪信息。 |
next_page_token |
|
Response
字段名称 |
类型 |
描述 |
|---|---|---|
traces |
TraceInfo 的数组 |
有关匹配搜索条件的追踪信息。 |
next_page_token |
|
Response
字段名称 |
类型 |
描述 |
|---|---|---|
npmi |
|
归一化逐点互信息分数(-1 到 1)。 |
npmi_smoothed |
|
使用 Jeffreys 先验进行稳健性平滑后的 NPMI 值。 |
filter1_count |
|
匹配第一个筛选器的追踪数量。 |
filter2_count |
|
匹配第二个筛选器的追踪数量。 |
joint_count |
|
同时匹配两个筛选器的追踪数量。 |
total_count |
|
实验中的追踪总数。 |
Response
字段名称 |
类型 |
描述 |
|---|---|---|
data_points |
MetricDataPoint 的数组 |
按维度分组的数据点。 |
next_page_token |
|
用于获取下一页的分页令牌。如果没有更多结果,则为空。 |
Response
字段名称 |
类型 |
描述 |
|---|---|---|
dataset_summaries |
DatasetSummary 的数组 |
返回最近创建的 N 个数据集的摘要(后端配置)。 |
Response
字段名称 |
类型 |
描述 |
|---|---|---|
models |
LoggedModel 的数组 |
匹配搜索条件的已记录模型。 |
next_page_token |
|
可用于检索下一页已记录模型的令牌。 |
Response
字段名称 |
类型 |
描述 |
|---|---|---|
root_uri |
|
已记录模型的根制品目录。 |
files |
FileInfo 的数组 |
制品的路径和元数据。 |
next_page_token |
|
可用于检索下一页制品结果的令牌 |
Response
字段名称 |
类型 |
描述 |
|---|---|---|
traces |
TraceInfoV3 的数组 |
有关匹配搜索条件的追踪信息。 |
next_page_token |
|
Response
字段名称 |
类型 |
描述 |
|---|---|---|
datasets |
Dataset 的数组 |
数据集列表(仅包含元数据)。 |
next_page_token |
|
如果有更多结果存在,则为下一页令牌。 |
Response
字段名称 |
类型 |
描述 |
|---|---|---|
workspaces |
Workspace 的数组 |
工作空间记录集合。 |
RunData
运行数据(指标、参数和标签)。
字段名称 |
类型 |
描述 |
|---|---|---|
metrics |
Metric 的数组 |
运行指标。 |
params |
Param 的数组 |
运行参数。 |
tags |
RunTag 的数组 |
附加的元数据键值对。 |
RunInfo
单个运行的元数据。
字段名称 |
类型 |
描述 |
|---|---|---|
run_id |
|
运行的唯一标识符。 |
run_uuid |
|
[已弃用,请使用 run_id] 运行的唯一标识符。此字段将在未来的 MLflow 版本中删除。 |
run_name |
|
运行名称。 |
experiment_id |
|
实验 ID。 |
user_id |
|
发起运行的用户。该字段在 MLflow 1.0 中已被弃用,并将于未来的 MLflow 版本中删除。请改用 ‘mlflow.user’ 标签。 |
status |
运行的当前状态。 |
|
start_time |
|
运行开始的 Unix 时间戳(毫秒)。 |
end_time |
|
运行结束的 Unix 时间戳(毫秒)。 |
artifact_uri |
|
应上传制品的目录 URI。可以是本地路径(以“/”开头),也可以是分布式文件系统 (DFS) 路径,例如 |
lifecycle_stage |
|
实验的当前生命周期阶段:OneOf(“active”, “deleted”) |
RunInputs
运行输入。
字段名称 |
类型 |
描述 |
|---|---|---|
dataset_inputs |
DatasetInput 的数组 |
运行的数据集输入。 |
model_inputs |
ModelInput 的数组 |
运行的模型输入。 |
Scorer
表示数据库中评分器的评分器实体。
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
实验 ID。 |
scorer_name |
|
评分器名称。 |
scorer_version |
|
评分器版本。 |
serialized_scorer |
|
序列化的评分器字符串。 |
creation_time |
|
评分器版本的创建时间(自纪元以来的毫秒数)。 |
scorer_id |
|
评分器的唯一标识符。 |
SearchEvaluationDatasets
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_ids |
STRING 的数组 |
关联的实验 ID(用于筛选)。 |
filter_string |
|
数据集名称的筛选字符串。 |
max_results |
|
最大结果数。 |
order_by |
STRING 的数组 |
排序标准。 |
page_token |
|
用于分页的页面令牌。 |
SearchLoggedModels
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_ids |
STRING 的数组 |
要搜索已记录模型的实验 ID。 |
filter |
|
对已记录模型信息和数据执行筛选的表达式,允许返回已记录模型的子集。语法是支持将二进制操作进行 AND 运算的 SQL 子集。示例: |
datasets |
Dataset 的数组 |
要应用指标筛选子句的数据集列表。例如,具有 metrics.accuracy > 0.9 的筛选器以及名为 “test_dataset” 的数据集信息,意味着我们将返回在 test_dataset 上准确度 > 0.9 的所有已记录模型。将考虑匹配条件的任何数据集中的指标值。如果没有指定数据集,则筛选器会考虑所有数据集的指标。 |
max_results |
|
要返回的已记录模型最大数量。最大阈值为 50。 |
order_by |
OrderBy 的数组 |
用于对结果进行排序的列列表,以及用于排序标准的附加字段。 |
page_token |
|
指示要获取的已记录模型页面的令牌。 |
SearchTraces
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_ids |
STRING 的数组 |
要搜索的实验 ID 列表。 |
filter |
|
对追踪属性和标签执行筛选的表达式,允许返回追踪的子集。语法是支持将二进制操作进行 AND 运算的 SQL 子集。示例: |
max_results |
|
所需的最大追踪数。最大阈值为 500。 |
order_by |
STRING 的数组 |
用于对结果进行排序的列列表,例如 |
page_token |
|
指示要获取的追踪页面的令牌。 |
SearchTracesV3
字段名称 |
类型 |
描述 |
|---|---|---|
locations |
TraceLocation 的数组 |
要搜索的 MLflow 实验列表。 |
filter |
|
对追踪属性和标签执行筛选的表达式,允许返回追踪的子集。语法是支持将二进制操作进行 AND 运算的 SQL 子集。示例: |
max_results |
|
所需的最大追踪数。最大阈值为 500。 |
order_by |
STRING 的数组 |
用于对结果进行排序的列列表,例如 |
page_token |
|
指示要获取的追踪页面的令牌。 |
SearchUnifiedTraces
字段名称 |
类型 |
描述 |
|---|---|---|
model_id |
|
此字段为必填项。 |
sql_warehouse_id |
|
此字段为必填项。 |
experiment_ids |
STRING 的数组 |
待办事项:最终我们希望提供一个仅使用 model_id 的 API。 |
filter |
|
对追踪属性和标签执行筛选的表达式,允许返回追踪的子集。语法是支持将二进制操作进行 AND 运算的 SQL 子集。示例: |
max_results |
|
所需的最大追踪数。最大阈值为 500。 |
order_by |
STRING 的数组 |
用于对结果进行排序的列列表,例如 |
page_token |
|
指示要获取的追踪页面的令牌。这是一个统一的令牌,编码了在线和离线追踪令牌。 |
SerializedValue
字段名称 |
类型 |
描述 |
|---|---|---|
serialization_format |
|
标记期望值的序列化格式。这是客户端特定的契约。服务不会尝试反序列化该值或验证格式。示例格式为 “JSON_FORMAT”。 |
value |
|
以 |
SetDatasetTags
字段名称 |
类型 |
描述 |
|---|---|---|
dataset_id |
|
要更新标签的数据集 ID。此字段为必填项。 |
tags |
|
要更新的标签(JSON 字符串)。此字段为必填项。 |
SetTraceTag
字段名称 |
类型 |
描述 |
|---|---|---|
request_id |
|
要设置标签的追踪 ID。 |
key |
|
标签名称。最大大小取决于存储后端。所有存储后端均保证支持最大 250 字节的键值。 |
value |
|
要记录的标签的字符串值。最大大小取决于存储后端。所有存储后端均保证支持最大 250 字节的键值。 |
SetTraceTagV3
字段名称 |
类型 |
描述 |
|---|---|---|
trace_id |
|
要设置标签的追踪 ID。 |
key |
|
标签名称。最大大小取决于存储后端。所有存储后端均保证支持最大 250 字节的键值。 |
value |
|
要记录的标签的字符串值。最大大小取决于存储后端。所有存储后端均保证支持最大 250 字节的键值。 |
StartTrace
字段名称 |
类型 |
描述 |
|---|---|---|
experiment_id |
|
关联实验的 ID。 |
timestamp_ms |
|
追踪开始的 Unix 时间戳(毫秒)。 |
request_metadata |
关于发起追踪的请求的元数据。 |
|
tags |
TraceTag 数组 |
追踪的标签。 |
TraceArchivalConfig
工作空间 API 接受并在工作空间元数据中返回的追踪存档设置。
字段名称 |
类型 |
描述 |
|---|---|---|
location |
|
可选的存档存储库根路径覆盖。 |
retention |
|
可选的存档保留期限覆盖。格式:<int><unit>,例如 30d。 |
TraceInfo
TraceInfo。代表追踪的元数据。
字段名称 |
类型 |
描述 |
|---|---|---|
request_id |
|
追踪的唯一标识符。 |
experiment_id |
|
包含该追踪的实验 ID。 |
timestamp_ms |
|
追踪开始的 Unix 时间戳(毫秒)。 |
execution_time_ms |
|
追踪持续时间的 Unix 时间戳(毫秒)。 |
status |
正在追踪的操作的整体状态(OK、错误等)。 |
|
request_metadata |
其他追踪元数据。 |
|
tags |
TraceTag 数组 |
追踪的标签。 |
TraceInfoV3
字段名称 |
类型 |
描述 |
|---|---|---|
trace_id |
|
与追踪关联的主键。 |
client_request_id |
|
客户端提供的与追踪关联的请求 ID。这可用于从产生追踪的外部系统识别追踪/请求。 |
trace_location |
||
request |
|
[已弃用,请改用 request_preview。] 对模型/代理的请求。等同于根跨度的输入,但为了易于访问而添加。表示为 JSON 字符串。 |
response |
|
[已弃用,请改用 request_preview。] 模型/代理的响应。等同于根跨度的输出,但为了易于访问而添加。表示为 JSON 字符串。 |
request_preview |
|
表示为 JSON 字符串的模型/代理请求预览。这等同于根跨度的输入。此预览值被截断为 10KB,而完整请求存储在 Blob 存储的追踪数据中。 |
response_preview |
|
表示为 JSON 字符串的模型/代理请求预览。这等同于根跨度的输出。此预览值被截断为 10KB,而完整响应存储在 Blob 存储的追踪数据中。 |
request_time |
|
追踪的开始时间。 |
execution_duration |
|
追踪执行时间。 |
state |
||
trace_metadata |
与追踪关联的元数据。示例包括:- run_id: 产生追踪的 mlflow 运行(即评估作业)的 ID。在某些情况下(如通过交互式验证创建追踪)可能不适用 - model_id: 产生追踪的关联模型 ID - dataset_id: mlflow 数据集的 ID(通常与 dataset_record_id 一起使用) - dataset_record_id: mlflow 数据集的 ID(通常与 dataset_record_id 一起使用) - session_id: 请求来源的会话 ID(例如聊天对话) |
|
assessments |
Assessment 的数组 |
|
tags |
TagsEntry 的数组 |
可变的、用户定义的追踪标签,例如 “question_topic”: “DBSQL” |
TraceLocation
存储和产生追踪的位置。
字段名称 |
类型 |
描述 |
|---|---|---|
type |
||
|
如果是 |
UpdateAssessment
更新现有评估的请求。
字段名称 |
类型 |
描述 |
|---|---|---|
assessment |
包含应更新字段的评估。此字段为必填项。 |
|
update_mask |
|
要更新的评估字段列表。这些应对应于 assessment 中存在的值(或缺失情况)。此字段为必填项。 |
UpdateWorkspace
更新工作空间元数据。
字段名称 |
类型 |
描述 |
|---|---|---|
workspace_name |
|
要更新的工作空间名称。此字段为必填项。 |
description |
|
可选的描述更新。 |
default_artifact_root |
|
可选的默认制品根路径覆盖更新。 |
trace_archival_config |
可选的追踪存档设置更新。 |
UpsertDatasetRecords
字段名称 |
类型 |
描述 |
|---|---|---|
dataset_id |
|
要更新记录的数据集 ID。此字段为必填项。 |
records |
|
要更新的记录(记录字典的 JSON 序列化列表)。此字段为必填项。 |
updated_by |
|
执行更新的用户。 |
Webhook
Webhook 实体。
字段名称 |
类型 |
描述 |
|---|---|---|
webhook_id |
|
Webhook 的唯一标识符。 |
name |
|
Webhook 名称。 |
description |
|
Webhook 的可选描述 |
url |
|
发送 Webhook 事件的 URL。 |
events |
WebhookEvent 数组 |
此 Webhook 订阅的事件列表。 |
status |
Webhook 的当前状态。 |
|
creation_timestamp |
|
Webhook 创建的时间戳。 |
last_updated_timestamp |
|
Webhook 最后更新的时间戳。 |
WebhookTestResult
测试 Webhook 结果。
字段名称 |
类型 |
描述 |
|---|---|---|
success |
|
测试是否成功。 |
response_status |
|
HTTP 响应状态码(如果可用)。 |
response_body |
|
响应体(如果可用)。 |
error_message |
|
如果测试失败,则显示错误信息。 |
Workspace
工作空间 API 返回的工作空间元数据。
字段名称 |
类型 |
描述 |
|---|---|---|
name |
|
唯一的工作空间名称。此字段为必填项。 |
description |
|
可选的工作区描述。 |
default_artifact_root |
|
此工作空间可选的默认制品根路径覆盖。 |
trace_archival_config |
此工作空间可选的追踪存档设置。 |
AggregationType
指标的聚合类型。
名称 |
描述 |
|---|---|
COUNT |
实体计数。 |
SUM |
值求和。 |
AVG |
值平均值。 |
PERCENTILE |
百分位数聚合(需要 percentile_value 参数)。 |
MIN |
最小值。 |
MAX |
最大值。 |
DeploymentJobRunState
名称 |
描述 |
|---|---|
DEPLOYMENT_JOB_RUN_STATE_UNSPECIFIED |
|
NO_VALID_DEPLOYMENT_JOB_FOUND |
|
RUNNING |
|
SUCCEEDED |
|
FAILED |
|
PENDING |
|
APPROVAL |
FallbackStrategy
路由的后备策略(面向未来,支持其他策略)。
名称 |
描述 |
|---|---|
FALLBACK_STRATEGY_UNSPECIFIED |
|
SEQUENTIAL |
顺序后备:按指定顺序尝试模型。 |
GatewayModelLinkageType
端点与模型定义之间的链接类型。
名称 |
描述 |
|---|---|
LINKAGE_TYPE_UNSPECIFIED |
|
PRIMARY |
主链接:用于路由流量。 |
FALLBACK |
后备链接:用于故障转移。 |
JobStatus
MLflow 作业的通用状态枚举。可用于不同的作业类型(优化、评分器等)。
名称 |
描述 |
|---|---|
JOB_STATUS_UNSPECIFIED |
|
JOB_STATUS_PENDING |
作业已排队,等待开始。 |
JOB_STATUS_IN_PROGRESS |
作业正在运行。 |
JOB_STATUS_COMPLETED |
作业成功完成。 |
JOB_STATUS_FAILED |
作业因错误失败。 |
JOB_STATUS_CANCELED |
作业已被用户取消。 |
LoggedModelStatus
LoggedModelStatus 枚举值表示已记录模型的状态。
名称 |
描述 |
|---|---|
LOGGED_MODEL_STATUS_UNSPECIFIED |
|
LOGGED_MODEL_PENDING |
LoggedModel 已创建,但 LoggedModel 文件尚未完全上传。 |
LOGGED_MODEL_READY |
LoggedModel 已创建,且 LoggedModel 文件已完全上传。 |
LOGGED_MODEL_UPLOAD_FAILED |
LoggedModel 已创建,但在上传 LoggedModel 文件(如模型权重/代理代码)时发生错误。 |
ModelVersionStatus
名称 |
描述 |
|---|---|
PENDING_REGISTRATION |
注册新模型版本的请求处于待处理状态,服务器正在执行后台任务。 |
FAILED_REGISTRATION |
注册新模型版本的请求失败。 |
READY |
模型版本已就绪,可以使用。 |
OptimizerType
要使用的优化算法类型。
名称 |
描述 |
|---|---|
OPTIMIZER_TYPE_UNSPECIFIED |
|
OPTIMIZER_TYPE_GEPA |
GEPA (Genetic Pareto) 优化器 (https://github.com/gepa-ai/gepa) |
OPTIMIZER_TYPE_METAPROMPT |
MetaPrompt 优化器 - 使用元提示词与 LLM,在单次传递中改进提示词。 |
RoutingStrategy
端点的路由策略。
名称 |
描述 |
|---|---|
ROUTING_STRATEGY_UNSPECIFIED |
|
REQUEST_BASED_TRAFFIC_SPLIT |
基于请求的流量分配:根据权重分配流量。 |
RunStatus
运行的状态。
名称 |
描述 |
|---|---|
RUNNING |
运行已发起。 |
SCHEDULED |
运行计划在稍后时间运行。 |
FINISHED |
运行已完成。 |
FAILED |
运行执行失败。 |
KILLED |
运行被用户终止。 |
SourceType
生成运行的源。
名称 |
描述 |
|---|---|
NOTEBOOK |
Databricks 笔记本环境。 |
JOB |
计划作业或立即运行作业。 |
PROJECT |
作为预打包项目:Docker 镜像或 GitHub 源等。 |
LOCAL |
本地运行:使用 CLI、IDE 或本地笔记本。 |
UNKNOWN |
未知的源类型。 |
SourceType
评估来源的类型。
名称 |
描述 |
|---|---|
SOURCE_TYPE_UNSPECIFIED |
|
HUMAN |
来自人类的评估。 |
LLM_JUDGE |
来自 LLM 评估器的评估。 |
CODE |
基于代码的评估(例如 Python UDF)。 |
SourceType
数据集记录来源的类型。
名称 |
描述 |
|---|---|
SOURCE_TYPE_UNSPECIFIED |
|
TRACE |
来自追踪/跨度的记录。 |
HUMAN |
来自人类注释的记录。 |
DOCUMENT |
来自文档的记录。 |
CODE |
来自代码/计算的记录。 |
State
记录时追踪的执行状态。
名称 |
描述 |
|---|---|
STATE_UNSPECIFIED |
|
OK |
被追踪的操作已成功。 |
ERROR |
被追踪的操作失败。 |
IN_PROGRESS |
被追踪的操作仍在进行中。这对于增量/分布式追踪记录非常有用,与仅在完成时记录完整追踪的情况相反。 |
State
名称 |
描述 |
|---|---|
DEPLOYMENT_JOB_CONNECTION_STATE_UNSPECIFIED |
|
NOT_SET_UP |
默认状态 |
CONNECTED |
已连接作业:作业存在,所有者拥有 ACL,且所需的作业参数均已具备。 |
NOT_FOUND |
作业已被删除或所有者的作业 ACL 已被移除。 |
REQUIRED_PARAMETERS_CHANGED |
必需的作业参数已更改。 |