summaryrefslogtreecommitdiffstats
path: root/src/client/model.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-04-29 05:22:23 +0000
committersigoden <sigoden@gmail.com>2024-04-29 05:24:10 +0000
commit3a00fb283e4f503901ce198ead65aebc46ae1875 (patch)
tree0440cdf2376d86bc91a9ff88269658f3b7bec92e /src/client/model.rs
parent602494b650a59b73b4548b9282375c99b471e11b (diff)
downloadaichat-3a00fb283e4f503901ce198ead65aebc46ae1875.tar.gz
refactor: user config models replace client builtin models
Diffstat (limited to 'src/client/model.rs')
-rw-r--r--src/client/model.rs13
1 files changed, 2 insertions, 11 deletions
diff --git a/src/client/model.rs b/src/client/model.rs
index f3fe97c..459d94e 100644
--- a/src/client/model.rs
+++ b/src/client/model.rs
@@ -3,7 +3,6 @@ use super::message::{Message, MessageContent};
use crate::utils::count_tokens;
use anyhow::{bail, Result};
-use indexmap::IndexMap;
use serde::Deserialize;
const PER_MESSAGES_TOKENS: usize = 5;
@@ -37,17 +36,9 @@ impl Model {
}
}
- pub fn from_config(
- client_name: &str,
- config_models: &[ModelConfig],
- client_models: &[ModelConfig],
- ) -> Vec<Self> {
- let mut models = IndexMap::new();
- for model in client_models.iter().chain(config_models.iter()) {
- models.insert(&model.name, model);
- }
+ pub fn from_config(client_name: &str, models: &[ModelConfig]) -> Vec<Self> {
models
- .values()
+ .iter()
.map(|v| {
Model::new(client_name, &v.name)
.set_max_input_tokens(v.max_input_tokens)