feat: add remain tokens indicator and max tokens guard (#50)

1 files changed, 29 insertions, 0 deletions

diff --git a/src/utils/mod.rs b/src/utils/mod.rs
new file mode 100644
index 0000000..fd178ba
--- /dev/null
+++ b/src/utils/mod.rs
@@ -0,0 +1,29 @@
+mod tiktoken;
+
+pub use self::tiktoken::{cl100k_base_singleton, count_tokens, text_to_tokens, tokens_to_text};
+
+use chrono::prelude::*;
+use crossterm::style::{Color, Stylize};
+use std::io::{stdout, Write};
+
+#[macro_export]
+macro_rules! print_now {
+    ($($arg:tt)*) => {
+        $crate::utils::print_now(&format!($($arg)*))
+    };
+}
+
+pub fn print_now<T: ToString>(text: T) {
+    print!("{}", text.to_string());
+    let _ = stdout().flush();
+}
+
+pub fn now() -> String {
+    let now = Local::now();
+    now.to_rfc3339_opts(SecondsFormat::Secs, false)
+}
+
+#[allow(unused)]
+pub fn emphasis(text: &str) -> String {
+    text.stylize().with(Color::White).to_string()
+}