Publish harness and TUI open-source
initial sync from the monorepo
This commit is contained in:
commit
c68e39f604
2734 changed files with 1437016 additions and 0 deletions
93
crates/common/xai-tool-protocol/tests/tool_id_derivation.rs
Normal file
93
crates/common/xai-tool-protocol/tests/tool_id_derivation.rs
Normal file
|
|
@ -0,0 +1,93 @@
|
|||
//! `ToolDescriptionWithSchema::derive_tool_id`: namespaced descriptions
|
||||
//! render as `"{ns}:{name}"`; bare names pass through; descriptions whose
|
||||
//! derived id fails [`ToolId`] validation return `Err`.
|
||||
|
||||
use xai_tool_protocol::{IdError, ToolDescriptionWithSchema, ToolId};
|
||||
use xai_tool_types::ToolDescription;
|
||||
|
||||
fn entry(name: &str, namespace: Option<&str>) -> ToolDescriptionWithSchema {
|
||||
let mut description = ToolDescription::new(name, "test description");
|
||||
if let Some(ns) = namespace {
|
||||
description = description.with_namespace(ns);
|
||||
}
|
||||
ToolDescriptionWithSchema {
|
||||
description,
|
||||
input_schema: None,
|
||||
capabilities: None,
|
||||
notification_schemas: None,
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn bare_name_derives_to_tool_id_without_namespace() {
|
||||
let derived = entry("read_file", None).derive_tool_id().unwrap();
|
||||
assert_eq!(derived, ToolId::new("read_file").unwrap());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn namespaced_name_derives_to_namespaced_tool_id() {
|
||||
let derived = entry("read_file", Some("GrokBuild"))
|
||||
.derive_tool_id()
|
||||
.unwrap();
|
||||
assert_eq!(derived, ToolId::new("GrokBuild:read_file").unwrap());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn invalid_name_propagates_id_validation_error() {
|
||||
let err = entry("foo bar", None).derive_tool_id().unwrap_err();
|
||||
assert!(
|
||||
matches!(err, IdError::InvalidFormat { .. }),
|
||||
"expected InvalidFormat, got {err:?}"
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn invalid_namespace_propagates_id_validation_error() {
|
||||
let err = entry("read_file", Some("bad ns"))
|
||||
.derive_tool_id()
|
||||
.unwrap_err();
|
||||
assert!(matches!(err, IdError::InvalidFormat { .. }));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn empty_name_yields_empty_id_error() {
|
||||
let err = entry("", None).derive_tool_id().unwrap_err();
|
||||
assert_eq!(err, IdError::Empty);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn duplicate_derivations_in_a_batch_are_detectable() {
|
||||
let batch = [
|
||||
entry("read_file", Some("GrokBuild")),
|
||||
entry("write_file", Some("GrokBuild")),
|
||||
entry("read_file", Some("GrokBuild")),
|
||||
];
|
||||
|
||||
let mut seen = std::collections::HashMap::new();
|
||||
let mut duplicates: Vec<(ToolId, Vec<usize>)> = Vec::new();
|
||||
for (i, e) in batch.iter().enumerate() {
|
||||
let id = e.derive_tool_id().unwrap();
|
||||
seen.entry(id).or_insert_with(Vec::new).push(i);
|
||||
}
|
||||
for (id, indices) in seen {
|
||||
if indices.len() > 1 {
|
||||
duplicates.push((id, indices));
|
||||
}
|
||||
}
|
||||
assert_eq!(duplicates.len(), 1, "exactly one duplicate id expected");
|
||||
let (id, indices) = &duplicates[0];
|
||||
assert_eq!(id, &ToolId::new("GrokBuild:read_file").unwrap());
|
||||
assert_eq!(indices, &vec![0, 2]);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn derivation_does_not_collide_across_namespaces() {
|
||||
let a = entry("read_file", Some("GrokBuild"))
|
||||
.derive_tool_id()
|
||||
.unwrap();
|
||||
let b = entry("read_file", Some("github")).derive_tool_id().unwrap();
|
||||
let c = entry("read_file", None).derive_tool_id().unwrap();
|
||||
assert_ne!(a, b);
|
||||
assert_ne!(a, c);
|
||||
assert_ne!(b, c);
|
||||
}
|
||||
Loading…
Reference in a new issue