From 3a00fb283e4f503901ce198ead65aebc46ae1875 Mon Sep 17 00:00:00 2001 From: sigoden Date: Mon, 29 Apr 2024 05:22:23 +0000 Subject: refactor: user config models replace client builtin models --- src/client/model.rs | 13 ++----------- 1 file changed, 2 insertions(+), 11 deletions(-) (limited to 'src/client/model.rs') 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 { - 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 { models - .values() + .iter() .map(|v| { Model::new(client_name, &v.name) .set_max_input_tokens(v.max_input_tokens) -- cgit v1.2.3