feat: Support Azure embedding model

2024-12-18 16:40:00 +08:00 · 2024-12-18 16:40:00 +08:00 · 097cabbf5b
parent 003f3d13e0
commit 097cabbf5b
1 changed files with 4 additions and 0 deletions
--- 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