diff options
| author | sigoden <sigoden@gmail.com> | 2024-05-14 15:02:58 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-05-14 15:02:58 +0800 |
| commit | a75aef459db3b28e24603617b2584a7f279a3e4d (patch) | |
| tree | 99761c2a86e8f2cffd4cdee53bc93623e36aef3a /src/main.rs | |
| parent | 369cf9a36aaf5b76914f8c2fd412124deb51ff92 (diff) | |
| download | aichat-a75aef459db3b28e24603617b2584a7f279a3e4d.tar.gz | |
feat: remove tiktoken (#506)
Diffstat (limited to 'src/main.rs')
| -rw-r--r-- | src/main.rs | 8 |
1 files changed, 1 insertions, 7 deletions
diff --git a/src/main.rs b/src/main.rs index b83d69d..a8ae389 100644 --- a/src/main.rs +++ b/src/main.rs @@ -19,10 +19,7 @@ use crate::config::{ }; use crate::render::{render_error, MarkdownRender}; use crate::repl::Repl; -use crate::utils::{ - cl100k_base_singleton, create_abort_signal, extract_block, run_command, run_spinner, - CODE_BLOCK_RE, -}; +use crate::utils::{create_abort_signal, extract_block, run_command, run_spinner, CODE_BLOCK_RE}; use anyhow::{bail, Result}; use clap::Parser; @@ -141,7 +138,6 @@ async fn start_directive( ) -> Result<()> { let mut client = input.create_client()?; ensure_model_capabilities(client.as_mut(), input.required_capabilities())?; - input.maybe_print_input_tokens(); let is_terminal_stdout = stdout().is_terminal(); let extract_code = !is_terminal_stdout && code_mode; let output = if no_stream || extract_code { @@ -170,7 +166,6 @@ async fn start_directive( } async fn start_interactive(config: &GlobalConfig) -> Result<()> { - cl100k_base_singleton(); let mut repl: Repl = Repl::init(config)?; repl.run().await } @@ -178,7 +173,6 @@ async fn start_interactive(config: &GlobalConfig) -> Result<()> { #[async_recursion::async_recursion] async fn execute(config: &GlobalConfig, mut input: Input) -> Result<()> { let client = input.create_client()?; - input.maybe_print_input_tokens(); let is_terminal_stdout = stdout().is_terminal(); let ret = if is_terminal_stdout { let (spinner_tx, spinner_rx) = oneshot::channel(); |
