minor adjustments

mem0ai · Nov 24, 2023 · bfd2eee · bfd2eee
1 parent 9b46d2c
commit bfd2eee
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 3 deletions.
diff --git a/embedchain/__init__.py b/embedchain/__init__.py
@@ -7,4 +7,5 @@
 from embedchain.pipeline import Pipeline  # noqa: F401
 from embedchain.vectordb.chroma import ChromaDB  # noqa: F401
 
+# Setup the user directory if doesn't exist already
 Client.setup_dir()
diff --git a/embedchain/embedchain.py b/embedchain/embedchain.py
@@ -16,8 +16,7 @@
 from embedchain.helpers.json_serializable import JSONSerializable
 from embedchain.llm.base import BaseLlm
 from embedchain.loaders.base_loader import BaseLoader
-from embedchain.models.data_type import (DataType, DirectDataType,
-                                         IndirectDataType, SpecialDataType)
+from embedchain.models.data_type import DataType, DirectDataType, IndirectDataType, SpecialDataType
 from embedchain.telemetry.posthog import AnonymousTelemetry
 from embedchain.utils import detect_datatype, is_valid_json_string
 from embedchain.vectordb.base import BaseVectorDB

diff --git a/embedchain/pipeline.py b/embedchain/pipeline.py
@@ -24,6 +24,10 @@
 from embedchain.vectordb.chroma import ChromaDB
 
 
+# Setup the user directory if doesn't exist already
+Client.setup_dir()
+
+
 @register_deserializable
 class Pipeline(EmbedChain):
     """

diff --git a/embedchain/store/assistants.py b/embedchain/store/assistants.py
@@ -10,7 +10,7 @@
 from openai import OpenAI
 from openai.types.beta.threads import MessageContentText, ThreadMessage
 
-from embedchain import Pipeline
+from embedchain import Client, Pipeline
 from embedchain.config import AddConfig
 from embedchain.data_formatter import DataFormatter
 from embedchain.models.data_type import DataType
@@ -19,6 +19,9 @@
 
 logging.basicConfig(level=logging.WARN)
 
+# Setup the user directory if doesn't exist already
+Client.setup_dir()
+
 
 class OpenAIAssistant:
     def __init__(