TokenCounterFactory

Struct TokenCounterFactory 

Source
pub struct TokenCounterFactory;
Expand description

Factory for creating token counters for different providers.

Use this factory to get the appropriate token counter for your provider and model. The factory handles selecting the correct tokenizer and context window size.

§Example

use multi_llm::{TokenCounterFactory, TokenCounter};

// Create counter for OpenAI
let openai = TokenCounterFactory::create_counter("openai", "gpt-4")?;

// Create counter for Anthropic
let anthropic = TokenCounterFactory::create_counter("anthropic", "claude-3-5-sonnet")?;

// Create counter with custom limit
let custom = TokenCounterFactory::create_counter_with_limit("openai", "gpt-4", 4096)?;

§Supported Providers

  • openai: Uses tiktoken for exact counts
  • anthropic: Uses approximation with safety margin
  • ollama: Uses cl100k_base (approximation)
  • lmstudio: Uses cl100k_base (approximation)

Implementations§

Source§

impl TokenCounterFactory

Source

pub fn create_counter( provider: &str, model: &str, ) -> LlmResult<Arc<dyn TokenCounter>>

Create token counter for specific provider and model

Source

pub fn create_counter_with_limit( provider: &str, model: &str, max_tokens: u32, ) -> LlmResult<Arc<dyn TokenCounter>>

Create counter with custom context window size

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<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> PolicyExt for T
where T: ?Sized,

Source§

fn and<P, B, E>(self, other: P) -> And<T, P>
where T: Policy<B, E>, P: Policy<B, E>,

Create a new Policy that returns Action::Follow only if self and other return Action::Follow. Read more
Source§

fn or<P, B, E>(self, other: P) -> Or<T, P>
where T: Policy<B, E>, P: Policy<B, E>,

Create a new Policy that returns Action::Follow if either self or other returns Action::Follow. Read more
Source§

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

Source§

type Error = Infallible

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

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

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<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