refactor: remove agent flavours, move provider to Agent (#2091)

This commit is contained in:
Salman Mohammed
2025-04-09 15:02:47 -04:00
committed by GitHub
parent a8cbd81c61
commit 513d5c8f5a
30 changed files with 1297 additions and 2272 deletions
+3 -3
View File
@@ -243,7 +243,7 @@ pub fn read_metadata(session_file: &Path) -> Result<SessionMetadata> {
pub async fn persist_messages(
session_file: &Path,
messages: &[Message],
provider: Option<Arc<Box<dyn Provider>>>,
provider: Option<Arc<dyn Provider>>,
) -> Result<()> {
// Count user messages
let user_message_count = messages
@@ -255,7 +255,7 @@ pub async fn persist_messages(
match provider {
Some(provider) if user_message_count < 4 => {
//generate_description is responsible for writing the messages
generate_description(session_file, messages, provider.as_ref().as_ref()).await
generate_description(session_file, messages, provider).await
}
_ => {
// Read existing metadata
@@ -298,7 +298,7 @@ pub fn save_messages_with_metadata(
pub async fn generate_description(
session_file: &Path,
messages: &[Message],
provider: &dyn Provider,
provider: Arc<dyn Provider>,
) -> Result<()> {
// Create a special message asking for a 3-word description
let mut description_prompt = "Based on the conversation so far, provide a concise description of this session in 4 words or less. This will be used for finding the session later in a UI with limited space - reply *ONLY* with the description".to_string();