Add SavingMemories variant

Cormanz · Jun 2, 2023 · 02b4c00 · 02b4c00
1 parent e94aae0
commit 02b4c00
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 2 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "smartgpt"
-version = "0.1.8"
+version = "0.1.9"
 edition = "2021"
 license = "MIT"
 description = "A crate that provides LLMs with the ability to complete complex tasks using plugins."

diff --git a/src/auto/agents/worker/methodical.rs b/src/auto/agents/worker/methodical.rs
@@ -69,6 +69,7 @@ pub fn add_memories(
     agent: &mut AgentInfo,
     listen_to_update: &mut impl FnMut(&Update) -> Result<(), Box<dyn Error>>
 ) -> Result<(), Box<dyn Error>> {
+    listen_to_update(&Update::StaticAgent(StaticUpdate::SavingMemories()))?;
     agent.llm.message_history.push(Message::User(
         SUMMARIZE_MEMORIES.fill(NoData)?  
     ));

diff --git a/src/auto/agents/worker/updates.rs b/src/auto/agents/worker/updates.rs
@@ -27,6 +27,8 @@ pub enum StaticUpdate {
     SelectedAsset(String),
     #[serde(rename = "added asset")]
     AddedAsset(NamedAsset),
+    #[serde(rename = "saving memories")]
+    SavingMemories(),
     #[serde(rename = "added memories")]
     SavedMemories(Memories)
 }