refactor: add is_cache_model method to various model classes to standardize cache behavior

Author: wxg0103

apps/models_provider/impl/aliyun_bai_lian_model_provider/model/image.py

@@ -8,6 +8,10 @@
 
 class QwenVLChatModel(MaxKBBaseModel, BaseChatOpenAI):
 
+    @staticmethod
+    def is_cache_model():
+        return False
+
     @staticmethod
     def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
         optional_params = MaxKBBaseModel.filter_optional_params(model_kwargs)

apps/models_provider/impl/aliyun_bai_lian_model_provider/model/stt.py

@@ -19,6 +19,10 @@ def __init__(self, **kwargs):
         self.api_key = kwargs.get('api_key')
         self.model = kwargs.get('model')
 
+    @staticmethod
+    def is_cache_model():
+        return False
+
     @staticmethod
     def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
         optional_params = {}

apps/models_provider/impl/aliyun_bai_lian_model_provider/model/tti.py

@@ -13,7 +13,6 @@
 from models_provider.impl.base_tti import BaseTextToImage
 
 
-
 class QwenTextToImageModel(MaxKBBaseModel, BaseTextToImage):
     api_key: str
     model_name: str
@@ -25,6 +24,10 @@ def __init__(self, **kwargs):
         self.model_name = kwargs.get('model_name')
         self.params = kwargs.get('params')
 
+    @staticmethod
+    def is_cache_model():
+        return False
+
     @staticmethod
     def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
         optional_params = {'params': {'size': '1024*1024', 'style': '<auto>', 'n': 1}}

apps/models_provider/impl/aliyun_bai_lian_model_provider/model/tts.py

@@ -20,6 +20,10 @@ def __init__(self, **kwargs):
         self.model = kwargs.get('model')
         self.params = kwargs.get('params')
 
+    @staticmethod
+    def is_cache_model():
+        return False
+
     @staticmethod
     def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
         optional_params = {'params': {'voice': 'longxiaochun', 'speech_rate': 1.0}}
@@ -52,5 +56,3 @@ def text_to_speech(self, text):
             raise Exception(audio)
         return audio
 
-    def is_cache_model(self):
-        return False

apps/models_provider/impl/anthropic_model_provider/model/image.py

@@ -13,6 +13,10 @@ def custom_get_token_ids(text: str):
 
 class AnthropicImage(MaxKBBaseModel, ChatAnthropic):
 
+    @staticmethod
+    def is_cache_model():
+        return False
+
     @staticmethod
     def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
         optional_params = MaxKBBaseModel.filter_optional_params(model_kwargs)

apps/models_provider/impl/azure_model_provider/model/image.py

@@ -14,6 +14,10 @@ def custom_get_token_ids(text: str):
 
 class AzureOpenAIImage(MaxKBBaseModel, AzureChatOpenAI):
 
+    @staticmethod
+    def is_cache_model():
+        return False
+
     @staticmethod
     def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
         optional_params = MaxKBBaseModel.filter_optional_params(model_kwargs)

apps/models_provider/impl/azure_model_provider/model/stt.py

@@ -25,6 +25,10 @@ def __init__(self, **kwargs):
         self.api_base = kwargs.get('api_base')
         self.api_version = kwargs.get('api_version')
 
+    @staticmethod
+    def is_cache_model():
+        return False
+
     @staticmethod
     def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
         optional_params = {}

apps/models_provider/impl/azure_model_provider/model/tti.py

@@ -27,6 +27,10 @@ def __init__(self, **kwargs):
         self.model = kwargs.get('model')
         self.params = kwargs.get('params')
 
+    @staticmethod
+    def is_cache_model():
+        return False
+
     @staticmethod
     def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
         optional_params = {'params': {'size': '1024x1024', 'quality': 'standard', 'n': 1}}

apps/models_provider/impl/azure_model_provider/model/tts.py

@@ -28,6 +28,10 @@ def __init__(self, **kwargs):
         self.model = kwargs.get('model')
         self.params = kwargs.get('params')
 
+    @staticmethod
+    def is_cache_model():
+        return False
+
     @staticmethod
     def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
         optional_params = {'params': {'voice': 'alloy'}}

apps/models_provider/impl/gemini_model_provider/model/image.py

@@ -13,6 +13,10 @@ def custom_get_token_ids(text: str):
 
 class GeminiImage(MaxKBBaseModel, ChatGoogleGenerativeAI):
 
+    @staticmethod
+    def is_cache_model():
+        return False
+
     @staticmethod
     def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
         optional_params = MaxKBBaseModel.filter_optional_params(model_kwargs)