From f6da06dad9b2a76016209a7d58ad923c2c72f150 Mon Sep 17 00:00:00 2001 From: sigoden Date: Wed, 1 Nov 2023 22:15:55 +0800 Subject: refactor: improve code quanity (#194) - extends ModelInfo for tokens calculating - refactor config/session.rs, improve export, render, getter/setter - modify main.rs, allow --model override session.model --- src/client/openai.rs | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'src/client/openai.rs') diff --git a/src/client/openai.rs b/src/client/openai.rs index b554f35..387baf0 100644 --- a/src/client/openai.rs +++ b/src/client/openai.rs @@ -38,7 +38,7 @@ impl OpenAIClient { let client = Self::name(local_config); MODELS .into_iter() - .map(|(name, max_tokens)| ModelInfo::new(client, name, Some(max_tokens), index)) + .map(|(name, max_tokens)| openai_tokens_formula(ModelInfo::new(index, client, name).set_max_tokens(Some(max_tokens)))) .collect() } @@ -135,3 +135,7 @@ pub fn openai_build_body(data: SendData, model: String) -> Value { } body } + +pub fn openai_tokens_formula(model: ModelInfo) -> ModelInfo { + model.set_tokens_formula(5, 2) +} -- cgit v1.2.3