dify插件初步构建

difyPlugin/app/schemas/__init__.py

@@ -0,0 +1,4 @@
+from app.schemas.base import ResultDomain
+from app.schemas.plugin import PluginRequest, PluginResponse
+
+__all__ = ["ResultDomain", "PluginRequest", "PluginResponse"]

difyPlugin/app/schemas/base.py

@@ -0,0 +1,52 @@
+"""统一返回类型定义"""
+from typing import TypeVar, Generic, Optional, List, Any
+from pydantic import BaseModel, Field
+
+T = TypeVar('T')
+
+
+class PageDomain(BaseModel, Generic[T]):
+    """分页数据模型"""
+    page: int = Field(default=1, description="当前页码")
+    pageSize: int = Field(default=10, description="每页大小")
+    total: int = Field(default=0, description="总记录数")
+    dataList: Optional[List[T]] = Field(default=None, description="数据列表")
+
+
+class ResultDomain(BaseModel, Generic[T]):
+    """统一返回类型"""
+    code: Optional[int] = Field(default=None, description="状态码")
+    success: Optional[bool] = Field(default=None, description="是否成功")
+    message: Optional[str] = Field(default=None, description="返回消息")
+    data: Optional[T] = Field(default=None, description="单条数据")
+    dataList: Optional[List[T]] = Field(default=None, description="数据列表")
+    pageDomain: Optional[PageDomain[T]] = Field(default=None, description="分页数据")
+    
+    @staticmethod
+    def ok(message: str = "success", data: Any = None) -> "ResultDomain":
+        """成功返回 - 单条数据"""
+        return ResultDomain(code=200, success=True, message=message, data=data)
+    
+    @staticmethod
+    def ok_list(message: str = "success", data_list: List[Any] = None) -> "ResultDomain":
+        """成功返回 - 数据列表"""
+        return ResultDomain(code=200, success=True, message=message, dataList=data_list)
+    
+    @staticmethod
+    def ok_page(message: str = "success", page_domain: "PageDomain" = None) -> "ResultDomain":
+        """成功返回 - 分页数据"""
+        result = ResultDomain(code=200, success=True, message=message, pageDomain=page_domain)
+        if page_domain:
+            result.dataList = page_domain.dataList
+        return result
+    
+    @staticmethod
+    def fail(message: str = "failure", code: int = 500) -> "ResultDomain":
+        """失败返回"""
+        return ResultDomain(code=code, success=False, message=message)
+    
+    model_config = {
+        "json_schema_extra": {
+            "examples": [{"code": 200, "success": True, "message": "操作成功"}]
+        }
+    }

difyPlugin/app/schemas/plugin.py

@@ -0,0 +1,43 @@
+"""插件相关数据模型"""
+from typing import Optional, Dict, Any
+from pydantic import BaseModel, Field
+
+
+class PluginRequest(BaseModel):
+    """
+    插件请求模型
+    
+    Attributes:
+        plugin_id: 插件ID
+        action: 执行动作
+        params: 请求参数
+    """
+    plugin_id: str = Field(..., description="插件ID", examples=["plugin_001"])
+    action: str = Field(..., description="执行动作", examples=["execute"])
+    params: Optional[Dict[str, Any]] = Field(default=None, description="请求参数")
+    
+    model_config = {
+        "json_schema_extra": {
+            "examples": [
+                {
+                    "plugin_id": "plugin_001",
+                    "action": "execute",
+                    "params": {"key": "value"}
+                }
+            ]
+        }
+    }
+
+
+class PluginResponse(BaseModel):
+    """
+    插件响应模型
+    
+    Attributes:
+        plugin_id: 插件ID
+        result: 执行结果
+        status: 执行状态
+    """
+    plugin_id: str = Field(..., description="插件ID")
+    result: Optional[Dict[str, Any]] = Field(default=None, description="执行结果")
+    status: str = Field(default="success", description="执行状态")