130 lines
5.2 KiB
Python
130 lines
5.2 KiB
Python
|
#
|
|||
|
|
# Copyright 2024 The InfiniFlow Authors. All Rights Reserved.
|
|||
|
|
#
|
|||
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|||
|
|
# you may not use this file except in compliance with the License.
|
|||
|
|
# You may obtain a copy of the License at
|
|||
|
|
#
|
|||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|||
|
|
#
|
|||
|
|
# Unless required by applicable law or agreed to in writing, software
|
|||
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|||
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|||
|
|
# See the License for the specific language governing permissions and
|
|||
|
|
# limitations under the License.
|
|||
|
|
#
|
|||
|
|
|
|||
|
|
from typing import Optional, List, Dict, Any, Union
|
|||
|
|
from pydantic import BaseModel, Field
|
|||
|
|
|
|||
|
|
|
|||
|
|
class DeleteCanvasRequest(BaseModel):
|
|||
|
|
"""删除画布请求"""
|
|||
|
|
canvas_ids: List[str] = Field(..., description="画布ID列表")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SaveCanvasRequest(BaseModel):
|
|||
|
|
"""保存画布请求"""
|
|||
|
|
dsl: Union[str, Dict[str, Any]] = Field(..., description="DSL配置")
|
|||
|
|
title: str = Field(..., description="画布标题")
|
|||
|
|
id: Optional[str] = Field(default=None, description="画布ID（更新时提供）")
|
|||
|
|
canvas_category: Optional[str] = Field(default=None, description="画布类别")
|
|||
|
|
description: Optional[str] = Field(default=None, description="描述")
|
|||
|
|
permission: Optional[str] = Field(default=None, description="权限")
|
|||
|
|
avatar: Optional[str] = Field(default=None, description="头像")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class CompletionRequest(BaseModel):
|
|||
|
|
"""完成/运行画布请求"""
|
|||
|
|
id: str = Field(..., description="画布ID")
|
|||
|
|
query: Optional[str] = Field(default="", description="查询内容")
|
|||
|
|
files: Optional[List[str]] = Field(default=[], description="文件列表")
|
|||
|
|
inputs: Optional[Dict[str, Any]] = Field(default={}, description="输入参数")
|
|||
|
|
user_id: Optional[str] = Field(default=None, description="用户ID")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class RerunRequest(BaseModel):
|
|||
|
|
"""重新运行请求"""
|
|||
|
|
id: str = Field(..., description="流水线ID")
|
|||
|
|
dsl: Dict[str, Any] = Field(..., description="DSL配置")
|
|||
|
|
component_id: str = Field(..., description="组件ID")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ResetCanvasRequest(BaseModel):
|
|||
|
|
"""重置画布请求"""
|
|||
|
|
id: str = Field(..., description="画布ID")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class InputFormQuery(BaseModel):
|
|||
|
|
"""获取输入表单查询参数"""
|
|||
|
|
id: str = Field(..., description="画布ID")
|
|||
|
|
component_id: str = Field(..., description="组件ID")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class DebugRequest(BaseModel):
|
|||
|
|
"""调试请求"""
|
|||
|
|
id: str = Field(..., description="画布ID")
|
|||
|
|
component_id: str = Field(..., description="组件ID")
|
|||
|
|
params: Dict[str, Any] = Field(..., description="参数")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class TestDBConnectRequest(BaseModel):
|
|||
|
|
"""测试数据库连接请求"""
|
|||
|
|
db_type: str = Field(..., description="数据库类型")
|
|||
|
|
database: str = Field(..., description="数据库名")
|
|||
|
|
username: str = Field(..., description="用户名")
|
|||
|
|
host: str = Field(..., description="主机")
|
|||
|
|
port: str = Field(..., description="端口")
|
|||
|
|
password: str = Field(..., description="密码")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ListCanvasQuery(BaseModel):
|
|||
|
|
"""列出画布查询参数"""
|
|||
|
|
keywords: Optional[str] = Field(default="", description="关键词")
|
|||
|
|
page: Optional[int] = Field(default=0, description="页码")
|
|||
|
|
page_size: Optional[int] = Field(default=0, description="每页大小")
|
|||
|
|
orderby: Optional[str] = Field(default="create_time", description="排序字段")
|
|||
|
|
desc: Optional[str] = Field(default="true", description="是否降序")
|
|||
|
|
canvas_category: Optional[str] = Field(default=None, description="画布类别")
|
|||
|
|
owner_ids: Optional[str] = Field(default="", description="所有者ID列表（逗号分隔）")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SettingRequest(BaseModel):
|
|||
|
|
"""画布设置请求"""
|
|||
|
|
id: str = Field(..., description="画布ID")
|
|||
|
|
title: str = Field(..., description="标题")
|
|||
|
|
permission: str = Field(..., description="权限")
|
|||
|
|
description: Optional[str] = Field(default=None, description="描述")
|
|||
|
|
avatar: Optional[str] = Field(default=None, description="头像")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class TraceQuery(BaseModel):
|
|||
|
|
"""追踪查询参数"""
|
|||
|
|
canvas_id: str = Field(..., description="画布ID")
|
|||
|
|
message_id: str = Field(..., description="消息ID")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ListSessionsQuery(BaseModel):
|
|||
|
|
"""列出会话查询参数"""
|
|||
|
|
user_id: Optional[str] = Field(default=None, description="用户ID")
|
|||
|
|
page: Optional[int] = Field(default=1, description="页码")
|
|||
|
|
page_size: Optional[int] = Field(default=30, description="每页大小")
|
|||
|
|
keywords: Optional[str] = Field(default=None, description="关键词")
|
|||
|
|
from_date: Optional[str] = Field(default=None, description="开始日期")
|
|||
|
|
to_date: Optional[str] = Field(default=None, description="结束日期")
|
|||
|
|
orderby: Optional[str] = Field(default="update_time", description="排序字段")
|
|||
|
|
desc: Optional[str] = Field(default="true", description="是否降序")
|
|||
|
|
dsl: Optional[str] = Field(default="true", description="是否包含DSL")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class DownloadQuery(BaseModel):
|
|||
|
|
"""下载查询参数"""
|
|||
|
|
id: str = Field(..., description="文件ID")
|
|||
|
|
created_by: str = Field(..., description="创建者ID")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class UploadQuery(BaseModel):
|
|||
|
|
"""上传查询参数"""
|
|||
|
|
url: Optional[str] = Field(default=None, description="URL（可选，用于从URL下载）")
|
|||
|
|
|