diff --git a/crates/extension/src/extension_host_proxy.rs b/crates/extension/src/extension_host_proxy.rs index 5f0e4f662fb580d3c2ff6ddfcf8a57658faf7119..b445878389015d4b3b8c3e25a0d103586462fd86 100644 --- a/crates/extension/src/extension_host_proxy.rs +++ b/crates/extension/src/extension_host_proxy.rs @@ -389,49 +389,6 @@ pub trait ExtensionContextServerProxy: Send + Sync + 'static { fn unregister_context_server(&self, server_id: Arc, cx: &mut App); } -/// A function that registers a language model provider with the registry. -/// This allows extension_host to create the provider (which requires WasmExtension) -/// and pass a registration closure to the language_models crate. -pub type LanguageModelProviderRegistration = Box; - -pub trait ExtensionLanguageModelProviderProxy: Send + Sync + 'static { - /// Register an LLM provider from an extension. - /// The `register_fn` closure will be called with the App context and should - /// register the provider with the LanguageModelRegistry. - fn register_language_model_provider( - &self, - provider_id: Arc, - register_fn: LanguageModelProviderRegistration, - cx: &mut App, - ); - - /// Unregister an LLM provider when an extension is unloaded. - fn unregister_language_model_provider(&self, provider_id: Arc, cx: &mut App); -} - -impl ExtensionLanguageModelProviderProxy for ExtensionHostProxy { - fn register_language_model_provider( - &self, - provider_id: Arc, - register_fn: LanguageModelProviderRegistration, - cx: &mut App, - ) { - let Some(proxy) = self.language_model_provider_proxy.read().clone() else { - return; - }; - - proxy.register_language_model_provider(provider_id, register_fn, cx) - } - - fn unregister_language_model_provider(&self, provider_id: Arc, cx: &mut App) { - let Some(proxy) = self.language_model_provider_proxy.read().clone() else { - return; - }; - - proxy.unregister_language_model_provider(provider_id, cx) - } -} - impl ExtensionContextServerProxy for ExtensionHostProxy { fn register_context_server( &self,