File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -26,12 +26,14 @@ class TextEmbedding(TextEmbeddingBase):
2626 PooledEmbedding ,
2727 CustomTextEmbedding ,
2828 ]
29+ _supported_models_cache : list [DenseModelDescription ] | None = None
2930 _embedding_type_cache : dict [str , type [TextEmbeddingBase ]] | None = None
3031 _embedding_description_cache : dict [str , DenseModelDescription ] | None = None
3132
3233 @classmethod
3334 def _clear_model_cache (cls ) -> None :
3435 cls ._embedding_type_cache = None
36+ cls ._supported_models_cache = None
3537 cls ._embedding_description_cache = None
3638
3739 @classmethod
@@ -62,9 +64,14 @@ def list_supported_models(cls) -> list[dict[str, Any]]:
6264
6365 @classmethod
6466 def _list_supported_models (cls ) -> list [DenseModelDescription ]:
67+ if cls ._supported_models_cache is not None :
68+ return cls ._supported_models_cache
69+
6570 result : list [DenseModelDescription ] = []
6671 for embedding in cls .EMBEDDINGS_REGISTRY :
6772 result .extend (embedding ._list_supported_models ())
73+
74+ cls ._supported_models_cache = result
6875 return result
6976
7077 @classmethod
You can’t perform that action at this time.
0 commit comments