Skip to content

Commit 0712141

Browse files
authored
fix: bug with loading directories (#1073)
1 parent 1097eee commit 0712141

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

memgpt/autogen/memgpt_agent.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,8 +62,6 @@ def load(self, name: str, type: str, **kwargs):
6262
load_directory(name=name, **kwargs)
6363
elif type == "webpage":
6464
load_webpage(name=name, **kwargs)
65-
elif type == "index":
66-
load_index(name=name, **kwargs)
6765
elif type == "database":
6866
load_database(name=name, **kwargs)
6967
elif type == "vector_database":

memgpt/data_sources/connectors.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ def __init__(self, input_files: List[str] = None, input_directory: str = None, r
8686
self.extensions = extensions
8787

8888
if self.recursive == True:
89-
assert self.input_dir is not None, "Must provide input directory if recursive is True."
89+
assert self.input_directory is not None, "Must provide input directory if recursive is True."
9090

9191
def generate_documents(self) -> Iterator[Tuple[str, Dict]]: # -> Iterator[Document]:
9292
from llama_index.core import SimpleDirectoryReader
@@ -101,7 +101,7 @@ def generate_documents(self) -> Iterator[Tuple[str, Dict]]: # -> Iterator[Docum
101101
assert self.input_files is not None, "Must provide input files if input_dir is None"
102102
reader = SimpleDirectoryReader(input_files=[str(f) for f in self.input_files])
103103

104-
llama_index_docs = reader.load_data()
104+
llama_index_docs = reader.load_data(show_progress=True)
105105
docs = []
106106
for llama_index_doc in llama_index_docs:
107107
# TODO: add additional metadata?

0 commit comments

Comments
 (0)