summaryrefslogtreecommitdiffstats
path: root/src/main.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2023-03-09 15:30:39 +0800
committerGitHub <noreply@github.com>2023-03-09 15:30:39 +0800
commit05d20f207fedf03c2a564438a0d44a2d85d4ed2d (patch)
treeeb7c9ae402632c5827cd4faa323b138510716f22 /src/main.rs
parentc7eb261abc32ac4d6fc5364244a6246bcdc462d9 (diff)
downloadaichat-05d20f207fedf03c2a564438a0d44a2d85d4ed2d.tar.gz
feat: add remain tokens indicator and max tokens guard (#50)
Diffstat (limited to 'src/main.rs')
-rw-r--r--src/main.rs2
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)
}