feat(tui): enhance system prompt with workspace structure information
This commit is contained in:
@@ -149,3 +149,45 @@ pub fn write_osc52(output: &mut impl Write, text: &str) -> std::io::Result<()> {
|
||||
output.flush()?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// Workspace Tree Builder
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
/// Build a string representing the directory tree of `root`, up to `max_files` limits.
|
||||
/// Automatically respects `.gitignore` and `.ignore` via the `ignore` crate.
|
||||
pub fn build_workspace_tree(root: &Path, max_files: usize) -> String {
|
||||
use ignore::WalkBuilder;
|
||||
let mut tree = String::new();
|
||||
let mut count = 0;
|
||||
|
||||
// hidden(false) allows files like .github to be seen, but it still respects .gitignore
|
||||
// and ignores .git directories by default.
|
||||
for result in WalkBuilder::new(root).hidden(false).build() {
|
||||
if let Ok(entry) = result {
|
||||
if count >= max_files {
|
||||
tree.push_str("\n... (truncated)");
|
||||
break;
|
||||
}
|
||||
|
||||
let path = entry.path();
|
||||
if let Ok(rel) = path.strip_prefix(root) {
|
||||
let name = rel.to_string_lossy();
|
||||
if name.is_empty() {
|
||||
tree.push_str(".\n");
|
||||
} else {
|
||||
let depth = entry.depth();
|
||||
let indent = " ".repeat(depth.saturating_sub(1));
|
||||
let is_dir = entry.file_type().map(|ft| ft.is_dir()).unwrap_or(false);
|
||||
let suffix = if is_dir { "/" } else { "" };
|
||||
|
||||
let filename = entry.file_name().to_string_lossy();
|
||||
tree.push_str(&format!("{indent}{filename}{suffix}\n"));
|
||||
count += 1;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
tree.trim_end().to_string()
|
||||
}
|
||||
|
||||
@@ -120,15 +120,21 @@ fn run_turn(params: TurnParams) {
|
||||
let tools = all_tools();
|
||||
let defs = tool_defs(&tools);
|
||||
|
||||
let sys_msg = ChatMessage::system(
|
||||
"You are Zesdex, an AI coding assistant. You have access to various tools via native function calling to help the user. \
|
||||
let mut sys_prompt = "You are Zesdex, an AI coding assistant. You have access to various tools via native function calling to help the user. \
|
||||
For complex problems, use `seq_think` to reason step-by-step. \
|
||||
For any non-trivial tasks, you MUST prioritize creating a structured plan (using `plan_enter`) and a list of TODOs (using `todowrite`) BEFORE executing any other tools or modifying files. \
|
||||
For large multi-step operations or delegating tasks, you MUST prioritize using `workflow_run` (to run a yaml workflow script) or `hive_mind` (to orchestrate multiple agents) to complete the task efficiently. \
|
||||
Use other tools when necessary. If a tool returns an error, fix the issue before retrying. \
|
||||
Respond conversationally, concisely, and helpfully."
|
||||
.to_string(),
|
||||
);
|
||||
Respond conversationally, concisely, and helpfully.".to_string();
|
||||
|
||||
if let Some(root) = params.workspace_roots.first() {
|
||||
let tree = zesdex_infrastructure::utils::build_workspace_tree(root, 800);
|
||||
sys_prompt.push_str("\n\nWorkspace structure:\n```\n");
|
||||
sys_prompt.push_str(&tree);
|
||||
sys_prompt.push_str("\n```\n");
|
||||
}
|
||||
|
||||
let sys_msg = ChatMessage::system(sys_prompt);
|
||||
params.messages.insert(0, sys_msg);
|
||||
|
||||
let tool_ctx = ToolCtx::builder()
|
||||
|
||||
Reference in New Issue
Block a user