51 lines
2.4 KiB
Python
51 lines
2.4 KiB
Python
from typing import Optional
|
||
from sqlmodel import Field, SQLModel
|
||
from datetime import datetime
|
||
|
||
class BotInstance(SQLModel, table=True):
|
||
id: str = Field(primary_key=True)
|
||
name: str
|
||
avatar_model: str = Field(default="base")
|
||
avatar_skin: Optional[str] = Field(default="blue_suit")
|
||
system_prompt: Optional[str] = Field(default=None)
|
||
soul_md: Optional[str] = Field(default=None)
|
||
agents_md: Optional[str] = Field(default=None)
|
||
user_md: Optional[str] = Field(default=None)
|
||
tools_md: Optional[str] = Field(default=None)
|
||
tools_config_json: Optional[str] = Field(default=None) # JSON string for runtime tools config
|
||
identity_md: Optional[str] = Field(default=None)
|
||
llm_provider: str
|
||
llm_model: str
|
||
api_key: str # Encrypted
|
||
api_base: Optional[str] = Field(default=None)
|
||
temperature: float = Field(default=0.2)
|
||
top_p: float = Field(default=1.0)
|
||
max_tokens: int = Field(default=8192)
|
||
presence_penalty: float = Field(default=0.0)
|
||
frequency_penalty: float = Field(default=0.0)
|
||
send_progress: bool = Field(default=False)
|
||
send_tool_hints: bool = Field(default=False)
|
||
workspace_dir: str = Field(unique=True)
|
||
docker_status: str = Field(default="STOPPED", index=True)
|
||
current_state: Optional[str] = Field(default="IDLE")
|
||
last_action: Optional[str] = Field(default=None)
|
||
image_tag: str = Field(default="nanobot-base:v0.1.4") # 记录该机器人使用的镜像版本
|
||
created_at: datetime = Field(default_factory=datetime.utcnow)
|
||
updated_at: datetime = Field(default_factory=datetime.utcnow)
|
||
|
||
class BotMessage(SQLModel, table=True):
|
||
id: Optional[int] = Field(default=None, primary_key=True)
|
||
bot_id: str = Field(foreign_key="botinstance.id", index=True)
|
||
role: str = Field(index=True) # user | assistant | system
|
||
text: str
|
||
media_json: Optional[str] = Field(default=None) # JSON string list of workspace-relative file paths
|
||
created_at: datetime = Field(default_factory=datetime.utcnow, index=True)
|
||
|
||
class NanobotImage(SQLModel, table=True):
|
||
tag: str = Field(primary_key=True) # e.g., nanobot-base:v0.1.4
|
||
image_id: Optional[str] = Field(default=None) # Docker 内部的 Image ID
|
||
version: str # e.g., 0.1.4
|
||
status: str = Field(default="READY") # READY, BUILDING, ERROR
|
||
source_dir: Optional[str] = Field(default=None) # 本地来源标识（如 manual）
|
||
created_at: datetime = Field(default_factory=datetime.utcnow)
|