diff --git a/llama-index-integrations/readers/llama-index-readers-google/BUILD b/llama-index-integrations/readers/llama-index-readers-google/BUILD index 1c1c977691cae26b186a15c923cba60a0349cadf..6e77d4b61552f1ebeb0a161b956322d980377c7a 100644 --- a/llama-index-integrations/readers/llama-index-readers-google/BUILD +++ b/llama-index-integrations/readers/llama-index-readers-google/BUILD @@ -5,7 +5,3 @@ poetry_requirements( "google-auth-oauthlib": ["google_auth_oauthlib"], }, ) - -python_requirements( - name="reqs", -) diff --git a/llama-index-integrations/readers/llama-index-readers-google/llama_index/readers/google/drive/base.py b/llama-index-integrations/readers/llama-index-readers-google/llama_index/readers/google/drive/base.py index 4a34da5be6785a6c02fa492dca9234ed69985d88..a6827256162c3d86b9b1cf2687e5afc25f283223 100644 --- a/llama-index-integrations/readers/llama-index-readers-google/llama_index/readers/google/drive/base.py +++ b/llama-index-integrations/readers/llama-index-readers-google/llama_index/readers/google/drive/base.py @@ -310,11 +310,12 @@ class GoogleDriveReader(BaseReader): metadata = {} for fileid_meta in fileids_meta: - filename = fileid_meta[2] - filepath = os.path.join(temp_dir, filename) + # Download files and name them with their fileid fileid = fileid_meta[0] + filepath = os.path.join(temp_dir, fileid) final_filepath = self._download_file(fileid, filepath) + # Add metadata of the file to metadata dictionary metadata[final_filepath] = { "file id": fileid_meta[0], "author": fileid_meta[1], diff --git a/llama-index-integrations/readers/llama-index-readers-google/pyproject.toml b/llama-index-integrations/readers/llama-index-readers-google/pyproject.toml index 469924d5d982e8b85f713cfd1aa128720b837cce..9a444a9ad1dadec9fac7fceaa4c3c034af5d57f9 100644 --- a/llama-index-integrations/readers/llama-index-readers-google/pyproject.toml +++ b/llama-index-integrations/readers/llama-index-readers-google/pyproject.toml @@ -45,7 +45,7 @@ maintainers = [ ] name = "llama-index-readers-google" readme = "README.md" -version = "0.1.6" +version = "0.1.7" [tool.poetry.dependencies] python = ">=3.10,<4.0" diff --git a/llama-index-integrations/readers/llama-index-readers-google/requirements.txt b/llama-index-integrations/readers/llama-index-readers-google/requirements.txt deleted file mode 100644 index 50658494606521c46502e5acb6be22c60165d0bb..0000000000000000000000000000000000000000 --- a/llama-index-integrations/readers/llama-index-readers-google/requirements.txt +++ /dev/null @@ -1,5 +0,0 @@ -google-api-python-client -google-auth-httplib2 -google-auth-oauthlib -PyDrive -gkeepapi diff --git a/llama-index-integrations/readers/llama-index-readers-google/tests/BUILD b/llama-index-integrations/readers/llama-index-readers-google/tests/BUILD new file mode 100644 index 0000000000000000000000000000000000000000..a1431c2350d7c0327cf8ed461ce5ad343a407dce --- /dev/null +++ b/llama-index-integrations/readers/llama-index-readers-google/tests/BUILD @@ -0,0 +1,3 @@ +python_tests( + interpreter_constraints=["==3.10.*"], +) diff --git a/llama-index-integrations/readers/llama-index-readers-google/tests/__init__.py b/llama-index-integrations/readers/llama-index-readers-google/tests/__init__.py new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/llama-index-integrations/readers/llama-index-readers-google/tests/test_readers_google_drive.py b/llama-index-integrations/readers/llama-index-readers-google/tests/test_readers_google_drive.py new file mode 100644 index 0000000000000000000000000000000000000000..07825d26623608fd7eeaa3d4a3bbe833fae8f5c1 --- /dev/null +++ b/llama-index-integrations/readers/llama-index-readers-google/tests/test_readers_google_drive.py @@ -0,0 +1,7 @@ +from llama_index.core.readers.base import BaseReader +from llama_index.readers.google import GoogleDriveReader + + +def test_class(): + names_of_base_classes = [b.__name__ for b in GoogleDriveReader.__mro__] + assert BaseReader.__name__ in names_of_base_classes