From 097cabbf5bd997aeccc1acf7b33ab3cf53a42724 Mon Sep 17 00:00:00 2001 From: CaptainB Date: Wed, 18 Dec 2024 16:40:00 +0800 Subject: [PATCH] feat: Support Azure embedding model --- .../impl/azure_model_provider/model/embedding.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/apps/setting/models_provider/impl/azure_model_provider/model/embedding.py b/apps/setting/models_provider/impl/azure_model_provider/model/embedding.py index 78ca2e8bb..c7acf595c 100644 --- a/apps/setting/models_provider/impl/azure_model_provider/model/embedding.py +++ b/apps/setting/models_provider/impl/azure_model_provider/model/embedding.py @@ -23,3 +23,7 @@ class AzureOpenAIEmbeddingModel(MaxKBBaseModel, AzureOpenAIEmbeddings): openai_api_version=model_credential.get('api_version'), openai_api_type="azure", ) + + + def is_cache_model(self): + return True \ No newline at end of file