wedata3-ml-runtime 0.0.1__tar.gz

wedata3_ml_runtime-0.0.1/PKG-INFO

@@ -0,0 +1,120 @@
+Metadata-Version: 2.4
+Name: wedata3-ml-runtime
+Version: 0.0.1
+Summary: WeData 3.0 机器学习 Notebook Kernel 运行时初始化库，提供 MLflow 深度桥接、Feast gRPC 代理、腾讯云 OpenAPI 签名调用及平台级多租户隔离。
+Author-email: WeData Team <wedata@tencent.com>
+License: MIT
+Project-URL: Homepage, https://wedata.tencent.com
+Project-URL: Documentation, https://wedata.tencent.com/docs
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: feast==0.49.0
+Requires-Dist: grpcio>=1.71.0
+Requires-Dist: mlflow<3.11.0,>=2.0.0
+Requires-Dist: pydantic>=2.10.6
+Requires-Dist: wedata-mlflow-header-plugin>=0.1.2
+Requires-Dist: mlflow-tclake-plugin>=3.0.15
+
+# wedata3-ml-runtime
+
+WeData 3.0 机器学习 Notebook Kernel 运行时初始化库。
+
+## 定位
+
+本包是 WeData 3.0 Notebook 场景下，运行在 DLC 计算容器中的 **Kernel 启动适配层**。它不改变用户代码，而是通过 monkey patching 把开源的 `mlflow` / `feast` 改造成带有 WeData 平台语义（多租户隔离、CAM 鉴权、审计、网关代理）的版本。
+
+> 前身：`wedata-pre-code`（同时支持 2.0 和 3.0）
+>
+> 本包仅承载 WeData 3.0 逻辑。WeData 2.0 仍沿用 `wedata-pre-code`，位于 `../wedata-pre-execute/`（已进入维护模式）。
+
+## 核心能力
+
+1. **装包标记**：配合 DLC 镜像预装 → 运行时刷新 `/opt/spark/pip_list.json` 系统包快照，消除重复装包告警
+2. **环境变量注入**：`WEDATA_WORKSPACE_ID` / `MLFLOW_TRACKING_URI` / `TENCENTCLOUD_SECRET_ID/KEY/TOKEN` / `KERNEL_WEDATA_*` 等 20+ 条
+3. **MLflow 客户端重写**：
+   - `create/get/search/rename/delete experiment` 改走腾讯云 OpenAPI（TC3-HMAC-SHA256 手签）
+   - `MlflowClient` 上所有写操作套装饰器做 workspace_id 校验
+4. **Feast gRPC 代理**：给 `RemoteRegistry` 加 `X-Target-Service-IP/PORT` 请求头走反向代理
+5. **自动 tag 注入**：所有 mlflow 对象自动打上 `wedata.workspace / wedata.datascience.type / mlflow.user`
+
+## 使用
+
+```python
+%pip install wedata3-ml-runtime
+from wedata_ml_runtime.client import Wedata3PreCodeClient
+
+client = Wedata3PreCodeClient(
+    workspace_id="...",
+    base_url="...",
+    region="...",
+    ap_region_id=1,
+    mlflow_gateway_url="...",
+    feast_gateway_url="...",
+    mlflow_proxy_ip="...",
+    mlflow_proxy_port="...",
+    feast_proxy_ip="...",
+    feast_proxy_port="...",
+    kernel_task_name="...",
+    kernel_task_id="...",
+    cloud_sdk_secret_id="...",
+    cloud_sdk_secret_key="...",
+    cloud_sdk_secret_token="...",
+    qcloud_uin="...",
+    qcloud_subuin="...",
+)
+client.init()
+```
+
+## 必传参数
+
+| 参数 | 说明 |
+|------|------|
+| `workspace_id` | 工作空间 ID |
+| `base_url` | WeData 控制台基础 URL |
+| `mlflow_gateway_url` | MLflow Serverless 网关地址 |
+| `feast_gateway_url` | Feast Serverless 网关地址 |
+| `mlflow_proxy_ip` / `mlflow_proxy_port` | MLflow 转发地址 |
+| `feast_proxy_ip` / `feast_proxy_port` | Feast 转发地址 |
+
+## 可选参数
+
+| 参数 | 说明 |
+|------|------|
+| `region` / `ap_region_id` | 地域标识 |
+| `kernel_task_name` / `kernel_task_id` | 任务身份 |
+| `kernel_submit_form_workflow` | 工作流标识 |
+| `cloud_sdk_secret_id` / `cloud_sdk_secret_key` / `cloud_sdk_secret_token` | CAM 临时凭证三元组（12 小时过期） |
+| `cloud_sdk_env` | SDK 环境（`dev` / `test` / `pre`） |
+| `cloud_sdk_user_id` | 测试账户 ID（`cloud_sdk_env=test` 时使用） |
+| `qcloud_uin` / `qcloud_subuin` | 腾讯云主账号 / 子账号 UIN |
+
+## 构建发布
+
+```bash
+bash build.sh
+```
+
+脚本内容：`rm -rf dist/ build/` → `uv build` → `twine upload dist/*`
+
+## 从 wedata-pre-code 迁移
+
+| 维度 | 老包 `wedata-pre-code` | 新包 `wedata3-ml-runtime` |
+|------|----------------------|------------------------|
+| PyPI 名 | `wedata-pre-code` | `wedata3-ml-runtime` |
+| Python 模块 | `wedata_pre_code.wedata3.client` | `wedata_ml_runtime.client` |
+| 客户端类 | `Wedata3PreCodeClient` | `Wedata3PreCodeClient`（保持兼容）|
+| extras | `pip install "wedata-pre-code[wedata-3]"` | 无需 extras，主依赖已覆盖 |
+| 覆盖平台 | 2.0 + 3.0 | **仅 3.0** |
+
+**迁移办法**：把 `%pip install wedata-pre-code[wedata-3]==X.Y.Z` 改为 `%pip install wedata3-ml-runtime==X.Y.Z`，并把 `from wedata_pre_code.wedata3.client import Wedata3PreCodeClient` 改为 `from wedata_ml_runtime.client import Wedata3PreCodeClient`。
+
+## 相关设计文档
+
+- `application/science/doc/plan/wedata-pre-code-image-baseline-plan.md`：DLC 镜像基线 + 运行时 Override 方案

