diff --git a/src/vectara/corpora/types/search_corpus_parameters.py b/src/vectara/corpora/types/search_corpus_parameters.py index 1563353..ec3affb 100644 --- a/src/vectara/corpora/types/search_corpus_parameters.py +++ b/src/vectara/corpora/types/search_corpus_parameters.py @@ -23,8 +23,9 @@ class Config: extra = pydantic.Extra.allow from ...types.chain_reranker import ChainReranker # noqa: E402 +from ...types.search_reranker import SearchReranker # noqa: E402 update_forward_refs(ChainReranker) if IS_PYDANTIC_V2: - SearchCorpusParameters.model_rebuild() \ No newline at end of file + SearchCorpusParameters.model_rebuild() diff --git a/src/vectara/types/search_corpora_parameters.py b/src/vectara/types/search_corpora_parameters.py index 3048448..7a1d551 100644 --- a/src/vectara/types/search_corpora_parameters.py +++ b/src/vectara/types/search_corpora_parameters.py @@ -28,6 +28,7 @@ class Config: extra = pydantic.Extra.allow from .chain_reranker import ChainReranker # noqa: E402 +from .search_reranker import SearchReranker # noqa: E402 update_forward_refs(ChainReranker)