From 227cd5891f7b997a0936b3302aa636935d9d6ace Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Wed, 31 Jul 2024 07:57:19 +0800
Subject: feat: add agent-scoped `agent_prelude` config (#770)

---
 src/config/agent.rs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

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

diff --git a/src/config/agent.rs b/src/config/agent.rs
index 0025095..b20e108 100644
--- a/src/config/agent.rs
+++ b/src/config/agent.rs
@@ -139,6 +139,10 @@ impl Agent {
         self.definition.interpolated_instructions()
     }
 
+    pub fn agent_prelude(&self) -> Option<&str> {
+        self.config.agent_prelude.as_deref()
+    }
+
     pub fn variables(&self) -> &[AgentVariable] {
         &self.definition.variables
     }
@@ -211,7 +215,9 @@ pub struct AgentConfig {
     #[serde(skip_serializing_if = "Option::is_none")]
     pub top_p: Option<f64>,
     #[serde(skip_serializing_if = "Option::is_none")]
-    use_tools: Option<String>,
+    pub use_tools: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub agent_prelude: Option<String>,
 }
 
 impl AgentConfig {
-- 
cgit v1.2.3