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.

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: Model, 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: Model, 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,

fn list_models<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Vec<Model>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: '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)