Skip to main content

ferrin_spec/dynamic/
speech_model.rs

1//! Object-safe speech model.
2
3use super::BoxFuture;
4use super::ModelRef;
5use super::model_ref::ref_conversions;
6use crate::error::ProviderError;
7use crate::shared::ModelId;
8use crate::shared::ProviderId;
9use crate::speech_model::SpeechModel;
10use crate::speech_model::SpeechOptions;
11use crate::speech_model::SpeechResult;
12
13/// Object-safe counterpart of [`SpeechModel`].
14pub trait DynSpeechModel: Send + Sync + 'static {
15    /// See [`SpeechModel::provider`].
16    fn provider(&self) -> &ProviderId;
17    /// See [`SpeechModel::model_id`].
18    fn model_id(&self) -> &ModelId;
19    /// See [`SpeechModel::do_generate`].
20    fn do_generate(
21        &self,
22        options: SpeechOptions,
23    ) -> BoxFuture<'_, Result<SpeechResult, ProviderError>>;
24}
25
26impl<T: SpeechModel> DynSpeechModel for T {
27    fn provider(&self) -> &ProviderId {
28        SpeechModel::provider(self)
29    }
30
31    fn model_id(&self) -> &ModelId {
32        SpeechModel::model_id(self)
33    }
34
35    fn do_generate(
36        &self,
37        options: SpeechOptions,
38    ) -> BoxFuture<'_, Result<SpeechResult, ProviderError>> {
39        Box::pin(SpeechModel::do_generate(self, options))
40    }
41}
42
43/// Shared reference to a speech model (or an unresolved model id).
44pub type SpeechModelRef = ModelRef<dyn DynSpeechModel>;
45
46ref_conversions!(SpeechModelRef, SpeechModel, DynSpeechModel);