diff options
| author | sigoden <sigoden@gmail.com> | 2023-03-09 15:30:39 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-03-09 15:30:39 +0800 |
| commit | 05d20f207fedf03c2a564438a0d44a2d85d4ed2d (patch) | |
| tree | eb7c9ae402632c5827cd4faa323b138510716f22 /src/main.rs | |
| parent | c7eb261abc32ac4d6fc5364244a6246bcdc462d9 (diff) | |
| download | aichat-05d20f207fedf03c2a564438a0d44a2d85d4ed2d.tar.gz | |
feat: add remain tokens indicator and max tokens guard (#50)
Diffstat (limited to 'src/main.rs')
| -rw-r--r-- | src/main.rs | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/main.rs b/src/main.rs index cc61790..b940053 100644 --- a/src/main.rs +++ b/src/main.rs @@ -21,6 +21,7 @@ use repl::{AbortSignal, Repl}; use std::io::{stdin, Read}; use std::sync::Arc; use std::{io::stdout, process::exit}; +use utils::cl100k_base_singleton; fn main() -> Result<()> { let cli = Cli::parse(); @@ -96,6 +97,7 @@ fn start_directive( } fn start_interactive(client: ChatGptClient, config: SharedConfig) -> Result<()> { + cl100k_base_singleton(); let mut repl = Repl::init(config.clone())?; repl.run(client, config) } |
