pub struct ModelRouter<S: LlmClient, F: LlmClient> { /* private fields */ }Expand description
Dual-model router that selects smart or fast model per request.
Implementations§
Source§impl<S: LlmClient, F: LlmClient> ModelRouter<S, F>
impl<S: LlmClient, F: LlmClient> ModelRouter<S, F>
pub fn new(smart: S, fast: F) -> Self
pub fn with_config(self, config: RouterConfig) -> Self
Sourcepub fn route_messages(&self, messages: &[Message]) -> ModelChoice
pub fn route_messages(&self, messages: &[Message]) -> ModelChoice
Decide which model to use based on request characteristics.
Sourcepub fn route_tools(
&self,
messages: &[Message],
tools: &[ToolDef],
) -> ModelChoice
pub fn route_tools( &self, messages: &[Message], tools: &[ToolDef], ) -> ModelChoice
Decide which model for tool calls based on tool count.
Sourcepub fn route_structured(
&self,
messages: &[Message],
_schema: &Value,
) -> ModelChoice
pub fn route_structured( &self, messages: &[Message], _schema: &Value, ) -> ModelChoice
Decide which model for structured calls.
Trait Implementations§
Source§impl<S: LlmClient, F: LlmClient> LlmClient for ModelRouter<S, F>
impl<S: LlmClient, F: LlmClient> LlmClient for ModelRouter<S, F>
Source§fn structured_call<'life0, 'life1, 'life2, 'async_trait>(
&'life0 self,
messages: &'life1 [Message],
schema: &'life2 Value,
) -> Pin<Box<dyn Future<Output = Result<(Option<Value>, Vec<ToolCall>, String), SgrError>> + Send + 'async_trait>>where
Self: 'async_trait,
'life0: 'async_trait,
'life1: 'async_trait,
'life2: 'async_trait,
fn structured_call<'life0, 'life1, 'life2, 'async_trait>(
&'life0 self,
messages: &'life1 [Message],
schema: &'life2 Value,
) -> Pin<Box<dyn Future<Output = Result<(Option<Value>, Vec<ToolCall>, String), SgrError>> + Send + 'async_trait>>where
Self: 'async_trait,
'life0: 'async_trait,
'life1: 'async_trait,
'life2: 'async_trait,
Structured call: send messages with schema injected into system prompt.
Returns (parsed_output, native_tool_calls, raw_text).
Source§fn tools_call<'life0, 'life1, 'life2, 'async_trait>(
&'life0 self,
messages: &'life1 [Message],
tools: &'life2 [ToolDef],
) -> Pin<Box<dyn Future<Output = Result<Vec<ToolCall>, SgrError>> + Send + 'async_trait>>where
Self: 'async_trait,
'life0: 'async_trait,
'life1: 'async_trait,
'life2: 'async_trait,
fn tools_call<'life0, 'life1, 'life2, 'async_trait>(
&'life0 self,
messages: &'life1 [Message],
tools: &'life2 [ToolDef],
) -> Pin<Box<dyn Future<Output = Result<Vec<ToolCall>, SgrError>> + Send + 'async_trait>>where
Self: 'async_trait,
'life0: 'async_trait,
'life1: 'async_trait,
'life2: 'async_trait,
Native function calling: send messages + tool defs, get tool calls.
Auto Trait Implementations§
impl<S, F> Freeze for ModelRouter<S, F>
impl<S, F> RefUnwindSafe for ModelRouter<S, F>where
S: RefUnwindSafe,
F: RefUnwindSafe,
impl<S, F> Send for ModelRouter<S, F>
impl<S, F> Sync for ModelRouter<S, F>
impl<S, F> Unpin for ModelRouter<S, F>
impl<S, F> UnsafeUnpin for ModelRouter<S, F>where
S: UnsafeUnpin,
F: UnsafeUnpin,
impl<S, F> UnwindSafe for ModelRouter<S, F>where
S: UnwindSafe,
F: UnwindSafe,
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more