File size: 2,798 Bytes
a21c316 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 | use serde_json::Value;
/// MCP 工具适配器 trait
///
/// 为不同的 MCP 工具提供定制化的 Schema 处理策略。
/// 每个工具可以实现自己的适配器来处理特定的需求。
pub trait ToolAdapter: Send + Sync {
/// 判断该适配器是否匹配给定的工具名称
///
/// # Arguments
/// * `tool_name` - 工具名称,通常格式为 "mcp__provider__function"
///
/// # Returns
/// 如果匹配返回 true,否则返回 false
fn matches(&self, tool_name: &str) -> bool;
/// 在通用清洗前执行的预处理
///
/// 可以在这里添加工具特定的字段处理、提示添加等
///
/// # Arguments
/// * `schema` - 待处理的 JSON Schema
///
/// # Returns
/// 成功返回 Ok(()), 失败返回错误信息
fn pre_process(&self, _schema: &mut Value) -> Result<(), String> {
Ok(())
}
/// 在通用清洗后执行的后处理
///
/// 可以在这里进行最终的调整和优化
///
/// # Arguments
/// * `schema` - 已清洗的 JSON Schema
///
/// # Returns
/// 成功返回 Ok(()), 失败返回错误信息
fn post_process(&self, _schema: &mut Value) -> Result<(), String> {
Ok(())
}
}
/// 辅助函数: 向 Schema 的 description 字段追加提示
pub fn append_hint_to_schema(schema: &mut Value, hint: &str) {
if let Value::Object(map) = schema {
let desc_val = map
.entry("description".to_string())
.or_insert_with(|| Value::String("".to_string()));
if let Value::String(s) = desc_val {
if s.is_empty() {
*s = hint.to_string();
} else if !s.contains(hint) {
*s = format!("{} {}", s, hint);
}
}
}
}
#[cfg(test)]
mod tests {
use super::*;
use serde_json::json;
struct TestAdapter;
impl ToolAdapter for TestAdapter {
fn matches(&self, tool_name: &str) -> bool {
tool_name.starts_with("test__")
}
fn pre_process(&self, schema: &mut Value) -> Result<(), String> {
append_hint_to_schema(schema, "[Test Adapter]");
Ok(())
}
}
#[test]
fn test_adapter_matches() {
let adapter = TestAdapter;
assert!(adapter.matches("test__function"));
assert!(!adapter.matches("other__function"));
}
#[test]
fn test_append_hint() {
let mut schema = json!({"type": "string"});
append_hint_to_schema(&mut schema, "Test hint");
assert_eq!(schema["description"], "Test hint");
append_hint_to_schema(&mut schema, "Another hint");
assert_eq!(schema["description"], "Test hint Another hint");
}
}
|