log character counts and estimated tokens

Author: Jephuff

crates/mcp-apollo-server/src/operations.rs

@@ -24,6 +24,7 @@ use crate::tree_shake::TreeShaker;
 pub struct Operation {
     tool: Tool,
     source_text: String,
+    character_count: usize,
 }
 impl AsRef<Tool> for Operation {
     fn as_ref(&self) -> &Tool {
@@ -99,9 +100,23 @@ impl Operation {
             ));
         };
 
+        let tool = Tool::new(operation_name.clone(), description, schema);
+        let character_count = tool_character_length(&tool);
+        match character_count {
+            Ok(length) => tracing::info!(
+                "Tool {} loaded with a character count of {}",
+                operation_name,
+                length
+            ),
+            Err(_) => tracing::info!(
+                "Tool {} loaded with an unknown character count",
+                operation_name
+            ),
+        }
         Ok(Operation {
-            tool: Tool::new(operation_name, description, schema),
+            tool,
             source_text: source_text.to_string(),
+            character_count: character_count.unwrap_or_default(),
         })
     }
 
@@ -225,6 +240,15 @@ impl Operation {
 
         lines.join("\n")
     }
+
+    pub fn tool_character_length(&self) -> usize {
+        self.character_count
+    }
+}
+
+fn tool_character_length(tool: &Tool) -> Result<usize, serde_json::Error> {
+    let tool_schema_string = serde_json::to_string_pretty(&serde_json::json!(tool.input_schema))?;
+    Ok(tool.name.len() + tool.description.len() + tool_schema_string.len())
 }
 
 fn get_json_schema(

crates/mcp-apollo-server/src/server.rs

@@ -66,11 +66,19 @@ impl Server {
             )?);
         }
 
-        info!(
+        tracing::debug!(
             "Loaded operations:\n{}",
             serde_json::to_string_pretty(&operations)?
         );
 
+        tracing::info!("Size of tools:");
+        let total_characters = operations
+            .iter()
+            .map(|op| op.tool_character_length())
+            .sum::<usize>();
+        tracing::info!("    characters: {}", total_characters);
+        tracing::info!("    estimated tokens: {}", total_characters / 4);
+
         // Load headers
         let mut default_headers = HeaderMap::new();
         default_headers.append(CONTENT_TYPE, HeaderValue::from_static("application/json"));