mirror of
https://github.com/storytold/craft-libs.git
synced 2026-11-03 01:00:48 +00:00
Jeremy
bd175be54f
Consent expiry is now per kind of data: renewing text no longer renews an expired image permission. A connection edit keeps models found by a test at the new address, and drops only the ones saved for the old one.
70 lines
3.0 KiB
Rust
70 lines
3.0 KiB
Rust
use craft_ai_core::{CapabilityId, InputRole, Role, RoleId, StoreFile};
|
|
|
|
fn tool(id: &str) -> Role {
|
|
Role {
|
|
id: RoleId::new(id).unwrap(),
|
|
label: "Describe photograph".into(),
|
|
capability: CapabilityId::parse("image.describe").unwrap(),
|
|
inputs: vec![InputRole::Source],
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn duplicate_user_tool_ids_are_refused_before_the_store_is_written() {
|
|
let store = StoreFile { custom_tools: vec![tool("describe-photo"), tool("describe-photo")], ..StoreFile::default() };
|
|
assert!(store.validate().is_err());
|
|
}
|
|
|
|
#[test]
|
|
fn old_store_loads_and_user_tool_round_trips() {
|
|
let old = r#"{"schema_version":1,"connections":[],"defaults":[],"apps":{}}"#;
|
|
let mut store = StoreFile::parse(old).unwrap();
|
|
assert!(store.custom_tools.is_empty());
|
|
store.custom_tools.push(tool("describe-photo"));
|
|
let again = StoreFile::parse(&store.to_json().unwrap()).unwrap();
|
|
assert_eq!(again.custom_tools, store.custom_tools);
|
|
assert_eq!(again.custom_tools[0].capability.as_str(), "image.describe");
|
|
}
|
|
|
|
#[test]
|
|
fn hostile_tool_declarations_are_refused_on_load() {
|
|
let mut s = StoreFile { custom_tools: vec![tool("describe-photo"); 65], ..StoreFile::default() };
|
|
assert!(s.validate().is_err());
|
|
s.custom_tools.truncate(1);
|
|
s.custom_tools[0].inputs.push(InputRole::Source);
|
|
assert!(StoreFile::parse(&s.to_json().unwrap()).is_err());
|
|
s.custom_tools[0].inputs = vec![InputRole::Source];
|
|
s.custom_tools[0].label = " \t ".into();
|
|
assert!(StoreFile::parse(&s.to_json().unwrap()).is_err());
|
|
s.custom_tools[0].label = "x".repeat(121);
|
|
assert!(StoreFile::parse(&s.to_json().unwrap()).is_err());
|
|
s.custom_tools[0].label = "Describe photograph".into();
|
|
s.custom_tools[0].inputs.clear();
|
|
assert!(StoreFile::parse(&s.to_json().unwrap()).is_err());
|
|
}
|
|
|
|
#[test]
|
|
fn an_app_cannot_reinterpret_the_inputs_of_a_shared_custom_tool() {
|
|
use craft_ai_core::role::ResolveError;
|
|
use craft_ai_core::{AppId, Binding, CapabilitySpec, Connection, ConnectionId, ModelId, ModelInfo};
|
|
let shared = tool("describe-photo");
|
|
let model = ModelInfo::new(ModelId::new("m").unwrap(), "M").with(CapabilitySpec::new(shared.capability.clone()).accepting(InputRole::Source, 1));
|
|
let connection: Connection = serde_json::from_value(serde_json::json!({
|
|
"id":"service", "label":"Service", "provider":"test",
|
|
"endpoint":"http://127.0.0.1:9", "locality":"local", "auth":{"kind":"none"},
|
|
"billing":"free", "models":[model]
|
|
}))
|
|
.unwrap();
|
|
let store = StoreFile {
|
|
connections: vec![connection],
|
|
defaults: vec![Binding { role: shared.id.clone(), connection: ConnectionId::new("service").unwrap(), model: ModelId::new("m").unwrap() }],
|
|
custom_tools: vec![shared.clone()],
|
|
..StoreFile::default()
|
|
};
|
|
let app = AppId::new("photocraft").unwrap();
|
|
assert!(store.resolve(&app, &shared).is_ok());
|
|
let mut redefined = shared;
|
|
redefined.inputs.clear();
|
|
assert!(matches!(store.resolve(&app, &redefined), Err(ResolveError::ToolMismatch(_))));
|
|
}
|