diff options
| author | sigoden <sigoden@gmail.com> | 2023-11-02 07:08:54 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-11-02 07:08:54 +0800 |
| commit | 5c7bfd92ff3e557477969be9db0638ef0d3d3659 (patch) | |
| tree | dc626111b7e7f9aa1a3c4b90aab22468d61e226f /src/client/mod.rs | |
| parent | 0238c8734ecadfc6172fdded0fbcf83949b198fa (diff) | |
| download | aichat-5c7bfd92ff3e557477969be9db0638ef0d3d3659.tar.gz | |
refactor: set tokens count factors (#195)
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), |