diff --git a/.gitignore b/.gitignore index 9b6769052..a69793cb1 100644 --- a/.gitignore +++ b/.gitignore @@ -56,3 +56,5 @@ FEATURE_SUMMARY.md GEMINI.md QWEN.md .omx/ +.opencode/ +CODEBUDDY.md diff --git a/swanlab/sdk/internal/core_python/api/__init__.py b/swanlab/sdk/internal/core_python/api/__init__.py index c3c3f6b31..7260e21dd 100644 --- a/swanlab/sdk/internal/core_python/api/__init__.py +++ b/swanlab/sdk/internal/core_python/api/__init__.py @@ -3,6 +3,7 @@ @file: __init__.py @time: 2026/3/7 18:19 @description: SwanLab 运行时 API 封装 + 绝大多数API使用 Client 对象,少部分API使用requests库直接调用 我们以rpc风格封装API,方便调用 """ diff --git a/swanlab/sdk/internal/core_python/api/experiment.py b/swanlab/sdk/internal/core_python/api/experiment.py index 1f276485f..3ba971506 100644 --- a/swanlab/sdk/internal/core_python/api/experiment.py +++ b/swanlab/sdk/internal/core_python/api/experiment.py @@ -5,7 +5,7 @@ @description: SwanLab 运行时实验API """ -from typing import List, Literal, Optional +from typing import Dict, List, Literal, Optional, Union from google.protobuf.timestamp_pb2 import Timestamp @@ -91,3 +91,12 @@ def stop_experiment(username: str, project: str, cuid: str, *, state: RunState, "from": "sdk", }, ) + + +def send_experiment_heartbeat(*, cuid: str, flag_id: str) -> None: + """ + 发送实验心跳,保持实验处于活跃状态 + :param cuid: 实验唯一标识符 + :param flag_id: 实验标记ID + """ + client.post(f"/house/experiments/{cuid}/heartbeat", {"flagId": flag_id}) diff --git a/swanlab/sdk/internal/core_python/api/project.py b/swanlab/sdk/internal/core_python/api/project.py index 81e7e03a9..871d11639 100644 --- a/swanlab/sdk/internal/core_python/api/project.py +++ b/swanlab/sdk/internal/core_python/api/project.py @@ -16,7 +16,7 @@ def get_project(*, username: str, name: str) -> ProjectType: """ - 获取项目信息 + 获取项目详情信息 :param username: 项目所属的用户名 :param name: 项目名称 :return: 项目信息 diff --git a/swanlab/sdk/typings/core_python/api/experiment.py b/swanlab/sdk/typings/core_python/api/experiment.py index b39212bd5..e32a9dc29 100644 --- a/swanlab/sdk/typings/core_python/api/experiment.py +++ b/swanlab/sdk/typings/core_python/api/experiment.py @@ -1,7 +1,7 @@ """ @author: cunyue @file: experiment.py -@time: 2026/4/18 17:43 +@time: 2026/3/10 19:02 @description: SwanLab 运行时实验API类型 """