Struct AgentClient 

pub struct AgentClient { /* private fields */ }

Unified agent client

Provides a single interface for:

• LLM inference via stakai (with Stakpak or direct providers)
• Session/checkpoint management via SessionStorage trait (Stakpak API or local SQLite)
• MCP tools, billing, rulebooks (Stakpak API only)

Implementations

impl AgentClient

pub async fn new(config: AgentClientConfig) -> Result<Self, String>

Create a new AgentClient

pub fn has_stakpak(&self) -> bool

Check if Stakpak API is available

pub fn get_stakpak_api_endpoint(&self) -> &str

Get the Stakpak API endpoint (with default fallback)

pub fn stakai(&self) -> &StakAIClient

Get reference to the StakAI client

pub fn stakpak_api(&self) -> Option<&StakpakApiClient>

Get reference to the Stakpak API client (if available)

pub fn hook_registry(&self) -> &Arc<HookRegistry<AgentState>>

Get reference to the hook registry

pub fn model_options(&self) -> &ModelOptions

Get the model options

pub fn session_storage(&self) -> &Arc<dyn SessionStorage>

Get reference to the session storage

Use this for all session and checkpoint operations.

pub fn get_model_string(&self, model: &AgentModel) -> LLMModel

Get the model string for the given agent model type

When Stakpak is available, routes through Stakpak provider. Otherwise, uses direct provider.

Trait Implementations

impl AgentProvider for AgentClient

fn get_my_account<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<GetMyAccountResponse, String>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

fn get_billing_info<'life0, 'life1, 'async_trait>( &'life0 self, account_username: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<BillingResponse, String>> + Send + 'async_trait>>

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

fn list_rulebooks<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<Vec<ListRuleBook>, String>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

fn get_rulebook_by_uri<'life0, 'life1, 'async_trait>( &'life0 self, uri: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<RuleBook, String>> + Send + 'async_trait>>

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

fn create_rulebook<'life0, 'life1, 'life2, 'life3, 'async_trait>( &'life0 self, uri: &'life1 str, description: &'life2 str, content: &'life3 str, tags: Vec<String>, visibility: Option<RuleBookVisibility>, ) -> Pin<Box<dyn Future<Output = Result<CreateRuleBookResponse, String>> + Send + 'async_trait>>

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

fn delete_rulebook<'life0, 'life1, 'async_trait>( &'life0 self, uri: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<(), String>> + Send + 'async_trait>>

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

fn chat_completion<'life0, 'async_trait>( &'life0 self, model: AgentModel, messages: Vec<ChatMessage>, tools: Option<Vec<Tool>>, session_id: Option<Uuid>, ) -> Pin<Box<dyn Future<Output = Result<ChatCompletionResponse, String>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

fn chat_completion_stream<'life0, 'async_trait>( &'life0 self, model: AgentModel, messages: Vec<ChatMessage>, tools: Option<Vec<Tool>>, _headers: Option<HeaderMap>, session_id: Option<Uuid>, ) -> Pin<Box<dyn Future<Output = Result<(Pin<Box<dyn Stream<Item = Result<ChatCompletionStreamResponse, ApiStreamError>> + Send>>, Option<String>), String>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

fn cancel_stream<'life0, 'async_trait>( &'life0 self, request_id: String, ) -> Pin<Box<dyn Future<Output = Result<(), String>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

fn search_docs<'life0, 'life1, 'async_trait>( &'life0 self, input: &'life1 SearchDocsRequest, ) -> Pin<Box<dyn Future<Output = Result<Vec<Content>, String>> + Send + 'async_trait>>

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

fn memorize_session<'life0, 'async_trait>( &'life0 self, checkpoint_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<(), String>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

fn search_memory<'life0, 'life1, 'async_trait>( &'life0 self, input: &'life1 SearchMemoryRequest, ) -> Pin<Box<dyn Future<Output = Result<Vec<Content>, String>> + Send + 'async_trait>>

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

fn slack_read_messages<'life0, 'life1, 'async_trait>( &'life0 self, input: &'life1 SlackReadMessagesRequest, ) -> Pin<Box<dyn Future<Output = Result<Vec<Content>, String>> + Send + 'async_trait>>

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

fn slack_read_replies<'life0, 'life1, 'async_trait>( &'life0 self, input: &'life1 SlackReadRepliesRequest, ) -> Pin<Box<dyn Future<Output = Result<Vec<Content>, String>> + Send + 'async_trait>>

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

fn slack_send_message<'life0, 'life1, 'async_trait>( &'life0 self, input: &'life1 SlackSendMessageRequest, ) -> Pin<Box<dyn Future<Output = Result<Vec<Content>, String>> + Send + 'async_trait>>

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

impl Clone for AgentClient

fn clone(&self) -> AgentClient

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for AgentClient

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl SessionStorage for AgentClient

fn list_sessions<'life0, 'life1, 'async_trait>( &'life0 self, query: &'life1 ListSessionsQuery, ) -> Pin<Box<dyn Future<Output = Result<ListSessionsResult, StorageError>> + Send + 'async_trait>>

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

List all sessions

fn get_session<'life0, 'async_trait>( &'life0 self, session_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<Session, StorageError>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Get a session by ID (includes active checkpoint)

fn create_session<'life0, 'life1, 'async_trait>( &'life0 self, request: &'life1 CreateSessionRequest, ) -> Pin<Box<dyn Future<Output = Result<CreateSessionResult, StorageError>> + Send + 'async_trait>>

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

Create a new session with initial checkpoint

fn update_session<'life0, 'life1, 'async_trait>( &'life0 self, session_id: Uuid, request: &'life1 UpdateSessionRequest, ) -> Pin<Box<dyn Future<Output = Result<Session, StorageError>> + Send + 'async_trait>>

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

Update session metadata (title, visibility)

fn delete_session<'life0, 'async_trait>( &'life0 self, session_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<(), StorageError>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Delete a session

fn list_checkpoints<'life0, 'life1, 'async_trait>( &'life0 self, session_id: Uuid, query: &'life1 ListCheckpointsQuery, ) -> Pin<Box<dyn Future<Output = Result<ListCheckpointsResult, StorageError>> + Send + 'async_trait>>

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

List checkpoints for a session

fn get_checkpoint<'life0, 'async_trait>( &'life0 self, checkpoint_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<Checkpoint, StorageError>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Get a checkpoint by ID

fn create_checkpoint<'life0, 'life1, 'async_trait>( &'life0 self, session_id: Uuid, request: &'life1 CreateCheckpointRequest, ) -> Pin<Box<dyn Future<Output = Result<Checkpoint, StorageError>> + Send + 'async_trait>>

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

Create a new checkpoint for a session

fn get_active_checkpoint<'life0, 'async_trait>( &'life0 self, session_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<Checkpoint, StorageError>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Get the latest/active checkpoint for a session

fn get_session_stats<'life0, 'async_trait>( &'life0 self, session_id: Uuid, ) -> Pin<Box<dyn Future<Output = Result<SessionStats, StorageError>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Get session stats (optional - returns default if not supported)