coral-rs 0.2.4

Rig and RMCP helper library for quickly building Rust powered Coral feature complete agents
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

pub mod agent;
pub mod agent_loop;
pub mod api;
pub mod claim_manager;
pub mod codegen;
pub mod completion_evaluated_prompt;
pub mod error;
pub mod mcp_server;
pub mod repeating_prompt_stream;

pub use rig;
pub use rmcp;
pub use serde;
use std::io;
use tracing::Level;
use tracing_subscriber::fmt::writer::MakeWriterExt;
use tracing_subscriber::layer::SubscriberExt;
use tracing_subscriber::util::{SubscriberInitExt, TryInitError};

///
/// Helper function that omits verbose logging information when CORAL_RUNTIME_ID is set.
/// This is useful for developing Coral agents because during dev-mode development, extra logging
/// information is desired, but when the agents are being orchestrated (during application
/// development), the extra information is duplicated with the server's logging information
pub fn init_tracing() -> Result<(), TryInitError> {
    if std::env::var("CORAL_RUNTIME_ID").is_ok() {
        let stderr = tracing_subscriber::fmt::layer()
            .with_target(false)
            .with_level(false)
            .without_time()
            .with_writer(
                io::stderr
                    .with_min_level(Level::ERROR)
                    .with_max_level(Level::WARN),
            );

        let stdout = tracing_subscriber::fmt::layer()
            .with_target(false)
            .with_level(false)
            .without_time()
            .with_writer(
                io::stdout
                    .with_min_level(Level::INFO)
                    .with_max_level(Level::INFO),
            );

        tracing_subscriber::registry()
            .with(stdout)
            .with(stderr)
            .try_init()
    } else {
        let stderr = tracing_subscriber::fmt::layer().with_writer(
            io::stderr
                .with_min_level(Level::ERROR)
                .with_max_level(Level::WARN),
        );

        let stdout = tracing_subscriber::fmt::layer().with_writer(
            io::stdout
                .with_min_level(Level::INFO)
                .with_max_level(Level::INFO),
        );

        tracing_subscriber::registry()
            .with(stdout)
            .with(stderr)
            .try_init()
    }
}