Skip to main content

TraceLayer

Struct TraceLayer 

Source
pub struct TraceLayer { /* private fields */ }
Expand description

The probe decorator. Wrap a model; every call is fingerprinted and recorded to the sink.

Implementations§

Source§

impl TraceLayer

Source

pub fn new(inner: Arc<dyn LanguageModel>, sink: Arc<dyn TraceSink>) -> Self

Wrap a model with a sink (records every call).

Source

pub fn with_default_session(self, session_id: String, key: [u8; 32]) -> Self

Default session id (used when CallOptions.session_id is absent) + session key (32-byte HMAC-style salt for fingerprint scoping).

Source

pub fn with_auditor(self, auditor: Arc<dyn CacheAuditor>) -> Self

Attach an auditor (default: none — verdict stays None).

Source

pub fn with_rules_auditor(self, strict: bool) -> Self

Convenience: attach the built-in rules auditor in strict/shared mode.

Trait Implementations§

Source§

impl LanguageModel for TraceLayer

Source§

fn config_snapshot(&self) -> ProviderRecord

RFC-0023 §3.3: transparent decorators must forward the inner snapshot (otherwise recording sees the decorator’s minimal record).

Source§

fn specification_version(&self) -> &'static str

Specification version (always "v4").
Source§

fn provider(&self) -> &str

Provider name, e.g. "openai".
Source§

fn model_id(&self) -> &str

Model identifier, e.g. "gpt-4o".
Source§

fn retry_config(&self) -> RetryConfig

Provider/model retry settings used when the caller does not override the retry count.
Source§

fn do_generate<'life0, 'life1, 'async_trait>( &'life0 self, options: &'life1 CallOptions, ) -> Pin<Box<dyn Future<Output = Result<GenerateResult, AiMuxError>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Generate a complete (non-streaming) response.
Source§

fn do_stream<'life0, 'life1, 'async_trait>( &'life0 self, options: &'life1 CallOptions, ) -> Pin<Box<dyn Future<Output = Result<StreamResult, AiMuxError>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Generate a streaming response.

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
where ST: ?Sized, DT: ?Sized,

Source§

impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
where ST: ?Sized, DT: ?Sized,

Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T> Instrument for T

Source§

fn instrument(self, span: Span) -> Instrumented<Self> ⓘ

Instruments this type with the provided Span, returning an Instrumented wrapper. Read more
Source§

fn in_current_span(self) -> Instrumented<Self> ⓘ

Instruments this type with the current Span, returning an Instrumented wrapper. Read more
Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> Read<Exclusive, BecauseExclusive> for T
where T: ?Sized,

Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = !

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, !>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

Source§

fn vzip(self) -> V

Source§

impl<T> WithSubscriber for T

Source§

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self> ⓘ
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a WithDispatch wrapper. Read more
Source§

fn with_current_subscriber(self) -> WithDispatch<Self> ⓘ

Attaches the current default Subscriber to this type, returning a WithDispatch wrapper. Read more