Skip to content

Commit 788e65d

Browse files
authored
fix: missing cache for supported models in TextEmbedding
1 parent 447d5e6 commit 788e65d

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

qwen3_embed/text/text_embedding.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)