diff options
Diffstat (limited to 'src/client/mod.rs')
| -rw-r--r-- | src/client/mod.rs | 6 |
1 files changed, 5 insertions, 1 deletions
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), |