formatting

crates/goose-cli/src/logging.rs

@@ -3,13 +3,8 @@ use std::fs;
 use std::path::PathBuf;
 use tracing_appender::rolling::Rotation;
 use tracing_subscriber::{
-    fmt,
-    Layer,
-    layer::SubscriberExt,
-    util::SubscriberInitExt,
-    EnvFilter,
+    filter::LevelFilter, fmt, layer::SubscriberExt, util::SubscriberInitExt, EnvFilter, Layer,
     Registry,
-    filter::LevelFilter,
 };
 
 use goose::tracing::langfuse_layer;
@@ -83,8 +78,8 @@ pub fn setup_logging() -> Result<()> {
 
     // Build the subscriber with required layers
     let subscriber = Registry::default()
-        .with(file_layer.with_filter(env_filter))    // Gets all logs
-        .with(console_layer.with_filter(LevelFilter::INFO));   // Controls log levels
+        .with(file_layer.with_filter(env_filter)) // Gets all logs
+        .with(console_layer.with_filter(LevelFilter::INFO)); // Controls log levels
 
     // Initialize with Langfuse if available
     if let Some(langfuse) = langfuse_layer::create_langfuse_observer() {
@@ -99,4 +94,4 @@ pub fn setup_logging() -> Result<()> {
     }
 
     Ok(())
-}
+}

crates/goose-cli/src/main.rs

@@ -4,10 +4,10 @@ mod commands {
     pub mod version;
 }
 pub mod agents;
+mod logging;
 mod profile;
 mod prompt;
 pub mod session;
-mod logging;
 
 mod systems;
 
@@ -16,9 +16,9 @@ use clap::{Parser, Subcommand};
 use commands::configure::handle_configure;
 use commands::session::build_session;
 use commands::version::print_version;
+use logging::setup_logging;
 use profile::has_no_profiles;
 use std::io::{self, Read};
-use logging::setup_logging;
 
 mod log_usage;
 
@@ -198,7 +198,7 @@ enum CliProviderVariant {
 #[tokio::main]
 async fn main() -> Result<()> {
     setup_logging();
-    
+
     let cli = Cli::parse();
 
     if cli.version {

crates/goose-server/src/logging.rs

@@ -3,13 +3,8 @@ use std::fs;
 use std::path::PathBuf;
 use tracing_appender::rolling::Rotation;
 use tracing_subscriber::{
-    fmt,
-    Layer,
-    layer::SubscriberExt,
-    util::SubscriberInitExt,
-    EnvFilter,
+    filter::LevelFilter, fmt, layer::SubscriberExt, util::SubscriberInitExt, EnvFilter, Layer,
     Registry,
-    filter::LevelFilter,
 };
 
 use goose::tracing::langfuse_layer;
@@ -84,8 +79,8 @@ pub fn setup_logging() -> Result<()> {
 
     // Build the subscriber with required layers
     let subscriber = Registry::default()
-        .with(file_layer)   
-        .with(console_layer.with_filter(env_filter)); 
+        .with(file_layer)
+        .with(console_layer.with_filter(env_filter));
 
     // Initialize with Langfuse if available
     if let Some(langfuse) = langfuse_layer::create_langfuse_observer() {
@@ -100,4 +95,4 @@ pub fn setup_logging() -> Result<()> {
     }
 
     Ok(())
-}
+}

crates/goose-server/src/main.rs

@@ -35,4 +35,4 @@ async fn main() -> anyhow::Result<()> {
     info!("listening on {}", listener.local_addr()?);
     axum::serve(listener, app).await?;
     Ok(())
-}
+}

crates/goose/src/agent.rs

@@ -129,11 +129,12 @@ impl Agent {
         let system_tool_call = ToolCall::new(tool_name, call.arguments);
         let result = system.call(system_tool_call.clone()).await;
 
-        debug!("input"=serde_json::to_string(&system_tool_call).unwrap(),
-                "output"=serde_json::to_string(&result).unwrap(),
+        debug!(
+            "input" = serde_json::to_string(&system_tool_call).unwrap(),
+            "output" = serde_json::to_string(&result).unwrap(),
         );
 
-        result 
+        result
     }
 
     fn get_system_prompt(&self) -> AgentResult<String> {
@@ -345,13 +346,14 @@ impl Agent {
         let reply_span = tracing::Span::current();
         let mut messages = messages.to_vec();
         let tools = self.get_prefixed_tools();
-        
+
         // Set the user_message field in the span instead of creating a new event
-        if let Some(content) = messages.last()
+        if let Some(content) = messages
+            .last()
             .and_then(|msg| msg.content.first())
-            .and_then(|c| c.as_text()) 
+            .and_then(|c| c.as_text())
         {
-            debug!("user_message"=&content);
+            debug!("user_message" = &content);
         }
 
         let system_prompt = self.get_system_prompt()?;
@@ -371,7 +373,7 @@ impl Agent {
 
         Ok(Box::pin(async_stream::try_stream! {
             let _reply_guard = reply_span.enter();
-            
+
             loop {
                 // Get completion from provider
                 let (response, usage) = self.provider.complete(
@@ -408,7 +410,7 @@ impl Agent {
                     })
                     .collect();
 
-                
+
 
                 // Process all the futures in parallel but wait until all are finished
                 let outputs = futures::future::join_all(futures)

crates/goose/src/providers/databricks.rs

@@ -15,8 +15,7 @@ use crate::providers::openai_utils::{
     openai_response_to_message, tools_to_openai_spec,
 };
 use mcp_core::tool::Tool;
-use tracing::{debug};
-
+use tracing::debug;
 
 pub const DATABRICKS_DEFAULT_MODEL: &str = "claude-3-5-sonnet-2";
 
@@ -76,7 +75,18 @@ impl Provider for DatabricksProvider {
         self.config.model_config()
     }
 
-    #[tracing::instrument(skip(self, system, messages, tools), fields(model_config, input, output, input_tokens, output_tokens, total_tokens, cost))]
+    #[tracing::instrument(
+        skip(self, system, messages, tools),
+        fields(
+            model_config,
+            input,
+            output,
+            input_tokens,
+            output_tokens,
+            total_tokens,
+            cost
+        )
+    )]
     async fn complete(
         &self,
         system: &str,