You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
437 lines
9.2 KiB
437 lines
9.2 KiB
import uuid
|
|
from datetime import datetime
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
# --- Auth ---
|
|
class LoginRequest(BaseModel):
|
|
username: str
|
|
password: str
|
|
|
|
|
|
class TokenResponse(BaseModel):
|
|
access_token: str
|
|
token_type: str = "bearer"
|
|
user: "UserOut"
|
|
|
|
|
|
# --- User ---
|
|
class UserCreate(BaseModel):
|
|
username: str
|
|
password: str
|
|
display_name: str
|
|
email: str | None = None
|
|
phone: str | None = None
|
|
wecom_user_id: str | None = None
|
|
department_id: uuid.UUID | None = None
|
|
position: str | None = None
|
|
manager_id: uuid.UUID | None = None
|
|
role_ids: list[uuid.UUID] = []
|
|
|
|
|
|
class UserUpdate(BaseModel):
|
|
display_name: str | None = None
|
|
email: str | None = None
|
|
phone: str | None = None
|
|
department_id: uuid.UUID | None = None
|
|
position: str | None = None
|
|
manager_id: uuid.UUID | None = None
|
|
status: str | None = None
|
|
role_ids: list[uuid.UUID] | None = None
|
|
|
|
|
|
class UserOut(BaseModel):
|
|
id: uuid.UUID
|
|
username: str
|
|
display_name: str
|
|
email: str | None = None
|
|
phone: str | None = None
|
|
wecom_user_id: str | None = None
|
|
department_id: uuid.UUID | None = None
|
|
position: str | None = None
|
|
manager_id: uuid.UUID | None = None
|
|
status: str
|
|
roles: list["RoleOut"] = []
|
|
created_at: datetime | None = None
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# --- Department ---
|
|
class DepartmentCreate(BaseModel):
|
|
name: str
|
|
parent_id: uuid.UUID | None = None
|
|
sort_order: int = 0
|
|
|
|
|
|
class DepartmentUpdate(BaseModel):
|
|
name: str | None = None
|
|
parent_id: uuid.UUID | None = None
|
|
sort_order: int | None = None
|
|
|
|
|
|
class DepartmentOut(BaseModel):
|
|
id: uuid.UUID
|
|
name: str
|
|
parent_id: uuid.UUID | None = None
|
|
path: str
|
|
level: int
|
|
sort_order: int
|
|
children: list["DepartmentOut"] = []
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# --- Role ---
|
|
class RoleCreate(BaseModel):
|
|
name: str
|
|
code: str = ""
|
|
description: str | None = None
|
|
data_scope: str = "self_only"
|
|
permission_ids: list[uuid.UUID] = []
|
|
|
|
|
|
class RoleUpdate(BaseModel):
|
|
name: str | None = None
|
|
description: str | None = None
|
|
data_scope: str | None = None
|
|
permission_ids: list[uuid.UUID] | None = None
|
|
|
|
|
|
class RoleOut(BaseModel):
|
|
id: uuid.UUID
|
|
name: str
|
|
code: str = ""
|
|
description: str | None = None
|
|
is_system: bool
|
|
data_scope: str
|
|
permissions: list[str] = []
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# --- Permission ---
|
|
class PermissionOut(BaseModel):
|
|
id: uuid.UUID
|
|
code: str
|
|
name: str
|
|
resource: str
|
|
action: str
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# --- Task ---
|
|
class TaskCreate(BaseModel):
|
|
title: str
|
|
content: str | None = None
|
|
assignee_id: uuid.UUID
|
|
priority: str = "normal"
|
|
deadline: datetime | None = None
|
|
push_to_wecom: bool = True
|
|
|
|
|
|
class TaskUpdate(BaseModel):
|
|
title: str | None = None
|
|
content: str | None = None
|
|
status: str | None = None
|
|
priority: str | None = None
|
|
deadline: datetime | None = None
|
|
|
|
|
|
class TaskOut(BaseModel):
|
|
id: uuid.UUID
|
|
title: str
|
|
content: str | None = None
|
|
assigner_id: uuid.UUID | None = None
|
|
assignee_id: uuid.UUID
|
|
status: str
|
|
priority: str
|
|
deadline: datetime | None = None
|
|
created_at: datetime | None = None
|
|
updated_at: datetime | None = None
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# --- Employee Analysis ---
|
|
class EmployeeAnalysis(BaseModel):
|
|
employee_name: str
|
|
department: str
|
|
period: str
|
|
task_completion_rate: float
|
|
active_days: int
|
|
total_interactions: int
|
|
main_topics: list[str]
|
|
efficiency_trend: str
|
|
efficiency_detail: str
|
|
strengths: list[str]
|
|
growth_suggestions: list[str]
|
|
personality_traits: str
|
|
|
|
|
|
# --- Flow ---
|
|
class TriggerNodeConfig(BaseModel):
|
|
event_type: str = "text_message"
|
|
|
|
class LLMNodeConfig(BaseModel):
|
|
system_prompt: str = ""
|
|
model: str = "gpt-4o-mini"
|
|
temperature: float = 0.7
|
|
agent_id: str = ""
|
|
|
|
class ToolNodeConfig(BaseModel):
|
|
tool_name: str = ""
|
|
|
|
class MCPNodeConfig(BaseModel):
|
|
mcp_server: str = ""
|
|
tool_name: str = ""
|
|
|
|
class WeComNotifyNodeConfig(BaseModel):
|
|
message_template: str = ""
|
|
target: str = ""
|
|
|
|
class ConditionNodeConfig(BaseModel):
|
|
condition: str = ""
|
|
|
|
class RAGNodeConfig(BaseModel):
|
|
knowledge_base: str = ""
|
|
top_k: int = 5
|
|
|
|
class OutputNodeConfig(BaseModel):
|
|
format: str = "text"
|
|
|
|
class FlowNode(BaseModel):
|
|
id: str | None = None
|
|
type: str
|
|
label: str | None = None
|
|
config: dict = {}
|
|
|
|
|
|
class FlowEdge(BaseModel):
|
|
source: str | None = None
|
|
target: str | None = None
|
|
from_field: str | None = Field(None, alias="from")
|
|
to_field: str | None = Field(None, alias="to")
|
|
|
|
class Config:
|
|
populate_by_name = True
|
|
|
|
|
|
class FlowDefinitionCreate(BaseModel):
|
|
name: str
|
|
description: str | None = None
|
|
trigger: dict = {}
|
|
nodes: list[FlowNode]
|
|
edges: list[FlowEdge]
|
|
|
|
|
|
class FlowDefinitionUpdate(BaseModel):
|
|
name: str | None = None
|
|
description: str | None = None
|
|
nodes: list[FlowNode] | None = None
|
|
edges: list[FlowEdge] | None = None
|
|
trigger: dict | None = None
|
|
|
|
|
|
class FlowDefinitionOut(BaseModel):
|
|
id: uuid.UUID
|
|
name: str
|
|
description: str | None = None
|
|
version: int
|
|
status: str
|
|
definition_json: dict
|
|
published_to_wecom: bool
|
|
created_at: datetime | None = None
|
|
updated_at: datetime | None = None
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# --- MCP ---
|
|
class MCPServiceCreate(BaseModel):
|
|
name: str
|
|
transport: str = "http"
|
|
url: str | None = None
|
|
command: str | None = None
|
|
args: list[str] = []
|
|
env: dict[str, str] = {}
|
|
|
|
|
|
class MCPServiceUpdate(BaseModel):
|
|
transport: str | None = None
|
|
url: str | None = None
|
|
command: str | None = None
|
|
args: list[str] | None = None
|
|
env: dict[str, str] | None = None
|
|
|
|
|
|
class MCPServiceOut(BaseModel):
|
|
id: uuid.UUID
|
|
name: str
|
|
transport: str
|
|
url: str | None = None
|
|
command: str | None = None
|
|
status: str = "disconnected"
|
|
tools: list[dict] = []
|
|
creator_id: uuid.UUID | None = None
|
|
created_at: datetime | None = None
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# --- Agent Config ---
|
|
class AgentConfigCreate(BaseModel):
|
|
name: str
|
|
description: str | None = None
|
|
system_prompt: str = ""
|
|
model: str = "gpt-4o-mini"
|
|
temperature: float = 0.7
|
|
tools: list[str] = []
|
|
|
|
|
|
class AgentConfigUpdate(BaseModel):
|
|
name: str | None = None
|
|
description: str | None = None
|
|
system_prompt: str | None = None
|
|
model: str | None = None
|
|
temperature: float | None = None
|
|
tools: list[str] | None = None
|
|
status: str | None = None
|
|
|
|
|
|
class AgentConfigOut(BaseModel):
|
|
id: uuid.UUID
|
|
name: str
|
|
description: str | None = None
|
|
system_prompt: str
|
|
model: str
|
|
temperature: float
|
|
tools: list[str] = []
|
|
status: str
|
|
creator_id: uuid.UUID | None = None
|
|
created_at: datetime | None = None
|
|
updated_at: datetime | None = None
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# --- Notification ---
|
|
class NotificationTemplateCreate(BaseModel):
|
|
name: str
|
|
code: str
|
|
channel: str = "wecom"
|
|
title_template: str | None = None
|
|
body_template: str
|
|
variables: list[str] = []
|
|
|
|
|
|
class NotificationTemplateOut(BaseModel):
|
|
id: uuid.UUID
|
|
name: str
|
|
code: str
|
|
channel: str
|
|
title_template: str | None = None
|
|
body_template: str
|
|
variables: list[str] = []
|
|
is_system: bool = False
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# --- Document ---
|
|
class DocumentUploadOut(BaseModel):
|
|
file_id: uuid.UUID
|
|
filename: str
|
|
file_size: int
|
|
content_type: str
|
|
upload_time: datetime
|
|
|
|
|
|
class DocumentParseResult(BaseModel):
|
|
file_id: uuid.UUID
|
|
filename: str
|
|
content: str
|
|
metadata: dict = {}
|
|
|
|
|
|
# --- Audit ---
|
|
class AuditQueryParams(BaseModel):
|
|
page: int = 1
|
|
page_size: int = 20
|
|
action: str | None = None
|
|
resource: str | None = None
|
|
operator_id: uuid.UUID | None = None
|
|
date_from: datetime | None = None
|
|
date_to: datetime | None = None
|
|
|
|
|
|
class AuditLogOut(BaseModel):
|
|
id: uuid.UUID
|
|
operator_id: uuid.UUID | None = None
|
|
action: str
|
|
resource: str | None = None
|
|
resource_id: str | None = None
|
|
detail: dict | None = None
|
|
ip_address: str | None = None
|
|
created_at: datetime | None = None
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class AuditLogPage(BaseModel):
|
|
items: list[AuditLogOut]
|
|
total: int
|
|
page: int
|
|
page_size: int
|
|
|
|
|
|
# --- System Metrics ---
|
|
class SystemMetricOut(BaseModel):
|
|
id: uuid.UUID
|
|
metric_type: str
|
|
value: dict
|
|
collected_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class SystemHealthOut(BaseModel):
|
|
status: str
|
|
service: str
|
|
uptime_seconds: float
|
|
db_connected: bool
|
|
redis_connected: bool
|
|
active_users: int
|
|
memory_mb: float
|
|
cpu_percent: float
|
|
|
|
|
|
class UsageStatsOut(BaseModel):
|
|
total_users: int
|
|
active_users_today: int
|
|
total_sessions: int
|
|
total_messages: int
|
|
total_tasks: int
|
|
total_flows: int
|
|
published_flows: int
|
|
api_calls_today: int
|
|
avg_response_time_ms: float
|
|
|
|
|
|
# --- Generic Response ---
|
|
class ApiResponse(BaseModel):
|
|
code: int = 200
|
|
message: str = "success"
|
|
data: dict | list | None = None
|