fix: check if data path exist

datvodinh · Oct 10, 2024 · 7d5355a · 7d5355a
1 parent 7a2bdba
commit 7d5355a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rag_chatbot/ui/ui.py b/rag_chatbot/ui/ui.py
@@ -89,6 +89,8 @@ def __init__(
         self._logger = logger
         self._host = host
         self._data_dir = os.path.join(os.getcwd(), data_dir)
+        if not os.path.exists(self._data_dir):
+            os.makedirs(self._data_dir, exist_ok = True)
         self._avatar_images = [
             os.path.join(os.getcwd(), image) for image in avatar_images
         ]