update assistant script

core/demorrha.py

@@ -22,6 +22,7 @@ class DemorrhaAssistant(object):
         self.client = self.initialize_client() # OpenAI(api_key = getenv("OPENAI_API_KEY"))
         self.vector_store = None
         self.load_assistant()
+        self.list_vector_store_ids = []
 
     def initialize_client(self):
         # Initialiser le client OpenAI avec la clé API
@@ -149,19 +150,41 @@ class DemorrhaAssistant(object):
                           system_prompt: Optional[str] = ""):
         self.system_prompt = system_prompt
         return self
+    
+    def add_vector_store_to_ressource(self, vector_store_id):
+        self.list_vector_store_ids.append(vector_store_id)
+        return self
+    
+    def get_vector_store_ids(self):
+        return self.list_vector_store_ids
+    
+    def empty_vector_store_ids(self):
+        self.list_vector_store_ids = []
+        return self
+
+    def update_vector_store_ids(self):
+        self.assistant = self.client.beta.assistants.update(
+            assistant_id=self.assistant.id,
+            tool_resources={"file_search": {"vector_store_ids": self.list_vector_store_ids}},
+        )
 
     def use_assistant(self):
         self.set_system_prompt(GlobalSystemPrompts.linguascribe())
         # Utiliser l'assistant ici
-        run = self.client.beta.threads.create_and_run(
+        with 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}
                 ]
-            }
-        )
+            },
+            stream=True
+        ) as stream:
+            for event in stream:
+                # Print the text from text delta events
+                if event.type == "thread.message.delta" and event.data.delta.content:
+                    print(event.data.delta.content[0].text)
     
 """
 
@@ -176,12 +199,14 @@ class DemorrhaAssistant(object):
 if __name__ == "__main__":
     demorrha = DemorrhaAssistant()
     vector_store = demorrha.load_vector_store().get_vector_store()
+    demorrha.empty_vector_store_ids()
     print(vector_store)
 
     if vector_store.status == "completed":
         if vector_store.file_counts.total > 0:
             if vector_store.file_counts.completed == vector_store.file_counts.total:
                 print("Le chargement du vecteur est terminé.")
+                demorrha.add_vector_store_to_ressource(vector_store.id)
         else:
             file_paths = ["style.txt"]
             file_batch = demorrha.add_file_to_vector_store(file_paths)
@@ -189,6 +214,8 @@ if __name__ == "__main__":
 
     # Exemple d'utilsation de l'assistant
     demorrha.load_assistant("Demorrha")
-    response = demorrha.use_assistant()
-    print("Réponse de l'assistant:", response)
+    demorrha.update_vector_store_ids()
+    demorrha.set_payload("Bonjour, comment ça va ?", "Traduit le texte en Anglais")
+    demorrha.use_assistant()
+    #print("Réponse de l'assistant:", response)