mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-06 02:48:13 +00:00
41df8cf4d0
* wip * feat(traces): establish shared Rust storage foundation * fix(traces): escape ClickHouse text parameters * test(traces): exercise response cap with bounded strings * fix(traces): remove unnecessary lint expectation * fix(traces): encode ClickHouse timestamp units in Rust * test(traces): mark exception match as a regex * refactor(traces): execute schema setup in Rust * refactor(traces): use shared logging execution wrapper * docs(traces): replace foundation README with boundary rules * fix(traces): use current bridge execution facade * fix(traces): account for protocol cast in lint budget Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --------- Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
40 lines
1.5 KiB
Rust
40 lines
1.5 KiB
Rust
use std::collections::BTreeMap;
|
|
|
|
use litellm_traces::encode_rows;
|
|
use rstest::rstest;
|
|
use serde_json::{Value, json};
|
|
|
|
#[rstest]
|
|
#[case::span("Timestamp", json!(1_234_567_890), json!("1970-01-01T00:00:01.23456789Z"))]
|
|
#[case::start("start_time", json!(1_234), json!("1970-01-01T00:00:01.234Z"))]
|
|
#[case::end("end_time", json!(2_345), json!("1970-01-01T00:00:02.345Z"))]
|
|
#[case::completion("completion_start_time", json!(1_345), json!("1970-01-01T00:00:01.345Z"))]
|
|
#[case::absent_completion("completion_start_time", Value::Null, Value::Null)]
|
|
#[case::before_epoch("Timestamp", json!(-1), json!("1969-12-31T23:59:59.999999999Z"))]
|
|
fn insert_encoding_preserves_timestamp_precision_and_other_fields(
|
|
#[case] field: &str,
|
|
#[case] value: Value,
|
|
#[case] expected: Value,
|
|
) {
|
|
let rows = vec![BTreeMap::from([
|
|
(field.to_owned(), value),
|
|
("SpanAttributes".into(), json!({"message": "a\nb\\c\"雪"})),
|
|
("InputTokens".into(), json!(42)),
|
|
])];
|
|
let encoded = encode_rows(rows).expect("valid row");
|
|
let actual: Value = serde_json::from_str(&encoded).expect("JSONEachRow record");
|
|
assert_eq!(
|
|
actual,
|
|
json!({
|
|
field: expected, "SpanAttributes": {"message": "a\nb\\c\"雪"}, "InputTokens": 42
|
|
})
|
|
);
|
|
}
|
|
|
|
#[rstest]
|
|
#[case::fractional(json!(1.25))]
|
|
#[case::out_of_range(json!(u64::MAX))]
|
|
#[case::null(Value::Null)]
|
|
fn insert_encoding_rejects_invalid_span_timestamps(#[case] timestamp: Value) {
|
|
assert!(encode_rows(vec![BTreeMap::from([("Timestamp".into(), timestamp)])]).is_err());
|
|
}
|