aether-telemetry 0.1.2

OpenTelemetry support for Aether GenAI calls
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

//! JSON encodings for the content-carrying `GenAI` span attributes
//! (`gen_ai.input.messages`, `gen_ai.output.messages`, `gen_ai.tool.definitions`).

use llm::ToolDefinition;

pub(crate) fn messages_json(role: &str, content: &str) -> String {
    serde_json::json!([
        {
            "role": role,
            "parts": [{ "type": "text", "content": content }]
        }
    ])
    .to_string()
}

pub(crate) fn tool_definitions_json(tools: &[ToolDefinition]) -> String {
    serde_json::Value::Array(
        tools
            .iter()
            .map(|tool| {
                let parameters = tool.parameters.clone();
                serde_json::json!({
                    "type": "function",
                    "name": tool.name,
                    "description": tool.description,
                    "parameters": parameters,
                })
            })
            .collect(),
    )
    .to_string()
}