TMP: tools allowed for all

szczyglis-dev · Jan 31, 2025 · 379646a · 379646a
1 parent c440dda
commit 379646a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/pygpt_net/core/command/__init__.py b/src/pygpt_net/core/command/__init__.py
@@ -641,16 +641,19 @@ def is_model_supports_tools(
         :param model: model item
         :return: True if model supports tools
         """
+        return True  # TMP allowed all
         if model is None:
             return False
         disabled_models = [
-            "deepseek",
+            "deepseek-r1:1.5b",
+            "deepseek-r1:7b",
             "llama2",
             "llama3.1",
             "codellama",
         ]
         if model.id is not None:
             for disabled_model in disabled_models:
-                if model.id.startswith(disabled_model):
+                if (model.llama_index['provider'] == "ollama"
+                        and model.id.startswith(disabled_model)):
                     return False
         return True