use crate::tool::SequentialThinkingTool;
use crate::types::SessionCommand;
use kodegen_mcp_schema::reasoning::{SequentialThinkingArgs, SequentialThinkingPromptArgs, SequentialThinkingOutput};
use kodegen_mcp_tool::error::McpError;
use kodegen_mcp_tool::{Tool, ToolExecutionContext, ToolResponse};
use rmcp::model::{PromptArgument, PromptMessage, PromptMessageContent, PromptMessageRole};
impl Tool for SequentialThinkingTool {
type Args = SequentialThinkingArgs;
type PromptArgs = SequentialThinkingPromptArgs;
fn name() -> &'static str {
kodegen_mcp_schema::reasoning::SEQUENTIAL_THINKING
}
fn description() -> &'static str {
"A detailed tool for dynamic and reflective problem-solving through thoughts.\n\
This tool helps analyze problems through a flexible thinking process that can adapt and evolve.\n\
Each thought can build on, question, or revise previous insights as understanding deepens.\n\n\
When to use this tool:\n\
- Breaking down complex problems into steps\n\
- Planning and design with room for revision\n\
- Analysis that might need course correction\n\
- Problems where the full scope might not be clear initially\n\
- Problems that require a multi-step solution\n\
- Tasks that need to maintain context over multiple steps\n\
- Situations where irrelevant information needs to be filtered out\n\n\
Key features:\n\
- Adjust total_thoughts up or down as you progress\n\
- Question or revise previous thoughts\n\
- Add more thoughts even after reaching what seemed like the end\n\
- Express uncertainty and explore alternative approaches\n\
- Branch or backtrack (not every thought needs to build linearly)\n\
- Generate and verify solution hypotheses\n\
- Repeat the process until satisfied"
}
fn read_only() -> bool {
true }
async fn execute(&self, args: Self::Args, _ctx: ToolExecutionContext) -> Result<ToolResponse<<Self::Args as kodegen_mcp_tool::ToolArgs>::Output>, McpError> {
let thought_data = Self::validate_thought(args.clone());
let (session_id, tx) = self.get_or_create_session(args.session_id).await?;
let (respond_to, rx) = tokio::sync::oneshot::channel();
let cmd = SessionCommand::AddThought {
thought: thought_data.clone(),
respond_to,
};
tx.send(cmd)
.await
.map_err(|_| McpError::Other(anyhow::anyhow!("Session actor terminated")))?;
let response = rx
.await
.map_err(|_| McpError::Other(anyhow::anyhow!("Session actor failed to respond")))?;
let narrative = format!(
"\x1b[36m **Thought {}/{}** recorded\x1b[0m\n\
Content: {}\n\
Next thought needed: {}\n\
Branches: {}\n\
Total thoughts in history: {}",
response.thought_number,
response.total_thoughts,
{
let words: Vec<&str> = thought_data.thought.split_whitespace().collect();
if words.len() > 15 {
format!("{}...", words[..15].join(" "))
} else {
thought_data.thought.clone()
}
},
if response.next_thought_needed { "Yes" } else { "No (complete)" },
if response.branches.is_empty() {
"None".to_string()
} else {
response.branches.join(", ")
},
response.thought_history_length
);
let output = SequentialThinkingOutput {
session_id,
thought_number: response.thought_number,
total_thoughts: response.total_thoughts,
thought: thought_data.thought,
next_thought_needed: response.next_thought_needed,
is_revision: args.is_revision,
revises_thought: args.revises_thought,
branch_id: args.branch_id,
branch_from_thought: args.branch_from_thought,
branches: response.branches,
thought_history_length: response.thought_history_length,
};
Ok(ToolResponse::new(narrative, output))
}
fn prompt_arguments() -> Vec<PromptArgument> {
vec![
PromptArgument {
name: "example_domain".to_string(),
title: Some("Example Domain".to_string()),
description: Some(
"Customize teaching examples for a specific domain (e.g., 'software engineering', 'mathematics', 'creative writing', 'data analysis'). \
Defaults to software engineering if not specified.".to_string()
),
required: Some(false),
},
PromptArgument {
name: "focus_feature".to_string(),
title: Some("Feature Focus".to_string()),
description: Some(
"Focus the teaching on specific features:\n\
- 'basic': Simple linear thinking examples\n\
- 'revision': How to revise and reconsider thoughts\n\
- 'branching': How to explore alternative paths\n\
- 'advanced': All features including dynamic adjustment\n\
- 'all': Comprehensive overview (default)".to_string()
),
required: Some(false),
},
]
}
async fn prompt(&self, _args: Self::PromptArgs) -> Result<Vec<PromptMessage>, McpError> {
Ok(vec![
PromptMessage {
role: PromptMessageRole::User,
content: PromptMessageContent::text(
"How do I use the sequential_thinking tool to solve a complex problem?",
),
},
PromptMessage {
role: PromptMessageRole::Assistant,
content: PromptMessageContent::text(
"The sequential_thinking tool helps you break down complex problems step by step:\n\n\
1. Start with initial estimate:\n\
sequential_thinking({\n\
\"thought\": \"First, I need to understand the problem scope\",\n\
\"thought_number\": 1,\n\
\"total_thoughts\": 5,\n\
\"next_thought_needed\": true\n\
})\n\n\
2. Continue building:\n\
sequential_thinking({\n\
\"thought\": \"Now analyzing the core requirements\",\n\
\"thought_number\": 2,\n\
\"total_thoughts\": 5,\n\
\"next_thought_needed\": true\n\
})\n\n\
3. Revise if needed:\n\
sequential_thinking({\n\
\"thought\": \"Wait, I need to reconsider my approach\",\n\
\"thought_number\": 3,\n\
\"total_thoughts\": 6,\n\
\"is_revision\": true,\n\
\"revises_thought\": 2,\n\
\"next_thought_needed\": true\n\
})\n\n\
4. Branch to explore alternatives:\n\
sequential_thinking({\n\
\"thought\": \"Alternative approach using pattern X\",\n\
\"thought_number\": 4,\n\
\"total_thoughts\": 6,\n\
\"branch_from_thought\": 2,\n\
\"branch_id\": \"alt-pattern-x\",\n\
\"next_thought_needed\": true\n\
})\n\n\
5. Conclude:\n\
sequential_thinking({\n\
\"thought\": \"Final solution: implement approach Y\",\n\
\"thought_number\": 6,\n\
\"total_thoughts\": 6,\n\
\"next_thought_needed\": false\n\
})\n\n\
The tool tracks your entire thinking process, allowing you to:\n\
- Adjust total_thoughts dynamically as you learn more\n\
- Revise earlier thoughts when you discover new information\n\
- Branch to explore multiple solution paths\n\
- See your complete thought history across all invocations",
),
},
])
}
}
#[cfg(feature = "server")]
use kodegen_server_http::ShutdownHook;
#[cfg(feature = "server")]
impl ShutdownHook for SequentialThinkingTool {
fn shutdown(&self) -> std::pin::Pin<Box<dyn std::future::Future<Output = Result<(), anyhow::Error>> + Send + '_>> {
Box::pin(async move {
SequentialThinkingTool::shutdown(self).await
.map_err(|e| anyhow::anyhow!("Failed to shutdown sequential thinking tool: {e}"))
})
}
}