Skip to main content

AiClient

Struct AiClient 

Source
pub struct AiClient {
    pub manifest: ProtocolManifest,
    pub transport: Arc<HttpTransport>,
    pub pipeline: Arc<Pipeline>,
    pub loader: Arc<ProtocolLoader>,
    /* private fields */
}
Expand description

Unified AI client that works with any provider through protocol configuration.

Fields§

§manifest: ProtocolManifest§transport: Arc<HttpTransport>§pipeline: Arc<Pipeline>§loader: Arc<ProtocolLoader>

Implementations§

Source§

impl AiClient

Source

pub async fn signals(&self) -> SignalsSnapshot

Snapshot current runtime signals (facts only) for application-layer orchestration.

Source

pub async fn new(model: &str) -> Result<Self>

Create a new client for a specific model.

The model identifier should be in the format “provider/model-name” (e.g., “anthropic/claude-3-5-sonnet”)

Source

pub fn chat(&self) -> ChatRequestBuilder<'_>

Create a chat request builder.

Source

pub async fn chat_batch( &self, requests: Vec<ChatBatchRequest>, concurrency_limit: Option<usize>, ) -> Vec<Result<UnifiedResponse>>

Execute multiple chat requests concurrently with an optional concurrency limit.

Notes:

  • Results preserve input order.
  • Internally uses the same “streaming → UnifiedResponse” path for consistency.
Source

pub async fn chat_batch_smart( &self, requests: Vec<ChatBatchRequest>, ) -> Vec<Result<UnifiedResponse>>

Smart batch execution with a conservative, developer-friendly default heuristic.

  • For very small batches, run sequentially to reduce overhead.
  • For larger batches, run with a bounded concurrency.

You can override the chosen concurrency via env:

  • AI_LIB_BATCH_CONCURRENCY
Source

pub async fn report_feedback(&self, event: FeedbackEvent) -> Result<()>

Report user feedback (optional). This delegates to the injected FeedbackSink.

Source

pub async fn update_rate_limits(&self, headers: &HeaderMap)

Update rate limiter state from response headers using protocol-mapped names.

This method is public for testing purposes.

Source

pub async fn call_model( &self, request: UnifiedRequest, ) -> Result<UnifiedResponse>

Unified entry point for calling a model. Handles text, streaming, and error fallback automatically.

Source

pub async fn call_model_with_stats( &self, request: UnifiedRequest, ) -> Result<(UnifiedResponse, CallStats)>

Call a model and also return per-call stats (latency, retries, request ids, endpoint, usage, etc.).

This is intended for higher-level model selection and observability. Call a model and also return per-call stats (latency, retries, request ids, endpoint, usage, etc.).

This is intended for higher-level model selection and observability.

Source

pub fn validate_request(&self, request: &ChatRequestBuilder<'_>) -> Result<()>

Validate request capabilities.

Trait Implementations§

Source§

impl EndpointExt for AiClient

Source§

async fn call_service(&self, service_name: &str) -> Result<Value>

Call a generic service by name.

Source§

async fn list_remote_models(&self) -> Result<Vec<String>>

List models available from the provider.

Source§

fn resolve_endpoint(&self, name: &str) -> Result<&EndpointConfig>

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> Same for T

Source§

type Output = T

Should always be Self
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<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