diff --git a/src/controller/agent/create/tests.rs b/src/controller/agent/create/tests.rs index 40cfc4c..3f1ccb8 100644 --- a/src/controller/agent/create/tests.rs +++ b/src/controller/agent/create/tests.rs @@ -9,7 +9,7 @@ fn agent_config_parsing_works() { let sample_config = crate::agent::default_config_for_provider(&provider); let sample_config_pretty_yaml = serde_yaml_ng::to_string(&sample_config).unwrap(); - let test_cases = vec![ + let test_cases = [ // Invalid input TestCase { input: r#"Hello"#.to_owned(), diff --git a/src/conversation/llm/tokenization.rs b/src/conversation/llm/tokenization.rs index 9616258..9865765 100644 --- a/src/conversation/llm/tokenization.rs +++ b/src/conversation/llm/tokenization.rs @@ -105,7 +105,8 @@ pub mod test { let bpe = super::get_bpe_for_model(model); - let max_response_tokens: Option = Some(5); + let max_response_tokens_value: u32 = 5; + let max_response_tokens: Option = Some(max_response_tokens_value); let prompt = super::Message { author: super::Author::Prompt, @@ -193,7 +194,7 @@ pub mod test { &Some(prompt), conversation_messages, max_response_tokens, - prompt_length + max_response_tokens.unwrap_or(0) + forth_length + third_length, + prompt_length + max_response_tokens_value + forth_length + third_length, ); assert_eq!(2, new_conversation_messages.len()); @@ -215,7 +216,8 @@ pub mod test { let bpe = super::get_bpe_for_model(model); - let max_response_tokens: Option = Some(5); + let max_response_tokens_value: u32 = 5; + let max_response_tokens: Option = Some(max_response_tokens_value); let prompt = super::Message { author: super::Author::User, @@ -303,7 +305,7 @@ pub mod test { &Some(prompt), conversation_messages, max_response_tokens, - prompt_length + max_response_tokens.unwrap_or(0) + forth_length + third_length, + prompt_length + max_response_tokens_value + forth_length + third_length, ); assert_eq!(2, new_conversation_messages.len());