modify: aibox_sophon_adapter plugins

example/models/__init__.py

@@ -5,31 +5,32 @@
 
 from .pydantic import AddGetRequest
 from .pydantic import AddGetResponse
+from .pydantic import AddMqttRequest
 from .pydantic import AddPostRequest
 from .pydantic import AddPostResponse
 from .pydantic import CountGetResponse
 from .pydantic import DivideGetRequest
 from .pydantic import DivideGetResponse
 from .pydantic import ExampleResponse
+from .pydantic import HeartbeatMqttRequest
 from .pydantic import MultiplyGetRequest
 from .pydantic import MultiplyGetResponse
 from .pydantic import SubtractGetRequest
 from .pydantic import SubtractGetResponse
 
-from .counter import Counter
-
 __all__ = [
     "AddGetRequest",
     "AddGetResponse",
+    "AddMqttRequest",
     "AddPostRequest",
     "AddPostResponse",
     "CountGetResponse",
     "DivideGetRequest",
     "DivideGetResponse",
     "ExampleResponse",
+    "HeartbeatMqttRequest",
     "MultiplyGetRequest",
     "MultiplyGetResponse",
     "SubtractGetRequest",
     "SubtractGetResponse",
-    "Counter"
 ]

example/models/pydantic.py

@@ -30,6 +30,11 @@ class AddPostResponse(BaseModel):
     operation: str = Field(default='add')
     timestamp: Optional[float] = Field(default=0.0)
 
+class AddMqttRequest(BaseModel):
+    """Add two numbers via MQTT — 请求体"""
+    a: float = Field(...)
+    b: float = Field(...)
+
 class CountGetResponse(BaseModel):
     """Get visit count and increment — 响应体"""
     name: str = Field(...)
@@ -47,6 +52,10 @@ class DivideGetResponse(BaseModel):
     timestamp: Optional[float] = Field(default=0.0)
     error: Optional[str] = Field(default='')
 
+class HeartbeatMqttRequest(BaseModel):
+    """Heartbeat ping (no response) — 请求体"""
+    worker_id: str = Field(...)
+
 class MultiplyGetRequest(BaseModel):
     """Multiply two numbers — 请求体"""
     a: float = Field(...)
@@ -71,4 +80,4 @@ class SubtractGetResponse(BaseModel):
 
 
 # 所有响应模型的联合类型
-ExampleResponse = SubtractGetResponse | CountGetResponse | MultiplyGetResponse | AddPostResponse | AddGetResponse | DivideGetResponse
+ExampleResponse = AddPostResponse | AddGetResponse | DivideGetResponse | SubtractGetResponse | CountGetResponse | MultiplyGetResponse