summaryrefslogtreecommitdiffstats
path: root/src/main.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-05-14 15:02:58 +0800
committerGitHub <noreply@github.com>2024-05-14 15:02:58 +0800
commita75aef459db3b28e24603617b2584a7f279a3e4d (patch)
tree99761c2a86e8f2cffd4cdee53bc93623e36aef3a /src/main.rs
parent369cf9a36aaf5b76914f8c2fd412124deb51ff92 (diff)
downloadaichat-a75aef459db3b28e24603617b2584a7f279a3e4d.tar.gz
feat: remove tiktoken (#506)
Diffstat (limited to 'src/main.rs')
-rw-r--r--src/main.rs8
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();