78 lines
1.8 KiB
Python
78 lines
1.8 KiB
Python
from datetime import datetime
|
|
from typing import Literal
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
TerminalMode = Literal[1, 2, 3]
|
|
SearchProvider = Literal["brave", "searxng"]
|
|
|
|
|
|
class HealthStatus(BaseModel):
|
|
service: str = "wiseclaw"
|
|
status: str = "ok"
|
|
timestamp: datetime = Field(default_factory=datetime.utcnow)
|
|
|
|
|
|
class SettingRecord(BaseModel):
|
|
key: str
|
|
value: str
|
|
updated_at: datetime = Field(default_factory=datetime.utcnow)
|
|
|
|
|
|
class ToolToggle(BaseModel):
|
|
name: str
|
|
enabled: bool
|
|
|
|
|
|
class UserRecord(BaseModel):
|
|
telegram_user_id: int
|
|
username: str | None = None
|
|
display_name: str | None = None
|
|
is_active: bool = True
|
|
|
|
|
|
class RuntimeSettings(BaseModel):
|
|
terminal_mode: TerminalMode = 3
|
|
search_provider: SearchProvider = "brave"
|
|
ollama_base_url: str = "http://127.0.0.1:11434"
|
|
default_model: str = "qwen3.5:4b"
|
|
tools: list[ToolToggle] = Field(
|
|
default_factory=lambda: [
|
|
ToolToggle(name="brave_search", enabled=True),
|
|
ToolToggle(name="searxng_search", enabled=False),
|
|
ToolToggle(name="web_fetch", enabled=True),
|
|
ToolToggle(name="apple_notes", enabled=True),
|
|
ToolToggle(name="files", enabled=True),
|
|
ToolToggle(name="terminal", enabled=True),
|
|
]
|
|
)
|
|
|
|
|
|
class DashboardSnapshot(BaseModel):
|
|
settings: RuntimeSettings
|
|
whitelist_count: int
|
|
memory_items: int
|
|
recent_logs: list[str]
|
|
|
|
|
|
class MemoryRecord(BaseModel):
|
|
id: int
|
|
content: str
|
|
kind: str
|
|
created_at: datetime
|
|
|
|
|
|
class OllamaStatus(BaseModel):
|
|
reachable: bool
|
|
base_url: str
|
|
model: str
|
|
installed_models: list[str] = Field(default_factory=list)
|
|
message: str
|
|
|
|
|
|
class TelegramStatus(BaseModel):
|
|
configured: bool
|
|
polling_active: bool
|
|
message: str
|