From bbf55f9547b799bf4fc624080051924f9ffa4ce1 Mon Sep 17 00:00:00 2001
From: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
Date: Sun, 26 May 2024 09:52:23 +0200
Subject: [PATCH] Updates to the configs

---
 cognee/infrastructure/databases/vector/config.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/cognee/infrastructure/databases/vector/config.py b/cognee/infrastructure/databases/vector/config.py
index a304b5bc5..dd8301401 100644
--- a/cognee/infrastructure/databases/vector/config.py
+++ b/cognee/infrastructure/databases/vector/config.py
@@ -4,6 +4,8 @@ from pydantic_settings import BaseSettings, SettingsConfigDict
 from cognee.base_config import get_base_config
 from cognee.infrastructure.databases.vector.lancedb.LanceDBAdapter import LanceDBAdapter
 from cognee.infrastructure.databases.vector.embeddings.config import get_embedding_config
+from cognee.infrastructure.files.storage import LocalStorage
+
 embeddings_config = get_embedding_config()
 base_config = get_base_config()
 
@@ -18,6 +20,8 @@ class VectorConfig(BaseSettings):
                     )
     vector_engine_choice:str = "lancedb"
 
+    LocalStorage.ensure_directory_exists(vector_db_path)
+
     model_config = SettingsConfigDict(env_file = ".env", extra = "allow")
 
     def to_dict(self) -> dict: