78 lines
2.2 KiB
Python
78 lines
2.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 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
|
|
|
|
# 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
|