From 2508d56598a37844e369ab623ceb5bc7b2c78d38 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Mon, 27 Nov 2023 15:39:55 +0800
Subject: feat: state-aware completer (#251)

---
 src/config/session.rs | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'src/config/session.rs')

diff --git a/src/config/session.rs b/src/config/session.rs
index 644255f..e135c58 100644
--- a/src/config/session.rs
+++ b/src/config/session.rs
@@ -20,6 +20,7 @@ pub struct Session {
     model_id: String,
     temperature: Option<f64>,
     messages: Vec<Message>,
+    #[serde(default)]
     data_urls: HashMap<String, String>,
     #[serde(skip)]
     pub name: String,
@@ -248,6 +249,7 @@ impl Session {
             role: MessageRole::Assistant,
             content: MessageContent::Text(output.to_string()),
         });
+        self.role = None;
         self.dirty = true;
         Ok(())
     }
-- 
cgit v1.2.3