github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-03 11:47:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed a source for documents uploaded from GCS (#8912)

Browse Source 
Sets source for documents uploaded from GCS to source on gcs
#8911

Co-authored-by: Leonid Kuligin <kuligin@google.com>
This commit is contained in:
Leonid Kuligin 2023-08-08 15:34:43 +02:00 committed by GitHub
parent e74a605379
commit 52d6b91c18
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/langchain/langchain/document_loaders/gcs_file.py
View File

@ -69,4 +69,8 @@ class GCSFileLoader(BaseLoader):
# Download the file to a destination
blob.download_to_filename(file_path)
loader = self._loader_func(file_path)
return loader.load()
docs = loader.load()
for doc in docs:
if "source" in doc.metadata:
doc.metadata["source"] = f"gs://{self.bucket}/{self.blob}"
return docs