update assistant script

core/demorrha.py

@@ -124,7 +124,13 @@ class DemorrhaAssistant(object):
             )
         else:
             self.assistant = self.client.beta.assistants.retrieve(assistant_id)
-
+        return self
+    
+    def get_assistant(self):
+        return self.assistant if not isinstance(self.assistant, None) else None
+    
+    def get_assistant_id(self):
+        return self.assistant.id if not isinstance(self.assistant, None) else None
 
     def add_file_to_vector_store(self, file_paths):
         file_streams = [open(path, "rb") for path in file_paths]
@@ -133,9 +139,29 @@ class DemorrhaAssistant(object):
         )
         return file_batch
     
+    def set_payload(self, 
+                    content_message:str, 
+                    operation_prompt: Optional[str] = ""):
+        self.payload_content = f'{operation_prompt} :\n"""\n{content_message}\n"""'
+        return self
+
+    def set_system_prompt(self, 
+                          system_prompt: Optional[str] = ""):
+        self.system_prompt = system_prompt
+        return self
+
     def use_assistant(self):
+        self.set_system_prompt(GlobalSystemPrompts.linguascribe())
         # Utiliser l'assistant ici
-        pass
+        run = self.client.beta.threads.create_and_run(
+            assistant_id=self.assistant.id,
+            thread={
+                "messages": [
+                    {"role": "system", "content": f"{self.system_prompt}"},
+                    {"role": "user", "content": self.payload_content}
+                ]
+            }
+        )
     
 """
 
@@ -160,12 +186,9 @@ if __name__ == "__main__":
             file_paths = ["style.txt"]
             file_batch = demorrha.add_file_to_vector_store(file_paths)
             print("Fichier ajouté au vector_store:", file_batch)
-"""
-    # Exemple d'utilisation de l'assistant
-    assistant_id = demorrha.search_assistant("Demorrha")
-    if assistant_id:
-        response = demorrha.use_assistant(assistant_id)
-        print("Réponse de l'assistant:", response)
-    else:
-        print("Assistant non trouvé.")
-"""
+
+    # Exemple d'utilsation de l'assistant
+    demorrha.load_assistant("Demorrha")
+    response = demorrha.use_assistant()
+    print("Réponse de l'assistant:", response)
+