ferrin_spec/dynamic/
reranking_model.rs

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

ferrin_spec/dynamic/
reranking_model.rs
