From 652b51509d461d2121e94304d6f00e714ebd0f9f Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 2 Nov 2023 15:33:22 +0800
Subject: refactor: improve repl completer (#199)

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

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

diff --git a/src/config/session.rs b/src/config/session.rs
index 446b682..92e8c2a 100644
--- a/src/config/session.rs
+++ b/src/config/session.rs
@@ -33,7 +33,7 @@ impl Session {
     pub fn new(name: &str, model_info: ModelInfo, role: Option<Role>) -> Self {
         let temperature = role.as_ref().and_then(|v| v.temperature);
         Self {
-            model: model_info.full_name(),
+            model: model_info.id(),
             temperature,
             messages: vec![],
             name: name.to_string(),
@@ -103,7 +103,7 @@ impl Session {
             items.push(("path", path.to_string()));
         }
 
-        items.push(("model", self.model_info.full_name()));
+        items.push(("model", self.model_info.id()));
 
         if let Some(temperature) = self.temperature() {
             items.push(("temperature", temperature.to_string()));
@@ -165,7 +165,7 @@ impl Session {
     }
 
     pub fn set_model(&mut self, model_info: ModelInfo) -> Result<()> {
-        self.model = model_info.full_name();
+        self.model = model_info.id();
         self.model_info = model_info;
         Ok(())
     }
-- 
cgit v1.2.3