Merge pull request #13 from rajivpant/identify-sources

Updated to call load_files function in updated format.
rajivpant · May 5, 2024 · 5ede7b1 · 5ede7b1
2 parents ea1d448 + 142a9c9
commit 5ede7b1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ragbot.py b/ragbot.py
@@ -163,7 +163,7 @@ def main():
         # Load default custom_instructions for profile
         default_custom_instructions_paths = custom_instruction_paths
         default_custom_instructions_paths = [path for path in default_custom_instructions_paths if path.strip() != '']
-        custom_instructions, custom_instructions_files = load_files(default_custom_instructions_paths + args.curated_dataset)
+        custom_instructions, custom_instructions_files = load_files(file_paths=default_custom_instructions_paths + args.curated_dataset, file_type="custom_instructions")
 
     if custom_instructions_files:
         print("Custom instructions being used:")
@@ -176,7 +176,7 @@ def main():
         # Load default curated_datasets profile
         default_curated_dataset_paths = curated_dataset_paths
         default_curated_dataset_paths = [path for path in default_curated_dataset_paths if path.strip() != '']
-        curated_datasets, curated_dataset_files = load_files(default_curated_dataset_paths + args.curated_dataset)
+        curated_datasets, curated_dataset_files = load_files(file_paths=default_curated_dataset_paths + args.curated_dataset, file_type="curated_datasets")
 
     if curated_dataset_files:
         print("Curated datasets being used:")