blob42/langchain
1
0
Fork 0
forked from Archives/langchain
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add load() to snowflake loader (#5956)

Browse Source 
Quick fix for recently added [snowflake data
loader](https://github.com/hwchase17/langchain/pull/5825/files).
This commit is contained in:
Lance Martin 2023-06-09 11:27:29 -07:00 committed by GitHub
parent 3678cba0be
commit f3e7ac0a2c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
langchain/document_loaders/snowflake_loader.py
View File

@ -118,3 +118,6 @@ class SnowflakeLoader(BaseLoader):
metadata = {k: v for k, v in row.items() if k in metadata_columns}
doc = Document(page_content=page_content, metadata=metadata)
yield doc
def load(self) -> List[Document]:
return list(self.lazy_load())