62 lines
2.2 KiB
Rust
62 lines
2.2 KiB
Rust
//! A singleton model for storing conversations by ID to enable restoration across terminal views.
|
|
|
|
use galaxyui::{Entity, SingletonEntity};
|
|
use std::collections::HashMap;
|
|
|
|
use crate::{
|
|
ai::{
|
|
agent::conversation::{AIConversation, AIConversationId},
|
|
blocklist::history_model::convert_persisted_conversation_to_ai_conversation_with_metadata,
|
|
},
|
|
persistence::model::AgentConversation,
|
|
};
|
|
|
|
/// Singleton model that holds restored agent conversations on app startup.
|
|
///
|
|
/// Loading restored conversations into this model is a means of propagating restored data from
|
|
/// sqlite (read at startup) to arbitrary consuming locations in the view/model hierarchy without
|
|
/// piping it all the way from the root view to the terminal view(s) that require it.
|
|
#[derive(Default)]
|
|
pub struct RestoredAgentConversations {
|
|
/// All conversations stored by their ID, available for restoration
|
|
conversations: HashMap<AIConversationId, AIConversation>,
|
|
}
|
|
|
|
impl RestoredAgentConversations {
|
|
pub fn new(conversations: Vec<AgentConversation>) -> Self {
|
|
let mut conversations_by_id = HashMap::new();
|
|
for conversation in conversations.into_iter() {
|
|
let conversation_id = conversation.conversation.conversation_id.clone();
|
|
let Some(conversation) =
|
|
convert_persisted_conversation_to_ai_conversation_with_metadata(conversation)
|
|
else {
|
|
log::warn!(
|
|
"Failed to convert persisted conversation {conversation_id} to AIConversation"
|
|
);
|
|
continue;
|
|
};
|
|
conversations_by_id.insert(conversation.id(), conversation);
|
|
}
|
|
|
|
Self {
|
|
conversations: conversations_by_id,
|
|
}
|
|
}
|
|
|
|
/// Gets a reference to a restored conversation without removing it.
|
|
pub fn get_conversation(&self, id: &AIConversationId) -> Option<&AIConversation> {
|
|
self.conversations.get(id)
|
|
}
|
|
|
|
/// Removes the restored conversation and returns it, if any.
|
|
pub fn take_conversation(&mut self, id: &AIConversationId) -> Option<AIConversation> {
|
|
self.conversations.remove(id)
|
|
}
|
|
}
|
|
|
|
impl Entity for RestoredAgentConversations {
|
|
type Event = ();
|
|
}
|
|
|
|
impl SingletonEntity for RestoredAgentConversations {}
|