diff --git a/src/lib.rs b/src/lib.rs index 45f0d2e..eae652c 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -32,10 +32,8 @@ impl exports::fai::platform::invoke::Guest for Orchestrator { fn invoke( ctx: fai::platform::types::InvocationContext, inputs: Vec<(String, fai::platform::types::Payload)>, - ) -> Result< - Vec<(String, fai::platform::types::Payload)>, - fai::platform::types::InvocationError, - > { + ) -> Result, fai::platform::types::InvocationError> + { use fai::platform::types::{InvocationError, Payload}; fai::platform::host::log( @@ -84,10 +82,7 @@ impl exports::fai::platform::invoke::Guest for Orchestrator { } #[cfg(target_arch = "wasm32")] -fn text_input( - inputs: &[(String, fai::platform::types::Payload)], - name: &str, -) -> Option { +fn text_input(inputs: &[(String, fai::platform::types::Payload)], name: &str) -> Option { use fai::platform::types::Payload; inputs.iter().find(|(k, _)| k == name).and_then(|(_, v)| { if let Payload::Text(s) = v { diff --git a/src/llm.rs b/src/llm.rs index a1d0987..c5bd1ee 100644 --- a/src/llm.rs +++ b/src/llm.rs @@ -224,7 +224,8 @@ mod tests { #[test] fn extracts_content_from_well_formed_response() { - let body = r#"{"model":"x","message":{"role":"assistant","content":"the-plan"},"done":true}"#; + let body = + r#"{"model":"x","message":{"role":"assistant","content":"the-plan"},"done":true}"#; assert_eq!(extract_ollama_content(body).unwrap(), "the-plan"); } @@ -239,7 +240,8 @@ mod tests { #[test] fn end_to_end_via_mock_client_returns_validated_plan() { - let canned_plan = r#"{"schema_version":1,"goal":"hi","steps":[{"kind":"explain","text":"ok"}]}"#; + let canned_plan = + r#"{"schema_version":1,"goal":"hi","steps":[{"kind":"explain","text":"ok"}]}"#; let canned_response = format!( r#"{{"message":{{"role":"assistant","content":{}}},"done":true}}"#, serde_json::to_string(canned_plan).unwrap(), @@ -260,7 +262,8 @@ mod tests { #[test] fn rejects_invalid_plan_from_llm() { - let bad_plan = r#"{"schema_version":99,"goal":"x","steps":[{"kind":"explain","text":"a"}]}"#; + let bad_plan = + r#"{"schema_version":99,"goal":"x","steps":[{"kind":"explain","text":"a"}]}"#; let canned_response = format!( r#"{{"message":{{"content":{}}}}}"#, serde_json::to_string(bad_plan).unwrap(), diff --git a/src/plan.rs b/src/plan.rs index baf5b8b..22be75b 100644 --- a/src/plan.rs +++ b/src/plan.rs @@ -164,7 +164,10 @@ mod tests { r#"{"schema_version":1,"goal":"x","steps":[{"kind":"save_flow","name":"","yaml":"a"}]}"#, ) .unwrap(); - assert!(matches!(validate(&plan), Err(PlanError::InvalidStep { .. }))); + assert!(matches!( + validate(&plan), + Err(PlanError::InvalidStep { .. }) + )); } #[test]