refactor: extract template and tools modules from agent
Split template selection and external tool management into dedicated modules for better separation of concerns.
This commit is contained in:
159
src/tools.rs
Normal file
159
src/tools.rs
Normal file
@@ -0,0 +1,159 @@
|
||||
use std::collections::HashMap;
|
||||
use std::os::unix::fs::PermissionsExt;
|
||||
use std::path::{Path, PathBuf};
|
||||
|
||||
use crate::llm::{Tool, ToolFunction};
|
||||
|
||||
struct ExternalTool {
|
||||
path: PathBuf,
|
||||
schema: Tool,
|
||||
}
|
||||
|
||||
pub struct ExternalToolManager {
|
||||
tools: HashMap<String, ExternalTool>,
|
||||
}
|
||||
|
||||
impl ExternalToolManager {
|
||||
/// Scan a tools/ directory, calling `--print-schema` on each executable to discover tools.
|
||||
pub async fn discover(tools_dir: &Path) -> Self {
|
||||
let mut tools = HashMap::new();
|
||||
|
||||
let mut entries = match tokio::fs::read_dir(tools_dir).await {
|
||||
Ok(e) => e,
|
||||
Err(_) => return Self { tools },
|
||||
};
|
||||
|
||||
while let Ok(Some(entry)) = entries.next_entry().await {
|
||||
let path = entry.path();
|
||||
|
||||
// Skip non-files
|
||||
let meta = match tokio::fs::metadata(&path).await {
|
||||
Ok(m) => m,
|
||||
Err(_) => continue,
|
||||
};
|
||||
if !meta.is_file() {
|
||||
continue;
|
||||
}
|
||||
|
||||
// Check executable bit
|
||||
if meta.permissions().mode() & 0o111 == 0 {
|
||||
tracing::debug!("Skipping non-executable: {}", path.display());
|
||||
continue;
|
||||
}
|
||||
|
||||
// Call --print-schema
|
||||
let output = match tokio::process::Command::new(&path)
|
||||
.arg("--print-schema")
|
||||
.output()
|
||||
.await
|
||||
{
|
||||
Ok(o) => o,
|
||||
Err(e) => {
|
||||
tracing::warn!("Failed to run --print-schema on {}: {}", path.display(), e);
|
||||
continue;
|
||||
}
|
||||
};
|
||||
|
||||
if !output.status.success() {
|
||||
tracing::warn!(
|
||||
"--print-schema failed for {}: {}",
|
||||
path.display(),
|
||||
String::from_utf8_lossy(&output.stderr)
|
||||
);
|
||||
continue;
|
||||
}
|
||||
|
||||
let schema: serde_json::Value = match serde_json::from_slice(&output.stdout) {
|
||||
Ok(v) => v,
|
||||
Err(e) => {
|
||||
tracing::warn!("Invalid schema JSON from {}: {}", path.display(), e);
|
||||
continue;
|
||||
}
|
||||
};
|
||||
|
||||
let name = schema["name"]
|
||||
.as_str()
|
||||
.unwrap_or_else(|| {
|
||||
path.file_name()
|
||||
.and_then(|n| n.to_str())
|
||||
.unwrap_or("unknown")
|
||||
})
|
||||
.to_string();
|
||||
|
||||
let description = schema["description"].as_str().unwrap_or("").to_string();
|
||||
let parameters = schema["parameters"].clone();
|
||||
|
||||
let tool = Tool {
|
||||
tool_type: "function".into(),
|
||||
function: ToolFunction {
|
||||
name: name.clone(),
|
||||
description,
|
||||
parameters,
|
||||
},
|
||||
};
|
||||
|
||||
tracing::info!("Discovered external tool: {}", name);
|
||||
tools.insert(
|
||||
name,
|
||||
ExternalTool {
|
||||
path: path.clone(),
|
||||
schema: tool,
|
||||
},
|
||||
);
|
||||
}
|
||||
|
||||
Self { tools }
|
||||
}
|
||||
|
||||
/// Return all discovered Tool definitions for LLM API calls.
|
||||
pub fn tool_definitions(&self) -> Vec<Tool> {
|
||||
self.tools.values().map(|t| t.schema.clone()).collect()
|
||||
}
|
||||
|
||||
/// Invoke an external tool by name, passing JSON args as the first argv.
|
||||
pub async fn invoke(
|
||||
&self,
|
||||
name: &str,
|
||||
args_json: &str,
|
||||
workdir: &str,
|
||||
) -> anyhow::Result<String> {
|
||||
let tool = self
|
||||
.tools
|
||||
.get(name)
|
||||
.ok_or_else(|| anyhow::anyhow!("External tool not found: {}", name))?;
|
||||
|
||||
let output = tokio::process::Command::new(&tool.path)
|
||||
.arg(args_json)
|
||||
.current_dir(workdir)
|
||||
.output()
|
||||
.await?;
|
||||
|
||||
let stdout = String::from_utf8_lossy(&output.stdout).to_string();
|
||||
let stderr = String::from_utf8_lossy(&output.stderr).to_string();
|
||||
|
||||
if output.status.success() {
|
||||
Ok(stdout)
|
||||
} else {
|
||||
let mut result = stdout;
|
||||
if !stderr.is_empty() {
|
||||
result.push_str("\nSTDERR: ");
|
||||
result.push_str(&stderr);
|
||||
}
|
||||
result.push_str(&format!(
|
||||
"\n[exit code: {}]",
|
||||
output.status.code().unwrap_or(-1)
|
||||
));
|
||||
Ok(result)
|
||||
}
|
||||
}
|
||||
|
||||
/// Check if a tool with the given name exists.
|
||||
pub fn has_tool(&self, name: &str) -> bool {
|
||||
self.tools.contains_key(name)
|
||||
}
|
||||
|
||||
/// Number of discovered tools.
|
||||
pub fn len(&self) -> usize {
|
||||
self.tools.len()
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user