浏览代码

[Bug fix] missing dir on first init for App (#934)

Sidharth Mohanty 1 年之前
父节点
当前提交
744ab5156f
共有 1 个文件被更改,包括 5 次插入2 次删除
  1. 5 2
      embedchain/apps/app.py

+ 5 - 2
embedchain/apps/app.py

@@ -2,8 +2,8 @@ from typing import Optional
 
 import yaml
 
-from embedchain.config import (AppConfig, BaseEmbedderConfig, BaseLlmConfig,
-                               ChunkerConfig)
+from embedchain.client import Client
+from embedchain.config import AppConfig, BaseEmbedderConfig, BaseLlmConfig, ChunkerConfig
 from embedchain.config.vectordb.base import BaseVectorDbConfig
 from embedchain.embedchain import EmbedChain
 from embedchain.embedder.base import BaseEmbedder
@@ -67,6 +67,9 @@ class App(EmbedChain):
         :type system_prompt: Optional[str], optional
         :raises TypeError: LLM, database or embedder or their config is not a valid class instance.
         """
+        # Setup user directory if it doesn't exist already
+        Client.setup_dir()
+
         # Type check configs
         if config and not isinstance(config, AppConfig):
             raise TypeError(