Add failed tooloutput that can be handled

bosun-ai · Nov 28, 2024 · 1039306 · 1039306
1 parent 97cc602
commit 1039306
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/swiftide-core/src/chat_completion/tools.rs b/swiftide-core/src/chat_completion/tools.rs
@@ -6,9 +6,10 @@ use derive_builder::Builder;
 pub enum ToolOutput {
     /// Adds the result of the toolcall to messages
     Text(String),
-    Ok,
+
+    /// Indicates that the toolcall failed, but can be handled by the llm
+    Fail(String),
     /// Stops an agent
-    ///
     Stop,
 }
 
@@ -31,7 +32,7 @@ impl std::fmt::Display for ToolOutput {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         match self {
             ToolOutput::Text(value) => write!(f, "{value}"),
-            ToolOutput::Ok => write!(f, "Ok"),
+            ToolOutput::Fail(value) => write!(f, "Tool call failed: {value}"),
             ToolOutput::Stop => write!(f, "Stop"),
         }
     }

diff --git a/swiftide-core/src/chat_completion/traits.rs b/swiftide-core/src/chat_completion/traits.rs
@@ -66,7 +66,7 @@ impl From<CommandOutput> for ToolOutput {
     fn from(value: CommandOutput) -> Self {
         match value {
             CommandOutput::Text(value) => ToolOutput::Text(value),
-            CommandOutput::Ok => ToolOutput::Ok,
+            CommandOutput::Ok => ToolOutput::Text("Tool successfuly completed".to_string()),
             CommandOutput::Shell {
                 stdout,
                 stderr,
@@ -76,7 +76,7 @@ impl From<CommandOutput> for ToolOutput {
                 if success {
                     ToolOutput::Text(stdout)
                 } else {
-                    ToolOutput::Text(stderr)
+                    ToolOutput::Fail(stderr)
                 }
             }
         }