Skip to content

Commit fcaa75d

Browse files
authored
Trim builder methods for GetSimN, add default to condition (#151)
* Trim builder methods for getsim, add default to condition * set default preprocess action to modelprocessing
1 parent 4ed3ad4 commit fcaa75d

File tree

2 files changed

+4
-9
lines changed
  • ahnlich/client/src/builders
  • examples/rust/image-search/src

2 files changed

+4
-9
lines changed

ahnlich/client/src/builders/ai.rs

+3-3
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,8 @@ pub struct GetSimNParams {
5151
#[builder(setter(into, transform = |s: String| StoreName(s)))]
5252
pub store: StoreName,
5353
pub search_input: StoreInput,
54+
55+
#[builder(default = None)]
5456
pub condition: Option<PredicateCondition>,
5557

5658
#[builder(setter(into, transform = |n: usize| NonZeroUsize::new(n).unwrap()),default=NonZeroUsize::new(1).unwrap())]
@@ -68,7 +70,6 @@ pub struct CreatePredIndexParams {
6870
#[builder(setter(into, transform = |s: String| StoreName(s)))]
6971
pub store: StoreName,
7072

71-
//#[builder(default = HashSet::new())]
7273
pub predicates: HashSet<MetadataKey>,
7374

7475
#[builder(default = None)]
@@ -92,7 +93,6 @@ pub struct DropPredIndexParams {
9293
#[builder(setter(into, transform = |s: String| StoreName(s)))]
9394
pub store: StoreName,
9495

95-
//#[builder(default = HashSet::new())]
9696
pub predicates: HashSet<MetadataKey>,
9797

9898
#[builder(default = true)]
@@ -109,7 +109,7 @@ pub struct SetParams {
109109

110110
pub inputs: Vec<(StoreInput, StoreValue)>,
111111

112-
#[builder(default = PreprocessAction::NoPreprocessing)]
112+
#[builder(default = PreprocessAction::ModelPreprocessing)]
113113
pub preprocess_action: PreprocessAction,
114114

115115
#[builder(default = None)]

examples/rust/image-search/src/main.rs

+1-6
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,9 @@ use std::{
66

77
use ahnlich_client_rs::{ai::AIClient, builders::ai as ai_params};
88
use ahnlich_types::{
9-
ai::{AIModel, AIServerResponse, PreprocessAction},
9+
ai::{AIModel, AIServerResponse},
1010
keyval::{StoreInput, StoreName},
1111
metadata::{MetadataKey, MetadataValue},
12-
similarity::Algorithm,
1312
};
1413
use clap::{Parser, Subcommand};
1514
use tokio;
@@ -84,7 +83,6 @@ async fn index_mode() {
8483
let set_params = ai_params::SetParams::builder()
8584
.store(storename.to_string())
8685
.inputs(inputs)
87-
.preprocess_action(PreprocessAction::ModelPreprocessing)
8886
.build();
8987
let res = ai_client
9088
.set(set_params)
@@ -113,9 +111,6 @@ async fn query_mode() {
113111
let get_sim_n_params = ai_params::GetSimNParams::builder()
114112
.store(storename.clone().to_string())
115113
.search_input(StoreInput::RawString(input.to_string()))
116-
.closest_n(1)
117-
.algorithm(Algorithm::CosineSimilarity)
118-
.condition(None)
119114
.build();
120115

121116
let res = ai_client

0 commit comments

Comments
 (0)