Fix getting sources html for UserData upload

h2oai · May 22, 2023 · 4d50f91 · 4d50f91
1 parent 8817ab5
commit 4d50f91
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git 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
@@ -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""")