Skip to content

Commit 1ba8d36

Browse files
committed
bugfix: qdrant best score strategy
1 parent c79de28 commit 1ba8d36

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

server/src/operators/qdrant_operator.rs

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -811,10 +811,10 @@ pub async fn recommend_qdrant_query(
811811

812812
let recommend_strategy = match strategy {
813813
Some(strategy) => match strategy.as_str() {
814-
"best_score" => RecommendStrategy::BestScore,
815-
_ => RecommendStrategy::AverageVector,
814+
"best_score" => Some(RecommendStrategy::BestScore.into()),
815+
_ => None,
816816
},
817-
None => RecommendStrategy::AverageVector,
817+
None => None,
818818
};
819819

820820
let filter = assemble_qdrant_filter(filters, None, None, dataset_id, pool).await?;
@@ -858,7 +858,7 @@ pub async fn recommend_qdrant_query(
858858
read_consistency: None,
859859
positive_vectors: vec![],
860860
negative_vectors: vec![],
861-
strategy: Some(recommend_strategy.into()),
861+
strategy: recommend_strategy,
862862
timeout: None,
863863
shard_key_selector: None,
864864
};
@@ -902,10 +902,10 @@ pub async fn recommend_qdrant_groups_query(
902902

903903
let recommend_strategy = match strategy {
904904
Some(strategy) => match strategy.as_str() {
905-
"best_score" => RecommendStrategy::BestScore,
906-
_ => RecommendStrategy::AverageVector,
905+
"best_score" => Some(RecommendStrategy::BestScore.into()),
906+
_ => None,
907907
},
908-
None => RecommendStrategy::AverageVector,
908+
None => None,
909909
};
910910

911911
let filters = assemble_qdrant_filter(filter, None, None, dataset_id, pool).await?;
@@ -948,7 +948,7 @@ pub async fn recommend_qdrant_groups_query(
948948
read_consistency: None,
949949
positive_vectors: vec![],
950950
negative_vectors: vec![],
951-
strategy: Some(recommend_strategy.into()),
951+
strategy: recommend_strategy,
952952
timeout: None,
953953
shard_key_selector: None,
954954
group_by: "group_ids".to_string(),

0 commit comments

Comments
 (0)