stynx-code 3.3.1

stynx-code — interactive AI coding assistant
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

use std::sync::{Arc, atomic::{AtomicU64, Ordering}};

use stynx_code_engine::{EngineEvent, QueryEngine};
use stynx_code_errors::AppError;
use stynx_code_types::Conversation;
use tokio::sync::mpsc;

pub(super) async fn run_engine_tui(
    engine: &Arc<QueryEngine>,
    conversation: Conversation,
    total_input: &Arc<AtomicU64>,
    total_output: &Arc<AtomicU64>,
    event_tx: mpsc::UnboundedSender<EngineEvent>,
) -> Result<Conversation, AppError> {
    let in_counter = total_input.clone();
    let out_counter = total_output.clone();
    let result = engine.run(conversation, move |event| {
        if let EngineEvent::Usage { input_tokens, output_tokens } = &event {
            if *input_tokens > 0 { in_counter.fetch_add(*input_tokens, Ordering::Relaxed); }
            if *output_tokens > 0 { out_counter.fetch_add(*output_tokens, Ordering::Relaxed); }
        }
        let _ = event_tx.send(event);
    });
    tokio::select! {
        r = result => r,
        _ = tokio::signal::ctrl_c() => Err(AppError::Interrupted),
    }
}