Add list files function

assistant.py

@@ -5,6 +5,8 @@ from openai import OpenAI
 from assistant_file_handler import FileHandler
 from openai.types.beta.thread import Thread
 from openai.types.beta.threads.message import Message
+from openai.types.beta.vector_stores.vector_store_file import VectorStoreFile
+
 
 import structlog
 from openai.pagination import SyncCursorPage
@@ -87,13 +89,12 @@ class OAIAssistant:
             return new_message
         except Exception as e:
             self.log.error("OAIAssistant: Error generating response", error=str(e))
-            import traceback
-
-            self.log.error(traceback.print_exc())
             return "OAIAssistant: An error occurred while generating the response."
 
     def create_thread(self) -> Thread:
-        thread: Thread = self.client.beta.threads.create(tool_resources={"file_search": {"vector_store_ids": [self.vectorstore_id]}})
+        thread: Thread = self.client.beta.threads.create(
+            tool_resources={"file_search": {"vector_store_ids": [self.vectorstore_id]}}
+        )
         return thread
 
     def delete_thread(self, thread_id: str):
@@ -139,10 +140,11 @@ class OAIAssistant:
                 )
                 if run.status == "failed":
                     self.log.error(
-                        "OAIAssistant:  Assistant run failed",
+                        "OAIAssistant: Assistant run failed",
                         run_id=run.id,
                     )
-                    self.log.error(run)
+                    self.log.info(run)
+
                     return "OAIAssistant: Error in generating response", []
 
             messages: SyncCursorPage[Message] = self.client.beta.threads.messages.list(
@@ -171,4 +173,12 @@ class OAIAssistant:
             return new_message, file_ids
         except Exception as e:
             self.log.error("OAIAssistant: Error extracting messages", error=str(e))
-            raise
+            raise
+
+    def get_files_list(self):
+        files: SyncCursorPage[VectorStoreFile] = (
+            self.client.beta.vector_stores.files.list(
+                vector_store_id=self.vectorstore_id
+            )
+        )
+        return [file.id for file in files]