modify: aibox_sophon_adapter plugins

This commit is contained in:
2026-07-27 14:28:07 +08:00
parent 2b7b8ee08c
commit 79eb7fe5cc
23 changed files with 1380 additions and 61 deletions
+210 -12
View File
@@ -1,19 +1,217 @@
# 插件目录
# Worker 插件开发指南
每个子目录代表一个 Worker 插件,包含以下约定结构:
每个子目录代表一个独立的 Worker 插件,封装特定领域的业务逻辑。插件通过 Zenoh RPC 对外提供服务,
HTTP 路由由框架根据 TOML 配置自动生成。
---
## 目录结构约定
```text
plugin_name/
├── init.py # 提供 create_workers 和 cleanup 工厂函数
├── router/ # RPC 方法实现(使用 @rpc_method)
├── models/ # 由 gen_models 生成的 Pydantic 模型(自动生成,勿手动修改)
├── global_var.py # 全局状态管理(如数据库连接)
└── README.md # 插件说明
├── __init__.py # 入口:create_workers() 工厂 + cleanup()
├── global_var.py # 模块级全局状态(数据库连接、API 客户端等)
├── router/ # RPC 端点实现
│ ├── __init__.py # 导入所有端点模块,__all__ 导出
│ └── <操作名>.py # 每个端点一个文件,使用 @rpc_method 装饰器
├── models/ # 数据模型
│ ├── __init__.py # 模型导出(自动生成,勿手动修改)
│ ├── pydantic.py # Pydantic 请求/响应模型(由 gen_models.py 自动生成)
│ └── <表名>.py # SQLModel ORM 模型(如需要数据库)
└── README.md # 插件说明文档
```
## 开发新插件
---
1. 在 `config/workers/` 下创建同名目录,编写路由 TOML
2. 运行 `uv run gen_models config/workers/<plugin> src/plugins/<plugin>`
3. 实现 `router/*.py` 中的业务逻辑
4. 在 `__init__.py` 中实现 `create_workers` 工厂函数
## 开发新插件:完整流程
### 第一步:创建目录结构
```bash
mkdir -p src/plugins/<plugin_name>/router
mkdir -p src/plugins/<plugin_name>/models
touch src/plugins/<plugin_name>/__init__.py
touch src/plugins/<plugin_name>/global_var.py
touch src/plugins/<plugin_name>/router/__init__.py
```
### 第二步:注册插件
在 `config/workers/config.toml` 中添加:
```toml
[[workers]]
type = "<plugin_name>"
module = "plugins.<plugin_name>"
factory = "create_workers"
cleanup = "cleanup"
count = 3
[workers.config]
type = "<plugin_name>"
# 自定义配置项...
```
### 第三步:编写端点 TOML 配置
创建 `config/workers/<plugin_name>/config.toml`(分组配置):
```toml
[worker]
defaults = { timeout = 3.0, auth_required = true, cacheable = false, internal = false }
[worker.registration]
router_prefix = "/<prefix>"
```
创建每个操作的 `.toml` 文件,如 `config/workers/<plugin_name>/my_action.toml`:
```toml
[POST]
summary = "我的操作"
auth_required = true
[POST.response_model]
result = { type = "str", required = true }
message = { type = "str", default = "" }
```
### 第四步:生成 Pydantic 模型
```bash
uv run python -m chongming.scripts.gen_models
```
这会自动在 `src/plugins/<plugin_name>/models/` 下生成 `pydantic.py` 和更新 `__init__.py`。
### 第五步:实现 RPC 端点
`src/plugins/<plugin_name>/router/my_action.py`:
```python
from chongming_worker.decorators import rpc_method
from ..models.pydantic import MyActionPostRequest, MyActionPostResponse
@rpc_method(
subject="<plugin_name>.my_action",
method="POST",
response_model=MyActionPostResponse,
)
async def my_action(request: MyActionPostRequest) -> MyActionPostResponse:
"""业务逻辑实现"""
result = await do_something(request)
return MyActionPostResponse(result="success", message="done")
```
### 第六步:更新 router 导出
`src/plugins/<plugin_name>/router/__init__.py`:
```python
from . import my_action
__all__ = ["my_action"]
```
### 第七步:实现工厂函数
`src/plugins/<plugin_name>/__init__.py`:
```python
from chongming_worker.runtime import create_workers as _create_workers
def create_workers(session, config, count, worker_manager):
"""插件入口:初始化全局资源,启动 Worker"""
from . import router # 触发 @rpc_method 注册
from .global_var import set_db
# 初始化数据库、客户端等
from chongming_database import Database
db = Database(config["data_dir"] / "mydb.db")
# await db.initialize() -- 在异步环境中
set_db(db)
_create_workers(session, config, count, worker_manager)
async def cleanup(worker_manager):
"""清理资源"""
from .global_var import db
if db:
await db.dispose()
```
---
## 为现有插件添加新端点
1. 在 `config/workers/<plugin_name>/` 下创建 `<新操作>.toml`
2. 运行 `uv run python -m chongming.scripts.gen_models`
3. 在 `router/` 下创建 `<新操作>.py`,引用生成的模型
4. 更新 `router/__init__.py`
5. 重启服务
---
## 全局变量模式
每个插件通过 `global_var.py` 管理模块级共享资源,避免跨请求的状态冲突:
```python
# global_var.py
db: "Database | None" = None
def set_db(database: "Database"):
global db
db = database
```
在端点中直接引用:
```python
from ..global_var import db
@rpc_method(subject="plugin.action", method="GET", ...)
async def handler(request: RequestModel):
async with db.get_session() as session:
# 执行数据库操作
...
```
---
## 装饰器说明
```python
@rpc_method(
subject="<plugin>.<operation>", # Zenoh 主题(必需)
method="GET", # HTTP 方法:GET 或 POST
response_model=ResponseCls, # Pydantic 响应模型类
)
async def handler(request: RequestCls) -> ResponseCls:
...
```
- **GET 端点**:参数通过 URL query string 传递,在 TOML 中用 `params` 定义
- **POST 端点**:参数通过 JSON body 传递,自动绑定到 Pydantic 请求模型
- **internal 端点**:设置 `internal = true` 后,不注册 HTTP 路由,仅可通过 Zenoh RPC 内部调用
---
## 命名约定
| 概念 | 约定 | 示例 |
| ----------------- | --------------- | ----------------------------------- |
| 插件名 | 小写下划线 | `doubao`, `auth`, `example` |
| 操作名 | 小写下划线 | `analyze`, `add`, `change_password` |
| Zenoh subject | `<插件>.<操作>` | `example.add`, `doubao.analyze` |
| TOML 文件名 | 与操作名一致 | `analyze.toml`, `add.toml` |
| Python 路由文件名 | 与操作名一致 | `analyze.py`, `add.py` |
---
## 现有插件
| 插件 | 说明 | 文档 |
| ---------------------------- | -------------------------------------- | ---------------- |
| [auth](auth/README.md) | 用户注册、登录、JWT 鉴权、用户信息管理 | 含 SQLite 数据库 |
| [doubao](doubao/README.md) | 火山引擎豆包视觉理解模型集成 | 第三方 API 调用 |
| [example](example/README.md) | 四则运算示例 + 访问计数器 | 入门参考 |