File tree Expand file tree Collapse file tree 2 files changed +41
-2
lines changed Expand file tree Collapse file tree 2 files changed +41
-2
lines changed Original file line number Diff line number Diff line change @@ -8,17 +8,35 @@ export type RerankerVoyageAIConfig = {
88 model ?: 'rerank-lite-1' | string ;
99} ;
1010
11+ export type RerankerJinaAIConfig = {
12+ model ?:
13+ | 'jina-reranker-v2-base-multilingual'
14+ | 'jina-reranker-v1-base-en'
15+ | 'jina-reranker-v1-turbo-en'
16+ | 'jina-reranker-v1-tiny-en'
17+ | 'jina-colbert-v1-en'
18+ | string ;
19+ } ;
20+
1121export type RerankerConfig =
1222 | RerankerCohereConfig
1323 | RerankerTransformersConfig
1424 | RerankerVoyageAIConfig
1525 | Record < string , any >
1626 | undefined ;
1727
18- export type Reranker = 'reranker-cohere' | 'reranker-transformers' | 'reranker-voyageai' | 'none' | string ;
28+ export type Reranker =
29+ | 'reranker-cohere'
30+ | 'reranker-jinaai'
31+ | 'reranker-transformers'
32+ | 'reranker-voyageai'
33+ | 'none'
34+ | string ;
1935
2036export type RerankerConfigType < R > = R extends 'reranker-cohere'
2137 ? RerankerCohereConfig
38+ : R extends 'reranker-jinaai'
39+ ? RerankerJinaAIConfig
2240 : R extends 'reranker-transformers'
2341 ? RerankerTransformersConfig
2442 : R extends 'reranker-voyageai'
Original file line number Diff line number Diff line change 1- import { ModuleConfig , RerankerCohereConfig , RerankerVoyageAIConfig } from '../config/types/index.js' ;
1+ import {
2+ ModuleConfig ,
3+ RerankerCohereConfig ,
4+ RerankerJinaAIConfig ,
5+ RerankerVoyageAIConfig ,
6+ } from '../config/types/index.js' ;
27
38export default {
49 /**
@@ -17,6 +22,22 @@ export default {
1722 config : config ,
1823 } ;
1924 } ,
25+ /**
26+ * Create a `ModuleConfig<'reranker-jinaai', RerankerJinaAIConfig>` object for use when reranking using the `reranker-jinaai` module.
27+ *
28+ * See the [documentation](https://weaviate.io/developers/weaviate/modules/retriever-vectorizer-modules/reranker-jinaai) for detailed usage.
29+ *
30+ * @param {RerankerJinaAIConfig } [config] The configuration for the `reranker-jinaai` module.
31+ * @returns {ModuleConfig<'reranker-jinaai', RerankerJinaAIConfig | undefined> } The configuration object.
32+ */
33+ jinaai : (
34+ config ?: RerankerJinaAIConfig
35+ ) : ModuleConfig < 'reranker-jinaai' , RerankerJinaAIConfig | undefined > => {
36+ return {
37+ name : 'reranker-jinaai' ,
38+ config : config ,
39+ } ;
40+ } ,
2041 /**
2142 * Create a `ModuleConfig<'reranker-transformers', Record<string, never>>` object for use when reranking using the `reranker-transformers` module.
2243 *
You can’t perform that action at this time.
0 commit comments