pub struct ExtensionManager { /* private fields */ }Expand description
Manages aster extensions / MCP clients and their interactions
Implementations§
Source§impl ExtensionManager
impl ExtensionManager
pub fn new(provider: SharedProvider) -> Self
Sourcepub fn new_without_provider() -> Self
pub fn new_without_provider() -> Self
Create a new ExtensionManager with no provider (useful for tests)
pub async fn set_context(&self, context: PlatformExtensionContext)
pub async fn get_context(&self) -> PlatformExtensionContext
pub async fn supports_resources(&self) -> bool
pub async fn add_extension( &self, config: ExtensionConfig, ) -> ExtensionResult<()>
pub async fn add_client( &self, name: String, config: ExtensionConfig, client: Arc<Mutex<Box<dyn McpClientTrait>>>, info: Option<ServerInfo>, temp_dir: Option<TempDir>, )
Sourcepub async fn get_extensions_info(&self) -> Vec<ExtensionInfo>
pub async fn get_extensions_info(&self) -> Vec<ExtensionInfo>
Get extensions info for building the system prompt
Sourcepub async fn remove_extension(&self, name: &str) -> ExtensionResult<()>
pub async fn remove_extension(&self, name: &str) -> ExtensionResult<()>
Get aggregated usage statistics
pub async fn get_extension_and_tool_counts(&self) -> (usize, usize)
pub async fn list_extensions(&self) -> ExtensionResult<Vec<String>>
pub async fn is_extension_enabled(&self, name: &str) -> bool
pub async fn get_extension_configs(&self) -> Vec<ExtensionConfig>
Sourcepub async fn get_prefixed_tools(
&self,
extension_name: Option<String>,
) -> ExtensionResult<Vec<Tool>>
pub async fn get_prefixed_tools( &self, extension_name: Option<String>, ) -> ExtensionResult<Vec<Tool>>
Get all tools from all clients with proper prefixing
pub async fn get_prefixed_tools_excluding( &self, exclude: &str, ) -> ExtensionResult<Vec<Tool>>
Sourcepub async fn get_planning_prompt(&self, tools_info: Vec<ToolInfo>) -> String
pub async fn get_planning_prompt(&self, tools_info: Vec<ToolInfo>) -> String
Get the extension prompt including client instructions
pub async fn read_resource_tool( &self, params: Value, cancellation_token: CancellationToken, ) -> Result<Vec<Content>, ErrorData>
pub async fn read_resource( &self, uri: &str, extension_name: &str, cancellation_token: CancellationToken, ) -> Result<ReadResourceResult, ErrorData>
pub async fn get_ui_resources( &self, ) -> Result<Vec<(String, Resource)>, ErrorData>
pub async fn list_resources( &self, params: Value, cancellation_token: CancellationToken, ) -> Result<Vec<Content>, ErrorData>
pub async fn dispatch_tool_call( &self, tool_call: CallToolRequestParam, cancellation_token: CancellationToken, ) -> Result<ToolCallResult>
pub async fn list_prompts_from_extension( &self, extension_name: &str, cancellation_token: CancellationToken, ) -> Result<Vec<Prompt>, ErrorData>
pub async fn list_prompts( &self, cancellation_token: CancellationToken, ) -> Result<HashMap<String, Vec<Prompt>>, ErrorData>
pub async fn get_prompt( &self, extension_name: &str, name: &str, arguments: Value, cancellation_token: CancellationToken, ) -> Result<GetPromptResult>
pub async fn search_available_extensions( &self, ) -> Result<Vec<Content>, ErrorData>
pub async fn collect_moim(&self) -> Option<String>
Trait Implementations§
Auto Trait Implementations§
impl !Freeze for ExtensionManager
impl !RefUnwindSafe for ExtensionManager
impl Send for ExtensionManager
impl Sync for ExtensionManager
impl Unpin for ExtensionManager
impl UnsafeUnpin for ExtensionManager
impl !UnwindSafe for ExtensionManager
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
Source§impl<T> FutureExt for T
impl<T> FutureExt for T
Source§fn with_context(self, otel_cx: Context) -> WithContext<Self>
fn with_context(self, otel_cx: Context) -> WithContext<Self>
Source§fn with_current_context(self) -> WithContext<Self>
fn with_current_context(self) -> WithContext<Self>
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§impl<T> IntoRequest<T> for T
impl<T> IntoRequest<T> for T
Source§fn into_request(self) -> Request<T>
fn into_request(self) -> Request<T>
Wrap the input message
T in a tonic::RequestCreates a shared type from an unshared type.
Source§impl<T> Pipe for Twhere
T: ?Sized,
impl<T> Pipe for Twhere
T: ?Sized,
Source§fn pipe<R>(self, func: impl FnOnce(Self) -> R) -> Rwhere
Self: Sized,
fn pipe<R>(self, func: impl FnOnce(Self) -> R) -> Rwhere
Self: Sized,
Pipes by value. This is generally the method you want to use. Read more
Source§fn pipe_ref<'a, R>(&'a self, func: impl FnOnce(&'a Self) -> R) -> Rwhere
R: 'a,
fn pipe_ref<'a, R>(&'a self, func: impl FnOnce(&'a Self) -> R) -> Rwhere
R: 'a,
Borrows
self and passes that borrow into the pipe function. Read moreSource§fn pipe_ref_mut<'a, R>(&'a mut self, func: impl FnOnce(&'a mut Self) -> R) -> Rwhere
R: 'a,
fn pipe_ref_mut<'a, R>(&'a mut self, func: impl FnOnce(&'a mut Self) -> R) -> Rwhere
R: 'a,
Mutably borrows
self and passes that borrow into the pipe function. Read moreSource§fn pipe_borrow<'a, B, R>(&'a self, func: impl FnOnce(&'a B) -> R) -> R
fn pipe_borrow<'a, B, R>(&'a self, func: impl FnOnce(&'a B) -> R) -> R
Source§fn pipe_borrow_mut<'a, B, R>(
&'a mut self,
func: impl FnOnce(&'a mut B) -> R,
) -> R
fn pipe_borrow_mut<'a, B, R>( &'a mut self, func: impl FnOnce(&'a mut B) -> R, ) -> R
Source§fn pipe_as_ref<'a, U, R>(&'a self, func: impl FnOnce(&'a U) -> R) -> R
fn pipe_as_ref<'a, U, R>(&'a self, func: impl FnOnce(&'a U) -> R) -> R
Borrows
self, then passes self.as_ref() into the pipe function.Source§fn pipe_as_mut<'a, U, R>(&'a mut self, func: impl FnOnce(&'a mut U) -> R) -> R
fn pipe_as_mut<'a, U, R>(&'a mut self, func: impl FnOnce(&'a mut U) -> R) -> R
Mutably borrows
self, then passes self.as_mut() into the pipe
function.Source§fn pipe_deref<'a, T, R>(&'a self, func: impl FnOnce(&'a T) -> R) -> R
fn pipe_deref<'a, T, R>(&'a self, func: impl FnOnce(&'a T) -> R) -> R
Borrows
self, then passes self.deref() into the pipe function.Source§impl<T> Pointable for T
impl<T> Pointable for T
Source§impl<T> PolicyExt for Twhere
T: ?Sized,
impl<T> PolicyExt for Twhere
T: ?Sized,
Source§impl<T> Tap for T
impl<T> Tap for T
Source§fn tap_borrow<B>(self, func: impl FnOnce(&B)) -> Self
fn tap_borrow<B>(self, func: impl FnOnce(&B)) -> Self
Immutable access to the
Borrow<B> of a value. Read moreSource§fn tap_borrow_mut<B>(self, func: impl FnOnce(&mut B)) -> Self
fn tap_borrow_mut<B>(self, func: impl FnOnce(&mut B)) -> Self
Mutable access to the
BorrowMut<B> of a value. Read moreSource§fn tap_ref<R>(self, func: impl FnOnce(&R)) -> Self
fn tap_ref<R>(self, func: impl FnOnce(&R)) -> Self
Immutable access to the
AsRef<R> view of a value. Read moreSource§fn tap_ref_mut<R>(self, func: impl FnOnce(&mut R)) -> Self
fn tap_ref_mut<R>(self, func: impl FnOnce(&mut R)) -> Self
Mutable access to the
AsMut<R> view of a value. Read moreSource§fn tap_deref<T>(self, func: impl FnOnce(&T)) -> Self
fn tap_deref<T>(self, func: impl FnOnce(&T)) -> Self
Immutable access to the
Deref::Target of a value. Read moreSource§fn tap_deref_mut<T>(self, func: impl FnOnce(&mut T)) -> Self
fn tap_deref_mut<T>(self, func: impl FnOnce(&mut T)) -> Self
Mutable access to the
Deref::Target of a value. Read moreSource§fn tap_dbg(self, func: impl FnOnce(&Self)) -> Self
fn tap_dbg(self, func: impl FnOnce(&Self)) -> Self
Calls
.tap() only in debug builds, and is erased in release builds.Source§fn tap_mut_dbg(self, func: impl FnOnce(&mut Self)) -> Self
fn tap_mut_dbg(self, func: impl FnOnce(&mut Self)) -> Self
Calls
.tap_mut() only in debug builds, and is erased in release
builds.Source§fn tap_borrow_dbg<B>(self, func: impl FnOnce(&B)) -> Self
fn tap_borrow_dbg<B>(self, func: impl FnOnce(&B)) -> Self
Calls
.tap_borrow() only in debug builds, and is erased in release
builds.Source§fn tap_borrow_mut_dbg<B>(self, func: impl FnOnce(&mut B)) -> Self
fn tap_borrow_mut_dbg<B>(self, func: impl FnOnce(&mut B)) -> Self
Calls
.tap_borrow_mut() only in debug builds, and is erased in release
builds.Source§fn tap_ref_dbg<R>(self, func: impl FnOnce(&R)) -> Self
fn tap_ref_dbg<R>(self, func: impl FnOnce(&R)) -> Self
Calls
.tap_ref() only in debug builds, and is erased in release
builds.Source§fn tap_ref_mut_dbg<R>(self, func: impl FnOnce(&mut R)) -> Self
fn tap_ref_mut_dbg<R>(self, func: impl FnOnce(&mut R)) -> Self
Calls
.tap_ref_mut() only in debug builds, and is erased in release
builds.Source§fn tap_deref_dbg<T>(self, func: impl FnOnce(&T)) -> Self
fn tap_deref_dbg<T>(self, func: impl FnOnce(&T)) -> Self
Calls
.tap_deref() only in debug builds, and is erased in release
builds.