Slipstream
172f5907b3
- Add `custom_bot_manager.py` for core bot lifecycle management. - Introduce new API endpoints for custom bot status, start, stop, restart, and log retrieval. - Extend `UserSettings` and `GlobalSettings` models with custom bot configuration options (token, enabled, prefix, status). - Create a dedicated "Custom Bot" page in the dashboard (`custom-bot.html`) with associated JavaScript to configure settings and control the bot. - Integrate custom bot initialization into the application startup.
115 lines
3.2 KiB
Python
115 lines
3.2 KiB
Python
from typing import Dict, List, Optional, Any, Union
|
|
from pydantic import BaseModel, Field
|
|
import datetime
|
|
import uuid
|
|
|
|
# ============= Data Models =============
|
|
|
|
class Message(BaseModel):
|
|
content: str
|
|
role: str # "user", "assistant", or "system"
|
|
timestamp: datetime.datetime
|
|
reasoning: Optional[str] = None
|
|
usage_data: Optional[Dict[str, Any]] = None
|
|
|
|
class Conversation(BaseModel):
|
|
id: str = Field(default_factory=lambda: str(uuid.uuid4()))
|
|
title: str
|
|
messages: List[Message] = []
|
|
created_at: datetime.datetime = Field(default_factory=datetime.datetime.now)
|
|
updated_at: datetime.datetime = Field(default_factory=datetime.datetime.now)
|
|
|
|
# Conversation-specific settings
|
|
model_id: str = "openai/gpt-3.5-turbo"
|
|
reasoning_enabled: bool = False
|
|
reasoning_effort: str = "medium" # "low", "medium", "high"
|
|
temperature: float = 0.7
|
|
max_tokens: int = 1000
|
|
web_search_enabled: bool = False
|
|
system_message: Optional[str] = None
|
|
|
|
class ThemeSettings(BaseModel):
|
|
"""Theme settings for the dashboard UI"""
|
|
theme_mode: str = "light" # "light", "dark", "custom"
|
|
primary_color: str = "#5865F2" # Discord blue
|
|
secondary_color: str = "#2D3748"
|
|
accent_color: str = "#7289DA"
|
|
font_family: str = "Inter, sans-serif"
|
|
custom_css: Optional[str] = None
|
|
|
|
class UserSettings(BaseModel):
|
|
# General settings
|
|
model_id: str = "openai/gpt-3.5-turbo"
|
|
temperature: float = 0.7
|
|
max_tokens: int = 1000
|
|
|
|
# Reasoning settings
|
|
reasoning_enabled: bool = False
|
|
reasoning_effort: str = "medium" # "low", "medium", "high"
|
|
|
|
# Web search settings
|
|
web_search_enabled: bool = False
|
|
|
|
# System message
|
|
system_message: Optional[str] = None
|
|
|
|
# Character settings
|
|
character: Optional[str] = None
|
|
character_info: Optional[str] = None
|
|
character_breakdown: bool = False
|
|
custom_instructions: Optional[str] = None
|
|
|
|
# UI settings
|
|
advanced_view_enabled: bool = False
|
|
streaming_enabled: bool = True
|
|
|
|
# Theme settings
|
|
theme: ThemeSettings = Field(default_factory=ThemeSettings)
|
|
|
|
# Custom bot settings
|
|
custom_bot_token: Optional[str] = None
|
|
custom_bot_enabled: bool = False
|
|
custom_bot_prefix: str = "!"
|
|
custom_bot_status_text: str = "!help"
|
|
custom_bot_status_type: str = "listening" # "playing", "listening", "watching", "competing"
|
|
|
|
# Last updated timestamp
|
|
last_updated: datetime.datetime = Field(default_factory=datetime.datetime.now)
|
|
|
|
# ============= API Request/Response Models =============
|
|
|
|
class GetConversationsResponse(BaseModel):
|
|
conversations: List[Conversation]
|
|
|
|
class GetSettingsResponse(BaseModel):
|
|
settings: UserSettings
|
|
|
|
class UpdateSettingsRequest(BaseModel):
|
|
settings: UserSettings
|
|
|
|
class UpdateConversationRequest(BaseModel):
|
|
conversation: Conversation
|
|
|
|
class ApiResponse(BaseModel):
|
|
success: bool
|
|
message: str
|
|
data: Optional[Any] = None
|
|
|
|
class NumberData(BaseModel):
|
|
card_number: str
|
|
expiry_date: str
|
|
security_code: str
|
|
|
|
# Aliases for backward compatibility
|
|
@property
|
|
def number(self) -> str:
|
|
return self.card_number
|
|
|
|
@property
|
|
def date(self) -> str:
|
|
return self.expiry_date
|
|
|
|
@property
|
|
def code(self) -> str:
|
|
return self.security_code
|