diff options
| author | sigoden <sigoden@gmail.com> | 2024-06-12 07:17:57 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-06-12 07:17:57 +0800 |
| commit | e9e48f23208fef3b334540f18d87feb8b17a2bda (patch) | |
| tree | 6da5ef751c2683708216654b5af2e151cd0f570d /src/config/role.rs | |
| parent | c0bfb90b204fccd1395e53e4fe977b34bcc6bca0 (diff) | |
| download | aichat-e9e48f23208fef3b334540f18d87feb8b17a2bda.tar.gz | |
refactor: rename `selected_functions` to `functions_filter` (#587)
Diffstat (limited to 'src/config/role.rs')
| -rw-r--r-- | src/config/role.rs | 28 |
1 files changed, 14 insertions, 14 deletions
diff --git a/src/config/role.rs b/src/config/role.rs index b6b7fc8..5b52587 100644 --- a/src/config/role.rs +++ b/src/config/role.rs @@ -20,11 +20,11 @@ pub trait RoleLike { fn model(&self) -> &Model; fn temperature(&self) -> Option<f64>; fn top_p(&self) -> Option<f64>; - fn selected_functions(&self) -> Option<FunctionsFilter>; + fn functions_filter(&self) -> Option<FunctionsFilter>; fn set_model(&mut self, model: &Model); fn set_temperature(&mut self, value: Option<f64>); fn set_top_p(&mut self, value: Option<f64>); - fn set_selected_functions(&mut self, value: Option<FunctionsFilter>); + fn set_functions_filter(&mut self, value: Option<FunctionsFilter>); } #[derive(Debug, Clone, Default, Deserialize, Serialize)] @@ -42,7 +42,7 @@ pub struct Role { #[serde(skip_serializing_if = "Option::is_none")] top_p: Option<f64>, #[serde(skip_serializing_if = "Option::is_none")] - selected_functions: Option<FunctionsFilter>, + functions_filter: Option<FunctionsFilter>, #[serde(skip)] model: Model, @@ -91,10 +91,10 @@ async function timeout(ms) { ), ] .into_iter() - .map(|(name, prompt, selected_functions)| Self { + .map(|(name, prompt, functions_filter)| Self { name: name.into(), prompt, - selected_functions, + functions_filter, ..Default::default() }) .collect() @@ -110,8 +110,8 @@ async function timeout(ms) { let model = role_like.model(); let temperature = role_like.temperature(); let top_p = role_like.top_p(); - let selected_functions = role_like.selected_functions(); - self.batch_set(model, temperature, top_p, selected_functions); + let functions_filter = role_like.functions_filter(); + self.batch_set(model, temperature, top_p, functions_filter); } pub fn batch_set( @@ -119,7 +119,7 @@ async function timeout(ms) { model: &Model, temperature: Option<f64>, top_p: Option<f64>, - selected_functions: Option<FunctionsFilter>, + functions_filter: Option<FunctionsFilter>, ) { self.set_model(model); if temperature.is_some() { @@ -128,8 +128,8 @@ async function timeout(ms) { if top_p.is_some() { self.set_top_p(top_p); } - if selected_functions.is_some() { - self.set_selected_functions(selected_functions); + if functions_filter.is_some() { + self.set_functions_filter(functions_filter); } } @@ -230,8 +230,8 @@ impl RoleLike for Role { self.top_p } - fn selected_functions(&self) -> Option<FunctionsFilter> { - self.selected_functions.clone() + fn functions_filter(&self) -> Option<FunctionsFilter> { + self.functions_filter.clone() } fn set_model(&mut self, model: &Model) { @@ -247,8 +247,8 @@ impl RoleLike for Role { self.top_p = value; } - fn set_selected_functions(&mut self, value: Option<FunctionsFilter>) { - self.selected_functions = value; + fn set_functions_filter(&mut self, value: Option<FunctionsFilter>) { + self.functions_filter = value; } } |
