From c752ba9b272ddf8186f7786ac32a88a410e6eee8 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Mon, 29 Apr 2024 15:34:24 +0800
Subject: feat: `.model` repl completions show max tokens and price (#462)

---
 src/repl/completer.rs | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

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

diff --git a/src/repl/completer.rs b/src/repl/completer.rs
index aea1ee6..2e7d79c 100644
--- a/src/repl/completer.rs
+++ b/src/repl/completer.rs
@@ -54,7 +54,7 @@ impl Completer for ReplCompleter {
                     .read()
                     .repl_complete(cmd, &args)
                     .iter()
-                    .map(|name| create_suggestion(name.clone(), None, span)),
+                    .map(|(value, description)| create_suggestion(value, description, span)),
             )
         }
 
@@ -69,7 +69,7 @@ impl Completer for ReplCompleter {
                 } else {
                     format!("{name} ")
                 };
-                create_suggestion(name, Some(description.to_string()), span)
+                create_suggestion(&name, description, span)
             }))
         }
         suggestions
@@ -105,9 +105,14 @@ impl ReplCompleter {
     }
 }
 
-fn create_suggestion(value: String, description: Option<String>, span: Span) -> Suggestion {
+fn create_suggestion(value: &str, description: &str, span: Span) -> Suggestion {
+    let description = if description.is_empty() {
+        None
+    } else {
+        Some(description.to_string())
+    };
     Suggestion {
-        value,
+        value: value.to_string(),
         description,
         style: None,
         extra: None,
-- 
cgit v1.2.3