diff --git a/gradio_runner.py b/gradio_runner.py index 7dcd23af0..89c5b742a 100644 --- a/gradio_runner.py +++ b/gradio_runner.py @@ -1311,7 +1311,7 @@ def _update_user_db(file, db1, x, y, dbs=None, db_type=None, langchain_mode='Use # NOTE we do not return db, because function call always same code path # return dbs[langchain_mode], x, y # db in this code path is updated in place - source_files_added = get_source_files(db1, exceptions=exceptions) + source_files_added = get_source_files(dbs[langchain_mode], exceptions=exceptions) return x, y, source_files_added diff --git a/tests/test_manual_test.py b/tests/test_manual_test.py index 0f549748c..53b2969b2 100644 --- a/tests/test_manual_test.py +++ b/tests/test_manual_test.py @@ -50,3 +50,7 @@ def test_no_db_dirs(): def test_upload_unsupported_file(): raise NotImplementedError("""MANUAL TEST FOR NOW -- e.g. json, ensure error correct and reasonable, no cascades""") + + +def test_upload_to_UserData_and_MyData(): + raise NotImplementedError("""MANUAL TEST FOR NOW Upload to each when enabled, ensure no failures""")