From 1ef97b2f32caecdb2eef9fe3755390f36d96eae5 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 16 Mar 2023 17:02:09 +0800
Subject: feat: support multiple models (#71)

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

(limited to 'src/repl/prompt.rs')

diff --git a/src/repl/prompt.rs b/src/repl/prompt.rs
index 1a59ba3..8c670f4 100644
--- a/src/repl/prompt.rs
+++ b/src/repl/prompt.rs
@@ -76,10 +76,10 @@ impl Prompt for ReplPrompt {
     }
 
     fn render_prompt_right(&self) -> Cow<str> {
-        if let Some(conversation) = self.config.read().conversation.as_ref() {
-            conversation.reamind_tokens().to_string().into()
-        } else {
+        if self.config.read().conversation.is_none() {
             Cow::Borrowed("")
+        } else {
+            self.config.read().get_reamind_tokens().to_string().into()
         }
     }
 
-- 
cgit v1.2.3