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

[MRG] Fixed the memory init before config error
MLSysOps · Nov 28, 2024 · b9765e7 · b9765e7
2 parents ea7393b + ff809bf
commit b9765e7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/mle/cli.py b/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]

diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 rich
 click
 py7zr~=0.22.0
-openai~=1.34.0
+openai
 pyyaml
 kaggle
 fastapi