From 123127893112803e985b44bf94b936ca31180e2a Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Fri, 18 Oct 2024 08:35:56 +0800
Subject: refactor: session role_prompt from agent (#933)

---
 src/config/agent.rs   | 4 ----
 src/config/session.rs | 2 +-
 2 files changed, 1 insertion(+), 5 deletions(-)

(limited to 'src/config')

diff --git a/src/config/agent.rs b/src/config/agent.rs
index 2f44477..9dcf62b 100644
--- a/src/config/agent.rs
+++ b/src/config/agent.rs
@@ -138,10 +138,6 @@ impl Agent {
         &self.functions
     }
 
-    pub fn definition(&self) -> &AgentDefinition {
-        &self.definition
-    }
-
     pub fn rag(&self) -> Option<Arc<Rag>> {
         self.rag.clone()
     }
diff --git a/src/config/session.rs b/src/config/session.rs
index c499abc..2ad517c 100644
--- a/src/config/session.rs
+++ b/src/config/session.rs
@@ -247,7 +247,7 @@ impl Session {
 
     pub fn set_agent(&mut self, agent: &Agent) {
         self.role_prompt
-            .clone_from(&agent.definition().instructions);
+            .clone_from(&agent.interpolated_instructions());
     }
 
     pub fn clear_role(&mut self) {
-- 
cgit v1.2.3