Merge pull request #274 from MLSysOps/fix/lancedb-init

[MRG] Fixed the memory init before config error

Author: HuaizhengZhang

mle/cli.py

@@ -19,11 +19,10 @@
     startup_web,
     print_in_box,
 )
-from mle.utils import LanceDBMemory, list_files, read_file
 from mle.utils import CodeChunker
+from mle.utils import LanceDBMemory, list_files, read_file
 
 console = Console()
-memory = LanceDBMemory(os.getcwd())
 
 
 @click.group()
@@ -199,6 +198,7 @@ def chat(model, build_mem):
     if not check_config(console):
         return
 
+    memory = LanceDBMemory(os.getcwd())
     if build_mem:
         working_dir = os.getcwd()
         table_name = 'mle_chat_' + working_dir.split('/')[-1]

requirements.txt

@@ -1,7 +1,7 @@
 rich
 click
 py7zr~=0.22.0
-openai~=1.34.0
+openai
 pyyaml
 kaggle
 fastapi