107 lines
2.9 KiB
Python
107 lines
2.9 KiB
Python
from typing import Optional, Dict, Any, List
|
|
|
|
from pydantic import BaseModel, ConfigDict
|
|
|
|
|
|
class ChannelConfigRequest(BaseModel):
|
|
channel_type: str
|
|
external_app_id: Optional[str] = None
|
|
app_secret: Optional[str] = None
|
|
internal_port: Optional[int] = None
|
|
is_active: bool = True
|
|
extra_config: Optional[Dict[str, Any]] = None
|
|
|
|
|
|
class ChannelConfigUpdateRequest(BaseModel):
|
|
channel_type: Optional[str] = None
|
|
external_app_id: Optional[str] = None
|
|
app_secret: Optional[str] = None
|
|
internal_port: Optional[int] = None
|
|
is_active: Optional[bool] = None
|
|
extra_config: Optional[Dict[str, Any]] = None
|
|
|
|
|
|
class BotCreateRequest(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
id: str
|
|
name: str
|
|
enabled: Optional[bool] = True
|
|
image_tag: str
|
|
access_password: Optional[str] = None
|
|
llm_provider: str
|
|
llm_model: str
|
|
api_key: str
|
|
api_base: str
|
|
temperature: float = 0.2
|
|
top_p: float = 1.0
|
|
max_tokens: int = 8192
|
|
cpu_cores: float = 1.0
|
|
memory_mb: int = 1024
|
|
storage_gb: int = 10
|
|
system_timezone: str
|
|
soul_md: str
|
|
agents_md: str
|
|
user_md: str
|
|
tools_md: str
|
|
tools_config: Optional[Dict[str, Any]] = None
|
|
env_params: Optional[Dict[str, str]] = None
|
|
identity_md: str
|
|
channels: Optional[List[ChannelConfigRequest]] = None
|
|
send_progress: Optional[bool] = None
|
|
send_tool_hints: Optional[bool] = None
|
|
|
|
|
|
class BotUpdateRequest(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
name: Optional[str] = None
|
|
enabled: Optional[bool] = None
|
|
image_tag: Optional[str] = None
|
|
access_password: Optional[str] = None
|
|
llm_provider: Optional[str] = None
|
|
llm_model: Optional[str] = None
|
|
api_key: Optional[str] = None
|
|
api_base: Optional[str] = None
|
|
temperature: Optional[float] = None
|
|
top_p: Optional[float] = None
|
|
max_tokens: Optional[int] = None
|
|
cpu_cores: Optional[float] = None
|
|
memory_mb: Optional[int] = None
|
|
storage_gb: Optional[int] = None
|
|
system_timezone: Optional[str] = None
|
|
agents_md: Optional[str] = None
|
|
soul_md: Optional[str] = None
|
|
user_md: Optional[str] = None
|
|
tools_md: Optional[str] = None
|
|
tools_config: Optional[Dict[str, Any]] = None
|
|
env_params: Optional[Dict[str, str]] = None
|
|
identity_md: Optional[str] = None
|
|
send_progress: Optional[bool] = None
|
|
send_tool_hints: Optional[bool] = None
|
|
|
|
|
|
class BotToolsConfigUpdateRequest(BaseModel):
|
|
tools_config: Optional[Dict[str, Any]] = None
|
|
|
|
|
|
class BotMcpConfigUpdateRequest(BaseModel):
|
|
mcp_servers: Optional[Dict[str, Any]] = None
|
|
|
|
|
|
class BotEnvParamsUpdateRequest(BaseModel):
|
|
env_params: Optional[Dict[str, str]] = None
|
|
|
|
|
|
class BotPageAuthLoginRequest(BaseModel):
|
|
password: str
|
|
|
|
|
|
class CommandRequest(BaseModel):
|
|
command: Optional[str] = None
|
|
attachments: Optional[List[str]] = None
|
|
|
|
|
|
class MessageFeedbackRequest(BaseModel):
|
|
feedback: Optional[str] = None
|