feat: conversation database

basilisk/config/main_config.py

@@ -45,6 +45,10 @@ class ConversationSettings(BaseModel):
 	shift_enter_mode: bool = Field(default=False)
 	use_accessible_output: bool = Field(default=True)
 	focus_history_after_send: bool = Field(default=False)
+	auto_save_to_db: bool = Field(default=True)
+	auto_save_draft: bool = Field(default=True)
+	reopen_last_conversation: bool = Field(default=False)
+	last_active_conversation_id: int | None = Field(default=None)
 
 
 class ImagesSettings(BaseModel):

basilisk/conversation/attached_file.py

@@ -226,6 +226,7 @@ class AttachmentFile(BaseModel):
 	description: str | None = None
 	size: int | None = None
 	mime_type: str | None = None
+	db_id: int | None = Field(default=None, exclude=True)
 
 	@field_serializer("location", mode="wrap")
 	@classmethod

basilisk/conversation/conversation_model.py

@@ -90,6 +90,7 @@ class SystemMessage(BaseMessage):
 	"""Represents a system message in a conversation. The system message is used to provide instructions or context to the assistant."""
 
 	role: MessageRoleEnum = Field(default=MessageRoleEnum.SYSTEM)
+	db_id: int | None = Field(default=None, exclude=True)
 
 	@field_validator("role", mode="after")
 	@classmethod
@@ -117,6 +118,19 @@ def __hash__(self):
 		"""
 		return hash((self.role, self.content))
 
+	def __eq__(self, other: object) -> bool:
+		"""Compare system messages by role and content, ignoring db_id.
+
+		Args:
+			other: The object to compare against.
+
+		Returns:
+			True if role and content match, False otherwise.
+		"""
+		if not isinstance(other, SystemMessage):
+			return NotImplemented
+		return self.role == other.role and self.content == other.content
+
 
 class MessageBlock(BaseModel):
 	"""Represents a block of messages in a conversation. The block may contain a user message, an AI model request, and an AI model response."""
@@ -131,6 +145,7 @@ class MessageBlock(BaseModel):
 	stream: bool = Field(default=False)
 	created_at: datetime = Field(default_factory=datetime.now)
 	updated_at: datetime = Field(default_factory=datetime.now)
+	db_id: int | None = Field(default=None, exclude=True)
 
 	@field_validator("response", mode="after")
 	@classmethod

basilisk/conversation/database/__init__.py

@@ -0,0 +1,5 @@
+"""Database package for conversation persistence."""
+
+from .manager import ConversationDatabase
+
+__all__ = ["ConversationDatabase"]