diff --git a/pebblo_saferetriever/langchain/pebblo-saferag/sharepoint-postgres/pebblo_saferag.py b/pebblo_saferetriever/langchain/pebblo-saferag/sharepoint-postgres/pebblo_saferag.py index 593257a8..7203d6c7 100644 --- a/pebblo_saferetriever/langchain/pebblo-saferag/sharepoint-postgres/pebblo_saferag.py +++ b/pebblo_saferetriever/langchain/pebblo-saferag/sharepoint-postgres/pebblo_saferag.py @@ -218,27 +218,30 @@ def select_drive(drives: list) -> tuple: client_secret=app_client_secret, tenant_id=tenant_id, ) - print("SharepointADHelper initialized.") + print("SharepointADHelper initialized.\n") site_url = ( - input(f"\nEnter Sharepoint Site URL (default={_site_url}): ") or _site_url + input(f"Enter Sharepoint Site URL (default={_site_url}): ") or _site_url ) if not site_url: print("\nSite URL is required. Exiting ...") exit(1) + # remove white spaces from the site url + site_url = site_url.strip() # Get SharePoint Site ID using URL site_id = sharepoint_helper.get_site_id(site_url) print(f"Derived Site Id: {site_id}\n") # Get drive info using site id - print("Fetching drive info ...") + print("Fetching drive info ...\n") drive_info = sharepoint_helper.get_drive_id(site_id) drive_id, drive_name = select_drive(drive_info) - print(f"\nSharePoint Drive name: {drive_name}, Drive Id: {drive_id}") + print(f"SharePoint Drive name: {drive_name}, Drive Id: {drive_id}\n") # Enter Folder path - folder_path = input("\nEnter folder path (default='/document'): ") or "/document" + def_folder_path = "documents" + folder_path = input(f"Enter folder path (default='{def_folder_path}'): ") or def_folder_path # Initialize PebbloSafeRAG app rag_app = PebbloSafeRAG(