From 50bbfc4541bf2cb2ecb28f2fed3b9b3ab7d5005b Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Sat, 30 Nov 2024 08:40:58 +0800
Subject: refactor: improve code regarding tools/agents (#1021)

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

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

diff --git a/src/config/session.rs b/src/config/session.rs
index 23658fd..4b3da76 100644
--- a/src/config/session.rs
+++ b/src/config/session.rs
@@ -35,7 +35,7 @@ pub struct Session {
     #[serde(skip_serializing_if = "Option::is_none")]
     role_name: Option<String>,
     #[serde(default, skip_serializing_if = "IndexMap::is_empty")]
-    agent_variables: IndexMap<String, String>,
+    agent_variables: AgentVariables,
 
     #[serde(default, skip_serializing_if = "Vec::is_empty")]
     compressed_messages: Vec<Message>,
@@ -284,7 +284,7 @@ impl Session {
         }
     }
 
-    pub fn agent_variables(&self) -> &IndexMap<String, String> {
+    pub fn agent_variables(&self) -> &AgentVariables {
         &self.agent_variables
     }
 
-- 
cgit v1.2.3