From 05d20f207fedf03c2a564438a0d44a2d85d4ed2d Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 9 Mar 2023 15:30:39 +0800
Subject: feat: add remain tokens indicator and max tokens guard (#50)

---
 src/main.rs | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'src/main.rs')

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)
 }
-- 
cgit v1.2.3