apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-06-17 15:25:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

pretty_eq

Browse source
This commit is contained in:
Adil Hafeez 2024-10-22 12:42:44 -07:00
parent e218f875d7
commit ffe6193a6e
3 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
crates/Cargo.lock generated
View file

@ -1120,6 +1120,7 @@ dependencies = [
"http",
"log",
"md5",
"pretty_assertions",
"proxy-wasm",
"proxy-wasm-test-framework",
"rand",

1
crates/prompt_gateway/Cargo.toml
View file

@ -26,3 +26,4 @@ sha2 = "0.10.8"
[dev-dependencies]
proxy-wasm-test-framework = { git = "https://github.com/katanemo/test-framework.git", branch = "new" }
serial_test = "3.1.1"
pretty_assertions = "1.4.1"

3
crates/prompt_gateway/src/hallucination.rs
View file

@ -40,6 +40,7 @@ pub fn extract_messages_for_hallucination(messages: &Vec<Message>) -> Vec<String
#[cfg(test)]
mod test {
use pretty_assertions::assert_eq;
use common::common_types::open_ai::Message;
use super::extract_messages_for_hallucination;
@ -157,5 +158,7 @@ mod test {
let messages_for_halluncination = extract_messages_for_hallucination(&messages);
println!("{:?}", messages_for_halluncination);
assert_eq!(messages_for_halluncination.len(), 3);
assert_eq!(["tell me about the weather", "Seattle", "7 days"], messages_for_halluncination.as_slice());
}
}