From bc44026ff8828f532e9eba138e2ba35f7c247271 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Sat, 28 Oct 2023 21:39:17 +0800
Subject: feat: enhance session/conversation (#162)

* feat: enhance session/conversation

* updates

* updates

* cut version v0.9.0-rc2

* add .session name completion
---
 src/repl/prompt.rs | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

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

diff --git a/src/repl/prompt.rs b/src/repl/prompt.rs
index 1d92407..be3763b 100644
--- a/src/repl/prompt.rs
+++ b/src/repl/prompt.rs
@@ -69,15 +69,17 @@ impl ReplPrompt {
 
 impl Prompt for ReplPrompt {
     fn render_prompt_left(&self) -> Cow<str> {
-        self.config
-            .read()
-            .role
-            .as_ref()
-            .map_or(Cow::Borrowed(""), |role| Cow::Owned(role.name.clone()))
+        if let Some(session) = &self.config.read().session {
+            Cow::Owned(session.name.clone())
+        } else if let Some(role) = &self.config.read().role {
+            Cow::Owned(role.name.clone())
+        } else {
+            Cow::Borrowed("")
+        }
     }
 
     fn render_prompt_right(&self) -> Cow<str> {
-        if self.config.read().conversation.is_none() {
+        if self.config.read().session.is_none() {
             Cow::Borrowed("")
         } else {
             self.config.read().get_reamind_tokens().to_string().into()
@@ -85,7 +87,7 @@ impl Prompt for ReplPrompt {
     }
 
     fn render_prompt_indicator(&self, _prompt_mode: reedline::PromptEditMode) -> Cow<str> {
-        if self.config.read().conversation.is_some() {
+        if self.config.read().session.is_some() {
             Cow::Borrowed("）")
         } else {
             Cow::Borrowed("〉")
-- 
cgit v1.2.3