From 746b087111fabc10ec3f3f3e9ef3628d1eb47fd8 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Fri, 14 Jun 2024 19:12:18 +0800
Subject: refactor: add/modify rag-related config (#599)

---
 src/config/session.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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

diff --git a/src/config/session.rs b/src/config/session.rs
index 38855cc..7400f30 100644
--- a/src/config/session.rs
+++ b/src/config/session.rs
@@ -73,7 +73,7 @@ impl Session {
         let mut session: Self =
             serde_yaml::from_str(&content).with_context(|| format!("Invalid session {}", name))?;
 
-        session.model = Model::retrieve(config, &session.model_id)?;
+        session.model = Model::retrieve_chat(config, &session.model_id)?;
         session.name = name.to_string();
         session.path = Some(path.display().to_string());
 
-- 
cgit v1.2.3