refactor: support both CSV and list formats for enabled_tools
This commit is contained in:
+3
-3
@@ -256,13 +256,13 @@ fn build_inline_role(
|
||||
}
|
||||
|
||||
if node.tools.as_deref().unwrap_or_default().is_empty() {
|
||||
role.set_enabled_tools(Some(String::new()));
|
||||
role.set_enabled_tools(Some(Vec::new()));
|
||||
role.set_enabled_mcp_servers(Some(Vec::new()));
|
||||
} else {
|
||||
if !regular_tools.is_empty() {
|
||||
role.set_enabled_tools(Some(regular_tools.join(",")));
|
||||
role.set_enabled_tools(Some(regular_tools.to_vec()));
|
||||
} else {
|
||||
role.set_enabled_tools(Some(String::new()));
|
||||
role.set_enabled_tools(Some(Vec::new()));
|
||||
}
|
||||
if !mcp_servers.is_empty() {
|
||||
role.set_enabled_mcp_servers(Some(mcp_servers.to_vec()));
|
||||
|
||||
@@ -55,7 +55,7 @@ async fn extract_via_extractor(
|
||||
|
||||
fn build_extractor_role() -> Result<Role> {
|
||||
let mut role = Role::new(EXTRACTOR_ROLE_NAME, EXTRACTOR_ROLE_PROMPT);
|
||||
role.set_enabled_tools(Some(String::new()));
|
||||
role.set_enabled_tools(Some(Vec::new()));
|
||||
role.set_enabled_mcp_servers(Some(Vec::new()));
|
||||
Ok(role)
|
||||
}
|
||||
@@ -183,7 +183,10 @@ mod tests {
|
||||
fn build_extractor_role_disables_tools_and_mcp() {
|
||||
let role = build_extractor_role().expect("builtin role must exist");
|
||||
|
||||
assert_eq!(role.enabled_tools().as_deref(), Some(""));
|
||||
assert_eq!(
|
||||
role.enabled_tools().as_deref(),
|
||||
Some([].as_slice())
|
||||
);
|
||||
assert_eq!(
|
||||
role.enabled_mcp_servers().as_deref(),
|
||||
Some([].as_slice())
|
||||
|
||||
Reference in New Issue
Block a user