rhine 0.1.8

code core of Rhine Lab
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
33
34
35
36
37

use tracing::log::info;
use crate::tests::prompt::test_prompt;
use crate::tests::chat::test_chat;

mod prompt;
mod message;
mod chat;


#[tokio::test]
pub async fn test() {
    // 初始化日志配置
    let _guard = clia_tracing_config::build()
        .filter_level("info")
        .with_ansi(true)
        .to_stdout(true)
        .directory("./logs")
        .file_name("test.log")
        .init();
    println!("log level: {}", "info");
    // test_prompt().await;
    test_chat().await;
}

pub fn format_test_block<F>(title: &str, content_fn: F)
where
    F: FnOnce() -> String,
{
    info!(
        "\n\
        [TEST]>>>>>>>>>>>>>>>>>>>>>>>>>>>\n\
        [{}]\n{}\n\
        --------------------------------------------------",
        title,
        content_fn()
    );
}