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

fix similarity bug (#63)

Browse source
This commit is contained in:
Sampreeth Sarma 2024-09-20 09:49:28 -07:00 • committed by GitHub
parent 97b47c2ab4
commit 941869ad24
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 13 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

15
envoyfilter/src/stream_context.rs
View file

@ -262,14 +262,25 @@ impl StreamContext {
debug!("zeroshot intent response: {:?}", zeroshot_intent_response);
let desc_emb_similarity_map: HashMap<String, f64> = callout_context
.similarity_scores
.clone()
.unwrap()
.into_iter()
.collect();
let pred_class_desc_emb_similarity = desc_emb_similarity_map
.get(&zeroshot_intent_response.predicted_class)
.unwrap();
let prompt_target_similarity_score = zeroshot_intent_response.predicted_class_score * 0.7
+ callout_context.similarity_scores.as_ref().unwrap()[0].1 * 0.3;
+ pred_class_desc_emb_similarity * 0.3;
debug!(
"similarity score: {:.3}, intent score: {:.3}, description embedding score: {:.3}",
prompt_target_similarity_score,
zeroshot_intent_response.predicted_class_score,
callout_context.similarity_scores.as_ref().unwrap()[0].1
pred_class_desc_emb_similarity
);
let prompt_target_name = zeroshot_intent_response.predicted_class.clone();