From e63441b1e4139007a244a8f419cda73368b11ec7 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Wed, 3 Jul 2024 10:15:47 +0800
Subject: refactor: improve function calling (#680)

---
 src/config/mod.rs | 4 +---
 src/function.rs   | 4 ++--
 2 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/src/config/mod.rs b/src/config/mod.rs
index 6725c93..375faba 100644
--- a/src/config/mod.rs
+++ b/src/config/mod.rs
@@ -532,9 +532,7 @@ impl Config {
             "function_calling" => {
                 let value = value.parse().with_context(|| "Invalid value")?;
                 if value && self.functions.is_empty() {
-                    bail!(
-                        "Cannot enable function_calling because there are no functions installed."
-                    )
+                    bail!("Function calling cannot be enabled because no functions are installed.")
                 }
                 self.function_calling = value;
             }
diff --git a/src/function.rs b/src/function.rs
index a8c193a..fcb4c93 100644
--- a/src/function.rs
+++ b/src/function.rs
@@ -222,9 +222,9 @@ impl ToolCall {
 
         let output = if is_dangerously {
             if *IS_STDOUT_TERMINAL {
-                println!("{prompt}");
+                println!("{}", dimmed_text(&prompt));
                 let answer = Text::new("[1] Run, [2] Run & Retrieve, [3] Skip:")
-                    .with_default("2")
+                    .with_default("1")
                     .with_validator(|input: &str| match matches!(input, "1" | "2" | "3") {
                         true => Ok(Validation::Valid),
                         false => Ok(Validation::Invalid(
-- 
cgit v1.2.3