kowalski-core 1.2.0

Kowalski Core Module: A Rust-based agent for interacting with Ollama models
Documentation 
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
29
30
31
32

//! Optional `[agent]` / `[tool]` line prefixes for CLI REPL (`kowalski run`).

use std::cell::Cell;

thread_local! {
    static ENABLED: Cell<bool> = const { Cell::new(false) };
}

/// When enabled, [`super::Agent::chat_with_tools`] labels LLM lines with `[agent]` and tool rounds with `[tool]`.
pub fn set_repl_trace(enabled: bool) {
    ENABLED.with(|c| c.set(enabled));
}

pub(crate) fn repl_trace_enabled() -> bool {
    ENABLED.with(|c| c.get())
}

/// RAII: enable trace for the current thread until dropped.
pub struct ReplTraceGuard;

impl ReplTraceGuard {
    pub fn enable() -> Self {
        set_repl_trace(true);
        Self
    }
}

impl Drop for ReplTraceGuard {
    fn drop(&mut self) {
        set_repl_trace(false);
    }
}