From 5c7bfd92ff3e557477969be9db0638ef0d3d3659 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 2 Nov 2023 07:08:54 +0800
Subject: refactor: set tokens count factors (#195)

---
 src/client/mod.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'src/client/mod.rs')

diff --git a/src/client/mod.rs b/src/client/mod.rs
index dba049d..5fa6146 100644
--- a/src/client/mod.rs
+++ b/src/client/mod.rs
@@ -3,7 +3,11 @@ mod common;
 
 pub use common::*;
 
-use crate::{config::ModelInfo, repl::ReplyStreamHandler, utils::PromptKind};
+use crate::{
+    config::{ModelInfo, TokensCountFactors},
+    repl::ReplyStreamHandler,
+    utils::PromptKind,
+};
 
 register_client!(
     (openai, "openai", OpenAI, OpenAIConfig, OpenAIClient),
-- 
cgit v1.2.3