aait 1.0.0.6__tar.gz → 1.0.2__tar.gz

{aait-1.0.0.6 → aait-1.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aait
-Version: 1.0.0.6
+Version: 1.0.2
 Summary: Advanced Artificial Intelligence Tools is a package meant to develop and enable advanced AI functionalities in Orange
 Home-page: 
 Author: Orange community

{aait-1.0.0.6 → aait-1.0.2}/aait.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aait
-Version: 1.0.0.6
+Version: 1.0.2
 Summary: Advanced Artificial Intelligence Tools is a package meant to develop and enable advanced AI functionalities in Orange
 Home-page: 
 Author: Orange community

{aait-1.0.0.6 → aait-1.0.2}/aait.egg-info/SOURCES.txt

@@ -1,4 +1,5 @@
 License.txt
+setup.cfg
 setup.py
 aait.egg-info/PKG-INFO
 aait.egg-info/SOURCES.txt

{aait-1.0.0.6 → aait-1.0.2}/orangecontrib/AAIT/llm/prompt_management.py

@@ -21,6 +21,13 @@ prompt_templates = {
 
 
 
+    "deepseek":
+"""### Instruction: {user_prompt}
+
+### Response:""",
+
+
+
     "qwen":
 """<|im_start|>system
 {system_prompt}<|im_end|>
@@ -44,9 +51,10 @@ model_types = {
     "Mistral-7B-Instruct-v0.3.Q6_K.gguf": "mistral",
     "Qwen2.5.1-Coder-7B-Instruct-Q6_K.gguf": "qwen",
     "qwen2.5-3b-instruct-q4_k_m.gguf": "qwen",
+    "deepseek-coder-6.7b-instruct.Q6_K.gguf": "deepseek"
 }
 
-model_keywords = ["qwen", "solar", "mistral", "llama"]
+model_keywords = ["qwen", "solar", "mistral", "llama", "deepseek"]
 
 def apply_prompt_template(model_name, user_prompt, assistant_prompt="", system_prompt=""):
     """

{aait-1.0.0.6 → aait-1.0.2}/orangecontrib/AAIT/widgets/OWModel_MPNET.py

@@ -54,10 +54,16 @@ class OWModelMPNET(widget.OWWidget):
                 SimpleDialogQt.BoxError("Unable to get the Model.")
                 return
         # Data Management
-        self.progressBarInit()
-        self.thread = thread_management.Thread(self.load_sentence_transformer, self.model_path)
-        self.thread.finish.connect(self.handle_loading_finish)
-        self.thread.start()
+        self.load_sentence_transformer(self.model_path)
+        if self.model is not None:
+            self.Outputs.out_model.send(self.model)
+        else:
+            SimpleDialogQt.BoxError("An Error Occurred when loading model.")
+            self.Outputs.out_model.send(None)
+        # self.progressBarInit()
+        # self.thread = thread_management.Thread(self.load_sentence_transformer, self.model_path)
+        # self.thread.finish.connect(self.handle_loading_finish)
+        # self.thread.start()
 
     def load_sentence_transformer(self, model_path):
         self.model = SentenceTransformer(model_path)

{aait-1.0.0.6 → aait-1.0.2}/setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 NAME = "aait"
 
 # Version du package PyPI
-VERSION = "1.0.0.6"  # la version doit être supérieure à la précédente sinon la publication sera refusée
+VERSION = "1.0.2"  # la version doit être supérieure à la précédente sinon la publication sera refusée
 
 # Facultatif / Adaptable à souhait
 AUTHOR = "Orange community"