wedata3_ml_runtime-0.0.1/README.md

@@ -0,0 +1,96 @@
+# wedata3-ml-runtime
+
+WeData 3.0 机器学习 Notebook Kernel 运行时初始化库。
+
+## 定位
+
+本包是 WeData 3.0 Notebook 场景下，运行在 DLC 计算容器中的 **Kernel 启动适配层**。它不改变用户代码，而是通过 monkey patching 把开源的 `mlflow` / `feast` 改造成带有 WeData 平台语义（多租户隔离、CAM 鉴权、审计、网关代理）的版本。
+
+> 前身：`wedata-pre-code`（同时支持 2.0 和 3.0）
+>
+> 本包仅承载 WeData 3.0 逻辑。WeData 2.0 仍沿用 `wedata-pre-code`，位于 `../wedata-pre-execute/`（已进入维护模式）。
+
+## 核心能力
+
+1. **装包标记**：配合 DLC 镜像预装 → 运行时刷新 `/opt/spark/pip_list.json` 系统包快照，消除重复装包告警
+2. **环境变量注入**：`WEDATA_WORKSPACE_ID` / `MLFLOW_TRACKING_URI` / `TENCENTCLOUD_SECRET_ID/KEY/TOKEN` / `KERNEL_WEDATA_*` 等 20+ 条
+3. **MLflow 客户端重写**：
+   - `create/get/search/rename/delete experiment` 改走腾讯云 OpenAPI（TC3-HMAC-SHA256 手签）
+   - `MlflowClient` 上所有写操作套装饰器做 workspace_id 校验
+4. **Feast gRPC 代理**：给 `RemoteRegistry` 加 `X-Target-Service-IP/PORT` 请求头走反向代理
+5. **自动 tag 注入**：所有 mlflow 对象自动打上 `wedata.workspace / wedata.datascience.type / mlflow.user`
+
+## 使用
+
+```python
+%pip install wedata3-ml-runtime
+from wedata_ml_runtime.client import Wedata3PreCodeClient
+
+client = Wedata3PreCodeClient(
+    workspace_id="...",
+    base_url="...",
+    region="...",
+    ap_region_id=1,
+    mlflow_gateway_url="...",
+    feast_gateway_url="...",
+    mlflow_proxy_ip="...",
+    mlflow_proxy_port="...",
+    feast_proxy_ip="...",
+    feast_proxy_port="...",
+    kernel_task_name="...",
+    kernel_task_id="...",
+    cloud_sdk_secret_id="...",
+    cloud_sdk_secret_key="...",
+    cloud_sdk_secret_token="...",
+    qcloud_uin="...",
+    qcloud_subuin="...",
+)
+client.init()
+```
+
+## 必传参数
+
+| 参数 | 说明 |
+|------|------|
+| `workspace_id` | 工作空间 ID |
+| `base_url` | WeData 控制台基础 URL |
+| `mlflow_gateway_url` | MLflow Serverless 网关地址 |
+| `feast_gateway_url` | Feast Serverless 网关地址 |
+| `mlflow_proxy_ip` / `mlflow_proxy_port` | MLflow 转发地址 |
+| `feast_proxy_ip` / `feast_proxy_port` | Feast 转发地址 |
+
+## 可选参数
+
+| 参数 | 说明 |
+|------|------|
+| `region` / `ap_region_id` | 地域标识 |
+| `kernel_task_name` / `kernel_task_id` | 任务身份 |
+| `kernel_submit_form_workflow` | 工作流标识 |
+| `cloud_sdk_secret_id` / `cloud_sdk_secret_key` / `cloud_sdk_secret_token` | CAM 临时凭证三元组（12 小时过期） |
+| `cloud_sdk_env` | SDK 环境（`dev` / `test` / `pre`） |
+| `cloud_sdk_user_id` | 测试账户 ID（`cloud_sdk_env=test` 时使用） |
+| `qcloud_uin` / `qcloud_subuin` | 腾讯云主账号 / 子账号 UIN |
+
+## 构建发布
+
+```bash
+bash build.sh
+```
+
+脚本内容：`rm -rf dist/ build/` → `uv build` → `twine upload dist/*`
+
+## 从 wedata-pre-code 迁移
+
+| 维度 | 老包 `wedata-pre-code` | 新包 `wedata3-ml-runtime` |
+|------|----------------------|------------------------|
+| PyPI 名 | `wedata-pre-code` | `wedata3-ml-runtime` |
+| Python 模块 | `wedata_pre_code.wedata3.client` | `wedata_ml_runtime.client` |
+| 客户端类 | `Wedata3PreCodeClient` | `Wedata3PreCodeClient`（保持兼容）|
+| extras | `pip install "wedata-pre-code[wedata-3]"` | 无需 extras，主依赖已覆盖 |
+| 覆盖平台 | 2.0 + 3.0 | **仅 3.0** |
+
+**迁移办法**：把 `%pip install wedata-pre-code[wedata-3]==X.Y.Z` 改为 `%pip install wedata3-ml-runtime==X.Y.Z`，并把 `from wedata_pre_code.wedata3.client import Wedata3PreCodeClient` 改为 `from wedata_ml_runtime.client import Wedata3PreCodeClient`。
+
+## 相关设计文档
+
+- `application/science/doc/plan/wedata-pre-code-image-baseline-plan.md`：DLC 镜像基线 + 运行时 Override 方案

