From 33a0049a81103c1f6c5639b9cec43cb14a9ef367 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 27 Jun 2024 08:21:49 +0800
Subject: refactor: saved messages ignore embedding context (#659)

---
 src/config/mod.rs | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

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

diff --git a/src/config/mod.rs b/src/config/mod.rs
index f82f36a..69609f4 100644
--- a/src/config/mod.rs
+++ b/src/config/mod.rs
@@ -1267,7 +1267,7 @@ impl Config {
 
     pub fn after_chat_completion(
         &mut self,
-        input: &mut Input,
+        input: &Input,
         output: &str,
         tool_results: &[ToolResult],
     ) -> Result<()> {
@@ -1280,9 +1280,11 @@ impl Config {
         Ok(())
     }
 
-    fn save_message(&mut self, input: &mut Input, output: &str) -> Result<()> {
+    fn save_message(&mut self, input: &Input, output: &str) -> Result<()> {
+        let mut input = input.clone();
+        input.clear_patch();
         if let Some(session) = input.session_mut(&mut self.session) {
-            session.add_message(input, output)?;
+            session.add_message(&input, output)?;
             return Ok(());
         }
 
-- 
cgit v1.2.3