Struct TracingChatModel 

pub struct TracingChatModel<M: BaseChatModel> { /* private fields */ }

Wrapper around any BaseChatModel that records LLM call traces.

Implementations

impl<M: BaseChatModel> TracingChatModel<M>

pub fn new( inner: M, store: Arc<dyn TracingStore>, event_bus: EventBus, trace_id: String, ) -> Self

pub fn with_parent_span(self, span_id: String) -> Self

Trait Implementations

impl<M: BaseChatModel + 'static> BaseChatModel for TracingChatModel<M>

fn name(&self) -> &str

The name of the model.

fn invoke( &self, messages: &[Message], config: &RunnableConfig, ) -> Result<Message, ModelError>

Invoke the model with a list of messages and get a response.

fn ainvoke<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, messages: &'life1 [Message], config: &'life2 RunnableConfig, ) -> Pin<Box<dyn Future<Output = Result<Message, ModelError>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait,

Invoke the model asynchronously. Default delegates to sync invoke via block_in_place.

fn astream<'a>( &'a self, messages: &'a [Message], config: &'a RunnableConfig, ) -> MessageStream<'a>

Stream tokens from the model. Returns a stream of partial Message chunks.

fn bind_tools(&self, tools: Vec<ToolDef>) -> Box<dyn BaseChatModel>

Bind tools to the model for tool-calling support.