Update agent/Retriever.py

Co-authored-by: andresdiverGPT <[email protected]>
deep-dive-mexico · Nov 24, 2023 · dbf40fa · dbf40fa
1 parent 451061f
commit dbf40fa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/agent/Retriever.py b/agent/Retriever.py
@@ -20,7 +20,7 @@ def __init__(self, index_path):
         self.llm = OpenAI(temperature=0, model="gpt-4-1106-preview")
     self.service_context = ServiceContext.from_defaults(llm=self.llm)
 
-    def load_documents(files=files):
+    def load_documents(self, files):
         self.docs = {}
         for file in files.keys():
             docs[file] = SimpleDirectoryReader(