wedata3_ml_runtime-0.0.1/pyproject.toml

@@ -0,0 +1,60 @@
+[project]
+name = "wedata3-ml-runtime"
+version = "0.0.1"
+description = "WeData 3.0 机器学习 Notebook Kernel 运行时初始化库，提供 MLflow 深度桥接、Feast gRPC 代理、腾讯云 OpenAPI 签名调用及平台级多租户隔离。"
+authors = [
+    {name = "WeData Team", email = "wedata@tencent.com"}
+]
+readme = "README.md"
+license = {text = "MIT"}
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+]
+requires-python = ">=3.10"
+# 依赖扁平化：wedata-pre-code 时期的 [wedata-3] extras 已直接并入主依赖，
+# 因为 wedata3-ml-runtime 仅服务 WeData 3.0，不需要按平台切换依赖集。
+dependencies = [
+    "feast==0.49.0",
+    "grpcio>=1.71.0",
+    "mlflow>=2.0.0,<3.11.0",
+    "pydantic>=2.10.6",
+    "wedata-mlflow-header-plugin>=0.1.2",
+    "mlflow-tclake-plugin>=3.0.15",
+]
+
+[project.urls]
+Homepage = "https://wedata.tencent.com"
+Documentation = "https://wedata.tencent.com/docs"
+
+[build-system]
+# 使用 setuptools 作为 build backend：
+# PyPI 包名（wedata3-ml-runtime）与 Python 包目录 / import 名（wedata_ml_runtime）
+# 故意不一致，uv_build 默认假设两者同名导致构建失败；setuptools 通过下方
+# `[tool.setuptools.packages.find]` 显式配置即可正确打包。
+requires = ["setuptools>=68", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[tool.setuptools]
+# 显式声明 src layout：包目录 wedata_ml_runtime 位于 src/ 下
+package-dir = {"" = "src"}
+
+[tool.setuptools.packages.find]
+where = ["src"]
+include = ["wedata_ml_runtime*"]
+
+[dependency-groups]
+dev = [
+    "pytest",
+    "python-dotenv",
+]
+
+[tool.pytest.ini_options]
+markers = [
+    "integration: 集成测试，需要真实的腾讯云 WeData 3.0 测试环境",
+]

wedata3_ml_runtime-0.0.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

wedata3_ml_runtime-0.0.1/src/wedata3_ml_runtime.egg-info/PKG-INFO

@@ -0,0 +1,120 @@
+Metadata-Version: 2.4
+Name: wedata3-ml-runtime
+Version: 0.0.1
+Summary: WeData 3.0 机器学习 Notebook Kernel 运行时初始化库，提供 MLflow 深度桥接、Feast gRPC 代理、腾讯云 OpenAPI 签名调用及平台级多租户隔离。
+Author-email: WeData Team <wedata@tencent.com>
+License: MIT
+Project-URL: Homepage, https://wedata.tencent.com
+Project-URL: Documentation, https://wedata.tencent.com/docs
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: feast==0.49.0
+Requires-Dist: grpcio>=1.71.0
+Requires-Dist: mlflow<3.11.0,>=2.0.0
+Requires-Dist: pydantic>=2.10.6
+Requires-Dist: wedata-mlflow-header-plugin>=0.1.2
+Requires-Dist: mlflow-tclake-plugin>=3.0.15
+
+# wedata3-ml-runtime
+
+WeData 3.0 机器学习 Notebook Kernel 运行时初始化库。
+
+## 定位
+
+本包是 WeData 3.0 Notebook 场景下，运行在 DLC 计算容器中的 **Kernel 启动适配层**。它不改变用户代码，而是通过 monkey patching 把开源的 `mlflow` / `feast` 改造成带有 WeData 平台语义（多租户隔离、CAM 鉴权、审计、网关代理）的版本。
+
+> 前身：`wedata-pre-code`（同时支持 2.0 和 3.0）
+>
+> 本包仅承载 WeData 3.0 逻辑。WeData 2.0 仍沿用 `wedata-pre-code`，位于 `../wedata-pre-execute/`（已进入维护模式）。
+
+## 核心能力
+
+1. **装包标记**：配合 DLC 镜像预装 → 运行时刷新 `/opt/spark/pip_list.json` 系统包快照，消除重复装包告警
+2. **环境变量注入**：`WEDATA_WORKSPACE_ID` / `MLFLOW_TRACKING_URI` / `TENCENTCLOUD_SECRET_ID/KEY/TOKEN` / `KERNEL_WEDATA_*` 等 20+ 条
+3. **MLflow 客户端重写**：
+   - `create/get/search/rename/delete experiment` 改走腾讯云 OpenAPI（TC3-HMAC-SHA256 手签）
+   - `MlflowClient` 上所有写操作套装饰器做 workspace_id 校验
+4. **Feast gRPC 代理**：给 `RemoteRegistry` 加 `X-Target-Service-IP/PORT` 请求头走反向代理
+5. **自动 tag 注入**：所有 mlflow 对象自动打上 `wedata.workspace / wedata.datascience.type / mlflow.user`
+
+## 使用
+
+```python
+%pip install wedata3-ml-runtime
+from wedata_ml_runtime.client import Wedata3PreCodeClient
+
+client = Wedata3PreCodeClient(
+    workspace_id="...",
+    base_url="...",
+    region="...",
+    ap_region_id=1,
+    mlflow_gateway_url="...",
+    feast_gateway_url="...",
+    mlflow_proxy_ip="...",
+    mlflow_proxy_port="...",
+    feast_proxy_ip="...",
+    feast_proxy_port="...",
+    kernel_task_name="...",
+    kernel_task_id="...",
+    cloud_sdk_secret_id="...",
+    cloud_sdk_secret_key="...",
+    cloud_sdk_secret_token="...",
+    qcloud_uin="...",
+    qcloud_subuin="...",
+)
+client.init()
+```
+
+## 必传参数
+
+| 参数 | 说明 |
+|------|------|
+| `workspace_id` | 工作空间 ID |
+| `base_url` | WeData 控制台基础 URL |
+| `mlflow_gateway_url` | MLflow Serverless 网关地址 |
+| `feast_gateway_url` | Feast Serverless 网关地址 |
+| `mlflow_proxy_ip` / `mlflow_proxy_port` | MLflow 转发地址 |
+| `feast_proxy_ip` / `feast_proxy_port` | Feast 转发地址 |
+
+## 可选参数
+
+| 参数 | 说明 |
+|------|------|
+| `region` / `ap_region_id` | 地域标识 |
+| `kernel_task_name` / `kernel_task_id` | 任务身份 |
+| `kernel_submit_form_workflow` | 工作流标识 |
+| `cloud_sdk_secret_id` / `cloud_sdk_secret_key` / `cloud_sdk_secret_token` | CAM 临时凭证三元组（12 小时过期） |
+| `cloud_sdk_env` | SDK 环境（`dev` / `test` / `pre`） |
+| `cloud_sdk_user_id` | 测试账户 ID（`cloud_sdk_env=test` 时使用） |
+| `qcloud_uin` / `qcloud_subuin` | 腾讯云主账号 / 子账号 UIN |
+
+## 构建发布
+
+```bash
+bash build.sh
+```
+
+脚本内容：`rm -rf dist/ build/` → `uv build` → `twine upload dist/*`
+
+## 从 wedata-pre-code 迁移
+
+| 维度 | 老包 `wedata-pre-code` | 新包 `wedata3-ml-runtime` |
+|------|----------------------|------------------------|
+| PyPI 名 | `wedata-pre-code` | `wedata3-ml-runtime` |
+| Python 模块 | `wedata_pre_code.wedata3.client` | `wedata_ml_runtime.client` |
+| 客户端类 | `Wedata3PreCodeClient` | `Wedata3PreCodeClient`（保持兼容）|
+| extras | `pip install "wedata-pre-code[wedata-3]"` | 无需 extras，主依赖已覆盖 |
+| 覆盖平台 | 2.0 + 3.0 | **仅 3.0** |
+
+**迁移办法**：把 `%pip install wedata-pre-code[wedata-3]==X.Y.Z` 改为 `%pip install wedata3-ml-runtime==X.Y.Z`，并把 `from wedata_pre_code.wedata3.client import Wedata3PreCodeClient` 改为 `from wedata_ml_runtime.client import Wedata3PreCodeClient`。
+
+## 相关设计文档
+
+- `application/science/doc/plan/wedata-pre-code-image-baseline-plan.md`：DLC 镜像基线 + 运行时 Override 方案

wedata3_ml_runtime-0.0.1/src/wedata3_ml_runtime.egg-info/SOURCES.txt

@@ -0,0 +1,13 @@
+README.md
+pyproject.toml
+src/wedata3_ml_runtime.egg-info/PKG-INFO
+src/wedata3_ml_runtime.egg-info/SOURCES.txt
+src/wedata3_ml_runtime.egg-info/dependency_links.txt
+src/wedata3_ml_runtime.egg-info/requires.txt
+src/wedata3_ml_runtime.egg-info/top_level.txt
+src/wedata_ml_runtime/__init__.py
+src/wedata_ml_runtime/client.py
+src/wedata_ml_runtime/experiment.py
+src/wedata_ml_runtime/common/__init__.py
+src/wedata_ml_runtime/common/base_client.py
+tests/test_refresh_sys_pkg.py

wedata3_ml_runtime-0.0.1/src/wedata3_ml_runtime.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

wedata3_ml_runtime-0.0.1/src/wedata3_ml_runtime.egg-info/requires.txt

@@ -0,0 +1,6 @@
+feast==0.49.0
+grpcio>=1.71.0
+mlflow<3.11.0,>=2.0.0
+pydantic>=2.10.6
+wedata-mlflow-header-plugin>=0.1.2
+mlflow-tclake-plugin>=3.0.15

wedata3_ml_runtime-0.0.1/src/wedata3_ml_runtime.egg-info/top_level.txt

@@ -0,0 +1 @@
+wedata_ml_runtime

wedata3_ml_runtime-0.0.1/src/wedata_ml_runtime/__init__.py

@@ -0,0 +1,121 @@
+"""WeData ML Runtime.
+
+WeData 3.0 Notebook Kernel runtime initialization library. Provides deep
+bridging with MLflow / Feast and platform-level multi-tenant isolation for
+Machine Learning experiments.
+
+History: this package is the WeData 3.0 portion split out of
+`wedata-pre-code` (which originally served both 2.0 and 3.0). Only 3.0 logic
+is carried here.
+"""
+
+# ---------------------------------------------------------------------------
+# MLflow 3.x 兼容垫片
+# ---------------------------------------------------------------------------
+# 背景：部分第三方 MLflow 插件（mlflow-tclake-plugin / wedata-mlflow-header-plugin
+# 等）仍在模块加载期 import 一些老版本 mlflow 的符号；在 mlflow>=3 环境里这些
+# 符号已被改名/移除，导致 Notebook 预执行脚本 ``ImportError`` 无法初始化。
+#
+# 本垫片必须在 ``wedata_ml_runtime`` 被 import 时立即生效（先于 client 里
+# ``import mlflow`` 触发的第三方插件注册链路），以回填别名/注入 stub 的方式
+# 让后续 ``from mlflow.xxx import YYY`` 能命中。
+#
+# 已覆盖符号：
+#   1. ``mlflow.types.llm.ChatResponse``
+#      mlflow 2.x 旧名，3.x 改名为 ``ChatCompletionResponse``。
+#   2. ``mlflow.utils.databricks_utils.is_in_databricks_serverless``
+#      mlflow 早期私有/内部符号，3.x 已移除。WeData/DLC 运行环境不是 Databricks
+#      Serverless，用永远返回 ``False`` 的 stub 兜底即可。
+#   3. ``mlflow.utils.model_utils.RECORD_ENV_VAR_ALLOWLIST``
+#      mlflow 新版本引入的 ``set``，用于标记允许记录到 model 元数据的环境
+#      变量白名单。在老版本 mlflow（如生产环境 Python 3.11 下仍在用的旧包）
+#      里缺失，第三方插件 eager import 会直接 ``ImportError``。这里用空
+#      ``set()`` 兜底——语义等价于 "不额外记录任何环境变量"，对 WeData/DLC
+#      场景安全无副作用。
+#   4. ``mlflow.models.display_utils``（整模块缺失）
+#      mlflow 较新版本才新增的子模块，用于在 Databricks Notebook 里渲染
+#      Agent Eval / ModelInfo 的 HTML 展示组件。老版本 mlflow 根本没有
+#      该子模块，第三方插件 ``import mlflow.models.display_utils`` 会抛
+#      ``ModuleNotFoundError``。这里向 ``sys.modules`` 注入一个伪模块，
+#      并提供 noop 版本的 ``maybe_render_agent_eval_recipe``，保证
+#      ``import``/``from ... import ...`` 两种用法都可过，且在非 Databricks
+#      环境下本就不需要真实渲染。
+#
+# 关联：2026-05 修复 Notebook 预执行脚本因第三方插件 import 老符号/老模块报
+#       ``ImportError: cannot import name 'ChatResponse' / 'is_in_databricks_serverless'
+#         / 'RECORD_ENV_VAR_ALLOWLIST'`` 以及
+#       ``ModuleNotFoundError: No module named 'mlflow.models.display_utils'`` 的问题。
+try:  # pragma: no cover - 防御性兜底，失败也不应影响主流程
+    import mlflow.types.llm as _mlflow_llm  # type: ignore
+
+    if not hasattr(_mlflow_llm, "ChatResponse"):
+        _fallback = getattr(_mlflow_llm, "ChatCompletionResponse", None)
+        if _fallback is not None:
+            _mlflow_llm.ChatResponse = _fallback  # type: ignore[attr-defined]
+except Exception:  # noqa: BLE001 - 不能让垫片报错阻断 runtime 初始化
+    pass
+
+try:  # pragma: no cover
+    import mlflow.utils.databricks_utils as _mlflow_dbx  # type: ignore
+
+    if not hasattr(_mlflow_dbx, "is_in_databricks_serverless"):
+        def _is_in_databricks_serverless_stub(*_args, **_kwargs) -> bool:  # noqa: D401
+            """WeData/DLC 运行环境恒定不是 Databricks Serverless，返回 False。"""
+            return False
+
+        _mlflow_dbx.is_in_databricks_serverless = _is_in_databricks_serverless_stub  # type: ignore[attr-defined]
+except Exception:  # noqa: BLE001
+    pass
+
+try:  # pragma: no cover
+    import mlflow.utils.model_utils as _mlflow_model_utils  # type: ignore
+
+    if not hasattr(_mlflow_model_utils, "RECORD_ENV_VAR_ALLOWLIST"):
+        # 空 set 语义：不额外把环境变量写入 model 元数据，对 WeData/DLC 安全无副作用
+        _mlflow_model_utils.RECORD_ENV_VAR_ALLOWLIST = set()  # type: ignore[attr-defined]
+except Exception:  # noqa: BLE001
+    pass
+
+try:  # pragma: no cover
+    # 整模块缺失：老 mlflow 版本没有 ``mlflow.models.display_utils`` 子模块，
+    # 必须向 sys.modules 注入伪模块，避免第三方插件
+    # ``import mlflow.models.display_utils`` / ``from ... import maybe_render_agent_eval_recipe``
+    # 直接抛 ``ModuleNotFoundError``。
+    import importlib as _importlib
+    import sys as _sys
+    import types as _types
+
+    _DISPLAY_UTILS_FQN = "mlflow.models.display_utils"
+    _need_stub_display_utils = False
+    try:
+        _importlib.import_module(_DISPLAY_UTILS_FQN)
+    except Exception:  # noqa: BLE001 - 任何导入异常都视为"模块不可用"，注入 stub
+        _need_stub_display_utils = True
+
+    if _need_stub_display_utils:
+        _stub = _types.ModuleType(_DISPLAY_UTILS_FQN)
+
+        def _maybe_render_agent_eval_recipe(*_args, **_kwargs):  # noqa: D401
+            """Noop：非 Databricks Notebook 环境下无需渲染 agent eval recipe。"""
+            return None
+
+        _stub.maybe_render_agent_eval_recipe = _maybe_render_agent_eval_recipe  # type: ignore[attr-defined]
+        _stub.__all__ = ["maybe_render_agent_eval_recipe"]  # type: ignore[attr-defined]
+        _sys.modules[_DISPLAY_UTILS_FQN] = _stub
+
+        # 同步把伪模块挂到父模块属性上，保持 ``mlflow.models.display_utils`` 可用
+        try:
+            import mlflow.models as _mlflow_models  # type: ignore
+
+            if not hasattr(_mlflow_models, "display_utils"):
+                _mlflow_models.display_utils = _stub  # type: ignore[attr-defined]
+        except Exception:  # noqa: BLE001
+            pass
+except Exception:  # noqa: BLE001
+    pass
+
+from wedata_ml_runtime.client import Wedata3PreCodeClient
+
+__all__ = ["Wedata3PreCodeClient"]
+
+__version__ = "0.1.8"